wave-code 0.14.0 → 0.14.1

package/src/contexts/useChat.tsx

@@ -117,6 +117,8 @@ export interface ChatContextType {
   workingDirectory: string;
   version?: string;
   workdir?: string;
+  // Agent recreation (e.g. after plugin install)
+  recreateAgent: () => void;
 }
 
 const ChatContext = createContext<ChatContextType | null>(null);
@@ -332,8 +334,11 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   );
 
   // Initialize AI manager
-  useEffect(() => {
-    const initializeAgent = async () => {
+  const initializeAgent = useCallback(
+    async (restoreSessionIdOverride?: string) => {
+      const effectiveRestoreSessionId =
+        restoreSessionIdOverride ?? restoreSessionId;
+
       const callbacks: AgentCallbacks = {
         onMessagesChange: () => {
           throttledSetMessages();
@@ -400,7 +405,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
 
         const agent = await Agent.create({
           callbacks,
-          restoreSessionId,
+          restoreSessionId: effectiveRestoreSessionId,
           continueLastSession,
           logger,
           permissionMode:
@@ -442,25 +447,53 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
       } catch (error) {
         console.error("Failed to initialize AI manager:", error);
       }
-    };
+    },
+    [
+      restoreSessionId,
+      continueLastSession,
+      bypassPermissions,
+      showConfirmation,
+      pluginDirs,
+      tools,
+      allowedTools,
+      disallowedTools,
+      workdir,
+      worktreeSession,
+      model,
+      initialPermissionMode,
+      throttledSetMessages,
+      throttledSetTokens,
+    ],
+  );
 
+  // 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
+      }
+    }
+    agentRef.current = null;
+    setMessages([]);
+    setMcpServers([]);
+    setSlashCommands([]);
+    setSessionId("");
+    setIsLoading(false);
+    setlatestTotalTokens(0);
+    setIsCommandRunning(false);
+    setIsCompressing(false);
+    if (currentSessionId) {
+      initializeAgent(currentSessionId);
+    }
+  }, [initializeAgent]);
+
+  // Run initial agent creation
+  useEffect(() => {
     initializeAgent();
-  }, [
-    restoreSessionId,
-    continueLastSession,
-    bypassPermissions,
-    showConfirmation,
-    pluginDirs,
-    tools,
-    allowedTools,
-    disallowedTools,
-    workdir,
-    worktreeSession,
-    model,
-    initialPermissionMode,
-    throttledSetMessages,
-    throttledSetTokens,
-  ]);
+  }, [initializeAgent]);
 
   // Cleanup on unmount
   useEffect(() => {
@@ -751,6 +784,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     workingDirectory,
     version,
     workdir,
+    recreateAgent,
   };
 
   return (

package/src/hooks/usePluginManager.ts

@@ -11,7 +11,9 @@ import {
   PluginManagerContextType,
 } from "../components/PluginManagerTypes.js";
 
-export function usePluginManager(): PluginManagerContextType {
+export function usePluginManager(options?: {
+  onPluginInstalled?: () => void;
+}): PluginManagerContextType {
   const [state, setState] = useState<PluginManagerState>({
     currentView: "DISCOVER",
     selectedId: null,
@@ -141,16 +143,20 @@ export function usePluginManager(): PluginManagerContextType {
   }, []);
 
   const addMarketplace = useCallback(
-    async (source: string) => {
+    async (source: string, scope: "user" | "project" | "local" = "user") => {
       clearPluginFeedback();
       setState((prev: PluginManagerState) => ({
         ...prev,
         isLoading: true,
       }));
       try {
-        await pluginCore.addMarketplace(source);
+        await pluginCore.addMarketplace(source, scope);
         await refresh();
-        setSuccessMessage(`Marketplace added successfully`);
+        setSuccessMessage(`Marketplace added successfully (${scope} scope)`);
+        setState((prev: PluginManagerState) => ({
+          ...prev,
+          currentView: "MARKETPLACES",
+        }));
       } catch (error) {
         setState((prev: PluginManagerState) => ({
           ...prev,
@@ -163,16 +169,20 @@ export function usePluginManager(): PluginManagerContextType {
   );
 
   const removeMarketplace = useCallback(
-    async (name: string) => {
+    async (name: string, scope?: "user" | "project" | "local") => {
       clearPluginFeedback();
       setState((prev: PluginManagerState) => ({
         ...prev,
         isLoading: true,
       }));
       try {
-        await pluginCore.removeMarketplace(name);
+        await pluginCore.removeMarketplace(name, scope);
         await refresh();
         setSuccessMessage(`Marketplace '${name}' removed successfully`);
+        setState((prev: PluginManagerState) => ({
+          ...prev,
+          currentView: "MARKETPLACES",
+        }));
       } catch (error) {
         setState((prev: PluginManagerState) => ({
           ...prev,
@@ -222,6 +232,7 @@ export function usePluginManager(): PluginManagerContextType {
         await pluginCore.installPlugin(pluginId, scope);
         await refresh();
         setSuccessMessage(`Plugin '${name}' installed successfully`);
+        options?.onPluginInstalled?.();
       } catch (error) {
         setState((prev: PluginManagerState) => ({
           ...prev,
@@ -230,7 +241,7 @@ export function usePluginManager(): PluginManagerContextType {
         }));
       }
     },
-    [pluginCore, refresh, clearPluginFeedback, setSuccessMessage],
+    [pluginCore, refresh, clearPluginFeedback, setSuccessMessage, options],
   );
 
   const uninstallPlugin = useCallback(

package/src/reducers/backgroundTaskManagerReducer.ts

@@ -0,0 +1,60 @@
+export interface Task {
+  id: string;
+  type: string;
+  description?: string;
+  status: "running" | "completed" | "failed" | "killed";
+  startTime: number;
+  exitCode?: number;
+  runtime?: number;
+  outputPath?: string;
+}
+
+export interface DetailOutput {
+  stdout: string;
+  stderr: string;
+  status: string;
+  outputPath?: string;
+}
+
+export interface BackgroundTaskManagerState {
+  tasks: Task[];
+  selectedIndex: number;
+  viewMode: "list" | "detail";
+  detailTaskId: string | null;
+  detailOutput: DetailOutput | null;
+}
+
+export type BackgroundTaskManagerAction =
+  | { type: "SET_TASKS"; tasks: Task[] }
+  | { type: "SELECT_INDEX"; index: number }
+  | { type: "SET_VIEW_MODE"; mode: "list" | "detail" }
+  | { type: "SET_DETAIL_TASK_ID"; id: string | null }
+  | { type: "SET_DETAIL_OUTPUT"; output: DetailOutput | null }
+  | { type: "RESET_DETAIL" };
+
+export function backgroundTaskManagerReducer(
+  state: BackgroundTaskManagerState,
+  action: BackgroundTaskManagerAction,
+): BackgroundTaskManagerState {
+  switch (action.type) {
+    case "SET_TASKS":
+      return { ...state, tasks: action.tasks };
+    case "SELECT_INDEX":
+      return { ...state, selectedIndex: action.index };
+    case "SET_VIEW_MODE":
+      return { ...state, viewMode: action.mode };
+    case "SET_DETAIL_TASK_ID":
+      return { ...state, detailTaskId: action.id };
+    case "SET_DETAIL_OUTPUT":
+      return { ...state, detailOutput: action.output };
+    case "RESET_DETAIL":
+      return {
+        ...state,
+        viewMode: "list",
+        detailTaskId: null,
+        detailOutput: null,
+      };
+    default:
+      return state;
+  }
+}

package/src/reducers/marketplaceAddFormReducer.ts

@@ -0,0 +1,35 @@
+export interface MarketplaceAddFormState {
+  source: string;
+  scopeIndex: number;
+  step: "source" | "scope";
+}
+
+export type MarketplaceAddFormAction =
+  | { type: "SET_SOURCE"; source: string }
+  | { type: "SET_SCOPE_INDEX"; index: number }
+  | { type: "SET_STEP"; step: "source" | "scope" }
+  | { type: "INSERT_CHAR"; text: string }
+  | { type: "DELETE_CHAR" }
+  | { type: "BACK_TO_SOURCE" };
+
+export function marketplaceAddFormReducer(
+  state: MarketplaceAddFormState,
+  action: MarketplaceAddFormAction,
+): MarketplaceAddFormState {
+  switch (action.type) {
+    case "SET_SOURCE":
+      return { ...state, source: action.source };
+    case "SET_SCOPE_INDEX":
+      return { ...state, scopeIndex: action.index };
+    case "SET_STEP":
+      return { ...state, step: action.step };
+    case "INSERT_CHAR":
+      return { ...state, source: state.source + action.text };
+    case "DELETE_CHAR":
+      return { ...state, source: state.source.slice(0, -1) };
+    case "BACK_TO_SOURCE":
+      return { ...state, step: "source" };
+    default:
+      return state;
+  }
+}

package/src/reducers/mcpManagerReducer.ts

@@ -0,0 +1,31 @@
+export interface McpManagerState {
+  selectedIndex: number;
+  viewMode: "list" | "detail";
+}
+
+export type McpManagerAction =
+  | { type: "MOVE_UP"; serverCount: number }
+  | { type: "MOVE_DOWN"; serverCount: number }
+  | { type: "SET_VIEW_MODE"; viewMode: "list" | "detail" };
+
+export function mcpManagerReducer(
+  state: McpManagerState,
+  action: McpManagerAction,
+): McpManagerState {
+  switch (action.type) {
+    case "MOVE_UP":
+      return { ...state, selectedIndex: Math.max(0, state.selectedIndex - 1) };
+    case "MOVE_DOWN":
+      return {
+        ...state,
+        selectedIndex: Math.min(
+          action.serverCount - 1,
+          state.selectedIndex + 1,
+        ),
+      };
+    case "SET_VIEW_MODE":
+      return { ...state, viewMode: action.viewMode };
+    default:
+      return state;
+  }
+}

package/src/reducers/pluginDetailReducer.ts

@@ -0,0 +1,58 @@
+export interface PluginDetailState {
+  selectedScopeIndex: number;
+  selectedActionIndex: number;
+}
+
+export type PluginDetailAction =
+  | { type: "SELECT_SCOPE_INDEX"; index: number }
+  | { type: "SELECT_ACTION_INDEX"; index: number }
+  | { type: "MOVE_SCOPE_UP"; maxIndex: number }
+  | { type: "MOVE_SCOPE_DOWN"; maxIndex: number }
+  | { type: "MOVE_ACTION_UP"; maxIndex: number }
+  | { type: "MOVE_ACTION_DOWN"; maxIndex: number };
+
+export function pluginDetailReducer(
+  state: PluginDetailState,
+  action: PluginDetailAction,
+): PluginDetailState {
+  switch (action.type) {
+    case "SELECT_SCOPE_INDEX":
+      return { ...state, selectedScopeIndex: action.index };
+    case "SELECT_ACTION_INDEX":
+      return { ...state, selectedActionIndex: action.index };
+    case "MOVE_SCOPE_UP":
+      return {
+        ...state,
+        selectedScopeIndex:
+          state.selectedScopeIndex > 0
+            ? state.selectedScopeIndex - 1
+            : action.maxIndex,
+      };
+    case "MOVE_SCOPE_DOWN":
+      return {
+        ...state,
+        selectedScopeIndex:
+          state.selectedScopeIndex < action.maxIndex
+            ? state.selectedScopeIndex + 1
+            : 0,
+      };
+    case "MOVE_ACTION_UP":
+      return {
+        ...state,
+        selectedActionIndex:
+          state.selectedActionIndex > 0
+            ? state.selectedActionIndex - 1
+            : action.maxIndex,
+      };
+    case "MOVE_ACTION_DOWN":
+      return {
+        ...state,
+        selectedActionIndex:
+          state.selectedActionIndex < action.maxIndex
+            ? state.selectedActionIndex + 1
+            : 0,
+      };
+    default:
+      return state;
+  }
+}