refactor: persist script executions to history

Change-Id: I4139072998a7b9d9130442e98ec968e960895e7b
Signed-off-by: Thomas Kosiewski <tk@coder.com>

Author: ThomasK33

.cmux/scripts/web_fetch

@@ -0,0 +1,13 @@
+#!/usr/bin/env bash
+# Description: Fetch a URL via curl and print the response body
+set -euo pipefail
+
+if [[ $# -ne 1 ]]; then
+  echo "Usage: /script web_fetch <url>" >&2
+  exit 1
+fi
+
+url="$1"
+
+echo "[web_fetch] Fetching: $url" >&2
+curl --fail --show-error --location "$url"

src/browser/components/ChatInput/index.tsx

@@ -722,14 +722,6 @@ export const ChatInput: React.FC<ChatInputProps> = (props) => {
             const toolResult = result.data;
             const exitCode = toolResult.exitCode;
 
-            workspaceStore.recordScriptExecution(currentWorkspaceId, {
-              rawCommand: messageText,
-              scriptName: parsed.scriptName,
-              args: parsed.args,
-              result: toolResult,
-              timestamp: Date.now(),
-            });
-
             // Use MUX_OUTPUT content if present, otherwise fall back to default message
             const toastMessage =
               toolResult.outputFile ??

src/browser/components/Messages/ScriptExecutionMessage.tsx

@@ -50,7 +50,7 @@ export const ScriptExecutionMessage: React.FC<ScriptExecutionMessageProps> = ({
         <span className="font-monospace max-w-96 truncate">
           {message.command || `/script ${message.scriptName}${argsPreview}`}
         </span>
-        <span className="text-[10px] text-foreground-secondary ml-2 whitespace-nowrap">
+        <span className="text-foreground-secondary ml-2 text-[10px] whitespace-nowrap">
           took {formatDuration(result.wall_duration_ms)}
         </span>
         <span className={exitBadgeClass}>exit {result.exitCode}</span>
@@ -66,11 +66,11 @@ export const ScriptExecutionMessage: React.FC<ScriptExecutionMessageProps> = ({
 
           <DetailSection>
             <DetailLabel>Runtime info</DetailLabel>
-            <div className="text-[11px] text-foreground-secondary">
+            <div className="text-foreground-secondary text-[11px]">
               {new Date(message.timestamp).toLocaleString()} •{" "}
               {formatDuration(result.wall_duration_ms)}
             </div>
-            <div className="text-[11px] text-foreground-secondary">
+            <div className="text-foreground-secondary text-[11px]">
               Visible only to you; never sent to the model.
             </div>
           </DetailSection>
@@ -108,7 +108,7 @@ export const ScriptExecutionMessage: React.FC<ScriptExecutionMessageProps> = ({
           {result.truncated && (
             <DetailSection>
               <DetailLabel>Truncation</DetailLabel>
-              <div className="text-[11px] text-foreground-secondary">
+              <div className="text-foreground-secondary text-[11px]">
                 Output truncated: {result.truncated.reason} ({result.truncated.totalLines} lines
                 preserved)
               </div>

src/browser/stores/WorkspaceStore.ts

@@ -1,13 +1,15 @@
 import assert from "@/common/utils/assert";
-import type { MuxMessage, DisplayedMessage, QueuedMessage } from "@/common/types/message";
+import type {
+  MuxMessage,
+  DisplayedMessage,
+  QueuedMessage,
+  MuxFrontendMetadata,
+} from "@/common/types/message";
 import { createMuxMessage } from "@/common/types/message";
 import type { FrontendWorkspaceMetadata } from "@/common/types/workspace";
 import type { WorkspaceChatMessage } from "@/common/types/ipc";
 import type { TodoItem } from "@/common/types/tools";
-import {
-  StreamingMessageAggregator,
-  type ScriptExecutionEventInput,
-} from "@/browser/utils/messages/StreamingMessageAggregator";
+import { StreamingMessageAggregator } from "@/browser/utils/messages/StreamingMessageAggregator";
 import { updatePersistedState } from "@/browser/hooks/usePersistedState";
 import { getRetryStateKey } from "@/common/constants/storage";
 import { CUSTOM_EVENTS, createCustomEvent } from "@/common/constants/events";
@@ -433,17 +435,6 @@ export class WorkspaceStore {
    * Get current TODO list for a workspace.
    * Returns empty array if workspace doesn't exist or has no TODOs.
    */
-  recordScriptExecution(workspaceId: string, details: ScriptExecutionEventInput): void {
-    const aggregator = this.aggregators.get(workspaceId);
-    if (!aggregator) {
-      return;
-    }
-
-    aggregator.addScriptEvent(details);
-    this.states.bump(workspaceId);
-    this.checkAndBumpRecencyIfChanged();
-  }
-
   getTodos(workspaceId: string): TodoItem[] {
     const aggregator = this.aggregators.get(workspaceId);
     return aggregator ? aggregator.getCurrentTodos() : [];
@@ -714,7 +705,7 @@ export class WorkspaceStore {
         // Store continueMessage in summary so it survives history replacement
         muxMetadata: continueMessage
           ? { type: "compaction-result", continueMessage, requestId: compactRequestMsg?.id }
-          : { type: "normal" },
+          : ({ type: "normal" } as unknown as MuxFrontendMetadata),
       }
     );
 

src/browser/utils/messages/StreamingMessageAggregator.test.ts

@@ -392,48 +392,57 @@ describe("StreamingMessageAggregator", () => {
   describe("script execution events", () => {
     test("adds script logs to displayed messages", () => {
       const aggregator = new StreamingMessageAggregator(TEST_CREATED_AT);
-      const userMessage = createMuxMessage("user-1", "user", "run script", {
-        historySequence: 1,
-        timestamp: Date.now(),
-      });
-      aggregator.addMessage(userMessage);
 
+      // Create a persisted message with script metadata
       const timestamp = Date.now();
-      aggregator.addScriptEvent({
-        rawCommand: "/script demo",
-        scriptName: "demo",
-        args: ["--flag"],
-        result: BASE_SCRIPT_RESULT,
+      const scriptMessage = createMuxMessage("script-1", "user", "Run script", {
+        historySequence: 1,
         timestamp,
+        muxMetadata: {
+          type: "script-execution",
+          id: "script-exec-1",
+          historySequence: 1,
+          timestamp,
+          command: "/script demo",
+          scriptName: "demo",
+          args: ["--flag"],
+          result: BASE_SCRIPT_RESULT,
+        }
       });
+      
+      aggregator.addMessage(scriptMessage);
 
       const displayed = aggregator.getDisplayedMessages();
       const scriptMsg = displayed.find((msg) => msg.type === "script-execution");
       expect(scriptMsg).toBeDefined();
       if (scriptMsg?.type === "script-execution") {
-        expect(scriptMsg.historySequence).toBeGreaterThan(1);
+        expect(scriptMsg.historySequence).toBe(1);
         expect(scriptMsg.timestamp).toBe(timestamp);
         expect(scriptMsg.result).toBe(BASE_SCRIPT_RESULT);
       }
     });
 
     test("removes script logs when history is truncated", () => {
       const aggregator = new StreamingMessageAggregator(TEST_CREATED_AT);
-      aggregator.addMessage(
-        createMuxMessage("user-1", "user", "first", {
+      
+      const timestamp = Date.now();
+      const scriptMessage = createMuxMessage("script-1", "user", "Run script", {
+        historySequence: 1,
+        timestamp,
+        muxMetadata: {
+          type: "script-execution",
+          id: "script-exec-1",
           historySequence: 1,
-          timestamp: Date.now(),
-        })
-      );
-
-      aggregator.addScriptEvent({
-        rawCommand: "/script cleanup",
-        scriptName: "cleanup",
-        args: [],
-        result: BASE_SCRIPT_RESULT,
-        timestamp: Date.now(),
+          timestamp,
+          command: "/script cleanup",
+          scriptName: "cleanup",
+          args: [],
+          result: BASE_SCRIPT_RESULT,
+        }
       });
 
+      aggregator.addMessage(scriptMessage);
+
       const deleteEvent: DeleteMessage = { type: "delete", historySequences: [1] };
       aggregator.handleDeleteMessage(deleteEvent);