stashes 0.1.28 → 0.1.29

package/dist/cli.js

@@ -1537,7 +1537,10 @@ class StashService {
   selectedComponent = null;
   messageQueue = [];
   isProcessingMessage = false;
+  activeChatId = null;
   chatSessions = new Map;
+  stashPollTimer = null;
+  knownStashIds = new Set;
   constructor(projectPath, worktreeManager, persistence, broadcast) {
     this.projectPath = projectPath;
     this.worktreeManager = worktreeManager;
@@ -1545,6 +1548,9 @@ class StashService {
     this.broadcast = broadcast;
     this.previewPool = new PreviewPool(worktreeManager, broadcast);
   }
+  getActiveChatId() {
+    return this.activeChatId;
+  }
   setSelectedComponent(component) {
     this.selectedComponent = component;
     if (component.filePath === "auto-detect") {
@@ -1599,8 +1605,10 @@ class StashService {
     this.isProcessingMessage = true;
     while (this.messageQueue.length > 0) {
       const msg = this.messageQueue.shift();
+      this.activeChatId = msg.chatId;
       await this.processMessage(msg.projectId, msg.chatId, msg.message, msg.referenceStashIds, msg.componentContext);
     }
+    this.activeChatId = null;
     this.isProcessingMessage = false;
   }
   async processMessage(projectId, chatId, message, referenceStashIds, componentContext) {
@@ -1660,20 +1668,22 @@ ${sourceCode.substring(0, 3000)}
     const aiProcess = startAiProcess("chat", chatPrompt, this.projectPath, existingSessionId);
     let thinkingBuf = "";
     let textBuf = "";
-    const pendingMessages = [];
     const now = new Date().toISOString();
-    function flushThinking() {
+    const save = (msg) => {
+      this.persistence.saveChatMessage(projectId, chatId, msg);
+    };
+    const flushThinking = () => {
       if (!thinkingBuf)
         return;
-      pendingMessages.push({ id: crypto.randomUUID(), role: "assistant", content: thinkingBuf, type: "thinking", createdAt: now });
+      save({ id: crypto.randomUUID(), role: "assistant", content: thinkingBuf, type: "thinking", createdAt: now });
       thinkingBuf = "";
-    }
-    function flushText() {
+    };
+    const flushText = () => {
       if (!textBuf)
         return;
-      pendingMessages.push({ id: crypto.randomUUID(), role: "assistant", content: textBuf, type: "text", createdAt: now });
+      save({ id: crypto.randomUUID(), role: "assistant", content: textBuf, type: "text", createdAt: now });
       textBuf = "";
-    }
+    };
     try {
       for await (const chunk of parseClaudeStream(aiProcess.process)) {
         if (chunk.type === "session_id" && chunk.sessionId) {
@@ -1707,7 +1717,7 @@ ${sourceCode.substring(0, 3000)}
             toolName = parsed.tool ?? "unknown";
             toolParams = parsed.input ?? {};
           } catch {}
-          pendingMessages.push({
+          save({
             id: crypto.randomUUID(),
             role: "assistant",
             content: chunk.content,
@@ -1726,7 +1736,11 @@ ${sourceCode.substring(0, 3000)}
             toolParams,
             toolStatus: "running"
           });
+          if (toolName.includes("stashes_generate") || toolName.includes("stashes_vary")) {
+            this.startStashPoll(projectId, chatId);
+          }
         } else if (chunk.type === "tool_result") {
+          this.stopStashPoll();
           let toolResult = chunk.content;
           let isError = false;
           try {
@@ -1734,7 +1748,7 @@ ${sourceCode.substring(0, 3000)}
             toolResult = parsed.result ?? chunk.content;
             isError = !!parsed.is_error;
           } catch {}
-          pendingMessages.push({
+          save({
             id: crypto.randomUUID(),
             role: "assistant",
             content: chunk.content,
@@ -1756,10 +1770,7 @@ ${sourceCode.substring(0, 3000)}
       await aiProcess.process.exited;
       flushThinking();
       flushText();
-      for (const msg of pendingMessages) {
-        this.persistence.saveChatMessage(projectId, chatId, msg);
-      }
-      this.syncStashesFromDisk(projectId);
+      this.syncStashesFromDisk(projectId, chatId);
     } catch (err) {
       this.broadcast({
         type: "ai_stream",
@@ -1768,12 +1779,17 @@ ${sourceCode.substring(0, 3000)}
         source: "chat"
       });
     } finally {
+      this.stopStashPoll();
       killAiProcess("chat");
     }
   }
-  syncStashesFromDisk(projectId) {
+  syncStashesFromDisk(projectId, chatId) {
     const diskStashes = this.persistence.listStashes(projectId);
     for (const stash of diskStashes) {
+      if (chatId && !stash.originChatId && !this.knownStashIds.has(stash.id)) {
+        const updated = { ...stash, originChatId: chatId };
+        this.persistence.saveStash(updated);
+      }
       this.broadcast({
         type: "stash:status",
         stashId: stash.id,
@@ -1788,6 +1804,48 @@ ${sourceCode.substring(0, 3000)}
           screenshots: stash.screenshots?.length ? [...stash.screenshots] : undefined
         });
       }
+      this.knownStashIds.add(stash.id);
+    }
+  }
+  startStashPoll(projectId, chatId) {
+    this.stopStashPoll();
+    const lastStatus = new Map;
+    for (const s of this.persistence.listStashes(projectId)) {
+      this.knownStashIds.add(s.id);
+      lastStatus.set(s.id, s.status);
+    }
+    this.stashPollTimer = setInterval(() => {
+      const stashes = this.persistence.listStashes(projectId);
+      for (const stash of stashes) {
+        const prev = lastStatus.get(stash.id);
+        if (!prev || prev !== stash.status) {
+          if (chatId && !stash.originChatId && !prev) {
+            this.persistence.saveStash({ ...stash, originChatId: chatId });
+          }
+          this.broadcast({
+            type: "stash:status",
+            stashId: stash.id,
+            status: stash.status,
+            number: stash.number
+          });
+          if (stash.screenshotUrl && prev !== stash.status) {
+            this.broadcast({
+              type: "stash:screenshot",
+              stashId: stash.id,
+              url: stash.screenshotUrl,
+              screenshots: stash.screenshots?.length ? [...stash.screenshots] : undefined
+            });
+          }
+          lastStatus.set(stash.id, stash.status);
+          this.knownStashIds.add(stash.id);
+        }
+      }
+    }, 2000);
+  }
+  stopStashPoll() {
+    if (this.stashPollTimer) {
+      clearInterval(this.stashPollTimer);
+      this.stashPollTimer = null;
     }
   }
   progressToBroadcast(event) {
@@ -1914,6 +1972,10 @@ function createWebSocketHandler(projectPath, userDevPort, appProxyPort) {
         projectId: project.id,
         projectName: project.name
       }));
+      const activeChatId = stashService.getActiveChatId();
+      if (activeChatId) {
+        ws.send(JSON.stringify({ type: "processing", chatId: activeChatId }));
+      }
     },
     async message(ws, message) {
       const raw = typeof message === "string" ? message : new TextDecoder().decode(message);

package/dist/mcp.js

@@ -1733,7 +1733,10 @@ class StashService {
   selectedComponent = null;
   messageQueue = [];
   isProcessingMessage = false;
+  activeChatId = null;
   chatSessions = new Map;
+  stashPollTimer = null;
+  knownStashIds = new Set;
   constructor(projectPath, worktreeManager, persistence, broadcast) {
     this.projectPath = projectPath;
     this.worktreeManager = worktreeManager;
@@ -1741,6 +1744,9 @@ class StashService {
     this.broadcast = broadcast;
     this.previewPool = new PreviewPool(worktreeManager, broadcast);
   }
+  getActiveChatId() {
+    return this.activeChatId;
+  }
   setSelectedComponent(component) {
     this.selectedComponent = component;
     if (component.filePath === "auto-detect") {
@@ -1795,8 +1801,10 @@ class StashService {
     this.isProcessingMessage = true;
     while (this.messageQueue.length > 0) {
       const msg = this.messageQueue.shift();
+      this.activeChatId = msg.chatId;
       await this.processMessage(msg.projectId, msg.chatId, msg.message, msg.referenceStashIds, msg.componentContext);
     }
+    this.activeChatId = null;
     this.isProcessingMessage = false;
   }
   async processMessage(projectId, chatId, message, referenceStashIds, componentContext) {
@@ -1856,20 +1864,22 @@ ${sourceCode.substring(0, 3000)}
     const aiProcess = startAiProcess("chat", chatPrompt, this.projectPath, existingSessionId);
     let thinkingBuf = "";
     let textBuf = "";
-    const pendingMessages = [];
     const now = new Date().toISOString();
-    function flushThinking() {
+    const save = (msg) => {
+      this.persistence.saveChatMessage(projectId, chatId, msg);
+    };
+    const flushThinking = () => {
       if (!thinkingBuf)
         return;
-      pendingMessages.push({ id: crypto.randomUUID(), role: "assistant", content: thinkingBuf, type: "thinking", createdAt: now });
+      save({ id: crypto.randomUUID(), role: "assistant", content: thinkingBuf, type: "thinking", createdAt: now });
       thinkingBuf = "";
-    }
-    function flushText() {
+    };
+    const flushText = () => {
       if (!textBuf)
         return;
-      pendingMessages.push({ id: crypto.randomUUID(), role: "assistant", content: textBuf, type: "text", createdAt: now });
+      save({ id: crypto.randomUUID(), role: "assistant", content: textBuf, type: "text", createdAt: now });
       textBuf = "";
-    }
+    };
     try {
       for await (const chunk of parseClaudeStream(aiProcess.process)) {
         if (chunk.type === "session_id" && chunk.sessionId) {
@@ -1903,7 +1913,7 @@ ${sourceCode.substring(0, 3000)}
             toolName = parsed.tool ?? "unknown";
             toolParams = parsed.input ?? {};
           } catch {}
-          pendingMessages.push({
+          save({
             id: crypto.randomUUID(),
             role: "assistant",
             content: chunk.content,
@@ -1922,7 +1932,11 @@ ${sourceCode.substring(0, 3000)}
             toolParams,
             toolStatus: "running"
           });
+          if (toolName.includes("stashes_generate") || toolName.includes("stashes_vary")) {
+            this.startStashPoll(projectId, chatId);
+          }
         } else if (chunk.type === "tool_result") {
+          this.stopStashPoll();
           let toolResult = chunk.content;
           let isError = false;
           try {
@@ -1930,7 +1944,7 @@ ${sourceCode.substring(0, 3000)}
             toolResult = parsed.result ?? chunk.content;
             isError = !!parsed.is_error;
           } catch {}
-          pendingMessages.push({
+          save({
             id: crypto.randomUUID(),
             role: "assistant",
             content: chunk.content,
@@ -1952,10 +1966,7 @@ ${sourceCode.substring(0, 3000)}
       await aiProcess.process.exited;
       flushThinking();
       flushText();
-      for (const msg of pendingMessages) {
-        this.persistence.saveChatMessage(projectId, chatId, msg);
-      }
-      this.syncStashesFromDisk(projectId);
+      this.syncStashesFromDisk(projectId, chatId);
     } catch (err) {
       this.broadcast({
         type: "ai_stream",
@@ -1964,12 +1975,17 @@ ${sourceCode.substring(0, 3000)}
         source: "chat"
       });
     } finally {
+      this.stopStashPoll();
       killAiProcess("chat");
     }
   }
-  syncStashesFromDisk(projectId) {
+  syncStashesFromDisk(projectId, chatId) {
     const diskStashes = this.persistence.listStashes(projectId);
     for (const stash of diskStashes) {
+      if (chatId && !stash.originChatId && !this.knownStashIds.has(stash.id)) {
+        const updated = { ...stash, originChatId: chatId };
+        this.persistence.saveStash(updated);
+      }
       this.broadcast({
         type: "stash:status",
         stashId: stash.id,
@@ -1984,6 +2000,48 @@ ${sourceCode.substring(0, 3000)}
           screenshots: stash.screenshots?.length ? [...stash.screenshots] : undefined
         });
       }
+      this.knownStashIds.add(stash.id);
+    }
+  }
+  startStashPoll(projectId, chatId) {
+    this.stopStashPoll();
+    const lastStatus = new Map;
+    for (const s of this.persistence.listStashes(projectId)) {
+      this.knownStashIds.add(s.id);
+      lastStatus.set(s.id, s.status);
+    }
+    this.stashPollTimer = setInterval(() => {
+      const stashes = this.persistence.listStashes(projectId);
+      for (const stash of stashes) {
+        const prev = lastStatus.get(stash.id);
+        if (!prev || prev !== stash.status) {
+          if (chatId && !stash.originChatId && !prev) {
+            this.persistence.saveStash({ ...stash, originChatId: chatId });
+          }
+          this.broadcast({
+            type: "stash:status",
+            stashId: stash.id,
+            status: stash.status,
+            number: stash.number
+          });
+          if (stash.screenshotUrl && prev !== stash.status) {
+            this.broadcast({
+              type: "stash:screenshot",
+              stashId: stash.id,
+              url: stash.screenshotUrl,
+              screenshots: stash.screenshots?.length ? [...stash.screenshots] : undefined
+            });
+          }
+          lastStatus.set(stash.id, stash.status);
+          this.knownStashIds.add(stash.id);
+        }
+      }
+    }, 2000);
+  }
+  stopStashPoll() {
+    if (this.stashPollTimer) {
+      clearInterval(this.stashPollTimer);
+      this.stashPollTimer = null;
     }
   }
   progressToBroadcast(event) {
@@ -2110,6 +2168,10 @@ function createWebSocketHandler(projectPath, userDevPort, appProxyPort) {
         projectId: project.id,
         projectName: project.name
       }));
+      const activeChatId = stashService.getActiveChatId();
+      if (activeChatId) {
+        ws.send(JSON.stringify({ type: "processing", chatId: activeChatId }));
+      }
     },
     async message(ws, message) {
       const raw = typeof message === "string" ? message : new TextDecoder().decode(message);