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

package/package/src/server/routes.ts

@@ -0,0 +1,342 @@
+import { Express, Request, Response, NextFunction } from 'express';
+import { AgentRpcService } from '../services/agent-rpc';
+import { getAgentConfig, saveUserConfig, loadConfig, findAicodeCommand } from '../config';
+import { AgentConfig, ConfigRequest, ApiResponse } from '../types';
+import { logger } from '../utils/logger';
+import * as fs from 'fs';
+import * as path from 'path';
+import * as os from 'os';
+import { execSync } from 'child_process';
+
+/**
+ * 注册 API 路由
+ */
+export function registerRoutes(app: Express, agentService: AgentRpcService): void {
+
+  // ==================== 配置相关 ====================
+
+  /**
+   * 获取当前配置
+   */
+  app.get('/api/aicode/config', (req: Request, res: Response) => {
+    logger.info('[API] GET /api/aicode/config');
+    const config = getAgentConfig();
+    const currentConfig = agentService.currentConfig;
+
+    // 读取 models.json
+    let models = null;
+    const modelsPath = path.join(os.homedir(), '.aicode-cli', 'agent', 'models.json');
+    if (fs.existsSync(modelsPath)) {
+      try {
+        models = JSON.parse(fs.readFileSync(modelsPath, 'utf-8'));
+      } catch (e) {
+        logger.warn('Failed to parse models.json:', e);
+      }
+    }
+
+    const response: ApiResponse<{
+      config: AgentConfig;
+      status: string;
+      models: any;
+    }> = {
+      success: true,
+      data: {
+        config: {
+          ...config,
+          command: config.command
+        },
+        status: agentService.status,
+        models
+      }
+    };
+
+    res.json(response);
+  });
+
+  /**
+   * 保存配置并初始化 agent
+   */
+  app.post('/api/aicode/config', async (req: Request, res: Response) => {
+    logger.info('[API] POST /api/aicode/config', req.body);
+
+    const body = req.body as ConfigRequest;
+
+    // command 字段可选，如果没有填则自动检测
+
+    // 自动检测 command 路径
+    let command = body.command;
+    if (!command || command === 'aicode') {
+      command = findAicodeCommand();
+      logger.info('[API] Auto-detected aicode command:', command);
+    }
+
+    const config: AgentConfig = {
+      enabled: true,
+      command: command,
+      provider: body.provider || 'minimax-custom',
+      model: body.model || 'MiniMax-M2.5',
+      noSession: body.noSession !== false,
+      sessionDir: body.sessionDir || '~/.aicode-cli/agent/sessions',
+      workingDir: body.workingDir || '~/.aicode-cli'
+    };
+
+    // 先停止已有 agent
+    if (agentService.isRunning()) {
+      logger.info('[API] Stopping existing agent...');
+      agentService.stop();
+    }
+
+    // 保存配置
+    saveUserConfig(config);
+
+    // 初始化并启动
+    const result = await agentService.initialize(config);
+
+    if (result.success) {
+      res.json({
+        success: true,
+        message: 'Agent initialized successfully'
+      } as ApiResponse);
+    } else {
+      res.status(500).json({
+        success: false,
+        message: result.message
+      } as ApiResponse);
+    }
+  });
+
+  // ==================== 状态相关 ====================
+
+  /**
+   * 获取 agent 状态
+   */
+  app.get('/api/aicode/status', (req: Request, res: Response) => {
+    logger.info('[API] GET /api/aicode/status');
+
+    res.json({
+      success: true,
+      data: {
+        status: agentService.status,
+        running: agentService.isRunning()
+      }
+    } as ApiResponse);
+  });
+
+  /**
+   * 健康检查
+   */
+  app.get('/api/aicode/health', (req: Request, res: Response) => {
+    res.json({
+      success: true,
+      data: {
+        status: agentService.status,
+        running: agentService.isRunning()
+      }
+    } as ApiResponse);
+  });
+
+  /**
+   * 检测 aicode 命令路径
+   */
+  app.get('/api/aicode/detect', (req: Request, res: Response) => {
+    logger.info('[API] GET /api/aicode/detect');
+
+    // 使用 findAicodeCommand 获取所有可能的路径
+    const detectedPaths: string[] = [];
+
+    try {
+      // 1. 尝试 whereis
+      const whereisResult = execSync('whereis aicode', { encoding: 'utf-8' });
+      const match = whereisResult.match(/aicode:\s*(.+)$/);
+      if (match && match[1] && !match[1].includes('not found')) {
+        const paths = match[1].trim().split(/\s+/);
+        paths.forEach(p => {
+          if (p && fs.existsSync(p) && !detectedPaths.includes(p)) {
+            detectedPaths.push(p);
+          }
+        });
+      }
+    } catch (e) {
+      // ignore
+    }
+
+    // 2. 尝试 which
+    try {
+      const whichResult = execSync('which -a aicode', { encoding: 'utf-8' }).trim();
+      if (whichResult) {
+        whichResult.split('\n').forEach(p => {
+          const trimmed = p.trim();
+          if (trimmed && fs.existsSync(trimmed) && !detectedPaths.includes(trimmed)) {
+            detectedPaths.push(trimmed);
+          }
+        });
+      }
+    } catch (e) {
+      // ignore
+    }
+
+    // 3. 检查常见路径
+    const commonPaths = [
+      '/usr/local/bin/aicode',
+      '/usr/bin/aicode',
+      path.join(os.homedir(), '.local/bin/aicode'),
+      path.join(os.homedir(), 'bin/aicode'),
+    ];
+    commonPaths.forEach(p => {
+      if (fs.existsSync(p) && !detectedPaths.includes(p)) {
+        detectedPaths.push(p);
+      }
+    });
+
+    res.json({
+      success: true,
+      data: {
+        paths: detectedPaths,
+        recommended: detectedPaths.length > 0 ? detectedPaths[0] : 'aicode'
+      }
+    } as ApiResponse);
+  });
+
+  // ==================== Agent 操作 ====================
+
+  /**
+   * 发送提示
+   */
+  app.post('/api/aicode/prompt', async (req: Request, res: Response) => {
+    const { message } = req.body;
+    logger.info(`[API] POST /api/aicode/prompt: ${message?.substring(0, 50)}...`);
+
+    if (!agentService.isRunning()) {
+      return res.status(503).json({
+        success: false,
+        message: 'Agent is not running. Please initialize first.'
+      } as ApiResponse);
+    }
+
+    try {
+      const result = await agentService.sendPrompt(message);
+      if (result.success) {
+        res.json({ success: true, data: result } as ApiResponse);
+      } else {
+        res.status(500).json({ success: false, message: result.message } as ApiResponse);
+      }
+    } catch (e: any) {
+      logger.error('[API] /prompt error:', e);
+      res.status(500).json({ success: false, message: e.message } as ApiResponse);
+    }
+  });
+
+  /**
+   * 发送引导消息
+   */
+  app.post('/api/aicode/steer', async (req: Request, res: Response) => {
+    const { message } = req.body;
+    logger.info(`[API] POST /api/aicode/steer: ${message?.substring(0, 50)}...`);
+
+    if (!agentService.isRunning()) {
+      return res.status(503).json({
+        success: false,
+        message: 'Agent is not running'
+      } as ApiResponse);
+    }
+
+    try {
+      const result = await agentService.sendSteer(message);
+      if (result.success) {
+        res.json({ success: true, data: result } as ApiResponse);
+      } else {
+        res.status(500).json({ success: false, message: result.message } as ApiResponse);
+      }
+    } catch (e: any) {
+      res.status(500).json({ success: false, message: e.message } as ApiResponse);
+    }
+  });
+
+  /**
+   * 发送后续消息
+   */
+  app.post('/api/aicode/follow-up', async (req: Request, res: Response) => {
+    const { message } = req.body;
+    logger.info(`[API] POST /api/aicode/follow-up`);
+
+    if (!agentService.isRunning()) {
+      return res.status(503).json({
+        success: false,
+        message: 'Agent is not running'
+      } as ApiResponse);
+    }
+
+    try {
+      const result = await agentService.sendFollowUp(message);
+      if (result.success) {
+        res.json({ success: true, data: result } as ApiResponse);
+      } else {
+        res.status(500).json({ success: false, message: result.message } as ApiResponse);
+      }
+    } catch (e: any) {
+      res.status(500).json({ success: false, message: e.message } as ApiResponse);
+    }
+  });
+
+  /**
+   * 中止操作
+   */
+  app.post('/api/aicode/abort', async (req: Request, res: Response) => {
+    logger.info('[API] POST /api/aicode/abort');
+
+    if (!agentService.isRunning()) {
+      return res.status(503).json({
+        success: false,
+        message: 'Agent is not running'
+      } as ApiResponse);
+    }
+
+    try {
+      const result = await agentService.sendAbort();
+      if (result.success) {
+        res.json({ success: true, data: result } as ApiResponse);
+      } else {
+        res.status(500).json({ success: false, message: result.message } as ApiResponse);
+      }
+    } catch (e: any) {
+      res.status(500).json({ success: false, message: e.message } as ApiResponse);
+    }
+  });
+
+  /**
+   * 创建新会话
+   */
+  app.post('/api/aicode/new-session', async (req: Request, res: Response) => {
+    const parentSession = req.query.parentSession as string | undefined;
+    logger.info(`[API] POST /api/aicode/new-session: parentSession=${parentSession}`);
+
+    if (!agentService.isRunning()) {
+      return res.status(503).json({
+        success: false,
+        message: 'Agent is not running'
+      } as ApiResponse);
+    }
+
+    try {
+      const result = await agentService.newSession(parentSession);
+      if (result.success) {
+        res.json({ success: true, data: result } as ApiResponse);
+      } else {
+        res.status(500).json({ success: false, message: result.message } as ApiResponse);
+      }
+    } catch (e: any) {
+      res.status(500).json({ success: false, message: e.message } as ApiResponse);
+    }
+  });
+
+  // 错误处理中间件
+  app.use((err: Error, req: Request, res: Response, next: NextFunction) => {
+    logger.error('[API] Error:', err);
+    res.status(500).json({
+      success: false,
+      message: err.message
+    } as ApiResponse);
+  });
+
+  logger.info('[API] Routes registered');
+}

package/package/src/server/websocket.ts

@@ -0,0 +1,82 @@
+import { WebSocketServer, WebSocket } from 'ws';
+import { AgentRpcService } from '../services/agent-rpc';
+import { AgentEvent } from '../types';
+import { logger } from '../utils/logger';
+
+/**
+ * WebSocket 客户端集合
+ */
+const clients: Set<WebSocket> = new Set();
+
+/**
+ * 设置 WebSocket 处理
+ */
+export function setupWebSocket(wss: WebSocketServer, agentService: AgentRpcService): void {
+
+  // 注册事件回调
+  agentService.registerCallback((event: AgentEvent) => {
+    broadcast(event);
+  });
+
+  wss.on('connection', (ws: WebSocket) => {
+    logger.info('[WebSocket] Client connected');
+    clients.add(ws);
+
+    // 发送客户端数量
+    broadcastClientCount();
+
+    // 发送欢迎消息
+    ws.send(JSON.stringify({
+      type: 'welcome',
+      message: 'Connected to AICode Agent WebSocket'
+    }));
+
+    ws.on('close', () => {
+      logger.info('[WebSocket] Client disconnected');
+      clients.delete(ws);
+      broadcastClientCount();
+    });
+
+    ws.on('error', (error) => {
+      logger.error('[WebSocket] Error:', error);
+      clients.delete(ws);
+    });
+  });
+
+  logger.info('[WebSocket] Server initialized');
+}
+
+/**
+ * 广播消息给所有客户端
+ */
+function broadcast(event: AgentEvent): void {
+  const message = JSON.stringify(event);
+  clients.forEach((client) => {
+    if (client.readyState === WebSocket.OPEN) {
+      try {
+        client.send(message);
+      } catch (e) {
+        logger.error('[WebSocket] Send error:', e);
+      }
+    }
+  });
+}
+
+/**
+ * 广播客户端数量
+ */
+function broadcastClientCount(): void {
+  const message = JSON.stringify({
+    type: 'client_count',
+    count: clients.size
+  });
+  clients.forEach((client) => {
+    if (client.readyState === WebSocket.OPEN) {
+      try {
+        client.send(message);
+      } catch (e) {
+        // 忽略
+      }
+    }
+  });
+}