feat: reduce events

toeverything · Oct 18, 2024 · 21d6099 · 21d6099
1 parent e9497c6
commit 21d6099
Show file tree

Hide file tree

Showing 6 changed files with 130 additions and 59 deletions.
diff --git a/packages/frontend/core/src/blocksuite/presets/ai/actions/doc-handler.ts b/packages/frontend/core/src/blocksuite/presets/ai/actions/doc-handler.ts
@@ -181,7 +181,10 @@ function updateAIPanelConfig<T extends keyof BlockSuitePresets.AIActions>(
   config.errorStateConfig = buildErrorConfig(aiPanel);
   config.copy = buildCopyConfig(aiPanel);
   config.discardCallback = () => {
-    getTracker(host).discardAction({ action: id });
+    getTracker(host).action_panel.discardAction({
+      action: id,
+      control: 'discard_button',
+    });
     reportResponse('result:discard');
   };
 }
@@ -202,7 +205,7 @@ export function actionToHandler<T extends keyof BlockSuitePresets.AIActions>(
     if (!blocks || blocks.length === 0) return;
     const block = blocks.at(-1);
     assertExists(block);
-    getTracker(host).startAction({ action: id });
+    getTracker(host).action_panel.invokeAction({ action: id });
     aiPanel.toggle(block, 'placeholder');
   };
 }

diff --git a/packages/frontend/core/src/blocksuite/presets/ai/actions/edgeless-handler.ts b/packages/frontend/core/src/blocksuite/presets/ai/actions/edgeless-handler.ts
@@ -1,4 +1,3 @@
-import track from '@affine/track';
 import type { EditorHost } from '@blocksuite/affine/block-std';
 import type {
   AffineAIPanelWidget,
@@ -41,6 +40,7 @@ import {
   getSelectedNoteAnchor,
   getSelections,
 } from '../utils/selection-utils';
+import { getTracker } from '../utils/track';
 import { EXCLUDING_COPY_ACTIONS, IMAGE_ACTIONS } from './consts';
 import { bindTextStream } from './doc-handler';
 import {
@@ -384,7 +384,10 @@ function updateEdgelessAIPanelConfig<
     },
   };
   config.discardCallback = () => {
-    track.copilot.edgeless.$.discardAction({ action: id });
+    getTracker(host).action_panel.discardAction({
+      action: id,
+      control: 'discard_button',
+    });
     reportResponse('result:discard');
   };
   config.hideCallback = () => {
@@ -495,7 +498,7 @@ export function actionToHandler<T extends keyof BlockSuitePresets.AIActions>(
 
     togglePanel()
       .then(isEmpty => {
-        track.copilot.edgeless.$.startAction({ action: id });
+        getTracker(host).action_panel.invokeAction({ action: id });
         aiPanel.toggle(referenceElement, isEmpty ? undefined : 'placeholder');
       })
       .catch(console.error);

diff --git a/packages/frontend/core/src/blocksuite/presets/ai/actions/edgeless-response.ts b/packages/frontend/core/src/blocksuite/presets/ai/actions/edgeless-response.ts
@@ -96,7 +96,10 @@ export function retry<T extends keyof BlockSuitePresets.AIActions>(
     name: 'Retry',
     icon: ResetIcon,
     handler: () => {
-      getTracker(panel.host).retryAction({ action: id });
+      getTracker(panel.host).action_panel.invokeAction({
+        action: id,
+        retry: true,
+      });
       reportResponse('result:retry');
       panel.generate();
     },
@@ -145,7 +148,10 @@ export function createInsertResp<T extends keyof BlockSuitePresets.AIActions>(
         );
       },
       handler: () => {
-        getTracker(host).finishAction({ action: id });
+        getTracker(host).action_panel.acceptAction({
+          action: id,
+          control: 'insert',
+        });
         reportResponse('result:insert');
         handler(host, ctx);
         const panel = getAIPanel(host);
@@ -167,7 +173,10 @@ export function asCaption<T extends keyof BlockSuitePresets.AIActions>(
       return id === 'generateCaption' && !!panel.answer;
     },
     handler: () => {
-      getTracker(host).finishAction({ action: id });
+      getTracker(host).action_panel.acceptAction({
+        action: id,
+        control: 'as_caption',
+      });
       reportResponse('result:use-as-caption');
       const panel = getAIPanel(host);
       const caption = panel.answer;
@@ -586,7 +595,10 @@ export function actionToResponse<T extends keyof BlockSuitePresets.AIActions>(
             name: 'Continue in chat',
             icon: ChatWithAIIcon,
             handler: () => {
-              getTracker(host).finishAction({ action: id });
+              getTracker(host).action_panel.acceptAction({
+                action: id,
+                control: 'continue_in_chat',
+              });
               reportResponse('result:continue-in-chat');
               const panel = getAIPanel(host);
               AIProvider.slots.requestOpenWithChat.emit({ host });
@@ -628,17 +640,26 @@ export function actionToErrorResponse<
 ): ErrorConfig {
   return {
     upgrade: () => {
-      getTracker(host).failureAction({ action: id });
+      getTracker(host).action_panel.discardAction({
+        action: id,
+        control: 'paywall',
+      });
       AIProvider.slots.requestUpgradePlan.emit({ host: panel.host });
       panel.hide();
     },
     login: () => {
-      getTracker(host).failureAction({ action: id });
+      getTracker(host).action_panel.discardAction({
+        action: id,
+        control: 'login_required',
+      });
       AIProvider.slots.requestLogin.emit({ host: panel.host });
       panel.hide();
     },
     cancel: () => {
-      getTracker(host).discardAction({ action: id });
+      getTracker(host).action_panel.discardAction({
+        action: id,
+        control: 'paywall',
+      });
       panel.hide();
     },
     responses: [

diff --git a/packages/frontend/core/src/blocksuite/presets/ai/ai-panel.ts b/packages/frontend/core/src/blocksuite/presets/ai/ai-panel.ts
@@ -66,7 +66,10 @@ function asCaption<T extends keyof BlockSuitePresets.AIActions>(
       return id === 'generateCaption' && !!panel.answer;
     },
     handler: () => {
-      getTracker(host).finishAction({ action: id });
+      getTracker(host).action_panel.acceptAction({
+        action: id,
+        control: 'as_caption',
+      });
       reportResponse('result:use-as-caption');
       const panel = getAIPanel(host);
       const caption = panel.answer;
@@ -96,7 +99,10 @@ function createNewNote<T extends keyof BlockSuitePresets.AIActions>(
       return !!panel.answer && isInsideEdgelessEditor(host);
     },
     handler: () => {
-      getTracker(host).finishAction({ action: id });
+      getTracker(host).action_panel.acceptAction({
+        action: id,
+        control: 'insert_note',
+      });
       reportResponse('result:add-note');
       // get the note block
       const { selectedBlocks } = getSelections(host);
@@ -202,7 +208,10 @@ export function buildTextResponseConfig<
           showWhen: () =>
             !!panel.answer && (!id || !INSERT_ABOVE_ACTIONS.includes(id)),
           handler: () => {
-            getTracker(host).finishAction({ action: id });
+            getTracker(host).action_panel.acceptAction({
+              action: id,
+              control: 'insert',
+            });
             reportResponse('result:insert');
             insertAnswerBelow(panel).catch(console.error);
           },
@@ -213,7 +222,10 @@ export function buildTextResponseConfig<
           showWhen: () =>
             !!panel.answer && !!id && INSERT_ABOVE_ACTIONS.includes(id),
           handler: () => {
-            getTracker(host).finishAction({ action: id });
+            getTracker(host).action_panel.acceptAction({
+              action: id,
+              control: 'insert',
+            });
             reportResponse('result:insert');
             insertAnswerAbove(panel).catch(console.error);
           },
@@ -224,7 +236,10 @@ export function buildTextResponseConfig<
           icon: ReplaceIcon,
           showWhen: () => !!panel.answer,
           handler: () => {
-            getTracker(host).finishAction({ action: id });
+            getTracker(host).action_panel.acceptAction({
+              action: id,
+              control: 'replace',
+            });
             reportResponse('result:replace');
             replaceWithAnswer(panel).catch(console.error);
           },
@@ -239,7 +254,10 @@ export function buildTextResponseConfig<
           name: 'Continue in chat',
           icon: ChatWithAIIcon,
           handler: () => {
-            getTracker(host).finishAction({ action: id });
+            getTracker(host).action_panel.acceptAction({
+              action: id,
+              control: 'continue_in_chat',
+            });
             reportResponse('result:continue-in-chat');
             AIProvider.slots.requestOpenWithChat.emit({ host });
             panel.hide();
@@ -249,7 +267,10 @@ export function buildTextResponseConfig<
           name: 'Regenerate',
           icon: RetryIcon,
           handler: () => {
-            getTracker(host).retryAction({ action: id });
+            getTracker(host).action_panel.invokeAction({
+              action: id,
+              retry: true,
+            });
             reportResponse('result:retry');
             panel.generate();
           },
@@ -280,7 +301,10 @@ export function buildErrorResponseConfig<
           icon: ReplaceIcon,
           showWhen: () => !!panel.answer,
           handler: () => {
-            getTracker(host).finishAction({ action: id });
+            getTracker(host).action_panel.acceptAction({
+              action: id,
+              control: 'replace',
+            });
             replaceWithAnswer(panel).catch(console.error);
           },
         },
@@ -290,7 +314,10 @@ export function buildErrorResponseConfig<
           showWhen: () =>
             !!panel.answer && (!id || !INSERT_ABOVE_ACTIONS.includes(id)),
           handler: () => {
-            getTracker(host).finishAction({ action: id });
+            getTracker(host).action_panel.acceptAction({
+              action: id,
+              control: 'insert',
+            });
             insertAnswerBelow(panel).catch(console.error);
           },
         },
@@ -300,7 +327,10 @@ export function buildErrorResponseConfig<
           showWhen: () =>
             !!panel.answer && !!id && INSERT_ABOVE_ACTIONS.includes(id),
           handler: () => {
-            getTracker(host).finishAction({ action: id });
+            getTracker(host).action_panel.acceptAction({
+              action: id,
+              control: 'insert',
+            });
             reportResponse('result:insert');
             insertAnswerAbove(panel).catch(console.error);
           },
@@ -317,7 +347,10 @@ export function buildErrorResponseConfig<
           icon: RetryIcon,
           showWhen: () => true,
           handler: () => {
-            getTracker(host).retryAction({ action: id });
+            getTracker(host).action_panel.invokeAction({
+              action: id,
+              retry: true,
+            });
             reportResponse('result:retry');
             panel.generate();
           },
@@ -353,17 +386,26 @@ export function buildErrorConfig<T extends keyof BlockSuitePresets.AIActions>(
 
   return {
     upgrade: () => {
-      getTracker(host).discardAction({ action: id });
+      getTracker(host).action_panel.discardAction({
+        action: id,
+        control: 'paywall',
+      });
       AIProvider.slots.requestUpgradePlan.emit({ host: panel.host });
       panel.hide();
     },
     login: () => {
-      getTracker(host).discardAction({ action: id });
+      getTracker(host).action_panel.discardAction({
+        action: id,
+        control: 'login_required',
+      });
       AIProvider.slots.requestLogin.emit({ host: panel.host });
       panel.hide();
     },
     cancel: () => {
-      getTracker(host).discardAction({ action: id });
+      getTracker(host).action_panel.discardAction({
+        action: id,
+        control: 'paywall',
+      });
       panel.hide();
     },
     responses: buildErrorResponseConfig(panel, id),

diff --git a/packages/frontend/core/src/blocksuite/presets/ai/utils/track.ts b/packages/frontend/core/src/blocksuite/presets/ai/utils/track.ts
@@ -3,5 +3,5 @@ import type { EditorHost } from '@blocksuite/affine/block-std';
 import { isInsideEdgelessEditor } from '@blocksuite/affine/blocks';
 
 export function getTracker(host: EditorHost) {
-  return track.copilot[isInsideEdgelessEditor(host) ? 'edgeless' : 'page'].$;
+  return track.copilot[isInsideEdgelessEditor(host) ? 'edgeless' : 'page'];
 }
diff --git a/packages/frontend/track/src/events.ts b/packages/frontend/track/src/events.ts
@@ -111,14 +111,11 @@ type PaymentEvents =
 type CopilotEvents =
   | 'startChat'
   | 'resetChat'
-  | 'retryChat'
-  | 'failureChat'
+  | 'abortChat'
   | 'addChatAttachment'
-  | 'startAction'
-  | 'retryAction'
+  | 'invokeAction'
   | 'discardAction'
-  | 'failureAction'
-  | 'finishAction';
+  | 'acceptAction';
 // END SECTION
 
 type UserEvents =
@@ -291,31 +288,17 @@ const PageEvents = {
   },
   copilot: {
     chat: {
-      $: [
-        'startChat',
-        'retryChat',
-        'resetChat',
-        'failureChat',
-        'addChatAttachment',
-      ],
+      $: ['startChat', 'abortChat', 'resetChat', 'addChatAttachment'],
     },
     page: {
-      $: [
-        'startAction',
-        'retryAction',
-        'discardAction',
-        'failureAction',
-        'finishAction',
-      ],
+      action_panel: ['invokeAction', 'discardAction', 'acceptAction'],
+      inline_panel: ['invokeAction', 'discardAction', 'acceptAction'],
+      chat: ['invokeAction', 'discardAction', 'acceptAction'],
     },
     edgeless: {
-      $: [
-        'startAction',
-        'retryAction',
-        'discardAction',
-        'failureAction',
-        'finishAction',
-      ],
+      action_panel: ['invokeAction', 'discardAction', 'acceptAction'],
+      inline_panel: ['invokeAction', 'discardAction', 'acceptAction'],
+      chat: ['invokeAction', 'discardAction', 'acceptAction'],
     },
   },
   // remove when type added
@@ -441,11 +424,30 @@ export type EventArgs = {
     type: string;
   };
   // copilot
-  startAction: { action: string };
-  retryAction: { action: string };
-  discardAction: { action: string };
-  failureAction: { action: string };
-  finishAction: { action: string };
+  invokeAction: {
+    action: string;
+    retry?: boolean;
+  };
+  discardAction: {
+    action: string;
+    control:
+      | 'stop_button'
+      | 'discard_button'
+      | 'paywall'
+      | 'backend_policy'
+      | 'backend_error'
+      | 'login_required'
+      | 'retry';
+  };
+  acceptAction: {
+    action: string;
+    control:
+      | 'insert'
+      | 'insert_note'
+      | 'replace'
+      | 'as_caption'
+      | 'continue_in_chat';
+  };
 };
 
 // for type checking