From c8bc905e7ee70ea817387bd40d4edf8012eb2945 Mon Sep 17 00:00:00 2001
From: yy-wow <yaoyun_bigger@outlook.com>
Date: Wed, 9 Oct 2024 18:55:42 -0700
Subject: [PATCH] feat: add hooks

---
 packages/toolbars/preview/src/Main.vue | 32 ++++++++++++++++++++++++--
 packages/toolbars/save/src/Main.vue    |  1 +
 packages/toolbars/save/src/js/index.js | 29 +++++++++++++++++++++++
 3 files changed, 60 insertions(+), 2 deletions(-)

diff --git a/packages/toolbars/preview/src/Main.vue b/packages/toolbars/preview/src/Main.vue
index df495f2f1..29cec14c0 100644
--- a/packages/toolbars/preview/src/Main.vue
+++ b/packages/toolbars/preview/src/Main.vue
@@ -13,7 +13,8 @@
 <script>
 import { previewPage, previewBlock } from '@opentiny/tiny-engine-common/js/preview'
 import { useBlock, useCanvas, useLayout, useNotify } from '@opentiny/tiny-engine-meta-register'
-import { getMergeMeta } from '@opentiny/tiny-engine-meta-register'
+import { getMergeMeta, getOptions } from '@opentiny/tiny-engine-meta-register'
+import meta from '../meta'
 import { ToolbarBase } from '@opentiny/tiny-engine-common'
 
 export default {
@@ -30,7 +31,26 @@ export default {
     const { isBlock, getCurrentPage, canvasApi } = useCanvas()
     const { getCurrentBlock } = useBlock()
 
-    const preview = () => {
+    const preview = async () => {
+      const { beforePreview, previewMethod, afterPreview } = getOptions(meta.id)
+      const printer = console
+
+      try {
+        if (typeof beforePreview === 'function') {
+          await beforePreview()
+        }
+
+        if (typeof previewMethod === 'function') {
+          const stop = await previewMethod()
+
+          if (stop) {
+            return
+          }
+        }
+      } catch (error) {
+        printer.log(error)
+      }
+
       if (useLayout().isEmptyPage()) {
         useNotify({
           type: 'warning',
@@ -59,6 +79,14 @@ export default {
         params.pageInfo.name = page?.name
         previewPage(params)
       }
+
+      if (typeof afterPreview === 'function') {
+        try {
+          await afterPreview()
+        } catch (error) {
+          printer.log('Error in afterPreview:', error)
+        }
+      }
     }
 
     return {
diff --git a/packages/toolbars/save/src/Main.vue b/packages/toolbars/save/src/Main.vue
index 6ffb464ae..8a6b1a142 100644
--- a/packages/toolbars/save/src/Main.vue
+++ b/packages/toolbars/save/src/Main.vue
@@ -65,6 +65,7 @@ import { useCanvas } from '@opentiny/tiny-engine-meta-register'
 import { ToolbarBase } from '@opentiny/tiny-engine-common'
 import { openCommon, saveCommon } from './js/index'
 import { isLoading } from './js/index'
+
 export const api = {
   saveCommon,
   openCommon
diff --git a/packages/toolbars/save/src/js/index.js b/packages/toolbars/save/src/js/index.js
index 7b951656b..e7eb97a54 100644
--- a/packages/toolbars/save/src/js/index.js
+++ b/packages/toolbars/save/src/js/index.js
@@ -17,11 +17,13 @@ import {
   useLayout,
   useNotify,
   usePage,
+  getOptions,
   getMetaApi,
   META_APP
 } from '@opentiny/tiny-engine-meta-register'
 import { constants } from '@opentiny/tiny-engine-utils'
 import { handlePageUpdate } from '@opentiny/tiny-engine-common/js/http'
+import meta from '../../meta'
 
 const { PAGE_STATUS } = constants
 const state = reactive({
@@ -95,6 +97,25 @@ export const openCommon = async () => {
     return
   }
 
+  const { beforeSave, saveMethod, saved } = getOptions(meta.id)
+  const printer = console
+
+  try {
+    if (typeof beforeSave === 'function') {
+      await beforeSave()
+    }
+
+    if (typeof saveMethod === 'function') {
+      const stop = await saveMethod()
+
+      if (stop) {
+        return
+      }
+    }
+  } catch (error) {
+    printer.log(error)
+  }
+
   const pageStatus = useLayout().layoutState?.pageStatus
   const curPageState = pageStatus?.state
   const pageInfo = pageStatus?.data
@@ -138,5 +159,13 @@ export const openCommon = async () => {
 
   saveCommon(state.code).finally(() => {
     state.disabled = false
+
+    if (typeof saved === 'function') {
+      try {
+        saved()
+      } catch (error) {
+        printer.log(error)
+      }
+    }
   })
 }