wave-code 1.1.5 → 1.3.0

package/src/constants/commands.ts

@@ -25,6 +25,12 @@ export const AVAILABLE_COMMANDS: SlashCommand[] = [
     description: "List available skills",
     handler: () => {}, // Handler here won't be used, actual processing is in the hook
   },
+  {
+    id: "hooks",
+    name: "hooks",
+    description: "View configured hooks",
+    handler: () => {}, // Handler here won't be used, actual processing is in the hook
+  },
   {
     id: "rewind",
     name: "rewind",
@@ -50,6 +56,12 @@ export const AVAILABLE_COMMANDS: SlashCommand[] = [
     description: "View and manage plugins",
     handler: () => {}, // Handler here won't be used, actual processing is in the hook
   },
+  {
+    id: "reload-plugins",
+    name: "reload-plugins",
+    description: "Reload plugins in place, without restarting the session",
+    handler: () => {}, // Handler here won't be used, actual processing is in the hook
+  },
   {
     id: "btw",
     name: "btw",
@@ -80,6 +92,24 @@ export const AVAILABLE_COMMANDS: SlashCommand[] = [
     description: "View and manage workflow runs",
     handler: () => {}, // Handler here won't be used, actual processing is in the hook
   },
+  {
+    id: "plan",
+    name: "plan",
+    description: "Enable plan mode or view the current session plan",
+    handler: () => {}, // Handler here won't be used, actual processing is in the hook
+  },
+  {
+    id: "resume",
+    name: "resume",
+    description: "Switch to another conversation without leaving the session",
+    handler: () => {}, // Handler here won't be used, actual processing is in the hook
+  },
+  {
+    id: "rename",
+    name: "rename",
+    description: "Set a custom title for the current conversation",
+    handler: () => {}, // Handler here won't be used, actual processing is in the hook
+  },
   {
     id: "clear",
     name: "clear",

package/src/contexts/useChat.tsx

@@ -17,6 +17,7 @@ import type {
   SlashCommand,
   SubagentConfiguration,
   SkillMetadata,
+  HookEventConfig,
   PermissionDecision,
   PermissionMode,
   QueuedMessage,
@@ -30,14 +31,23 @@ import {
   type WorktreeSession,
   OPERATION_CANCELLED_BY_USER,
   extractLatestTotalTokens,
+  PLUGIN_CHANGE_PENDING_MESSAGE,
+  PLUGIN_RELOADED_MESSAGE,
 } from "wave-agent-sdk";
 import { logger } from "../utils/logger.js";
 import { displayUsageSummary } from "../utils/usageSummary.js";
 import { expandLongTextPlaceholders } from "../managers/inputHandlers.js";
+import { readFile } from "node:fs/promises";
 
 import { BaseAppProps } from "../types.js";
 
 // Main Chat Context
+export interface PlanViewData {
+  path?: string;
+  content?: string;
+  message?: string;
+}
+
 export interface ChatContextType {
   messages: Message[];
   isLoading: boolean;
@@ -69,6 +79,17 @@ export interface ChatContextType {
   clearMessages: () => Promise<void>;
   compact: (instructions?: string) => Promise<void>;
   addDir: (args?: string) => Promise<void>;
+  /**
+   * Set the current session's custom title (`/rename <title>`). A blank title
+   * only prints the usage hint — no AI-generated titles (spec: session-management.md).
+   */
+  renameSession: (args?: string) => Promise<void>;
+  /**
+   * Switch to another conversation in-process (`/resume`). Pass `resumeWorkdir`
+   * when the target session lives in a sibling worktree of the same repo — the
+   * session then moves into that directory too.
+   */
+  resumeSession: (sessionId: string, resumeWorkdir?: string) => Promise<void>;
   abortMessage: () => void;
   recallQueuedMessage: () => QueuedMessage | null;
   removeQueuedMessageById: (id: string) => boolean;
@@ -104,9 +125,17 @@ export interface ChatContextType {
   agentDefinitions: SubagentConfiguration[];
   // Skill metadata (for /skills overlay)
   skills: SkillMetadata[];
+  // Hook configs per scope (for /hooks overlay)
+  hooks: Partial<
+    Record<"user" | "project" | "plugin", Record<string, HookEventConfig[]>>
+  >;
   // Permission functionality
   permissionMode: PermissionMode;
   setPermissionMode: (mode: PermissionMode) => void;
+  // /plan overlay state + handler
+  planView: PlanViewData | null;
+  setPlanView: (view: PlanViewData | null) => void;
+  handlePlanCommand: (args?: string) => Promise<void>;
   // Permission confirmation state
   isConfirmationVisible: boolean;
   hasPendingConfirmations: boolean;
@@ -115,6 +144,7 @@ export interface ChatContextType {
     input?: Record<string, unknown>;
     suggestedPrefix?: string;
     hidePersistentOption?: boolean;
+    outsideSafeZoneDirectory?: string;
     planContent?: string;
     permissionMode?: PermissionMode;
     warning?: string;
@@ -127,6 +157,7 @@ export interface ChatContextType {
     planContent?: string,
     permissionMode?: PermissionMode,
     warning?: string,
+    outsideSafeZoneDirectory?: string,
   ) => Promise<PermissionDecision>;
   hideConfirmation: () => void;
   handleConfirmationDecision: (decision: PermissionDecision) => void;
@@ -148,8 +179,10 @@ export interface ChatContextType {
   workingDirectory: string;
   version?: string;
   workdir?: string;
-  // Agent recreation (e.g. after plugin install)
-  recreateAgent: () => void;
+  // Plugin in-place reload (docs/specs/ecosystem/plugin.md「插件变更的就地重载」)
+  reloadPlugins: () => Promise<void>;
+  notifyPluginChange: () => void;
+  pluginHint: string | null;
 }
 
 const ChatContext = createContext<ChatContextType | null>(null);
@@ -472,6 +505,12 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   >([]);
   // Skill metadata (for /skills overlay)
   const [skills, setSkills] = useState<SkillMetadata[]>([]);
+  // Hook configs per scope (for /hooks overlay)
+  const [hooks, setHooks] = useState<
+    Partial<
+      Record<"user" | "project" | "plugin", Record<string, HookEventConfig[]>>
+    >
+  >({});
 
   // Permission state
   const [permissionMode, setPermissionModeState] = useState<PermissionMode>(
@@ -479,6 +518,9 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
       (bypassPermissions ? "bypassPermissions" : "default"),
   );
 
+  // Plan view state for the /plan command overlay
+  const [planView, setPlanView] = useState<PlanViewData | null>(null);
+
   // Confirmation state with queue-based architecture
   const [isConfirmationVisible, setIsConfirmationVisible] = useState(false);
   const [confirmingTool, setConfirmingTool] = useState<
@@ -487,6 +529,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         input?: Record<string, unknown>;
         suggestedPrefix?: string;
         hidePersistentOption?: boolean;
+        outsideSafeZoneDirectory?: string;
         planContent?: string;
         permissionMode?: PermissionMode;
         warning?: string;
@@ -499,6 +542,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
       toolInput?: Record<string, unknown>;
       suggestedPrefix?: string;
       hidePersistentOption?: boolean;
+      outsideSafeZoneDirectory?: string;
       planContent?: string;
       permissionMode?: PermissionMode;
       warning?: string;
@@ -511,6 +555,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     toolInput?: Record<string, unknown>;
     suggestedPrefix?: string;
     hidePersistentOption?: boolean;
+    outsideSafeZoneDirectory?: string;
     planContent?: string;
     permissionMode?: PermissionMode;
     warning?: string;
@@ -540,7 +585,9 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   // the incremental callbacks in initializeAgent below.
   const refreshMessages = useCallback(() => {
     if (!isExpandedRef.current && agentRef.current) {
-      const msgs = agentRef.current.messages.map(snapshotMessage);
+      // Pull the full UI display stream (keeps pre-compaction history);
+      // `agent.messages` would only expose the folded API context.
+      const msgs = agentRef.current.displayMessages.map(snapshotMessage);
       // Snapshot-safe: the full-list replacement makes the SDK state
       // authoritative. Any update still queued inside the 500ms throttle
       // window was applied to the SDK before this pull, so it is already
@@ -565,6 +612,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
       planContent?: string,
       permissionMode?: PermissionMode,
       warning?: string,
+      outsideSafeZoneDirectory?: string,
     ): Promise<PermissionDecision> => {
       return new Promise<PermissionDecision>((resolve, reject) => {
         const queueItem = {
@@ -572,6 +620,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
           toolInput,
           suggestedPrefix,
           hidePersistentOption,
+          outsideSafeZoneDirectory,
           planContent,
           permissionMode,
           warning,
@@ -674,73 +723,6 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
           // delayed by a streaming window).
           updateMessages.flush();
         },
-        onAddBangMessage: (command, messageId) => {
-          if (isExpandedRef.current) return;
-          updateMessages((prev) =>
-            prev.some((m) => m.id === messageId)
-              ? prev
-              : [
-                  ...prev,
-                  {
-                    id: messageId,
-                    role: "user",
-                    timestamp: new Date().toISOString(),
-                    blocks: [
-                      {
-                        type: "bang",
-                        command,
-                        output: "",
-                        stage: "running",
-                        exitCode: null,
-                      },
-                    ],
-                  },
-                ],
-          );
-          updateMessages.flush();
-        },
-        onUpdateBangMessage: (command, output, messageId) => {
-          if (isExpandedRef.current) return;
-          updateMessages((prev) =>
-            prev.map((m) =>
-              m.id === messageId
-                ? {
-                    ...m,
-                    blocks: m.blocks.map((b, idx) =>
-                      idx === m.blocks.length - 1 && b.type === "bang"
-                        ? { ...b, command, output }
-                        : b,
-                    ),
-                  }
-                : m,
-            ),
-          );
-        },
-        onCompleteBangMessage: (command, exitCode, messageId, output) => {
-          if (isExpandedRef.current) return;
-          updateMessages((prev) =>
-            prev.map((m) =>
-              m.id === messageId
-                ? {
-                    ...m,
-                    blocks: m.blocks.map((b, idx) =>
-                      idx === m.blocks.length - 1 && b.type === "bang"
-                        ? {
-                            ...b,
-                            command,
-                            exitCode,
-                            stage: "end",
-                            ...(output !== undefined ? { output } : {}),
-                          }
-                        : b,
-                    ),
-                  }
-                : m,
-            ),
-          );
-          // Completion signal — flush so the final state applies immediately.
-          updateMessages.flush();
-        },
         onLatestTotalTokensChange: (tokens) => {
           setLatestTotalTokens(tokens);
         },
@@ -815,6 +797,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
               context.planContent,
               context.permissionMode,
               context.warning,
+              context.outsideSafeZoneDirectory,
             );
           } catch {
             // If confirmation was cancelled or failed, deny the operation
@@ -865,9 +848,10 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         }
 
         // Get initial state — snapshot the SDK messages (never hold live
-        // references; see snapshotMessage)
+        // references; see snapshotMessage). Uses the full display stream so a
+        // restored session shows pre-compaction history too.
         setSessionId(agent.sessionId);
-        setMessages(agent.messages.map(snapshotMessage));
+        setMessages(agent.displayMessages.map(snapshotMessage));
         setIsLoading(agent.isLoading);
         setLatestTotalTokens(extractLatestTotalTokens(agent.messages));
         setIsCommandRunning(agent.isCommandRunning);
@@ -894,6 +878,19 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         // Get initial skill metadata
         const initialSkills = agent.getSkillMetadata?.() || [];
         setSkills(initialSkills);
+
+        // Get initial hooks (user/project from settings.json, plugin from
+        // the hook manager's plugin-registered set) — snapshot for /hooks.
+        const [userHooks, projectHooks, pluginHooks] = await Promise.all([
+          agent.getHooksByScope("user"),
+          agent.getHooksByScope("project"),
+          agent.getHooksByScope("plugin"),
+        ]);
+        setHooks({
+          user: userHooks,
+          project: projectHooks,
+          plugin: pluginHooks,
+        });
       } catch (error) {
         console.error("Failed to initialize AI manager:", error);
       }
@@ -918,32 +915,50 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     ],
   );
 
-  // Recreate agent (e.g. after plugin install) — destroys current agent and reinitializes
-  const recreateAgent = useCallback(() => {
-    const currentSessionId = agentRef.current?.sessionId;
-    if (agentRef.current) {
-      try {
-        agentRef.current.destroy();
-      } catch {
-        // Ignore destroy errors
-      }
+  // 插件变更落盘后只置一个「待应用」信号并提示一次，不再重建 Agent；真正换装由
+  // 用户敲 /reload-plugins 触发，在当前会话内完成（docs/specs/ecosystem/plugin.md
+  // 「插件变更的就地重载」：会…不重建、不打断回合、不新建会话）。
+  const [pluginHint, setPluginHint] = useState<string | null>(null);
+  const pluginHintTimer = useRef<ReturnType<typeof setTimeout> | null>(null);
+
+  const showPluginHint = useCallback((message: string) => {
+    setPluginHint(message);
+    if (pluginHintTimer.current) {
+      clearTimeout(pluginHintTimer.current);
     }
-    agentRef.current = null;
-    setMessages([]);
-    setMcpServerStatuses([]);
-    setSlashCommands([]);
-    setAgentDefinitions([]);
-    setSkills([]);
-    setSessionId("");
-    setIsLoading(false);
-    setLatestTotalTokens(0);
-    setMaxInputTokens(200000);
-    setIsCommandRunning(false);
-    setIsCompacting(false);
-    if (currentSessionId) {
-      initializeAgent(currentSessionId);
+    // 一次性提示：数秒后自行消失，不留常驻标识（spec「插件变更提示」）。
+    pluginHintTimer.current = setTimeout(() => setPluginHint(null), 8000);
+  }, []);
+
+  const notifyPluginChange = useCallback(() => {
+    showPluginHint(PLUGIN_CHANGE_PENDING_MESSAGE);
+  }, [showPluginHint]);
+
+  const reloadPlugins = useCallback(async () => {
+    const agent = agentRef.current;
+    if (!agent) return;
+    try {
+      const result = await agent.reloadPlugins();
+      // 换装后本进程各会话的能力已变，UI 侧的命令 / 子代理 / 技能列表同步重读。
+      setSlashCommands(agent.getSlashCommands?.() || []);
+      setAgentDefinitions(agent.getSubagentConfigurations?.() || []);
+      setSkills(agent.getSkillMetadata?.() || []);
+      if (result.failures.length > 0) {
+        // 失败不回滚：如实提示，会话继续可用（spec 场景 11）。
+        showPluginHint(
+          `${PLUGIN_RELOADED_MESSAGE}部分插件加载失败：${result.failures
+            .map((failure) => `${failure.path}: ${failure.error}`)
+            .join("；")}`,
+        );
+      } else {
+        showPluginHint(PLUGIN_RELOADED_MESSAGE);
+      }
+    } catch (error) {
+      showPluginHint(
+        `插件重载失败：${error instanceof Error ? error.message : String(error)}`,
+      );
     }
-  }, [initializeAgent]);
+  }, [showPluginHint]);
 
   // Run initial agent creation
   useEffect(() => {
@@ -1087,6 +1102,90 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     );
   }, []);
 
+  // /rename: set the current conversation's custom title. The title is stored
+  // as a reserved `custom-title` entry appended to the session's own JSONL
+  // file, so `wave -r` and every GUI host read the same value. A blank title
+  // only prints the usage hint — no AI-generated titles (spec:
+  // session-management.md "不做 AI 自动标题").
+  const renameSession = useCallback(async (args?: string) => {
+    const agent = agentRef.current;
+    if (!agent) return;
+
+    const title = (args ?? "").trim();
+    const showResult = (content: string) => {
+      setMessages((prev) => [
+        ...prev,
+        {
+          id: `msg-${Date.now()}-${Math.random().toString(36).slice(2, 11)}`,
+          role: "assistant",
+          timestamp: new Date().toISOString(),
+          blocks: [{ type: "text", content, stage: "end" }],
+        },
+      ]);
+    };
+
+    if (!title) {
+      showResult("Usage: /rename <title>");
+      return;
+    }
+
+    try {
+      await agent.renameSession(title);
+      showResult(`Conversation renamed to "${title}".`);
+    } catch (error) {
+      showResult(
+        `Failed to rename conversation: ${
+          error instanceof Error ? error.message : String(error)
+        }`,
+      );
+    }
+  }, []);
+
+  // /resume: switch to another conversation without leaving the process. The
+  // session id and its on-disk project directory move together, so the next
+  // message is appended to the target's own transcript. A failed restore leaves
+  // the current session untouched and only surfaces a notice.
+  const resumeSession = useCallback(
+    async (sessionId: string, resumeWorkdir?: string) => {
+      const agent = agentRef.current;
+      if (!agent) return;
+
+      try {
+        await agent.restoreSession(
+          sessionId,
+          resumeWorkdir ? { workdir: resumeWorkdir } : undefined,
+        );
+      } catch (error) {
+        // UI-only notice (same display path as /add-dir): nothing was switched,
+        // so nothing about the failed target is persisted.
+        setMessages((prev) => [
+          ...prev,
+          {
+            id: `msg-${Date.now()}-${Math.random().toString(36).slice(2, 11)}`,
+            role: "assistant",
+            timestamp: new Date().toISOString(),
+            blocks: [
+              {
+                type: "error",
+                content: `Failed to resume conversation: ${
+                  error instanceof Error ? error.message : String(error)
+                }`,
+              },
+            ],
+          },
+        ]);
+        return;
+      }
+
+      setSessionId(agent.sessionId);
+      setWorkingDirectory(agent.workingDirectory);
+      setIsLoading(agent.isLoading);
+      refreshMessages();
+      forceRemount();
+    },
+    [forceRemount, refreshMessages],
+  );
+
   // Unified interrupt method, interrupt both AI messages and command execution
   const abortMessage = useCallback(() => {
     agentRef.current?.abortMessage();
@@ -1111,6 +1210,52 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     });
   }, []);
 
+  // /plan command handler:
+  // - Outside plan mode: switch to plan mode, wait for the plan file path
+  //   (generated asynchronously by PlanManager), then either send the given
+  //   description as a message (starting a plan query) or show a confirmation.
+  // - Inside plan mode: display the current plan file contents; `/plan open`
+  //   is not supported on any end (spec: plan-mode.md "三端均不支持").
+  const handlePlanCommand = useCallback(
+    async (args?: string) => {
+      const agent = agentRef.current;
+      const description = args?.trim() ?? "";
+      const argList = description.split(/\s+/).filter(Boolean);
+      const wantsOpen = argList[0] === "open";
+
+      if (permissionMode !== "plan") {
+        setPermissionMode("plan");
+        // Wait for the plan file path to be generated before triggering a
+        // query so the model always receives the plan file location (spec:
+        // plan-mode.md "路径生成必须先于查询触发").
+        const planPath = await agent?.awaitPlanFilePath();
+
+        if (description && !wantsOpen) {
+          await sendMessage(description);
+          return;
+        }
+        setPlanView({ message: "Enabled plan mode", path: planPath });
+        return;
+      }
+
+      // Already in plan mode — display the current plan file contents.
+      const planPath =
+        agent?.getPlanFilePath() ?? (await agent?.awaitPlanFilePath());
+      if (!planPath) {
+        setPlanView({ message: "No plan written yet." });
+        return;
+      }
+      try {
+        const content = await readFile(planPath, "utf8");
+        setPlanView({ path: planPath, content });
+      } catch (error) {
+        logger.warn("Failed to read plan file:", error);
+        setPlanView({ message: "No plan written yet." });
+      }
+    },
+    [permissionMode, sendMessage, setPermissionMode],
+  );
+
   // MCP management methods - delegate to Agent
   const connectMcpServer = useCallback(async (serverName: string) => {
     return (await agentRef.current?.connectMcpServer(serverName)) ?? false;
@@ -1150,6 +1295,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         input: next.toolInput,
         suggestedPrefix: next.suggestedPrefix,
         hidePersistentOption: next.hidePersistentOption,
+        outsideSafeZoneDirectory: next.outsideSafeZoneDirectory,
         planContent: next.planContent,
         permissionMode: next.permissionMode,
         warning: next.warning,
@@ -1237,7 +1383,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
 
   const setModel = useCallback((model: string) => {
     if (agentRef.current) {
-      agentRef.current.setModel(model);
+      void agentRef.current.setModel(model);
       setCurrentModelState(model);
     }
   }, []);
@@ -1287,6 +1433,8 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     clearMessages,
     compact,
     addDir,
+    renameSession,
+    resumeSession,
     abortMessage,
     recallQueuedMessage,
     removeQueuedMessageById,
@@ -1311,8 +1459,12 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     hasSlashCommand,
     agentDefinitions,
     skills,
+    hooks,
     permissionMode,
     setPermissionMode,
+    planView,
+    setPlanView,
+    handlePlanCommand,
     isConfirmationVisible,
     hasPendingConfirmations: confirmationQueue.length > 0,
     confirmingTool,
@@ -1331,7 +1483,9 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     workingDirectory,
     version,
     workdir,
-    recreateAgent,
+    reloadPlugins,
+    notifyPluginChange,
+    pluginHint,
   };
 
   return (