finish toast part remove let chatbar handle polling

Author: pengyu

frontend/src/components/chat/code-engine/code-engine.tsx

@@ -46,7 +46,15 @@ export function CodeEngine({
   const [isCompleting, setIsCompleting] = useState(false);
   // 添加一个ref来持久跟踪项目状态，避免重新渲染时丢失
   const isProjectLoadedRef = useRef(false);
+  const context = useContext(ProjectContext);
+  if (!context) throw new Error('Must be used inside ProjectProvider');
+  const { setRecentlyCompletedProjectId } = context;
 
+  useEffect(() => {
+    if (projectCompleted) {
+      setRecentlyCompletedProjectId(curProject?.id || localProject?.id);
+    }
+  }, [projectCompleted]);
   // 在组件挂载时从localStorage检查项目是否已完成
   useEffect(() => {
     try {

frontend/src/components/chat/code-engine/project-context.tsx

@@ -50,6 +50,10 @@ export interface ProjectContextType {
   takeProjectScreenshot: (projectId: string, url: string) => Promise<void>;
   refreshProjects: () => Promise<void>;
   editorRef?: React.MutableRefObject<any>;
+  recentlyCompletedProjectId: string | null;
+  setRecentlyCompletedProjectId: (id: string | null) => void;
+  chatId: string | null;
+  setChatId: (chatId: string | null) => void;
 }
 
 export const ProjectContext = createContext<ProjectContextType | undefined>(
@@ -105,7 +109,10 @@ export function ProjectProvider({ children }: { children: ReactNode }) {
   const [filePath, setFilePath] = useState<string | null>(null);
   const [isLoading, setIsLoading] = useState<boolean>(false);
   const editorRef = useRef<any>(null);
-
+  const [recentlyCompletedProjectId, setRecentlyCompletedProjectId] = useState<
+    string | null
+  >(null);
+  const [chatId, setChatId] = useState<string | null>(null);
   interface ChatProjectCacheEntry {
     project: Project | null;
     timestamp: number;
@@ -142,6 +149,24 @@ export function ProjectProvider({ children }: { children: ReactNode }) {
       pendingOperations.current.clear();
     };
   }, []);
+  // Poll project data every 5 seconds
+  useEffect(() => {
+    if (!chatId) return;
+    let stopped = false;
+
+    const interval = setInterval(async () => {
+      const project = await pollChatProject(chatId);
+      if (project?.projectPath) {
+        setCurProject(project);
+        clearInterval(interval);
+        stopped = true;
+      }
+    }, 5000);
+
+    return () => {
+      if (!stopped) clearInterval(interval);
+    };
+  }, [chatId]);
 
   // Function to clean expired cache entries
   const cleanCache = useCallback(() => {
@@ -914,6 +939,8 @@ export function ProjectProvider({ children }: { children: ReactNode }) {
       takeProjectScreenshot,
       refreshProjects,
       editorRef,
+      chatId,
+      setChatId,
     }),
     [
       projects,
@@ -930,11 +957,19 @@ export function ProjectProvider({ children }: { children: ReactNode }) {
       takeProjectScreenshot,
       refreshProjects,
       editorRef,
+      chatId,
+      setChatId,
     ]
   );
 
   return (
-    <ProjectContext.Provider value={contextValue}>
+    <ProjectContext.Provider
+      value={{
+        ...contextValue,
+        recentlyCompletedProjectId,
+        setRecentlyCompletedProjectId,
+      }}
+    >
       {children}
     </ProjectContext.Provider>
   );

frontend/src/components/sidebar.tsx

@@ -96,20 +96,17 @@ function ChatSideBarComponent({
 }: SidebarProps) {
   const router = useRouter();
   const [currentChatid, setCurrentChatid] = useState('');
-  const { setCurProject, pollChatProject } = useContext(ProjectContext);
+  const { setChatId } = useContext(ProjectContext);
 
   const handleChatSelect = useCallback(
     (chatId: string) => {
       setCurrentChatid(chatId);
       router.push(`/chat?id=${chatId}`);
-      setCurProject(null);
-      pollChatProject(chatId).then((p) => {
-        setCurProject(p);
-      });
+      setChatId(chatId);
       const event = new Event(EventEnum.CHAT);
       window.dispatchEvent(event);
     },
-    [router, setCurProject, pollChatProject]
+    [router]
   );
 
   if (loading) return <SidebarSkeleton />;

frontend/src/providers/BaseProvider.tsx

@@ -4,11 +4,8 @@ import dynamic from 'next/dynamic';
 import { ThemeProvider } from 'next-themes';
 import { Toaster } from 'sonner';
 import { AuthProvider } from './AuthProvider';
-import {
-  ProjectContext,
-  ProjectProvider,
-} from '@/components/chat/code-engine/project-context';
-
+import { ProjectProvider } from '@/components/chat/code-engine/project-context';
+import GlobalToastListener from '@/components/global-toast-listener';
 const DynamicApolloProvider = dynamic(() => import('./DynamicApolloProvider'), {
   ssr: false, // disables SSR for the ApolloProvider
 });
@@ -23,6 +20,7 @@ export function BaseProviders({ children }: ProvidersProps) {
       <DynamicApolloProvider>
         <AuthProvider>
           <ProjectProvider>
+            <GlobalToastListener />
             {children}
             <Toaster position="bottom-right" />
           </ProjectProvider>