wave-code 0.9.2 → 0.9.4

package/src/components/InputBox.tsx

@@ -9,6 +9,7 @@ import { McpManager } from "./McpManager.js";
 import { RewindCommand } from "./RewindCommand.js";
 import { HelpView } from "./HelpView.js";
 import { StatusCommand } from "./StatusCommand.js";
+import { PluginManagerShell } from "./PluginManagerShell.js";
 import { useInputManager } from "../hooks/useInputManager.js";
 import { useChat } from "../contexts/useChat.js";
 
@@ -58,6 +59,7 @@ export const InputBox: React.FC<InputBoxProps> = ({
     getFullMessageThread,
     clearMessages,
     sessionId,
+    workingDirectory,
   } = useChat();
 
   // Input manager with all input state and functionality (including images)
@@ -91,11 +93,13 @@ export const InputBox: React.FC<InputBoxProps> = ({
     showRewindManager,
     showHelp,
     showStatusCommand,
+    showPluginManager,
     setShowBackgroundTaskManager,
     setShowMcpManager,
     setShowRewindManager,
     setShowHelp,
     setShowStatusCommand,
+    setShowPluginManager,
     // Permission mode
     permissionMode,
     setPermissionMode,
@@ -111,6 +115,8 @@ export const InputBox: React.FC<InputBoxProps> = ({
     onPermissionModeChange: setChatPermissionMode,
     onClearMessages: clearMessages,
     sessionId,
+    workdir: workingDirectory,
+    getFullMessageThread,
   });
 
   // Sync permission mode from useChat to InputManager
@@ -180,6 +186,10 @@ export const InputBox: React.FC<InputBoxProps> = ({
     return <StatusCommand onCancel={() => setShowStatusCommand(false)} />;
   }
 
+  if (showPluginManager) {
+    return <PluginManagerShell onCancel={() => setShowPluginManager(false)} />;
+  }
+
   return (
     <Box flexDirection="column">
       {showFileSelector && (
@@ -229,7 +239,8 @@ export const InputBox: React.FC<InputBoxProps> = ({
         showMcpManager ||
         showRewindManager ||
         showHelp ||
-        showStatusCommand || (
+        showStatusCommand ||
+        showPluginManager || (
           <Box flexDirection="column">
             <Box
               borderStyle="single"

package/src/components/MarketplaceDetail.tsx

@@ -9,6 +9,10 @@ export const MarketplaceDetail: React.FC = () => {
   const marketplace = marketplaces.find((m) => m.name === state.selectedId);
 
   const ACTIONS = [
+    {
+      id: "toggle-auto-update",
+      label: `${marketplace?.autoUpdate ? "Disable" : "Enable"} auto-update`,
+    },
     { id: "update", label: "Update marketplace" },
     { id: "remove", label: "Remove marketplace" },
   ] as const;
@@ -26,7 +30,9 @@ export const MarketplaceDetail: React.FC = () => {
       );
     } else if (key.return && marketplace && !state.isLoading) {
       const action = ACTIONS[selectedActionIndex].id;
-      if (action === "update") {
+      if (action === "toggle-auto-update") {
+        actions.toggleAutoUpdate(marketplace.name, !marketplace.autoUpdate);
+      } else if (action === "update") {
         actions.updateMarketplace(marketplace.name);
       } else {
         actions.removeMarketplace(marketplace.name);
@@ -55,6 +61,15 @@ export const MarketplaceDetail: React.FC = () => {
         <Text>Source: {JSON.stringify(marketplace.source)}</Text>
       </Box>
 
+      <Box marginBottom={1}>
+        <Text>
+          Auto-update:{" "}
+          <Text color={marketplace.autoUpdate ? "green" : "red"}>
+            {marketplace.autoUpdate ? "Enabled" : "Disabled"}
+          </Text>
+        </Text>
+      </Box>
+
       {state.isLoading && (
         <Box marginBottom={1}>
           <Text color="yellow">⌛ Processing operation...</Text>

package/src/components/PluginManagerShell.tsx

@@ -10,9 +10,10 @@ import { PluginDetail } from "./PluginDetail.js";
 import { MarketplaceAddForm } from "./MarketplaceAddForm.js";
 import { PluginManagerContext } from "../contexts/PluginManagerContext.js";
 
-export const PluginManagerShell: React.FC<{ children?: React.ReactNode }> = ({
-  children,
-}) => {
+export const PluginManagerShell: React.FC<{
+  children?: React.ReactNode;
+  onCancel?: () => void;
+}> = ({ children, onCancel }) => {
   const pluginManager = usePluginManager();
   const { state, actions, discoverablePlugins } = pluginManager;
 
@@ -62,6 +63,8 @@ export const PluginManagerShell: React.FC<{ children?: React.ReactNode }> = ({
         state.currentView === "ADD_MARKETPLACE"
       ) {
         setView("MARKETPLACES");
+      } else {
+        onCancel?.();
       }
     }
   });

package/src/components/PluginManagerTypes.ts

@@ -43,6 +43,7 @@ export interface PluginManagerContextType {
     ) => Promise<void>;
     uninstallPlugin: (name: string, marketplace: string) => Promise<void>;
     updatePlugin: (name: string, marketplace: string) => Promise<void>;
+    toggleAutoUpdate: (name: string, enabled: boolean) => Promise<void>;
     refresh: () => Promise<void>;
     clearPluginFeedback: () => void;
   };

package/src/constants/commands.ts

@@ -38,4 +38,10 @@ export const AVAILABLE_COMMANDS: SlashCommand[] = [
     description: "Show agent status and configuration",
     handler: () => {}, // Handler here won't be used, actual processing is in the hook
   },
+  {
+    id: "plugin",
+    name: "plugin",
+    description: "View and manage plugins",
+    handler: () => {}, // Handler here won't be used, actual processing is in the hook
+  },
 ];

package/src/contexts/useChat.tsx

@@ -21,6 +21,7 @@ import {
   Agent,
   AgentCallbacks,
   type ToolPermissionContext,
+  OPERATION_CANCELLED_BY_USER,
 } from "wave-agent-sdk";
 import { logger } from "../utils/logger.js";
 import { displayUsageSummary } from "../utils/usageSummary.js";
@@ -317,7 +318,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
                 // If confirmation was cancelled or failed, deny the operation
                 return {
                   behavior: "deny",
-                  message: "Operation cancelled by user",
+                  message: OPERATION_CANCELLED_BY_USER,
                 };
               }
             };
@@ -372,6 +373,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     tools,
     workdir,
     worktreeSession,
+    model,
   ]);
 
   // Cleanup on unmount

package/src/hooks/useInputManager.ts

@@ -142,6 +142,10 @@ export const useInputManager = (
     callbacksRef.current.onStatusCommandStateChange?.(state.showStatusCommand);
   }, [state.showStatusCommand]);
 
+  useEffect(() => {
+    callbacksRef.current.onPluginManagerStateChange?.(state.showPluginManager);
+  }, [state.showPluginManager]);
+
   useEffect(() => {
     callbacksRef.current.onImagesStateChange?.(state.attachedImages);
   }, [state.attachedImages]);
@@ -291,6 +295,10 @@ export const useInputManager = (
     dispatch({ type: "SET_SHOW_STATUS_COMMAND", payload: show });
   }, []);
 
+  const setShowPluginManager = useCallback((show: boolean) => {
+    dispatch({ type: "SET_SHOW_PLUGIN_MANAGER", payload: show });
+  }, []);
+
   const setPermissionMode = useCallback((mode: PermissionMode) => {
     dispatch({ type: "SET_PERMISSION_MODE", payload: mode });
     callbacksRef.current.onPermissionModeChange?.(mode);
@@ -384,6 +392,7 @@ export const useInputManager = (
     showRewindManager: state.showRewindManager,
     showHelp: state.showHelp,
     showStatusCommand: state.showStatusCommand,
+    showPluginManager: state.showPluginManager,
     permissionMode: state.permissionMode,
     attachedImages: state.attachedImages,
     isManagerReady: true,
@@ -423,6 +432,7 @@ export const useInputManager = (
     setShowRewindManager,
     setShowHelp,
     setShowStatusCommand,
+    setShowPluginManager,
     setPermissionMode,
 
     // Image management

package/src/hooks/usePluginManager.ts

@@ -126,7 +126,7 @@ export function usePluginManager(): PluginManagerContextType {
         error: error instanceof Error ? error.message : String(error),
       }));
     }
-  }, [pluginCore]);
+  }, [pluginCore, clearPluginFeedback]);
 
   useEffect(() => {
     refresh();
@@ -279,6 +279,30 @@ export function usePluginManager(): PluginManagerContextType {
     [pluginCore, refresh, clearPluginFeedback, setSuccessMessage],
   );
 
+  const toggleAutoUpdate = useCallback(
+    async (name: string, enabled: boolean) => {
+      clearPluginFeedback();
+      setState((prev: PluginManagerState) => ({
+        ...prev,
+        isLoading: true,
+      }));
+      try {
+        await pluginCore.toggleAutoUpdate(name, enabled);
+        await refresh();
+        setSuccessMessage(
+          `Auto-update for '${name}' ${enabled ? "enabled" : "disabled"}`,
+        );
+      } catch (error) {
+        setState((prev: PluginManagerState) => ({
+          ...prev,
+          isLoading: false,
+          error: error instanceof Error ? error.message : String(error),
+        }));
+      }
+    },
+    [pluginCore, refresh, clearPluginFeedback, setSuccessMessage],
+  );
+
   return {
     state,
     marketplaces,
@@ -293,6 +317,7 @@ export function usePluginManager(): PluginManagerContextType {
       installPlugin,
       uninstallPlugin,
       updatePlugin,
+      toggleAutoUpdate,
       refresh,
       clearPluginFeedback,
     },

package/src/index.ts

@@ -70,165 +70,135 @@ export async function main() {
         type: "string",
         global: false,
       })
-      .command(
-        "plugin",
-        "Manage plugins and marketplaces",
-        (yargs) => {
-          return yargs
-            .help()
-            .command(
-              "ui",
-              "Open interactive plugin manager UI",
-              {},
-              async () => {
-                const { startPluginManagerCli } = await import(
-                  "./plugin-manager-cli.js"
-                );
-                const shouldExit = await startPluginManagerCli();
-                if (shouldExit) {
-                  process.exit(0);
-                }
-              },
-            )
-            .command(
-              "marketplace",
-              "Manage plugin marketplaces",
-              (yargs) => {
-                return yargs
-                  .help()
-                  .command(
-                    "add <input>",
-                    "Add a plugin marketplace (local path, owner/repo, or Git URL)",
-                    (yargs) => {
-                      return yargs.positional("input", {
-                        describe:
-                          "Path to local marketplace, GitHub owner/repo, or full Git URL (with optional #ref)",
-                        type: "string",
-                      });
-                    },
-                    async (argv) => {
-                      const { addMarketplaceCommand } = await import(
-                        "./commands/plugin/marketplace.js"
-                      );
-                      await addMarketplaceCommand(argv as { input: string });
-                    },
-                  )
-                  .command(
-                    "update [name]",
-                    "Update registered marketplace(s)",
-                    (yargs) => {
-                      return yargs.positional("name", {
-                        describe: "Name of the marketplace to update",
-                        type: "string",
-                      });
-                    },
-                    async (argv) => {
-                      const { updateMarketplaceCommand } = await import(
-                        "./commands/plugin/marketplace.js"
-                      );
-                      await updateMarketplaceCommand(argv as { name?: string });
-                    },
-                  )
-                  .command(
-                    "list",
-                    "List registered marketplaces",
-                    {},
-                    async () => {
-                      const { listMarketplacesCommand } = await import(
-                        "./commands/plugin/marketplace.js"
-                      );
-                      await listMarketplacesCommand();
-                    },
-                  )
-                  .demandCommand(1, "Please specify a marketplace subcommand");
-              },
-              () => {},
-            )
-            .command(
-              "install <plugin>",
-              "Install a plugin from a marketplace",
-              (yargs) => {
-                return yargs
-                  .positional("plugin", {
-                    describe: "Plugin to install (format: name@marketplace)",
-                    type: "string",
-                  })
-                  .option("scope", {
-                    alias: "s",
-                    describe: "Scope to enable the plugin in",
-                    choices: ["user", "project", "local"],
-                    type: "string",
-                  });
-              },
-              async (argv) => {
-                const { installPluginCommand } = await import(
-                  "./commands/plugin/install.js"
-                );
-                await installPluginCommand(
-                  argv as {
-                    plugin: string;
-                    scope?: Scope;
+      .command("plugin", "Manage plugins and marketplaces", (yargs) => {
+        return yargs
+          .help()
+          .command(
+            "marketplace",
+            "Manage plugin marketplaces",
+            (yargs) => {
+              return yargs
+                .help()
+                .command(
+                  "add <input>",
+                  "Add a plugin marketplace (local path, owner/repo, or Git URL)",
+                  (yargs) => {
+                    return yargs.positional("input", {
+                      describe:
+                        "Path to local marketplace, GitHub owner/repo, or full Git URL (with optional #ref)",
+                      type: "string",
+                    });
                   },
-                );
-              },
-            )
-            .command(
-              "list",
-              "List all available plugins from marketplaces",
-              {},
-              async () => {
-                const { listPluginsCommand } = await import(
-                  "./commands/plugin/list.js"
-                );
-                await listPluginsCommand();
-              },
-            )
-            .command(
-              "uninstall <plugin>",
-              "Uninstall a plugin",
-              (yargs) => {
-                return yargs.positional("plugin", {
-                  describe: "Plugin to uninstall (format: name@marketplace)",
+                  async (argv) => {
+                    const { addMarketplaceCommand } = await import(
+                      "./commands/plugin/marketplace.js"
+                    );
+                    await addMarketplaceCommand(argv as { input: string });
+                  },
+                )
+                .command(
+                  "update [name]",
+                  "Update registered marketplace(s)",
+                  (yargs) => {
+                    return yargs.positional("name", {
+                      describe: "Name of the marketplace to update",
+                      type: "string",
+                    });
+                  },
+                  async (argv) => {
+                    const { updateMarketplaceCommand } = await import(
+                      "./commands/plugin/marketplace.js"
+                    );
+                    await updateMarketplaceCommand(argv as { name?: string });
+                  },
+                )
+                .command(
+                  "list",
+                  "List registered marketplaces",
+                  {},
+                  async () => {
+                    const { listMarketplacesCommand } = await import(
+                      "./commands/plugin/marketplace.js"
+                    );
+                    await listMarketplacesCommand();
+                  },
+                )
+                .demandCommand(1, "Please specify a marketplace subcommand");
+            },
+            () => {},
+          )
+          .command(
+            "install <plugin>",
+            "Install a plugin from a marketplace",
+            (yargs) => {
+              return yargs
+                .positional("plugin", {
+                  describe: "Plugin to install (format: name@marketplace)",
                   type: "string",
-                });
-              },
-              async (argv) => {
-                const { uninstallPluginCommand } = await import(
-                  "./commands/plugin/uninstall.js"
-                );
-                await uninstallPluginCommand(argv as { plugin: string });
-              },
-            )
-            .command(
-              "update <plugin>",
-              "Update a plugin (uninstall followed by install)",
-              (yargs) => {
-                return yargs.positional("plugin", {
-                  describe: "Plugin to update (format: name@marketplace)",
+                })
+                .option("scope", {
+                  alias: "s",
+                  describe: "Scope to enable the plugin in",
+                  choices: ["user", "project", "local"],
                   type: "string",
                 });
-              },
-              async (argv) => {
-                const { updatePluginCommand } = await import(
-                  "./commands/plugin/update.js"
-                );
-                await updatePluginCommand(argv as { plugin: string });
-              },
-            );
-        },
-        async (argv) => {
-          // If no subcommand is provided, launch the UI
-          if (argv._.length === 1 && argv._[0] === "plugin") {
-            const { startPluginManagerCli } = await import(
-              "./plugin-manager-cli.js"
-            );
-            const shouldExit = await startPluginManagerCli();
-            if (shouldExit) {
-              process.exit(0);
-            }
-          }
-        },
-      )
+            },
+            async (argv) => {
+              const { installPluginCommand } = await import(
+                "./commands/plugin/install.js"
+              );
+              await installPluginCommand(
+                argv as {
+                  plugin: string;
+                  scope?: Scope;
+                },
+              );
+            },
+          )
+          .command(
+            "list",
+            "List all available plugins from marketplaces",
+            {},
+            async () => {
+              const { listPluginsCommand } = await import(
+                "./commands/plugin/list.js"
+              );
+              await listPluginsCommand();
+            },
+          )
+          .command(
+            "uninstall <plugin>",
+            "Uninstall a plugin",
+            (yargs) => {
+              return yargs.positional("plugin", {
+                describe: "Plugin to uninstall (format: name@marketplace)",
+                type: "string",
+              });
+            },
+            async (argv) => {
+              const { uninstallPluginCommand } = await import(
+                "./commands/plugin/uninstall.js"
+              );
+              await uninstallPluginCommand(argv as { plugin: string });
+            },
+          )
+          .command(
+            "update <plugin>",
+            "Update a plugin (uninstall followed by install)",
+            (yargs) => {
+              return yargs.positional("plugin", {
+                describe: "Plugin to update (format: name@marketplace)",
+                type: "string",
+              });
+            },
+            async (argv) => {
+              const { updatePluginCommand } = await import(
+                "./commands/plugin/update.js"
+              );
+              await updatePluginCommand(argv as { plugin: string });
+            },
+          );
+      })
       .version()
       .alias("v", "version")
       .example("$0", "Start CLI with default settings")

package/src/managers/inputHandlers.ts

@@ -63,6 +63,7 @@ export const handleSubmit = async (
       contentWithPlaceholders,
       callbacks.sessionId,
       state.longTextMap,
+      callbacks.workdir,
     ).catch((err: unknown) => {
       callbacks.logger?.error("Failed to save prompt history", err);
     });
@@ -285,7 +286,6 @@ export const handlePasteInput = (
       char = "!";
     }
 
-    dispatch({ type: "RESET_HISTORY_NAVIGATION" });
     dispatch({ type: "INSERT_TEXT", payload: char });
 
     processSelectorInput(state, dispatch, char);
@@ -342,6 +342,8 @@ export const handleCommandSelect = (
           dispatch({ type: "SET_SHOW_HELP", payload: true });
         } else if (command === "status") {
           dispatch({ type: "SET_SHOW_STATUS_COMMAND", payload: true });
+        } else if (command === "plugin") {
+          dispatch({ type: "SET_SHOW_PLUGIN_MANAGER", payload: true });
         }
       }
     })();
@@ -458,8 +460,12 @@ export const handleSelectorInput = (
     return true;
   }
 
-  if (input === " " && state.showFileSelector) {
-    dispatch({ type: "CANCEL_FILE_SELECTOR" });
+  if (input === " ") {
+    if (state.showFileSelector) {
+      dispatch({ type: "CANCEL_FILE_SELECTOR" });
+    } else if (state.showCommandSelector) {
+      dispatch({ type: "CANCEL_COMMAND_SELECTOR" });
+    }
   }
 
   if (
@@ -511,9 +517,26 @@ export const handleNormalInput = async (
 
   if (key.upArrow) {
     if (state.history.length === 0) {
-      const history = await PromptHistoryManager.getHistory(
-        callbacks.sessionId,
-      );
+      let sessionIds: string[] | undefined = callbacks.sessionId
+        ? [callbacks.sessionId]
+        : undefined;
+
+      if (callbacks.getFullMessageThread) {
+        try {
+          const thread = await callbacks.getFullMessageThread();
+          sessionIds = thread.sessionIds;
+        } catch (error) {
+          callbacks.logger?.error(
+            "Failed to fetch ancestor session IDs",
+            error,
+          );
+        }
+      }
+
+      const history = await PromptHistoryManager.getHistory({
+        sessionId: sessionIds,
+        workdir: callbacks.workdir,
+      });
       dispatch({ type: "SET_HISTORY_ENTRIES", payload: history });
     }
     dispatch({ type: "NAVIGATE_HISTORY", payload: "up" });
@@ -536,7 +559,6 @@ export const handleNormalInput = async (
       const newInputText = beforeCursor + afterCursor;
 
       dispatch({ type: "DELETE_CHAR" });
-      dispatch({ type: "RESET_HISTORY_NAVIGATION" });
 
       checkForAtDeletion(state, dispatch, newCursorPosition);
       checkForSlashDeletion(state, dispatch, newCursorPosition);
@@ -657,7 +679,8 @@ export const handleInput = async (
         state.showMcpManager ||
         state.showRewindManager ||
         state.showHelp ||
-        state.showStatusCommand
+        state.showStatusCommand ||
+        state.showPluginManager
       )
     ) {
       callbacks.onAbortMessage?.();
@@ -678,14 +701,16 @@ export const handleInput = async (
     state.showMcpManager ||
     state.showRewindManager ||
     state.showHelp ||
-    state.showStatusCommand
+    state.showStatusCommand ||
+    state.showPluginManager
   ) {
     if (
       state.showBackgroundTaskManager ||
       state.showMcpManager ||
       state.showRewindManager ||
       state.showHelp ||
-      state.showStatusCommand
+      state.showStatusCommand ||
+      state.showPluginManager
     ) {
       return true;
     }