wave-code 0.2.0 → 0.4.0

package/src/index.ts

@@ -1,306 +1,267 @@
 import yargs from "yargs";
 import { hideBin } from "yargs/helpers";
 import { startCli } from "./cli.js";
-import {
-  listSessions,
-  getSessionFilePath,
-  getFirstMessageContent,
-} from "wave-agent-sdk";
+import { Scope } 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) => {
-              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",
-                    });
+  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(
+              "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;
                   },
-                  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"],
-                  default: "user",
+                );
+              },
+            )
+            .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 { installPluginCommand } = await import(
-                "./commands/plugin/install.js"
-              );
-              await installPluginCommand(
-                argv as {
-                  plugin: string;
-                  scope?: "user" | "project" | "local";
-                },
-              );
-            },
-          )
-          .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",
+              },
+              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 { enablePluginCommand } = await import(
-                "./commands/plugin/enable.js"
-              );
-              await enablePluginCommand(
-                argv as {
-                  plugin: string;
-                  scope: "user" | "project" | "local";
-                },
-              );
-            },
-          )
-          .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",
-                });
-            },
-            async (argv) => {
-              const { disablePluginCommand } = await import(
-                "./commands/plugin/disable.js"
-              );
-              await disablePluginCommand(
-                argv as {
-                  plugin: string;
-                  scope: "user" | "project" | "local";
-                },
-              );
-            },
-          )
-          .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);
+              },
+              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);
+            }
+          }
+        },
+      )
+      .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();
 
-      if (sessions.length === 0) {
-        console.log(`No sessions found for workdir: ${currentWorkdir}`);
+    // 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;
       }
+      // Continue with the selected session
+      return startCli({
+        restoreSessionId: selectedSessionId,
+        bypassPermissions: argv.dangerouslySkipPermissions,
+        pluginDirs: argv.pluginDir as string[],
+      });
+    }
 
-      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) + "...";
-        }
-
-        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`);
-      }
-
-      return;
-    } 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 as string[],
+      });
     }
-  }
 
-  // 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 as string[],
     });
+  } catch (error) {
+    console.error("Failed to start WAVE Code:", error);
+    process.exit(1);
   }
-
-  await startCli({
-    restoreSessionId: argv.restore,
-    continueLastSession: argv.continue,
-    bypassPermissions: argv.dangerouslySkipPermissions,
-    pluginDirs: argv.pluginDir as string[],
-  });
 }
 
 // Export CLI function