wave-code 0.17.1 → 0.17.2

package/src/contexts/useChat.tsx

@@ -18,6 +18,7 @@ import type {
   PermissionDecision,
   PermissionMode,
   QueuedMessage,
+  WorkflowRun,
 } from "wave-agent-sdk";
 import {
   Agent,
@@ -53,6 +54,8 @@ export interface ChatContextType {
   ) => Promise<void>;
   askBtw: (question: string) => Promise<string>;
   abortMessage: () => void;
+  recallQueuedMessage: () => QueuedMessage | null;
+  removeQueuedMessageById: (id: string) => boolean;
   latestTotalTokens: number;
   maxInputTokens: number;
   // Model functionality
@@ -66,6 +69,9 @@ export interface ChatContextType {
   disconnectMcpServer: (serverName: string) => Promise<boolean>;
   // Background tasks
   backgroundTasks: BackgroundTask[];
+  // Workflow runs
+  workflowRuns: WorkflowRun[];
+  stopWorkflowRun: (runId: string) => void;
   // Tasks
   tasks: Task[];
   getBackgroundTaskOutput: (taskId: string) => {
@@ -122,6 +128,10 @@ export interface ChatContextType {
   recreateAgent: () => void;
   // Trigger WorktreeRemove hook BEFORE agent destruction
   triggerWorktreeRemoveHook: (worktreePath: string) => Promise<void>;
+  // Goal state
+  isGoalActive: boolean;
+  goalElapsed?: string;
+  isGoalEvaluating: boolean;
 }
 
 const ChatContext = createContext<ChatContextType | null>(null);
@@ -201,6 +211,28 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   const [currentModel, setCurrentModelState] = useState("");
   const [configuredModels, setConfiguredModels] = useState<string[]>([]);
   const [queuedMessages, setQueuedMessages] = useState<QueuedMessage[]>([]);
+  const [isGoalActive, setIsGoalActive] = useState(false);
+  const [goalElapsed, setGoalElapsed] = useState<string | undefined>();
+  const [isGoalEvaluating, setIsGoalEvaluating] = useState(false);
+  const goalStartedAt = useRef<number | null>(null);
+
+  // Update goal elapsed time every 30s while active
+  useEffect(() => {
+    if (!isGoalActive || goalStartedAt.current === null) return;
+    const formatElapsed = (ms: number): string => {
+      const minutes = Math.floor(ms / 60000);
+      if (minutes < 1) return "<1m";
+      if (minutes < 60) return `${minutes}m`;
+      const hours = Math.floor(minutes / 60);
+      const remainingMin = minutes % 60;
+      return `${hours}h${remainingMin}m`;
+    };
+    const update = () =>
+      setGoalElapsed(formatElapsed(Date.now() - goalStartedAt.current!));
+    update();
+    const timer = setInterval(update, 30_000);
+    return () => clearInterval(timer);
+  }, [isGoalActive]);
 
   // MCP State
   const [mcpServerStatuses, setMcpServerStatuses] = useState<McpServerStatus[]>(
@@ -209,6 +241,8 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
 
   // Background tasks state
   const [backgroundTasks, setBackgroundTasks] = useState<BackgroundTask[]>([]);
+  // Workflow runs state
+  const [workflowRuns, setWorkflowRuns] = useState<WorkflowRun[]>([]);
   // Tasks state
   const [tasks, setTasks] = useState<Task[]>([]);
 
@@ -344,11 +378,25 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         onCompactionStateChange: (isCompactingState) => {
           setIsCompacting(isCompactingState);
         },
-        onBackgroundTasksChange: (tasks) => {
+        onBackgroundTasksChange: async (tasks) => {
           setBackgroundTasks([...tasks]);
+          // Also refresh workflow runs since workflows are background tasks
+          const runs = await agentRef.current?.getWorkflowRuns();
+          if (runs) setWorkflowRuns([...runs]);
         },
-        onTasksChange: (tasks) => {
-          setTasks([...tasks]);
+        onTasksChange: (newTasks) => {
+          setTasks((prev) => {
+            if (
+              prev.length === newTasks.length &&
+              prev.every(
+                (t, i) =>
+                  t.id === newTasks[i].id && t.status === newTasks[i].status,
+              )
+            ) {
+              return prev;
+            }
+            return [...newTasks];
+          });
         },
         onPermissionModeChange: (mode) => {
           setPermissionModeState(mode);
@@ -368,6 +416,18 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         onQueuedMessagesChange: (messages) => {
           setQueuedMessages([...messages]);
         },
+        onGoalStateChange: (active, _condition, elapsed) => {
+          setIsGoalActive(active);
+          if (active) {
+            goalStartedAt.current = Date.now();
+          } else {
+            goalStartedAt.current = null;
+          }
+          setGoalElapsed(elapsed);
+        },
+        onGoalEvaluating: (evaluating) => {
+          setIsGoalEvaluating(evaluating);
+        },
       };
 
       try {
@@ -424,7 +484,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         setIsCompacting(agent.isCompacting);
         setPermissionModeState(agent.getPermissionMode());
         setWorkingDirectory(agent.workingDirectory);
-        setCurrentModelState(agent.getModelConfig().model);
+        setCurrentModelState(agent.getModelConfig().model || "");
         setConfiguredModels(agent.getConfiguredModels());
         setMaxInputTokens(agent.getMaxInputTokens());
 
@@ -568,6 +628,14 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     agentRef.current?.abortMessage();
   }, []);
 
+  const recallQueuedMessage = useCallback((): QueuedMessage | null => {
+    return agentRef.current?.recallQueuedMessage() ?? null;
+  }, []);
+
+  const removeQueuedMessageById = useCallback((id: string): boolean => {
+    return agentRef.current?.removeQueuedMessageById(id) ?? false;
+  }, []);
+
   // Permission management methods
   const setPermissionMode = useCallback((mode: PermissionMode) => {
     setPermissionModeState((prev) => {
@@ -599,6 +667,10 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     return agentRef.current.stopBackgroundTask(taskId);
   }, []);
 
+  const stopWorkflowRun = useCallback((runId: string) => {
+    agentRef.current?.stopWorkflowRun(runId);
+  }, []);
+
   const hasSlashCommand = useCallback((commandId: string) => {
     if (!agentRef.current) return false;
     return agentRef.current.hasSlashCommand(commandId);
@@ -753,6 +825,8 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     sendMessage,
     askBtw,
     abortMessage,
+    recallQueuedMessage,
+    removeQueuedMessageById,
     latestTotalTokens,
     maxInputTokens,
     currentModel,
@@ -764,6 +838,8 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     connectMcpServer,
     disconnectMcpServer,
     backgroundTasks,
+    workflowRuns,
+    stopWorkflowRun,
     tasks,
     getBackgroundTaskOutput,
     stopBackgroundTask,
@@ -791,6 +867,9 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     workdir,
     recreateAgent,
     triggerWorktreeRemoveHook,
+    isGoalActive,
+    goalElapsed,
+    isGoalEvaluating,
   };
 
   return (

package/src/hooks/useInputManager.ts

@@ -31,6 +31,7 @@ export const useInputManager = (
     onStatusCommandStateChange,
     onPluginManagerStateChange,
     onModelSelectorStateChange,
+    onWorkflowManagerStateChange,
     onImagesStateChange,
     onSendMessage,
     onHasSlashCommand,
@@ -42,18 +43,10 @@ export const useInputManager = (
     workdir,
     getFullMessageThread,
     logger,
+    hasQueuedMessages: hasQueuedMessagesProp,
+    onRecallQueuedMessage,
   } = callbacks;
 
-  // Handle selectorJustUsed reset
-  useEffect(() => {
-    if (state.selectorJustUsed) {
-      const timer = setTimeout(() => {
-        dispatch({ type: "SET_SELECTOR_JUST_USED", payload: false });
-      }, 0);
-      return () => clearTimeout(timer);
-    }
-  }, [state.selectorJustUsed]);
-
   // Handle debounced file search
   useEffect(() => {
     if (state.showFileSelector) {
@@ -217,6 +210,8 @@ export const useInputManager = (
                 dispatch({ type: "SET_SHOW_PLUGIN_MANAGER", payload: true });
               } else if (command === "model") {
                 dispatch({ type: "SET_SHOW_MODEL_SELECTOR", payload: true });
+              } else if (command === "workflows") {
+                dispatch({ type: "SET_SHOW_WORKFLOW_MANAGER", payload: true });
               } else if (command === "btw") {
                 dispatch({
                   type: "SET_BTW_STATE",
@@ -225,6 +220,9 @@ export const useInputManager = (
               }
             }
             break;
+          case "RECALL_QUEUED_MESSAGE":
+            onRecallQueuedMessage?.();
+            break;
         }
       } catch (error) {
         console.error("Effect execution error:", error);
@@ -244,6 +242,7 @@ export const useInputManager = (
     getFullMessageThread,
     logger,
     onHasSlashCommand,
+    onRecallQueuedMessage,
   ]);
 
   useEffect(() => {
@@ -313,6 +312,10 @@ export const useInputManager = (
     onModelSelectorStateChange?.(state.showModelSelector);
   }, [state.showModelSelector, onModelSelectorStateChange]);
 
+  useEffect(() => {
+    onWorkflowManagerStateChange?.(state.showWorkflowManager);
+  }, [state.showWorkflowManager, onWorkflowManagerStateChange]);
+
   useEffect(() => {
     onImagesStateChange?.(state.attachedImages);
   }, [state.attachedImages, onImagesStateChange]);
@@ -479,6 +482,10 @@ export const useInputManager = (
     dispatch({ type: "SET_SHOW_MODEL_SELECTOR", payload: show });
   }, []);
 
+  const setShowWorkflowManager = useCallback((show: boolean) => {
+    dispatch({ type: "SET_SHOW_WORKFLOW_MANAGER", payload: show });
+  }, []);
+
   const setPermissionMode = useCallback(
     (mode: PermissionMode) => {
       dispatch({ type: "SET_PERMISSION_MODE", payload: mode });
@@ -518,10 +525,11 @@ export const useInputManager = (
           input,
           key: {} as Key,
           hasSlashCommand: (cmd) => !!onHasSlashCommand?.(cmd),
+          hasQueuedMessages: hasQueuedMessagesProp ?? false,
         },
       });
     },
-    [onHasSlashCommand],
+    [onHasSlashCommand, hasQueuedMessagesProp],
   );
 
   const handleSubmit = useCallback(async () => {
@@ -531,9 +539,10 @@ export const useInputManager = (
         input: "",
         key: { return: true } as Key,
         hasSlashCommand: (cmd) => !!onHasSlashCommand?.(cmd),
+        hasQueuedMessages: hasQueuedMessagesProp ?? false,
       },
     });
-  }, [onHasSlashCommand]);
+  }, [onHasSlashCommand, hasQueuedMessagesProp]);
 
   const expandLongTextPlaceholders = useCallback(
     (text: string) => {
@@ -564,11 +573,12 @@ export const useInputManager = (
           input,
           key,
           hasSlashCommand: (cmd) => !!onHasSlashCommand?.(cmd),
+          hasQueuedMessages: hasQueuedMessagesProp ?? false,
         },
       });
       return true;
     },
-    [onHasSlashCommand],
+    [onHasSlashCommand, hasQueuedMessagesProp],
   );
 
   return {
@@ -593,6 +603,7 @@ export const useInputManager = (
     showLoginCommand: state.showLoginCommand,
     showPluginManager: state.showPluginManager,
     showModelSelector: state.showModelSelector,
+    showWorkflowManager: state.showWorkflowManager,
     permissionMode: state.permissionMode,
     attachedImages: state.attachedImages,
     btwState: state.btwState,
@@ -636,6 +647,7 @@ export const useInputManager = (
     setShowLoginCommand,
     setShowPluginManager,
     setShowModelSelector,
+    setShowWorkflowManager,
     setPermissionMode,
     setBtwState,
 

package/src/managers/inputHandlers.ts

@@ -371,6 +371,8 @@ export const handleCommandSelect = (
           dispatch({ type: "SET_SHOW_PLUGIN_MANAGER", payload: true });
         } else if (command === "model") {
           dispatch({ type: "SET_SHOW_MODEL_SELECTOR", payload: true });
+        } else if (command === "workflows") {
+          dispatch({ type: "SET_SHOW_WORKFLOW_MANAGER", payload: true });
         } else if (command === "btw") {
           dispatch({
             type: "SET_BTW_STATE",
@@ -699,10 +701,6 @@ export const handleInput = async (
   key: Key,
   clearImages?: () => void,
 ): Promise<boolean> => {
-  if (state.selectorJustUsed) {
-    return true;
-  }
-
   if (state.btwState.isActive) {
     if (key.escape) {
       dispatch({
@@ -756,6 +754,7 @@ export const handleInput = async (
         state.showStatusCommand ||
         state.showPluginManager ||
         state.showModelSelector ||
+        state.showWorkflowManager ||
         state.btwState.isActive
       )
     ) {
@@ -780,6 +779,7 @@ export const handleInput = async (
     state.showStatusCommand ||
     state.showPluginManager ||
     state.showModelSelector ||
+    state.showWorkflowManager ||
     state.btwState.isActive
   ) {
     if (
@@ -790,6 +790,7 @@ export const handleInput = async (
       state.showStatusCommand ||
       state.showPluginManager ||
       state.showModelSelector ||
+      state.showWorkflowManager ||
       state.btwState.isActive
     ) {
       return true;

package/src/managers/inputReducer.ts

@@ -13,6 +13,7 @@ import {
   SELECTOR_TRIGGERS,
   getProjectedState,
 } from "../utils/inputUtils.js";
+import { AVAILABLE_COMMANDS } from "../constants/commands.js";
 
 export interface AttachedImage {
   id: number;
@@ -40,7 +41,8 @@ export type PendingEffect =
   | { type: "PERMISSION_MODE_CHANGE"; mode: PermissionMode }
   | { type: "FETCH_HISTORY" }
   | { type: "PASTE_IMAGE" }
-  | { type: "EXECUTE_COMMAND"; command: string };
+  | { type: "EXECUTE_COMMAND"; command: string }
+  | { type: "RECALL_QUEUED_MESSAGE" };
 
 export interface InputManagerCallbacks {
   onInputTextChange?: (text: string) => void;
@@ -64,6 +66,7 @@ export interface InputManagerCallbacks {
   onStatusCommandStateChange?: (show: boolean) => void;
   onPluginManagerStateChange?: (show: boolean) => void;
   onModelSelectorStateChange?: (show: boolean) => void;
+  onWorkflowManagerStateChange?: (show: boolean) => void;
   onImagesStateChange?: (images: AttachedImage[]) => void;
   onSendMessage?: (
     content: string,
@@ -82,6 +85,8 @@ export interface InputManagerCallbacks {
     sessionIds: string[];
   }>;
   logger?: Logger;
+  hasQueuedMessages?: boolean;
+  onRecallQueuedMessage?: () => void;
 }
 
 export interface InputState {
@@ -108,6 +113,7 @@ export interface InputState {
   showLoginCommand: boolean;
   showPluginManager: boolean;
   showModelSelector: boolean;
+  showWorkflowManager: boolean;
   permissionMode: PermissionMode;
   selectorJustUsed: boolean;
   isPasting: boolean;
@@ -146,6 +152,7 @@ export const initialState: InputState = {
   showLoginCommand: false,
   showPluginManager: false,
   showModelSelector: false,
+  showWorkflowManager: false,
   permissionMode: "default",
   selectorJustUsed: false,
   isPasting: false,
@@ -191,6 +198,7 @@ export type InputAction =
   | { type: "SET_SHOW_LOGIN_COMMAND"; payload: boolean }
   | { type: "SET_SHOW_PLUGIN_MANAGER"; payload: boolean }
   | { type: "SET_SHOW_MODEL_SELECTOR"; payload: boolean }
+  | { type: "SET_SHOW_WORKFLOW_MANAGER"; payload: boolean }
   | { type: "SET_PERMISSION_MODE"; payload: PermissionMode }
   | { type: "SET_SELECTOR_JUST_USED"; payload: boolean }
   | { type: "INSERT_TEXT_WITH_PLACEHOLDER"; payload: string }
@@ -222,6 +230,7 @@ export type InputAction =
         input: string;
         key: Key;
         hasSlashCommand: (cmd: string) => boolean;
+        hasQueuedMessages?: boolean;
       };
     };
 
@@ -413,6 +422,12 @@ export function inputReducer(
         showModelSelector: action.payload,
         selectorJustUsed: !action.payload ? true : state.selectorJustUsed,
       };
+    case "SET_SHOW_WORKFLOW_MANAGER":
+      return {
+        ...state,
+        showWorkflowManager: action.payload,
+        selectorJustUsed: !action.payload ? true : state.selectorJustUsed,
+      };
     case "SET_PERMISSION_MODE":
       return { ...state, permissionMode: action.payload };
     case "SET_SELECTOR_JUST_USED":
@@ -671,10 +686,7 @@ export function inputReducer(
       return { ...state, pendingEffect: null };
     case "HANDLE_KEY": {
       const { input, key } = action.payload;
-
-      if (state.selectorJustUsed) {
-        return state;
-      }
+      const hasQueuedMessages = action.payload.hasQueuedMessages ?? false;
 
       // 1. BTW State Handling
       if (state.btwState.isActive) {
@@ -760,7 +772,8 @@ export function inputReducer(
             state.showStatusCommand ||
             state.showLoginCommand ||
             state.showPluginManager ||
-            state.showModelSelector
+            state.showModelSelector ||
+            state.showWorkflowManager
           )
         ) {
           return { ...state, pendingEffect: { type: "ABORT_MESSAGE" } };
@@ -812,6 +825,13 @@ export function inputReducer(
         !state.showFileSelector &&
         !state.showCommandSelector
       ) {
+        // If queue has messages, recall from queue first (before history)
+        if (hasQueuedMessages) {
+          return {
+            ...state,
+            pendingEffect: { type: "RECALL_QUEUED_MESSAGE" },
+          };
+        }
         if (state.history.length === 0) {
           return { ...state, pendingEffect: { type: "FETCH_HISTORY" } };
         }
@@ -1021,6 +1041,34 @@ export function inputReducer(
             };
           }
 
+          // Check if the content is a CLI-internal slash command (help, tasks,
+          // etc.) that should be executed locally rather than sent as a message.
+          // Agent slash commands and unknown /commands always go to SEND_MESSAGE.
+          if (contentWithPlaceholders.startsWith("/")) {
+            const spaceIndex = contentWithPlaceholders.indexOf(" ");
+            const commandName =
+              spaceIndex === -1
+                ? contentWithPlaceholders.substring(1)
+                : contentWithPlaceholders.substring(1, spaceIndex);
+
+            const isInternalCommand = AVAILABLE_COMMANDS.some(
+              (cmd) => cmd.id === commandName,
+            );
+            if (isInternalCommand) {
+              return {
+                ...state,
+                inputText: "",
+                cursorPosition: 0,
+                historyIndex: -1,
+                longTextMap: {},
+                pendingEffect: {
+                  type: "EXECUTE_COMMAND",
+                  command: commandName,
+                },
+              };
+            }
+          }
+
           return {
             ...state,
             inputText: "",

package/src/print-cli.ts

@@ -65,15 +65,15 @@ export async function startPrintCli(options: PrintCliOptions): Promise<void> {
   let agent: Agent;
   let isReasoning = false;
   let isContent = false;
-  const subagentReasoningStates = new Map<string, boolean>();
-  const subagentContentStates = new Map<string, boolean>();
 
   // Setup callbacks for agent
+  // Print mode only outputs the main agent's response (matching Claude Code's
+  // behavior). Subagent output is internal — the main agent incorporates
+  // relevant results in its own response.
   const callbacks: AgentCallbacks = {
     onAssistantMessageAdded: () => {
       isReasoning = false;
       isContent = false;
-      // Assistant message started - no content to output yet
     },
     onAssistantReasoningUpdated: (chunk: string) => {
       if (!isReasoning) {
@@ -91,43 +91,9 @@ export async function startPrintCli(options: PrintCliOptions): Promise<void> {
         }
         isContent = true;
       }
-      // FR-001: Stream content updates for real-time display - output only the new chunk
       process.stdout.write(chunk);
     },
 
-    // Subagent message callbacks
-    onSubagentAssistantMessageAdded: (subagentId: string) => {
-      subagentReasoningStates.set(subagentId, false);
-      subagentContentStates.set(subagentId, false);
-    },
-    onSubagentAssistantReasoningUpdated: (
-      subagentId: string,
-      chunk: string,
-    ) => {
-      if (!subagentReasoningStates.get(subagentId)) {
-        process.stdout.write("\n   💭 Reasoning: ");
-        subagentReasoningStates.set(subagentId, true);
-      }
-      process.stdout.write(chunk);
-    },
-    onSubagentAssistantContentUpdated: (subagentId: string, chunk: string) => {
-      if (!subagentContentStates.get(subagentId)) {
-        if (subagentReasoningStates.get(subagentId)) {
-          process.stdout.write("\n   📝 Response: ");
-        } else {
-          process.stdout.write("\n   ");
-        }
-        subagentContentStates.set(subagentId, true);
-      }
-      process.stdout.write(chunk);
-    },
-    onSubagentUserMessageAdded: (
-      _subagentId: string,
-      params: { content: string },
-    ) => {
-      process.stdout.write(`\n   👤 User: ${params.content}\n`);
-    },
-
     // Tool block callback - display tool name when tool starts
     onToolBlockUpdated: (params) => {
       // Print tool name only during 'running' stage (happens once per tool call)