wave-code 0.2.1 → 0.5.0

package/dist/hooks/usePluginManager.js

@@ -0,0 +1,227 @@
+import { useState, useCallback, useEffect, useMemo } from "react";
+import { MarketplaceService, PluginScopeManager, ConfigurationService, PluginManager, } from "wave-agent-sdk";
+export function usePluginManager() {
+    const [state, setState] = useState({
+        currentView: "DISCOVER",
+        selectedId: null,
+        isLoading: true,
+        error: null,
+        searchQuery: "",
+    });
+    const [marketplaces, setMarketplaces] = useState([]);
+    const [installedPlugins, setInstalledPlugins] = useState([]);
+    const [discoverablePlugins, setDiscoverablePlugins] = useState([]);
+    const marketplaceService = useMemo(() => new MarketplaceService(), []);
+    const configurationService = useMemo(() => new ConfigurationService(), []);
+    const pluginManager = useMemo(() => new PluginManager({
+        workdir: process.cwd(),
+        configurationService,
+    }), [configurationService]);
+    const pluginScopeManager = useMemo(() => new PluginScopeManager({
+        workdir: process.cwd(),
+        configurationService,
+        pluginManager,
+    }), [configurationService, pluginManager]);
+    const refresh = useCallback(async () => {
+        setState((prev) => ({
+            ...prev,
+            isLoading: true,
+            error: null,
+        }));
+        try {
+            const [mks, installed, enabledMap] = await Promise.all([
+                marketplaceService.listMarketplaces(),
+                marketplaceService.getInstalledPlugins(),
+                Promise.resolve(pluginScopeManager.getMergedEnabledPlugins()),
+            ]);
+            setMarketplaces(mks);
+            const allInstalledWithEnabled = installed.plugins.map((p) => {
+                const pluginId = `${p.name}@${p.marketplace}`;
+                return {
+                    ...p,
+                    enabled: !!enabledMap[pluginId],
+                    scope: pluginScopeManager.findPluginScope(pluginId) || undefined,
+                };
+            });
+            // Only show enabled plugins in the "Installed" view
+            setInstalledPlugins(allInstalledWithEnabled.filter((p) => p.enabled));
+            const allDiscoverable = [];
+            for (const mk of mks) {
+                try {
+                    const manifest = await marketplaceService.loadMarketplaceManifest(marketplaceService.getMarketplacePath(mk));
+                    manifest.plugins.forEach((p) => {
+                        const pluginId = `${p.name}@${mk.name}`;
+                        const isInstalled = installed.plugins.find((ip) => ip.name === p.name && ip.marketplace === mk.name);
+                        const isEnabled = !!enabledMap[pluginId];
+                        // Show in Discover if not installed OR if installed but not enabled in current scope
+                        if (!isInstalled || !isEnabled) {
+                            allDiscoverable.push({
+                                ...p,
+                                marketplace: mk.name,
+                                installed: !!isInstalled,
+                            });
+                        }
+                    });
+                }
+                catch {
+                    // Skip marketplaces that fail to load
+                }
+            }
+            setDiscoverablePlugins(allDiscoverable);
+            setState((prev) => ({ ...prev, isLoading: false }));
+        }
+        catch (error) {
+            setState((prev) => ({
+                ...prev,
+                isLoading: false,
+                error: error instanceof Error ? error.message : String(error),
+            }));
+        }
+    }, [marketplaceService, pluginScopeManager]);
+    useEffect(() => {
+        refresh();
+    }, [refresh]);
+    const setView = useCallback((view) => {
+        setState((prev) => ({ ...prev, currentView: view }));
+    }, []);
+    const setSelectedId = useCallback((id) => {
+        setState((prev) => ({ ...prev, selectedId: id }));
+    }, []);
+    const addMarketplace = useCallback(async (source) => {
+        setState((prev) => ({
+            ...prev,
+            isLoading: true,
+            error: null,
+        }));
+        try {
+            await marketplaceService.addMarketplace(source);
+            await refresh();
+        }
+        catch (error) {
+            setState((prev) => ({
+                ...prev,
+                isLoading: false,
+                error: error instanceof Error ? error.message : String(error),
+            }));
+        }
+    }, [marketplaceService, refresh]);
+    const removeMarketplace = useCallback(async (name) => {
+        setState((prev) => ({
+            ...prev,
+            isLoading: true,
+            error: null,
+        }));
+        try {
+            await marketplaceService.removeMarketplace(name);
+            await refresh();
+        }
+        catch (error) {
+            setState((prev) => ({
+                ...prev,
+                isLoading: false,
+                error: error instanceof Error ? error.message : String(error),
+            }));
+        }
+    }, [marketplaceService, refresh]);
+    const updateMarketplace = useCallback(async (name) => {
+        setState((prev) => ({
+            ...prev,
+            isLoading: true,
+            error: null,
+        }));
+        try {
+            await marketplaceService.updateMarketplace(name);
+            await refresh();
+        }
+        catch (error) {
+            setState((prev) => ({
+                ...prev,
+                isLoading: false,
+                error: error instanceof Error ? error.message : String(error),
+            }));
+        }
+    }, [marketplaceService, refresh]);
+    const installPlugin = useCallback(async (name, marketplace, scope = "project") => {
+        setState((prev) => ({
+            ...prev,
+            isLoading: true,
+            error: null,
+        }));
+        try {
+            const pluginId = `${name}@${marketplace}`;
+            const workdir = process.cwd();
+            await marketplaceService.installPlugin(pluginId, workdir);
+            await pluginScopeManager.enablePlugin(scope, pluginId);
+            await refresh();
+        }
+        catch (error) {
+            setState((prev) => ({
+                ...prev,
+                isLoading: false,
+                error: error instanceof Error ? error.message : String(error),
+            }));
+        }
+    }, [marketplaceService, pluginScopeManager, refresh]);
+    const uninstallPlugin = useCallback(async (name, marketplace) => {
+        setState((prev) => ({
+            ...prev,
+            isLoading: true,
+            error: null,
+        }));
+        try {
+            const pluginId = `${name}@${marketplace}`;
+            const workdir = process.cwd();
+            // 1. Remove from global registry and potentially clean up cache
+            await marketplaceService.uninstallPlugin(pluginId, workdir);
+            // 2. Find the scope where it's currently enabled and remove it from there
+            const scope = pluginScopeManager.findPluginScope(pluginId);
+            if (scope) {
+                await configurationService.removeEnabledPlugin(workdir, scope, pluginId);
+            }
+            await refresh();
+        }
+        catch (error) {
+            setState((prev) => ({
+                ...prev,
+                isLoading: false,
+                error: error instanceof Error ? error.message : String(error),
+            }));
+        }
+    }, [configurationService, marketplaceService, pluginScopeManager, refresh]);
+    const updatePlugin = useCallback(async (name, marketplace) => {
+        setState((prev) => ({
+            ...prev,
+            isLoading: true,
+            error: null,
+        }));
+        try {
+            const pluginId = `${name}@${marketplace}`;
+            await marketplaceService.updatePlugin(pluginId);
+            await refresh();
+        }
+        catch (error) {
+            setState((prev) => ({
+                ...prev,
+                isLoading: false,
+                error: error instanceof Error ? error.message : String(error),
+            }));
+        }
+    }, [marketplaceService, refresh]);
+    return {
+        state,
+        marketplaces,
+        installedPlugins,
+        discoverablePlugins,
+        actions: {
+            setView,
+            setSelectedId,
+            addMarketplace,
+            removeMarketplace,
+            updateMarketplace,
+            installPlugin,
+            uninstallPlugin,
+            updatePlugin,
+            refresh,
+        },
+    };
+}

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAUA,wBAAsB,IAAI,kBAqSzB;AAGD,OAAO,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAC;AAGpC,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAG3C,OAAO,EACL,kBAAkB,EAClB,gBAAgB,EAChB,iBAAiB,EACjB,KAAK,oBAAoB,GAC1B,MAAM,sBAAsB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAMA,wBAAsB,IAAI,kBAkQzB;AAGD,OAAO,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAC;AAGpC,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAG3C,OAAO,EACL,kBAAkB,EAClB,gBAAgB,EAChB,iBAAiB,EACjB,KAAK,oBAAoB,GAC1B,MAAM,sBAAsB,CAAC"}

package/dist/index.js

@@ -1,208 +1,181 @@
 import yargs from "yargs";
 import { hideBin } from "yargs/helpers";
 import { startCli } from "./cli.js";
-import { listSessions, getSessionFilePath, getFirstMessageContent, } from "wave-agent-sdk";
 // Export main function for external use
 export async function main() {
-    const argv = await yargs(hideBin(process.argv))
-        .option("restore", {
-        alias: "r",
-        description: "Restore session by ID",
-        type: "string",
-        global: false,
-    })
-        .option("continue", {
-        alias: "c",
-        description: "Continue from last session",
-        type: "boolean",
-        global: false,
-    })
-        .option("print", {
-        alias: "p",
-        description: "Print response without interactive mode",
-        type: "string",
-        global: false,
-    })
-        .option("show-stats", {
-        description: "Show timing and usage statistics in print mode",
-        type: "boolean",
-        global: false,
-    })
-        .option("list-sessions", {
-        description: "List all available sessions",
-        type: "boolean",
-        global: false,
-    })
-        .option("dangerously-skip-permissions", {
-        description: "Skip all permission checks (dangerous)",
-        type: "boolean",
-        default: false,
-        global: false,
-    })
-        .option("plugin-dir", {
-        description: "Load a plugin from a specific directory",
-        type: "array",
-        string: true,
-        global: false,
-    })
-        .command("plugin", "Manage plugins and marketplaces", (yargs) => {
-        return yargs
-            .help()
-            .command("marketplace", "Manage plugin marketplaces", (yargs) => {
+    try {
+        const argv = await yargs(hideBin(process.argv))
+            .option("restore", {
+            alias: "r",
+            description: "Restore session by ID (or list sessions if no ID provided)",
+            type: "string",
+            global: false,
+        })
+            .option("continue", {
+            alias: "c",
+            description: "Continue from last session",
+            type: "boolean",
+            global: false,
+        })
+            .option("print", {
+            alias: "p",
+            description: "Print response without interactive mode",
+            type: "string",
+            global: false,
+        })
+            .option("show-stats", {
+            description: "Show timing and usage statistics in print mode",
+            type: "boolean",
+            global: false,
+        })
+            .option("dangerously-skip-permissions", {
+            description: "Skip all permission checks (dangerous)",
+            type: "boolean",
+            default: false,
+            global: false,
+        })
+            .option("plugin-dir", {
+            description: "Load a plugin from a specific directory",
+            type: "array",
+            string: true,
+            global: false,
+        })
+            .command("plugin", "Manage plugins and 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)",
+                .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);
+                })
+                    .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);
+                })
+                    .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 { addMarketplaceCommand } = await import("./commands/plugin/marketplace.js");
-                await addMarketplaceCommand(argv);
+                const { installPluginCommand } = await import("./commands/plugin/install.js");
+                await installPluginCommand(argv);
+            })
+                .command("list", "List all available plugins from marketplaces", {}, async () => {
+                const { listPluginsCommand } = await import("./commands/plugin/list.js");
+                await listPluginsCommand();
             })
-                .command("update [name]", "Update registered marketplace(s)", (yargs) => {
-                return yargs.positional("name", {
-                    describe: "Name of the marketplace to update",
+                .command("uninstall <plugin>", "Uninstall a plugin", (yargs) => {
+                return yargs.positional("plugin", {
+                    describe: "Plugin to uninstall (format: name@marketplace)",
                     type: "string",
                 });
             }, async (argv) => {
-                const { updateMarketplaceCommand } = await import("./commands/plugin/marketplace.js");
-                await updateMarketplaceCommand(argv);
+                const { uninstallPluginCommand } = await import("./commands/plugin/uninstall.js");
+                await uninstallPluginCommand(argv);
             })
-                .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"],
-                default: "user",
-                type: "string",
-            });
-        }, async (argv) => {
-            const { installPluginCommand } = await import("./commands/plugin/install.js");
-            await installPluginCommand(argv);
-        })
-            .command("list", "List all available plugins from marketplaces", {}, async () => {
-            const { listPluginsCommand } = await import("./commands/plugin/list.js");
-            await listPluginsCommand();
-        })
-            .command("enable <plugin>", "Enable a plugin in a specific scope", (yargs) => {
-            return yargs
-                .positional("plugin", {
-                describe: "Plugin ID (format: name@marketplace)",
-                type: "string",
-            })
-                .option("scope", {
-                alias: "s",
-                describe: "Scope to enable the plugin in",
-                choices: ["user", "project", "local"],
-                default: "user",
-                type: "string",
-            });
-        }, async (argv) => {
-            const { enablePluginCommand } = await import("./commands/plugin/enable.js");
-            await enablePluginCommand(argv);
-        })
-            .command("disable <plugin>", "Disable a plugin in a specific scope", (yargs) => {
-            return yargs
-                .positional("plugin", {
-                describe: "Plugin ID (format: name@marketplace)",
-                type: "string",
-            })
-                .option("scope", {
-                alias: "s",
-                describe: "Scope to disable the plugin in",
-                choices: ["user", "project", "local"],
-                default: "user",
-                type: "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);
             });
         }, async (argv) => {
-            const { disablePluginCommand } = await import("./commands/plugin/disable.js");
-            await disablePluginCommand(argv);
-        })
-            .demandCommand(1, "Please specify a plugin subcommand");
-    }, () => { })
-        .version()
-        .alias("v", "version")
-        .example("$0", "Start CLI with default settings")
-        .example("$0 --restore session_123", "Restore specific session")
-        .example("$0 --continue", "Continue from last session")
-        .example("$0 --print 'Hello'", "Send message in print mode")
-        .example("$0 -p 'Hello' --show-stats", "Send message in print mode with statistics")
-        .example("$0 --list-sessions", "List all available sessions")
-        .help("h")
-        .recommendCommands()
-        .strict()
-        .parseAsync();
-    // Handle list sessions command
-    if (argv.listSessions) {
-        try {
-            const currentWorkdir = process.cwd();
-            const sessions = await listSessions(currentWorkdir);
-            if (sessions.length === 0) {
-                console.log(`No sessions found for workdir: ${currentWorkdir}`);
-                return;
-            }
-            console.log(`Available sessions for: ${currentWorkdir}`);
-            console.log("==========================================");
-            // Get last 5 sessions
-            const lastSessions = sessions.slice(0, 5);
-            for (const session of lastSessions) {
-                const lastActiveAt = new Date(session.lastActiveAt).toLocaleString();
-                const filePath = await getSessionFilePath(session.id, session.workdir);
-                // Get first message content
-                const firstMessageContent = await getFirstMessageContent(session.id, session.workdir);
-                // Truncate content if too long
-                let truncatedContent = firstMessageContent || "No first message content";
-                if (truncatedContent.length > 30) {
-                    truncatedContent = truncatedContent.substring(0, 30) + "...";
+            // 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);
                 }
-                console.log(`ID: ${session.id}`);
-                console.log(`  Workdir: ${session.workdir}`);
-                console.log(`  File Path: ${filePath}`);
-                console.log(`  Last Active: ${lastActiveAt}`);
-                console.log(`  Last Message Tokens: ${session.latestTotalTokens}`);
-                console.log(`  First Message: ${truncatedContent}`);
-                console.log("");
             }
-            if (sessions.length > 5) {
-                console.log(`... and ${sessions.length - 5} more sessions`);
+        })
+            .version()
+            .alias("v", "version")
+            .example("$0", "Start CLI with default settings")
+            .example("$0 --restore session_123", "Restore specific session")
+            .example("$0 --continue", "Continue from last session")
+            .example("$0 --print 'Hello'", "Send message in print mode")
+            .example("$0 -p 'Hello' --show-stats", "Send message in print mode with statistics")
+            .help("h")
+            .recommendCommands()
+            .strict()
+            .parseAsync();
+        // Handle restore session command
+        if (argv.restore === "" ||
+            (process.argv.includes("-r") && argv.restore === undefined) ||
+            (process.argv.includes("--restore") && argv.restore === undefined)) {
+            // Interactive session selection
+            const { startSessionSelectorCli } = await import("./session-selector-cli.js");
+            const selectedSessionId = await startSessionSelectorCli();
+            if (!selectedSessionId) {
+                return;
             }
-            return;
+            // Continue with the selected session
+            return startCli({
+                restoreSessionId: selectedSessionId,
+                bypassPermissions: argv.dangerouslySkipPermissions,
+                pluginDirs: argv.pluginDir,
+            });
         }
-        catch (error) {
-            console.error("Failed to list sessions:", error);
-            process.exit(1);
+        // Handle print mode directly
+        if (argv.print !== undefined) {
+            const { startPrintCli } = await import("./print-cli.js");
+            return startPrintCli({
+                restoreSessionId: argv.restore,
+                continueLastSession: argv.continue,
+                message: argv.print,
+                showStats: argv.showStats,
+                bypassPermissions: argv.dangerouslySkipPermissions,
+                pluginDirs: argv.pluginDir,
+            });
         }
-    }
-    // Handle print mode directly
-    if (argv.print !== undefined) {
-        const { startPrintCli } = await import("./print-cli.js");
-        return startPrintCli({
+        await startCli({
             restoreSessionId: argv.restore,
             continueLastSession: argv.continue,
-            message: argv.print,
-            showStats: argv.showStats,
             bypassPermissions: argv.dangerouslySkipPermissions,
             pluginDirs: argv.pluginDir,
         });
     }
-    await startCli({
-        restoreSessionId: argv.restore,
-        continueLastSession: argv.continue,
-        bypassPermissions: argv.dangerouslySkipPermissions,
-        pluginDirs: argv.pluginDir,
-    });
+    catch (error) {
+        console.error("Failed to start WAVE Code:", error);
+        process.exit(1);
+    }
 }
 // Export CLI function
 export { startCli } from "./cli.js";