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

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

@@ -0,0 +1,397 @@
+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

@@ -0,0 +1,60 @@
+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

@@ -0,0 +1,13 @@
+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

@@ -0,0 +1,19 @@
+{
+  "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"]
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ia-ccun/code-agent-claw",
-  "version": "0.0.1",
+  "version": "0.0.3",
   "description": "Code Agent Claw WebUI - Node.js implementation with frontend configuration",
   "main": "dist/index.js",
   "bin": {

package/public/index.html

@@ -3,12 +3,7 @@
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>AICode Agent</title>
-    <link rel="preconnect" href="https://fonts.googleapis.com">
-    <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
-    <link href="https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600;700&family=JetBrains+Mono:wght@400;500;600&display=swap" rel="stylesheet">
-    <link rel="stylesheet" href="juejin.css">
-    <script src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"></script>
+    <title>AICode Claw</title>
     <style>
         /* ========================================
            ANTHROPIC STYLE - AI Agent Console
@@ -64,8 +59,8 @@
             --space-2xl: 32px;
 
             /* Typography - Inter for clean readability */
-            --font-display: 'Inter', -apple-system, BlinkMacSystemFont, 'Segoe UI', sans-serif;
-            --font-mono: 'JetBrains Mono', 'SF Mono', Consolas, monospace;
+            --font-display: -apple-system, BlinkMacSystemFont, 'Segoe UI', 'PingFang SC', 'Microsoft YaHei', sans-serif;
+            --font-mono: 'SF Mono', Consolas, 'PingFang SC', 'Microsoft YaHei Mono', monospace;
 
             /* Effects */
             --shadow-sm: 0 1px 3px rgba(61, 61, 58, 0.08);
@@ -262,7 +257,7 @@
             background: var(--bg-tertiary);
             padding: 2px 6px;
             border-radius: 4px;
-            font-family: 'JetBrains Mono', monospace;
+            font-family: var(--font-mono);
             font-size: 0.75rem;
             color: var(--accent-green);
         }
@@ -1075,7 +1070,7 @@
             <div class="header-left">
                 <div class="logo">
                     <img class="logo-icon" src="aicode.svg" alt="AICode">
-                    <span class="logo-text">AICode Agent</span>
+                    <span class="logo-text">AICode Claw</span>
                 </div>
                 <div class="status-badge">
                     <div class="status-dot" id="statusDot"></div>
@@ -1163,7 +1158,7 @@
                 <div id="responseTab" class="panel-content">
                     <div class="empty-state" id="emptyState">
                         <img class="empty-logo" src="aicode.svg" alt="AICode">
-                        <div class="empty-title">你好，我是 AICode Agent</div>
+                        <div class="empty-title">你好，我是 AICode Claw</div>
                         <div class="empty-desc">我可以帮助你完成以下任务：</div>
                         <div class="empty-features">
                             <div class="feature-item"><span class="feature-icon">📝</span> 代码编写与调试</div>
@@ -1236,12 +1231,6 @@
     <script>
         const API_BASE = '';
         let messages = [];
-
-        // Configure marked
-        marked.setOptions({
-            breaks: true,
-            gfm: true
-        });
         let isStreaming = false;
         let lastRenderTime = 0;
         const RENDER_THROTTLE = 100;