@ai-setting/roy-agent-cli 1.0.0

package/src/commands/plugin.ts

@@ -0,0 +1,267 @@
+/**
+ * Plugin Module
+ *
+ * Provides plugin registry and hook integration for coder enhancements.
+ */
+
+import type { Tool, ToolResult } from "@ai-setting/roy-agent-core";
+
+// ============================================================================
+// Types
+// ============================================================================
+
+/**
+ * Tool hook context - passed to plugin hooks
+ */
+export interface ToolHookContext {
+  tool?: Tool;
+  args?: Record<string, unknown>;
+  result?: ToolResult;
+}
+
+/**
+ * Tool hook points
+ */
+export const ToolHookPoints = {
+  AFTER_EXECUTE: "tool.after-execute",
+  BEFORE_EXECUTE: "tool.before-execute",
+} as const;
+
+/**
+ * Base plugin interface
+ */
+export interface BasePlugin {
+  name: string;
+  description?: string;
+  version?: string;
+}
+
+/**
+ * Tool plugin interface - hooks into tool execution
+ */
+export interface ToolPlugin extends BasePlugin {
+  getHookPoint(): (typeof ToolHookPoints)[keyof typeof ToolHookPoints];
+  shouldHandle?(context: ToolHookContext): boolean;
+  onAfterExecute?(context: ToolHookContext): void | Promise<void>;
+  onBeforeExecute?(context: ToolHookContext): void | Promise<void>;
+}
+
+/**
+ * Plugin config from CLI
+ */
+export interface PluginConfig {
+  name: string;
+  config?: string;
+}
+
+/**
+ * Parsed plugin arguments
+ */
+export interface PluginArgs {
+  plugins: (string | PluginConfig)[];
+  fileTypes: string[];
+}
+
+// ============================================================================
+// Built-in Plugins
+// ============================================================================
+
+/**
+ * Built-in plugin registry
+ */
+export const BUILTIN_PLUGINS = {
+  lsp: "LSP code intelligence (diagnostics, completions)",
+  "code-check": "External command code checking (linter + type check)",
+  reminder: "Max iterations reminder for ReAct loop",
+} as const;
+
+/**
+ * File type mappings for built-in plugins
+ */
+const FILE_TYPE_MAP: Record<string, string[]> = {
+  typescript: [".ts", ".tsx"],
+  javascript: [".js", ".jsx", ".mjs"],
+  python: [".py"],
+  rust: [".rs"],
+  go: [".go"],
+  java: [".java"],
+  csharp: [".cs"],
+};
+
+// ============================================================================
+// Plugin Manager
+// ============================================================================
+
+/**
+ * Plugin registry for managing registered plugins
+ */
+export class PluginRegistry {
+  private plugins: Map<string, ToolPlugin> = new Map();
+
+  /**
+   * Register a plugin
+   */
+  register(plugin: ToolPlugin): void {
+    if (this.plugins.has(plugin.name)) {
+      throw new Error(`Plugin '${plugin.name}' already registered`);
+    }
+    this.plugins.set(plugin.name, plugin);
+  }
+
+  /**
+   * Unregister a plugin
+   */
+  unregister(name: string): boolean {
+    return this.plugins.delete(name);
+  }
+
+  /**
+   * Get a plugin by name
+   */
+  get(name: string): ToolPlugin | undefined {
+    return this.plugins.get(name);
+  }
+
+  /**
+   * Get all registered plugins
+   */
+  getAll(): ToolPlugin[] {
+    return Array.from(this.plugins.values());
+  }
+
+  /**
+   * Get plugins for a specific hook point
+   */
+  getByHookPoint(hookPoint: string): ToolPlugin[] {
+    return this.getAll().filter((p) => p.getHookPoint() === hookPoint);
+  }
+
+  /**
+   * Check if plugin is registered
+   */
+  has(name: string): boolean {
+    return this.plugins.has(name);
+  }
+
+  /**
+   * Clear all plugins
+   */
+  clear(): void {
+    this.plugins.clear();
+  }
+}
+
+// ============================================================================
+// CLI Argument Parsing
+// ============================================================================
+
+/**
+ * Parse plugin CLI arguments
+ *
+ * Formats:
+ * - plugin-name (e.g., "lsp", "linter")
+ * - plugin-name:config (e.g., "lsp:typescript", "linter:/path/to/eslintrc.json")
+ * - /path/to/plugin.js (custom plugin)
+ */
+export function parseArgs(args: string[]): PluginArgs {
+  const plugins: (string | PluginConfig)[] = [];
+  const fileTypes = new Set<string>();
+
+  for (const arg of args) {
+    if (arg.startsWith("/") || arg.startsWith(".")) {
+      // It's a file path
+      plugins.push(arg);
+
+      // Extract file type from path
+      const ext = arg.match(/\.[^.]+$/)?.[0];
+      if (ext) {
+        fileTypes.add(ext);
+      }
+    } else if (arg.includes(":")) {
+      // It's a plugin with config
+      const [name, config] = arg.split(":", 2);
+      plugins.push({ name, config: config || "" });
+
+      // Extract file types from config
+      if (FILE_TYPE_MAP[config]) {
+        FILE_TYPE_MAP[config].forEach((t) => fileTypes.add(t));
+      } else {
+        // Try to extract extension from config
+        const ext = config.match(/\.[^.]+$/)?.[0];
+        if (ext) {
+          fileTypes.add(ext);
+        }
+      }
+    } else {
+      // It's just a plugin name
+      plugins.push(arg);
+    }
+  }
+
+  return {
+    plugins,
+    fileTypes: Array.from(fileTypes),
+  };
+}
+
+// ============================================================================
+// Plugin Execution
+// ============================================================================
+
+/**
+ * Execute a plugin hook
+ */
+export async function executePlugin(
+  plugin: ToolPlugin,
+  context: ToolHookContext
+): Promise<void> {
+  // Check if plugin should handle this context
+  if (plugin.shouldHandle && !plugin.shouldHandle(context)) {
+    return;
+  }
+
+  // Execute the appropriate hook
+  const hookPoint = plugin.getHookPoint();
+  if (hookPoint === ToolHookPoints.AFTER_EXECUTE && plugin.onAfterExecute) {
+    await plugin.onAfterExecute(context);
+  } else if (
+    hookPoint === ToolHookPoints.BEFORE_EXECUTE &&
+    plugin.onBeforeExecute
+  ) {
+    await plugin.onBeforeExecute(context);
+  }
+}
+
+// ============================================================================
+// Help
+// ============================================================================
+
+/**
+ * Get the help message for plugin command
+ */
+export function getPluginHelp(): string {
+  const lines = [
+    "",
+    "  --plugin <plugins>   Enable coder plugin",
+    "",
+    "  Built-in plugins:",
+  ];
+
+  for (const [name, desc] of Object.entries(BUILTIN_PLUGINS)) {
+    lines.push(`    ${name.padEnd(12)} ${desc}`);
+  }
+
+  lines.push("", "  Plugin formats:", "    --plugin lsp                    # Enable LSP");
+  lines.push("    --plugin lsp:typescript          # LSP for TypeScript");
+  lines.push("    --plugin lsp:python              # LSP for Python");
+  lines.push("    --plugin code-check             # Enable external command checking");
+  lines.push("    --plugin reminder               # Enable iteration reminder");
+  lines.push(
+    "    --plugin /path/to/plugin.js          # Custom plugin"
+  );
+  lines.push(
+    "    --plugin lsp code-check reminder    # Multiple plugins"
+  );
+
+  return lines.join("\n");
+}

package/src/commands/sessions/active.ts

@@ -0,0 +1,96 @@
+/**
+ * @fileoverview Sessions Active Command
+ * 
+ * 命令：roy sessions active [options]
+ */
+
+import { CommandModule } from "yargs";
+import { EnvironmentService } from "../../services/environment.service";
+import { OutputService } from "../../services/output.service";
+import chalk from "chalk";
+import type { SessionComponent, Session } from "@ai-setting/roy-agent-core";
+
+interface ActiveOptions {
+  clear?: boolean;
+  json?: boolean;
+  config?: string;
+}
+
+export const ActiveCommand: CommandModule<object, ActiveOptions> = {
+  command: "active",
+  aliases: ["current"],
+  describe: "查看/设置当前活跃会话",
+
+  builder: (yargs) =>
+    yargs
+      .option("clear", { type: "boolean", default: false })
+      .option("json", { alias: "j", type: "boolean", default: false }),
+
+  async handler(args) {
+    const output = new OutputService();
+    const envService = new EnvironmentService(output);
+
+    try {
+      await envService.create({ configPath: args.config });
+      const env = envService.getEnvironment();
+      if (!env) {
+        output.error("Failed to create environment");
+        process.exit(1);
+      }
+      const sessionComponent = env.getComponent("session") as SessionComponent | undefined;
+
+      if (!sessionComponent) {
+        output.error("SessionComponent not available");
+        process.exit(1);
+      }
+
+      const activeId = sessionComponent.getActiveSessionId();
+
+      if (args.clear) {
+        // 清除活跃会话
+        // 注意：需要通过内部机制清除，暂时不支持
+        output.log("Active session cleared");
+        return;
+      }
+
+      if (!activeId) {
+        output.log(chalk.gray("No active session set"));
+        return;
+      }
+
+      const session: Session | undefined = await sessionComponent.get(activeId);
+
+      if (args.json) {
+        output.json({
+          activeSessionId: activeId,
+          session: session ? {
+            id: session.id,
+            title: session.title,
+            directory: session.directory,
+            messageCount: session.messageCount,
+            createdAt: new Date(session.createdAt).toISOString(),
+            updatedAt: new Date(session.updatedAt).toISOString(),
+          } : null,
+        });
+      } else {
+        if (!session) {
+          output.log(chalk.gray("Active session not found: " + activeId));
+          return;
+        }
+
+        output.log(chalk.bold.green("┌─ Active Session ─────────────────────────────────┐"));
+        output.log(`│ ID:          ${session.id}`);
+        output.log(`│ Title:       ${session.title}`);
+        output.log(`│ Directory:   ${session.directory}`);
+        output.log(`│ Messages:    ${session.messageCount}`);
+        output.log(`│ Updated:     ${new Date(session.updatedAt).toLocaleString("zh-CN")}`);
+        output.log("└" + "─".repeat(51) + "┘");
+      }
+    } catch (error) {
+      output.error(`Failed to get active session: ${error instanceof Error ? error.message : String(error)}`);
+      process.exit(1);
+    } finally {
+      await envService.dispose();
+    }
+  },
+};

package/src/commands/sessions/add-message.ts

@@ -0,0 +1,96 @@
+/**
+ * @fileoverview Sessions Add Message Command
+ * 
+ * 命令：roy sessions add-message <session-id> <role> <content>
+ */
+
+import { CommandModule } from "yargs";
+import { EnvironmentService } from "../../services/environment.service";
+import { OutputService } from "../../services/output.service";
+import chalk from "chalk";
+import type { SessionComponent, Session } from "@ai-setting/roy-agent-core";
+
+interface AddMessageOptions {
+  sessionId: string;
+  role: "user" | "assistant" | "system" | "tool";
+  content: string;
+  json?: boolean;
+  config?: string;
+}
+
+export const AddMessageCommand: CommandModule<object, object> = {
+  command: "add-message <session-id> <role> <content>",
+  aliases: ["add"],
+  describe: "添加消息到会话 [test]",
+
+  builder: (yargs) =>
+    yargs
+      .positional("session-id", {
+        describe: "会话 ID",
+        type: "string",
+        demandOption: true,
+      })
+      .positional("role", {
+        describe: "消息角色",
+        type: "string",
+        choices: ["user", "assistant", "system", "tool"],
+        demandOption: true,
+      })
+      .positional("content", {
+        describe: "消息内容",
+        type: "string",
+        demandOption: true,
+      })
+      .option("json", { alias: "j", type: "boolean", default: false }),
+
+  async handler(args) {
+    const a = args as any;
+    const output = new OutputService();
+    const envService = new EnvironmentService(output);
+
+    try {
+      await envService.create({ configPath: a.config });
+      const env = envService.getEnvironment();
+      if (!env) {
+        output.error("Failed to create environment");
+        process.exit(1);
+      }
+      const sessionComponent = env.getComponent("session") as SessionComponent | undefined;
+
+      if (!sessionComponent) {
+        output.error("SessionComponent not available");
+        process.exit(1);
+      }
+
+      // 检查会话是否存在
+      const session: Session | undefined = await sessionComponent.get(a.sessionId);
+      if (!session) {
+        output.error(`Session not found: ${a.sessionId}`);
+        process.exit(1);
+      }
+
+      // 添加消息
+      const messageId = await sessionComponent.addMessage(a.sessionId, {
+        role: a.role,
+        content: a.content,
+      });
+
+      if (a.json) {
+        output.json({
+          success: true,
+          messageId,
+          sessionId: a.sessionId,
+          role: a.role,
+          contentLength: a.content.length,
+        });
+      } else {
+        output.success(`Message added to session ${a.sessionId}`);
+        output.log(`Message ID: ${messageId}`);
+        output.log(`Role: ${a.role}`);
+        output.log(`Content: ${a.content.substring(0, 100)}${a.content.length > 100 ? "..." : ""}`);
+      }
+    } finally {
+      await envService.dispose();
+    }
+  },
+};

package/src/commands/sessions/checkpoints.ts

@@ -0,0 +1,154 @@
+/**
+ * @fileoverview Sessions Checkpoints Command
+ * 
+ * 命令：roy sessions checkpoints <session-id>
+ */
+
+import { CommandModule } from "yargs";
+import { EnvironmentService } from "../../services/environment.service";
+import { OutputService } from "../../services/output.service";
+import chalk from "chalk";
+import type { SessionComponent, Session, SessionCheckpoint } from "@ai-setting/roy-agent-core";
+
+interface CheckpointsOptions {
+  sessionId: string;
+  detail?: boolean;
+  json?: boolean;
+  config?: string;
+}
+
+export const CheckpointsCommand: CommandModule<object, object> = {
+  command: "checkpoints <session-id>",
+  aliases: ["cps"],
+  describe: "查看会话检查点列表 [new]",
+
+  builder: (yargs) =>
+    yargs
+      .positional("session-id", {
+        describe: "会话 ID",
+        type: "string",
+        demandOption: true,
+      })
+      .option("detail", { alias: "d", type: "boolean", default: false })
+      .option("json", { alias: "j", type: "boolean", default: false }),
+
+  async handler(args) {
+    const a = args as any;
+    const output = new OutputService();
+    const envService = new EnvironmentService(output);
+
+    try {
+      await envService.create({ configPath: a.config });
+      const env = envService.getEnvironment();
+      if (!env) {
+        output.error("Failed to create environment");
+        process.exit(1);
+      }
+      const sessionComponent = env.getComponent("session") as SessionComponent | undefined;
+
+      if (!sessionComponent) {
+        output.error("SessionComponent not available");
+        process.exit(1);
+      }
+
+      const session: Session | undefined = await sessionComponent.get(a.sessionId);
+      if (!session) {
+        output.error(`Session not found: ${a.sessionId}`);
+        process.exit(1);
+      }
+
+      const checkpoints: SessionCheckpoint[] = await sessionComponent.getCheckpoints(a.sessionId);
+
+      if (checkpoints.length === 0) {
+        output.log(chalk.gray("No checkpoints for this session"));
+        return;
+      }
+
+      if (a.json) {
+        output.json({
+          sessionId: a.sessionId,
+          checkpoints: checkpoints.map((cp: SessionCheckpoint) => ({
+            id: cp.id,
+            title: cp.title,
+            type: cp.type,
+            messageIndex: cp.messageIndex,
+            processKeyPoints: cp.processKeyPoints,
+            currentState: cp.currentState,
+            nextSteps: cp.nextSteps,
+            userIntents: cp.userIntents,  // Include user intents
+            messageCountBefore: cp.messageCountBefore,
+            createdAt: new Date(cp.createdAt).toISOString(),
+          })),
+          total: checkpoints.length,
+        });
+        return;
+      }
+
+      if (a.detail) {
+        // 详细模式
+        checkpoints.reverse().forEach((cp: SessionCheckpoint, i: number) => {
+          const isLatest = i === 0;
+          output.log(chalk.bold(`┌─ Checkpoint: ${cp.title} ${isLatest ? chalk.green("✓") : ""} ──────` + "─".repeat(Math.max(0, 30 - cp.title.length)) + "┐"));
+          output.log(`│ ID:            ${cp.id}`);
+          output.log(`│ Type:          ${cp.type}`);
+          output.log(`│ Message Index: ${cp.messageIndex} (${cp.messageCountBefore} messages)`);
+          output.log(`│ Created:       ${new Date(cp.createdAt).toLocaleString("zh-CN")}`);
+          output.log("├─ Process Key Points ─────────────────────────────────┤");
+          cp.processKeyPoints.forEach((p: string, j: number) => {
+            output.log(`│ ${j + 1}. ${p}`);
+          });
+          output.log("├─ Current State ─────────────────────────────────────┤");
+          output.log("│ " + cp.currentState);
+          if (cp.nextSteps?.length) {
+            output.log("├─ Next Steps ───────────────────────────────────────┤");
+            cp.nextSteps.forEach((s: string, j: number) => {
+              output.log(`│ ${j + 1}. ${s}`);
+            });
+          }
+          if (cp.userIntents?.length) {
+            output.log("├─ User Intents ─────────────────────────────────────┤");
+            cp.userIntents.forEach((intent: string, j: number) => {
+              output.log(`│ ${j + 1}. ${intent}`);
+            });
+          }
+          output.log("└" + "─".repeat(51) + "┘");
+          output.log("");
+        });
+      } else {
+        // 简洁模式
+        const header = [
+          chalk.bold("#"),
+          chalk.bold("Title"),
+          chalk.bold("Type"),
+          chalk.bold("Messages"),
+          chalk.bold("Created"),
+        ].join(" │ ");
+
+        output.log(`┌─ Checkpoints (${checkpoints.length}) ${"─".repeat(40)}┐`);
+        output.log(`│${header}│`);
+        output.log("├" + "─".repeat(header.length + 2) + "┤");
+
+        checkpoints.reverse().forEach((cp: any, i: number) => {
+          const isLatest = i === 0;
+          const marker = isLatest ? chalk.green("✓ ") : "  ";
+          const title = cp.title.length > 25 ? cp.title.slice(0, 22) + "..." : cp.title;
+          const created = new Date(cp.createdAt).toLocaleString("zh-CN", {
+            month: "2-digit",
+            day: "2-digit",
+            hour: "2-digit",
+            minute: "2-digit",
+          });
+
+          output.log(`│${marker}${i + 1}  ${title.padEnd(25)} │ ${cp.type.padEnd(5)} │ ${cp.messageCountBefore.toString().padStart(8)} │ ${chalk.gray(created)}│`);
+        });
+
+        output.log("└" + "─".repeat(header.length + 2) + "┘");
+      }
+    } catch (error) {
+      output.error(`Failed to get checkpoints: ${error instanceof Error ? error.message : String(error)}`);
+      process.exit(1);
+    } finally {
+      await envService.dispose();
+    }
+  },
+};