From 118b7341c426c1051932e149dffc5453879644f0 Mon Sep 17 00:00:00 2001
From: mertmit <mertmit99@gmail.com>
Date: Sat, 25 Mar 2023 00:11:27 +0300
Subject: [PATCH] feat: undo/redo set display value

Signed-off-by: mertmit <mertmit99@gmail.com>
---
 .../components/smartsheet/header/Menu.vue     | 35 +++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/packages/nc-gui/components/smartsheet/header/Menu.vue b/packages/nc-gui/components/smartsheet/header/Menu.vue
index 3a982fdc6d..457c6046cf 100644
--- a/packages/nc-gui/components/smartsheet/header/Menu.vue
+++ b/packages/nc-gui/components/smartsheet/header/Menu.vue
@@ -18,6 +18,7 @@ import {
   useMetas,
   useNuxtApp,
   useSmartsheetStoreOrThrow,
+  useUndoRedo,
 } from '#imports'
 
 const { virtual = false } = defineProps<{ virtual?: boolean }>()
@@ -42,6 +43,8 @@ const { t } = useI18n()
 
 const { getMeta } = useMetas()
 
+const { addUndo } = useUndoRedo()
+
 const deleteColumn = () =>
   Modal.confirm({
     title: h('div', ['Do you want to delete ', h('span', { class: 'font-weight-bold' }, [column?.value?.title]), ' column ?']),
@@ -69,6 +72,8 @@ const deleteColumn = () =>
 
 const setAsDisplayValue = async () => {
   try {
+    const currentDisplayValue = meta?.value?.columns?.find((f) => f.pv)
+
     await $api.dbTableColumn.primaryColumnSet(column?.value?.id as string)
 
     await getMeta(meta?.value?.id as string, true)
@@ -79,6 +84,36 @@ const setAsDisplayValue = async () => {
     message.success(t('msg.success.primaryColumnUpdated'))
 
     $e('a:column:set-primary')
+
+    addUndo({
+      redo: {
+        fn: async (id: string) => {
+          await $api.dbTableColumn.primaryColumnSet(id)
+
+          await getMeta(meta?.value?.id as string, true)
+
+          eventBus.emit(SmartsheetStoreEvents.FIELD_RELOAD)
+
+          // Successfully updated as primary column
+          message.success(t('msg.success.primaryColumnUpdated'))
+        },
+        args: [column?.value?.id as string],
+      },
+      undo: {
+        fn: async (id: string) => {
+          await $api.dbTableColumn.primaryColumnSet(id)
+
+          await getMeta(meta?.value?.id as string, true)
+
+          eventBus.emit(SmartsheetStoreEvents.FIELD_RELOAD)
+
+          // Successfully updated as primary column
+          message.success(t('msg.success.primaryColumnUpdated'))
+        },
+        args: [currentDisplayValue?.id],
+      },
+      scope: meta.value?.id,
+    })
   } catch (e) {
     message.error(t('msg.error.primaryColumnUpdateFailed'))
   }