feat(cli): add plugin support

Author: hopelynd

packages/cli/src/commands/catalog.ts

@@ -35,6 +35,10 @@ import consoleCall from "./console/call.ts";
 import usageFree from "./usage/free.ts";
 import pipelineRun from "./pipeline/run.ts";
 import pipelineValidate from "./pipeline/validate.ts";
+import pluginsList from "./plugins/list.ts";
+import pluginsInstall from "./plugins/install.ts";
+import pluginsLink from "./plugins/link.ts";
+import pluginsRemove from "./plugins/remove.ts";
 
 /** Command registry map (no dependency on registry.ts — safe for build-time import). */
 export const commands: Record<string, Command> = {
@@ -69,6 +73,10 @@ export const commands: Record<string, Command> = {
   "usage free": usageFree,
   "pipeline run": pipelineRun,
   "pipeline validate": pipelineValidate,
+  "plugins list": pluginsList,
+  "plugins install": pluginsInstall,
+  "plugins link": pluginsLink,
+  "plugins remove": pluginsRemove,
   "config show": configShow,
   "config set": configSet,
   "config export-schema": configExportSchema,

packages/cli/src/commands/config/export-schema.ts

@@ -1,13 +1,14 @@
-import { defineCommand, generateToolSchema } from "bailian-cli-core";
+import { defineCommand, generateToolSchema, type Command } from "bailian-cli-core";
 import type { Config } from "bailian-cli-core";
 import type { GlobalFlags } from "bailian-cli-core";
 import { BailianError } from "bailian-cli-core";
 import { ExitCode } from "bailian-cli-core";
+import { loadCommandCatalog } from "../../load-commands.ts";
 
 /**
  * Commands that are infrastructure/auth-related and not suitable as Agent tools.
  */
-const SKIP_PREFIXES = ["auth ", "config ", "update"];
+const SKIP_PREFIXES = ["auth ", "config ", "update", "plugins "];
 
 export default defineCommand({
   name: "config export-schema",
@@ -22,7 +23,7 @@ export default defineCommand({
   ],
   examples: ["bl config export-schema", 'bl config export-schema --command "video generate"'],
   async run(config: Config, flags: GlobalFlags) {
-    const { commands } = await import("../catalog.ts");
+    const { commands } = await loadCommandCatalog();
     const targetCommand = flags.command as string | undefined;
 
     if (targetCommand) {
@@ -35,8 +36,7 @@ export default defineCommand({
       return;
     }
 
-    // Export all suitable commands
-    const allCommands = Object.values(commands);
+    const allCommands = Object.values(commands) as Command[];
     const schemas = allCommands
       .filter((c) => !SKIP_PREFIXES.some((p) => c.name.startsWith(p)))
       .map((c) => generateToolSchema(c));

packages/cli/src/commands/plugins/install.ts

@@ -0,0 +1,40 @@
+import {
+  BailianError,
+  ExitCode,
+  defineCommand,
+  type Config,
+  type GlobalFlags,
+} from "bailian-cli-core";
+import { installPlugin } from "../../plugins/manager.ts";
+import { resetCommandCatalogCache } from "../../load-commands.ts";
+import { createRegistry, resetRegistry } from "../../registry.ts";
+
+export default defineCommand({
+  name: "plugins install",
+  description: "Install a bailian-cli plugin package into ~/.bailian/plugins",
+  usage: "bl plugins install <package>",
+  options: [],
+  examples: ["bl plugins install @alife/bailian-agent", "bl plugins install bailian-agent"],
+  async run(_config: Config, flags: GlobalFlags) {
+    const positional = flags._positional as string[] | undefined;
+    const packageSpec = positional?.[0];
+    if (!packageSpec) {
+      throw new BailianError(
+        "Missing plugin package name.",
+        ExitCode.USAGE,
+        "bl plugins install <package>",
+      );
+    }
+
+    const name = await installPlugin(packageSpec);
+    resetCommandCatalogCache();
+    resetRegistry();
+    await createRegistry();
+
+    if (flags.output === "json") {
+      process.stdout.write(JSON.stringify({ installed: name }, null, 2) + "\n");
+      return;
+    }
+    process.stdout.write(`Installed plugin: ${name}\n`);
+  },
+});

packages/cli/src/commands/plugins/link.ts

@@ -0,0 +1,35 @@
+import {
+  BailianError,
+  ExitCode,
+  defineCommand,
+  type Config,
+  type GlobalFlags,
+} from "bailian-cli-core";
+import { linkPlugin } from "../../plugins/manager.ts";
+import { resetCommandCatalogCache } from "../../load-commands.ts";
+import { createRegistry, resetRegistry } from "../../registry.ts";
+
+export default defineCommand({
+  name: "plugins link",
+  description: "Link a local bailian-cli plugin directory",
+  usage: "bl plugins link <path>",
+  examples: ["bl plugins link ../bailian-plugin-agent"],
+  async run(_config: Config, flags: GlobalFlags) {
+    const positional = flags._positional as string[] | undefined;
+    const pluginPath = positional?.[0];
+    if (!pluginPath) {
+      throw new BailianError("Missing plugin path.", ExitCode.USAGE, "bl plugins link <path>");
+    }
+
+    await linkPlugin(pluginPath);
+    resetCommandCatalogCache();
+    resetRegistry();
+    await createRegistry();
+
+    if (flags.output === "json") {
+      process.stdout.write(JSON.stringify({ linked: pluginPath }, null, 2) + "\n");
+      return;
+    }
+    process.stdout.write(`Linked plugin at ${pluginPath}\n`);
+  },
+});

packages/cli/src/commands/plugins/list.ts

@@ -0,0 +1,35 @@
+import { defineCommand, type Config, type GlobalFlags } from "bailian-cli-core";
+import { listPlugins } from "../../plugins/manager.ts";
+
+export default defineCommand({
+  name: "plugins list",
+  description: "List installed and discovered bailian-cli plugins",
+  usage: "bl plugins list",
+  examples: ["bl plugins list"],
+  async run(_config: Config, flags: GlobalFlags) {
+    const { plugins, errors } = await listPlugins();
+    const format = flags.output === "json" ? "json" : "text";
+
+    if (format === "json") {
+      process.stdout.write(JSON.stringify({ plugins, errors }, null, 2) + "\n");
+      return;
+    }
+
+    if (plugins.length === 0) {
+      process.stdout.write("No plugins installed.\n");
+    } else {
+      process.stdout.write("Plugins:\n");
+      for (const p of plugins) {
+        process.stdout.write(`  ${p.name} (${p.source})${p.version ? ` @ ${p.version}` : ""}\n`);
+        process.stdout.write(`    ${p.root}\n`);
+      }
+    }
+
+    if (errors.length > 0) {
+      process.stderr.write("\nPlugin errors:\n");
+      for (const e of errors) {
+        process.stderr.write(`  ${e.plugin}: ${e.message}\n`);
+      }
+    }
+  },
+});

packages/cli/src/commands/plugins/remove.ts

@@ -0,0 +1,35 @@
+import {
+  BailianError,
+  ExitCode,
+  defineCommand,
+  type Config,
+  type GlobalFlags,
+} from "bailian-cli-core";
+import { removePlugin } from "../../plugins/manager.ts";
+import { resetCommandCatalogCache } from "../../load-commands.ts";
+import { createRegistry, resetRegistry } from "../../registry.ts";
+
+export default defineCommand({
+  name: "plugins remove",
+  description: "Remove an installed bailian-cli plugin",
+  usage: "bl plugins remove <name>",
+  examples: ["bl plugins remove @alife/bailian-agent"],
+  async run(_config: Config, flags: GlobalFlags) {
+    const positional = flags._positional as string[] | undefined;
+    const name = positional?.[0];
+    if (!name) {
+      throw new BailianError("Missing plugin name.", ExitCode.USAGE, "bl plugins remove <name>");
+    }
+
+    await removePlugin(name);
+    resetCommandCatalogCache();
+    resetRegistry();
+    await createRegistry();
+
+    if (flags.output === "json") {
+      process.stdout.write(JSON.stringify({ removed: name }, null, 2) + "\n");
+      return;
+    }
+    process.stdout.write(`Removed plugin: ${name}\n`);
+  },
+});

packages/cli/src/load-commands.ts

@@ -0,0 +1,44 @@
+import { commands as builtinCommands } from "./commands/catalog.ts";
+import { discoverPlugins } from "./plugins/discover.ts";
+import { mergeCommands } from "./plugins/priority.ts";
+import { scanAllPluginCommands } from "./plugins/scan.ts";
+import type { CommandCatalog } from "./plugins/types.ts";
+
+let cachedCatalog: CommandCatalog | undefined;
+
+/**
+ * Load the command catalog.
+ * @returns The command catalog.
+ */
+export async function loadCommandCatalog(): Promise<CommandCatalog> {
+  if (cachedCatalog) return cachedCatalog;
+
+  const plugins = await discoverPlugins();
+  const {
+    entries: pluginEntries,
+    pluginErrors: scanErrors,
+    noAuthSetup: pluginNoAuth,
+  } = await scanAllPluginCommands(plugins);
+
+  const { commands, pluginErrors: mergeErrors } = mergeCommands(builtinCommands, pluginEntries);
+
+  cachedCatalog = {
+    commands,
+    noAuthSetup: pluginNoAuth,
+    plugins,
+    pluginErrors: [...scanErrors, ...mergeErrors],
+  };
+
+  return cachedCatalog;
+}
+
+/**
+ * Reset the command catalog cache.
+ */
+export function resetCommandCatalogCache(): void {
+  cachedCatalog = undefined;
+}
+
+export function getCachedCommandCatalog(): CommandCatalog | undefined {
+  return cachedCatalog;
+}

packages/cli/src/main.ts

@@ -1,5 +1,6 @@
 import { scanCommandPath, parseFlags } from "./args.ts";
-import { registry } from "./registry.ts";
+import { createRegistry, getActiveRegistry } from "./registry.ts";
+import { loadCommandCatalog } from "./load-commands.ts";
 import {
   GLOBAL_OPTIONS,
   loadConfig,
@@ -21,32 +22,8 @@ import {
   setExecutingCommandPath,
 } from "./utils/command-help.ts";
 
-registerCommandHelpPrinter((commandPath, out) => {
-  const a = process.argv.slice(2);
-  const ri = a.indexOf("--region");
-  const region = ((ri >= 0 && a[ri + 1]) ||
-    process.env.DASHSCOPE_REGION ||
-    readConfigFile().region ||
-    "cn") as Region;
-  registry.printHelp(commandPath, out, region);
-});
-
-// 优雅处理 Ctrl+C
-// 退出前尝试 best-effort 刷出埋点，让去抖队列中 / 在途的 fetch 请求有机会
-// 落网络；flush 与较短超时 race，保证 SIGINT 仍然响应及时。
-process.on("SIGINT", () => {
-  process.stderr.write("\nInterrupted. Exiting.\n");
-  void flushTelemetry(500).finally(() => process.exit(130));
-});
-
-// 优雅处理 stdout EPIPE（例如管道到提前退出的 `mpv`）
-process.stdout.on("error", (e: NodeJS.ErrnoException) => {
-  if (e.code === "EPIPE") process.exit(0);
-  else throw e;
-});
-
 // 自己接管鉴权 或 根本不需要 API key 的命令
-const NO_AUTH_SETUP = [
+const BUILTIN_NO_AUTH_SETUP: string[][] = [
   ["auth", "login"],
   ["auth", "logout"],
   ["config", "show"],
@@ -60,9 +37,31 @@ const NO_AUTH_SETUP = [
   ["app", "list"],
   ["console", "call"],
   ["usage", "free"],
+  ["plugins", "list"],
+  ["plugins", "install"],
+  ["plugins", "link"],
+  ["plugins", "remove"],
 ];
 
+function matchesNoAuthSetup(commandPath: string[], rules: string[][]): boolean {
+  return rules.some((cmd) => cmd.every((c, i) => commandPath[i] === c));
+}
+
 async function main() {
+  const registry = await createRegistry();
+  const catalog = await loadCommandCatalog();
+  const noAuthSetup = [...BUILTIN_NO_AUTH_SETUP, ...catalog.noAuthSetup];
+
+  registerCommandHelpPrinter((commandPath, out) => {
+    const a = process.argv.slice(2);
+    const ri = a.indexOf("--region");
+    const region = ((ri >= 0 && a[ri + 1]) ||
+      process.env.DASHSCOPE_REGION ||
+      readConfigFile().region ||
+      "cn") as Region;
+    getActiveRegistry().printHelp(commandPath, out, region);
+  });
+
   const argv = process.argv.slice(2);
 
   if (argv.includes("--version") || argv.includes("-v")) {
@@ -122,7 +121,7 @@ async function main() {
   config.clientName = "bailian-cli";
   config.clientVersion = CLI_VERSION;
 
-  const needsAuthSetup = !NO_AUTH_SETUP.some((cmd) => cmd.every((c, i) => commandPath[i] === c));
+  const needsAuthSetup = !matchesNoAuthSetup(commandPath, noAuthSetup);
   if (needsAuthSetup) {
     await ensureApiKey(config);
     try {
@@ -168,10 +167,16 @@ async function main() {
   await flushTelemetry(1000);
 }
 
+process.on("SIGINT", () => {
+  process.stderr.write("\nInterrupted. Exiting.\n");
+  void flushTelemetry(500).finally(() => process.exit(130));
+});
+
+process.stdout.on("error", (e: NodeJS.ErrnoException) => {
+  if (e.code === "EPIPE") process.exit(0);
+  else throw e;
+});
+
 main().catch((err) => {
-  // 在 handleError() 调用 process.exit() 之前刷出在途埋点。
-  // 命令抛出的错误已被 trackCommandExecution 的 finally 块记录，
-  // 但底层 tracker 有 ~500ms 的发送去抖。不主动 flush 的话，
-  // 错误事件会随进程退出丢掉。
   void flushTelemetry(1000).finally(() => handleError(err));
 });