feat(frontend): support tracking chat's creation for widget (#591)

pingcap · Jan 15, 2025 · f70022e · f70022e
1 parent ab557b2
commit f70022e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/frontend/app/src/components/chat/chat-hooks.tsx b/frontend/app/src/components/chat/chat-hooks.tsx
@@ -30,7 +30,7 @@ const ChatsContext = createContext<ChatsProviderValues>({
 
 const ChatControllerContext = createContext<ChatController | null>(null);
 
-export function ChatsProvider ({ children }: { children: ReactNode }) {
+export function ChatsProvider ({ onChatCreated, children }: { children: ReactNode, onChatCreated?: (id: string, chat: Chat, controller: ChatController) => void }) {
   const bootstrapStatusRef = useLatestRef(useBootstrapStatus());
   const [chats, setChats] = useState(() => new Map<string, ChatController>);
 
@@ -42,6 +42,7 @@ export function ChatsProvider ({ children }: { children: ReactNode }) {
     const controller = new ChatController(...args);
     controller.once('created', (chat) => {
       setChats(chats => new Map(chats).set(chat.id, controller));
+      onChatCreated?.(chat.id, chat, controller);
     });
 
     return controller;

diff --git a/frontend/packages/widget-react/src/Widget.tsx b/frontend/packages/widget-react/src/Widget.tsx
@@ -131,7 +131,13 @@ export const Widget = forwardRef<WidgetInstance, WidgetProps>(({ container, trig
     <PortalProvider container={container}>
       <BootstrapStatusProvider bootstrapStatus={bootstrapStatus}>
         <ExperimentalFeaturesProvider features={experimentalFeatures}>
-          <ChatsProvider>
+          <ChatsProvider
+            onChatCreated={id => {
+              window.dispatchEvent(new CustomEvent('tidbainewchat', {
+                detail: { id },
+              }));
+            }}
+          >
             <Dialog open={open} onOpenChange={(open) => {
               setOpen(open);
               if (!open) {