feat: support custom behavior after harbor upload

packages/page-spy-plugin-data-harbor/src/index.ts

@@ -46,7 +46,10 @@ interface DataHarborConfig {
   filename?: () => string;
 
   // Custom download behavior.
-  onDownload?: ((data: CacheMessageItem[]) => void) | null;
+  onDownload?: (data: CacheMessageItem[]) => void;
+
+  // Custom behavior after upload.
+  onAfterUpload?: (replayUrl: string, remark: string) => void;
 }
 
 const defaultConfig: Required<DataHarborConfig> = {
@@ -61,7 +64,8 @@ const defaultConfig: Required<DataHarborConfig> = {
   filename: () => {
     return new Date().toLocaleString();
   },
-  onDownload: null,
+  onDownload: () => {},
+  onAfterUpload: () => {},
 };
 
 export default class DataHarborPlugin implements PageSpyPlugin {
@@ -263,29 +267,24 @@ export default class DataHarborPlugin implements PageSpyPlugin {
       remark: '',
     },
   ): Promise<void | string> {
-    let result;
-    if (type === 'upload' || type === 'download') {
-      const args: any = await this.getParams(type as any, params);
-      result =
-        type === 'upload' ? await startUpload(args) : await startDownload(args);
-
-      if ((params as WholeActionParams).clearCache === true) {
-        this.clearAndNotify();
-      }
+    if (!isPeriodActionParams(params) || params.startTime > params.endTime) {
+      throw new Error(t.invalidParams);
     }
-    if (type === 'upload-periods' || type === 'download-periods') {
-      if (!isPeriodActionParams(params) || params.startTime > params.endTime) {
-        throw new Error(t.invalidParams);
-      }
-      const args: any = await this.getParams(type as any, params);
-      result =
-        type === 'upload-periods'
-          ? await startUpload(args)
-          : await startDownload(args);
+
+    const args: any = await this.getParams(type as any, params);
+    const isUpload = ['upload', 'upload-periods'].includes(type);
+    const result = isUpload
+      ? await startUpload(args)
+      : await startDownload(args);
+
+    if ((params as WholeActionParams).clearCache === true) {
+      this.clearAndNotify();
     }
 
     if (result) {
-      return this.getDebugUrl(result);
+      const url = this.getDebugUrl(result);
+      this.$harborConfig.onAfterUpload(url, args.remark);
+      return url;
     }
   }
 

packages/page-spy-plugin-mp-data-harbor/src/index.ts

@@ -1,16 +1,11 @@
 /* eslint-disable consistent-return */
 import type {
   SpyMessage,
-  OnInitParams,
   PageSpyPlugin,
   PluginOrder,
   InitConfigBase,
 } from '@huolala-tech/page-spy-types';
 import { removeEndSlash } from '@huolala-tech/page-spy-base/dist/utils';
-import { MemoryHarbor } from './harbor/memoryHarbor';
-import { startUpload } from './utils/upload';
-import { buildSearchParams, getDeviceId, makeData, makeFile } from './utils';
-import { DataType, WholeActionParams } from './harbor/base';
 import {
   getMPSDK,
   type Client,
@@ -19,6 +14,10 @@ import {
   MPPluginInitParams,
   setMPSDK,
 } from '@huolala-tech/page-spy-mp-base';
+import { MemoryHarbor } from './harbor/memoryHarbor';
+import { startUpload } from './utils/upload';
+import { buildSearchParams, getDeviceId, makeData, makeFile } from './utils';
+import { DataType, WholeActionParams } from './harbor/base';
 
 interface DataHarborConfig {
   // Specify which types of data to collect.
@@ -27,6 +26,9 @@ interface DataHarborConfig {
   // Custom uploaded filename by this.
   // Default value is `new Date().toLocaleString()`.
   filename?: () => string;
+
+  // Custom behavior after upload
+  onAfterUpload?: (replayUrl: string) => void;
 }
 
 const defaultConfig: DataHarborConfig = {
@@ -40,6 +42,7 @@ const defaultConfig: DataHarborConfig = {
   filename: () => {
     return new Date().toLocaleString();
   },
+  onAfterUpload: () => {},
 };
 
 export default class MPDataHarborPlugin implements PageSpyPlugin {
@@ -120,6 +123,8 @@ export default class MPDataHarborPlugin implements PageSpyPlugin {
             const result = await this.upload();
             mp.hideLoading();
             if (result) {
+              this.$harborConfig.onAfterUpload(result);
+
               if (mp.setClipboardData) {
                 mp.showModal({
                   title: '上传成功',