@ia-ccun/code-agent-claw 0.0.3 → 0.0.5

package/package/src/services/agent-rpc.ts

@@ -1,397 +0,0 @@
-import { spawn, ChildProcess } from 'child_process';
-import * as readline from 'readline';
-import * as fs from 'fs';
-import * as path from 'path';
-import { AgentConfig, CommandResult, AgentEvent, EventCallback, AgentStatus } from '../types';
-import { logger } from '../utils/logger';
-
-/**
- * aicode RPC 服务
- * 负责管理 aicode 子进程，并通过 JSONL 协议进行通信
- */
-export class AgentRpcService {
-  private process: ChildProcess | null = null;
-  private rl: readline.Interface | null = null;
-  private pendingCommands: Map<string, {
-    resolve: (r: CommandResult) => void;
-    reject: (e: Error) => void;
-  }> = new Map();
-  private callbacks: Set<EventCallback> = new Set();
-  private config: AgentConfig | null = null;
-  private _status: AgentStatus = 'uninitialized';
-  private healthCheckInterval: NodeJS.Timeout | null = null;
-
-  /**
-   * 获取当前状态
-   */
-  get status(): AgentStatus {
-    return this._status;
-  }
-
-  /**
-   * 获取当前配置
-   */
-  get currentConfig(): AgentConfig | null {
-    return this.config;
-  }
-
-  /**
-   * 初始化 agent（保存配置并启动）
-   */
-  async initialize(config: AgentConfig): Promise<CommandResult> {
-    logger.info('[AgentRpc] Initializing with config:', config);
-
-    // 验证命令是否存在
-    const commandPath = config.command;
-    if (!fs.existsSync(commandPath)) {
-      const error = `aicode command not found: ${commandPath}`;
-      logger.error('[AgentRpc]', error);
-      return { success: false, message: error };
-    }
-
-    // 确保目录存在
-    this.ensureDir(config.sessionDir);
-    this.ensureDir(config.workingDir);
-
-    // 保存配置
-    this.config = config;
-
-    // 启动进程
-    return this.start();
-  }
-
-  /**
-   * 启动 aicode 进程
-   */
-  async start(): Promise<CommandResult> {
-    if (!this.config) {
-      return { success: false, message: 'Config not set' };
-    }
-
-    if (this._status === 'running') {
-      return { success: true, message: 'Already running' };
-    }
-
-    this._status = 'starting';
-    logger.info('[AgentRpc] Starting aicode process...');
-
-    try {
-      // 构建命令参数
-      const args: string[] = [
-        '--mode', 'rpc',
-        '--provider', this.config.provider,
-        '--model', this.config.model,
-        '--session-dir', this.config.sessionDir
-      ];
-
-      // noSession 模式下不传 session 参数或使用特殊处理
-      if (!this.config.noSession) {
-        // 如果需要会话模式，可以在这里添加
-      }
-
-      logger.info(`[AgentRpc] Command: ${this.config.command} ${args.join(' ')}`);
-      logger.info(`[AgentRpc] Working dir: ${this.config.workingDir}`);
-
-      // 启动子进程
-      this.process = spawn(this.config.command, args, {
-        cwd: this.config.workingDir,
-        env: { ...process.env },
-        stdio: ['pipe', 'pipe', 'pipe']
-      });
-
-      // 创建 readline 接口监听 stdout
-      this.rl = readline.createInterface({
-        input: this.process.stdout!,
-        crlfDelay: Infinity
-      });
-
-      // 监听 stdout 事件
-      this.rl.on('line', (line: string) => {
-        this.handleEvent(line);
-      });
-
-      // 监听 stderr
-      this.process.stderr?.on('data', (data: Buffer) => {
-        const line = data.toString().replace(/\x1B\[[0-9;]*[a-zA-Z]/g, '');
-        if (line.trim()) {
-          logger.warn(`[aicode-stderr] ${line}`);
-        }
-      });
-
-      // 监听进程退出
-      this.process.on('exit', (code, signal) => {
-        logger.info(`[AgentRpc] Process exited with code ${code}, signal ${signal}`);
-        this._status = 'stopped';
-        this.stopHealthCheck();
-      });
-
-      this.process.on('error', (err) => {
-        logger.error('[AgentRpc] Process error:', err);
-        this._status = 'error';
-      });
-
-      // 启动健康检查
-      this.startHealthCheck();
-
-      this._status = 'running';
-      logger.info('[AgentRpc] aicode process started successfully');
-
-      return { success: true, message: 'Agent started' };
-    } catch (error: any) {
-      this._status = 'error';
-      logger.error('[AgentRpc] Failed to start:', error);
-      return { success: false, message: error.message };
-    }
-  }
-
-  /**
-   * 处理事件
-   */
-  private handleEvent(jsonLine: string): void {
-    try {
-      const trimmed = jsonLine.trim();
-      if (!trimmed || !trimmed.startsWith('{')) {
-        return;
-      }
-
-      const event = JSON.parse(jsonLine);
-      const type = event.type;
-
-      logger.debug(`[AgentRpc] <<< type=${type}`);
-
-      // 处理命令响应 - RPC 模式特有
-      if (type === 'response') {
-        this.handleCommandResponse(event);
-        return;
-      }
-
-      // 构建 AgentEvent
-      const agentEvent: AgentEvent = {
-        type,
-        data: event
-      };
-
-      if (type === 'message_update') {
-        const msgEvent = event.assistantMessageEvent;
-        if (msgEvent) {
-          agentEvent.messageType = msgEvent.type;
-          agentEvent.delta = msgEvent.delta;
-        }
-      } else if (type === 'agent_end') {
-        agentEvent.success = true;
-        agentEvent.usage = event.usage;
-      }
-
-      // 通知回调
-      this.callbacks.forEach(cb => {
-        try {
-          cb(agentEvent);
-        } catch (e) {
-          logger.error('[AgentRpc] Event callback error:', e);
-        }
-      });
-    } catch (e) {
-      // 忽略解析错误
-    }
-  }
-
-  /**
-   * 处理命令响应
-   */
-  private handleCommandResponse(event: any): void {
-    const commandId = event.id;
-    const commandType = event.command;
-    const success = event.success;
-
-    logger.info(`[AgentRpc] Command response: id=${commandId}, command=${commandType}, success=${success}`);
-
-    const pending = this.pendingCommands.get(commandId);
-    if (pending) {
-      this.pendingCommands.delete(commandId);
-      if (success) {
-        pending.resolve({ success: true, message: commandType });
-      } else {
-        pending.resolve({ success: false, message: event.error || 'Command failed' });
-      }
-    } else {
-      logger.warn(`[AgentRpc] No pending command for id: ${commandId}`);
-    }
-  }
-
-  /**
-   * 发送提示
-   */
-  async sendPrompt(message: string): Promise<CommandResult> {
-    return this.sendCommand('prompt', message);
-  }
-
-  /**
-   * 发送引导消息
-   */
-  async sendSteer(message: string): Promise<CommandResult> {
-    return this.sendCommand('steer', message);
-  }
-
-  /**
-   * 发送后续消息
-   */
-  async sendFollowUp(message: string): Promise<CommandResult> {
-    return this.sendCommand('follow_up', message);
-  }
-
-  /**
-   * 中止操作
-   */
-  async sendAbort(): Promise<CommandResult> {
-    return this.sendCommand('abort');
-  }
-
-  /**
-   * 创建新会话
-   */
-  async newSession(parentSession?: string): Promise<CommandResult> {
-    return this.sendCommand('new_session', undefined, parentSession);
-  }
-
-  /**
-   * 发送命令
-   */
-  private sendCommand(type: string, message?: string, extra?: string): Promise<CommandResult> {
-    return new Promise((resolve, reject) => {
-      if (!this.isRunning() || !this.process || !this.process.stdin) {
-        return resolve({ success: false, message: 'Agent is not running' });
-      }
-
-      const command: any = { type };
-      if (message) command.message = message;
-      if (extra) command.parentSession = extra;
-
-      // 生成命令ID
-      const commandId = Date.now() + '-' + Math.floor(Math.random() * 10000);
-      command.id = commandId;
-
-      this.pendingCommands.set(commandId, { resolve, reject });
-
-      try {
-        const jsonLine = JSON.stringify(command) + '\n';
-        logger.debug(`[AgentRpc] >>> ${jsonLine.trim()}`);
-        this.process.stdin.write(jsonLine);
-      } catch (e: any) {
-        this.pendingCommands.delete(commandId);
-        reject(e);
-      }
-    });
-  }
-
-  /**
-   * 检查是否运行中
-   */
-  isRunning(): boolean {
-    return this._status === 'running' && this.process?.pid !== undefined;
-  }
-
-  /**
-   * 停止 agent
-   */
-  stop(): void {
-    logger.info('[AgentRpc] Stopping agent...');
-    this._status = 'stopped';
-    this.stopHealthCheck();
-
-    try {
-      if (this.process?.stdin) {
-        this.process.stdin.end();
-      }
-      if (this.process) {
-        this.process.kill('SIGTERM');
-        setTimeout(() => {
-          if (this.process && !this.process.killed) {
-            this.process.kill('SIGKILL');
-          }
-        }, 3000);
-      }
-    } catch (e) {
-      logger.error('[AgentRpc] Error stopping:', e);
-    }
-
-    this.process = null;
-    this.rl = null;
-    logger.info('[AgentRpc] Agent stopped');
-  }
-
-  /**
-   * 注册事件回调
-   */
-  registerCallback(callback: EventCallback): void {
-    this.callbacks.add(callback);
-  }
-
-  /**
-   * 移除事件回调
-   */
-  removeCallback(callback: EventCallback): void {
-    this.callbacks.delete(callback);
-  }
-
-  /**
-   * 启动健康检查
-   */
-  private startHealthCheck(): void {
-    this.stopHealthCheck();
-
-    this.healthCheckInterval = setInterval(async () => {
-      if (!this.isRunning() && this._status === 'running') {
-        logger.warn('[AgentRpc] Process died, auto-restarting...');
-        this._status = 'restarting';
-
-        // 自动重启
-        if (this.config) {
-          try {
-            await this.start();
-            logger.info('[AgentRpc] Auto-restart successful');
-          } catch (e) {
-            logger.error('[AgentRpc] Auto-restart failed:', e);
-            this._status = 'error';
-          }
-        } else {
-          this._status = 'stopped';
-        }
-      }
-    }, 5000);
-  }
-
-  /**
-   * 停止健康检查
-   */
-  private stopHealthCheck(): void {
-    if (this.healthCheckInterval) {
-      clearInterval(this.healthCheckInterval);
-      this.healthCheckInterval = null;
-    }
-  }
-
-  /**
-   * 确保目录存在
-   */
-  private ensureDir(dirPath: string): void {
-    if (!fs.existsSync(dirPath)) {
-      logger.info(`[AgentRpc] Creating directory: ${dirPath}`);
-      fs.mkdirSync(dirPath, { recursive: true });
-    }
-  }
-}
-
-/**
- * 单例实例
- */
-let agentRpcService: AgentRpcService | null = null;
-
-/**
- * 获取 AgentRpcService 单例
- */
-export function getAgentRpcService(): AgentRpcService {
-  if (!agentRpcService) {
-    agentRpcService = new AgentRpcService();
-  }
-  return agentRpcService;
-}

package/package/src/types/index.ts

@@ -1,60 +0,0 @@
-export interface ServerConfig {
-  port: number;
-  host: string;
-}
-
-export interface AgentConfig {
-  enabled: boolean;
-  command: string;
-  provider: string;
-  model: string;
-  noSession: boolean;
-  sessionDir: string;
-  workingDir: string;
-}
-
-export interface LoggingConfig {
-  level: string;
-}
-
-export interface Config {
-  server: ServerConfig;
-  agent: AgentConfig;
-  logging: LoggingConfig;
-}
-
-export interface CommandResult {
-  success: boolean;
-  message: string;
-}
-
-export interface AgentEvent {
-  type: string;
-  data?: any;
-  messageType?: string;
-  delta?: string;
-  success?: boolean;
-  usage?: {
-    completionTokens?: number;
-    promptTokens?: number;
-  };
-}
-
-export type EventCallback = (event: AgentEvent) => void;
-
-export type AgentStatus = 'uninitialized' | 'starting' | 'running' | 'stopped' | 'error' | 'restarting';
-
-export interface ConfigRequest {
-  command?: string;
-  provider?: string;
-  model?: string;
-  noSession?: boolean;
-  sessionDir?: string;
-  workingDir?: string;
-}
-
-export interface ApiResponse<T = any> {
-  success: boolean;
-  message?: string;
-  data?: T;
-}

package/package/src/utils/logger.ts

@@ -1,13 +0,0 @@
-import pino from 'pino';
-
-export const logger = pino({
-  level: process.env.LOG_LEVEL || 'info',
-  transport: process.env.NODE_ENV !== 'production' ? {
-    target: 'pino-pretty',
-    options: {
-      colorize: true,
-      translateTime: 'HH:MM:ss',
-      ignore: 'pid,hostname'
-    }
-  } : undefined
-});

package/package/tsconfig.json

@@ -1,19 +0,0 @@
-{
-  "compilerOptions": {
-    "target": "ES2020",
-    "module": "commonjs",
-    "lib": ["ES2020"],
-    "outDir": "./dist",
-    "rootDir": "./src",
-    "strict": true,
-    "esModuleInterop": true,
-    "skipLibCheck": true,
-    "forceConsistentCasingInFileNames": true,
-    "resolveJsonModule": true,
-    "declaration": true,
-    "declarationMap": true,
-    "sourceMap": true
-  },
-  "include": ["src/**/*"],
-  "exclude": ["node_modules", "dist"]
-}