From 73f59e5d040f675e6f2e11f584eef0040bb041e5 Mon Sep 17 00:00:00 2001
From: DarkPhoenix2704 <anbarasun123@gmail.com>
Date: Thu, 28 Nov 2024 14:48:00 +0000
Subject: [PATCH] feat: restore snapshots

---
 packages/nc-gui/components/dlg/Snapshot/Restore.vue | 13 ++++++++++++-
 packages/nc-gui/composables/useBaseSettings.ts      | 11 +++++++----
 2 files changed, 19 insertions(+), 5 deletions(-)

diff --git a/packages/nc-gui/components/dlg/Snapshot/Restore.vue b/packages/nc-gui/components/dlg/Snapshot/Restore.vue
index b74bfade72..e9edd5acba 100644
--- a/packages/nc-gui/components/dlg/Snapshot/Restore.vue
+++ b/packages/nc-gui/components/dlg/Snapshot/Restore.vue
@@ -17,7 +17,18 @@ const { snapshot } = props
 
 const vModel = useVModel(props, 'modelValue', emits)
 
-const { restoreSnapshot, isRestoringSnapshot } = useBaseSettings()
+const { restoreSnapshot: _restoreSnapshot, isRestoringSnapshot } = useBaseSettings()
+
+const restoreSnapshot = async (snapshot: SnapshotExtendedType) => {
+  try {
+    await _restoreSnapshot(snapshot, () => {
+      vModel.value = false
+      emits('update:modelValue', false)
+    })
+  } catch (error) {
+    console.error(error)
+  }
+}
 </script>
 
 <template>
diff --git a/packages/nc-gui/composables/useBaseSettings.ts b/packages/nc-gui/composables/useBaseSettings.ts
index aac295099b..4a7f0886e2 100644
--- a/packages/nc-gui/composables/useBaseSettings.ts
+++ b/packages/nc-gui/composables/useBaseSettings.ts
@@ -126,14 +126,14 @@ export const useBaseSettings = createSharedComposable(() => {
     }
   }
 
-  const restoreSnapshot = async (snapshot: SnapshotExtendedType) => {
+  const restoreSnapshot = async (snapshot: SnapshotExtendedType, onRestoreSuccess?: () => void | Promise<void>) => {
     if (!baseId.value) return
     try {
       isRestoringSnapshot.value = true
-      await $api.snapshot.restore(baseId.value, snapshot.id!)
+      const response = await $api.snapshot.restore(baseId.value, snapshot.id!)
 
       $poller.subscribe(
-        { id: snapshot.id! },
+        { id: response.id! },
         async (data: {
           id: string
           status?: string
@@ -147,7 +147,8 @@ export const useBaseSettings = createSharedComposable(() => {
         }) => {
           if (data.status === JobStatus.COMPLETED) {
             await loadProjects('workspace')
-            const base = bases.value.get(data.id)
+
+            const base = bases.value.get(data.data?.result.id)
 
             isRestoringSnapshot.value = false
 
@@ -161,6 +162,8 @@ export const useBaseSettings = createSharedComposable(() => {
             }
             message.info('Snapshot restored successfully')
 
+            onRestoreSuccess?.()
+
             refreshCommandPalette()
           } else if (data.status === JobStatus.FAILED) {
             message.error('Failed to restore snapshot')