imtoagent 0.2.0

package/modules/agent/opencode.ts

@@ -0,0 +1,247 @@
+// OpenCode Agent 模块
+// 对接 opencode serve HTTP API，通过 :18899 Anthropic Proxy 调 Provider
+//
+// 设计：薄模块 — oc serve 管理 session/工具/provider，Gateway 只做 IM 翻译
+
+import type { AgentContext, SessionData } from '../types';
+import { parseToBlocks } from '../capabilities';
+import { resolveCapabilities, buildSystemPrompt } from '../prompt-builder';
+import { calculateCost } from '../proxy/anthropic-proxy';
+import * as path from 'path';
+import * as fs from 'fs';
+import { getDataDir } from '../utils/paths';
+
+// ================================================================
+// 配置（从 config.json 读取）
+// ================================================================
+
+interface OpenCodeConfig {
+  serverUrl: string;
+  defaultModel: { providerID: string; modelID: string };
+}
+
+let _ocConfig: OpenCodeConfig | null = null;
+
+export function initOpenCodeConfig(cfg: OpenCodeConfig) {
+  _ocConfig = cfg;
+}
+
+function getOcConfig(): OpenCodeConfig {
+  if (!_ocConfig) {
+    try {
+      
+      const raw = JSON.parse(fs.readFileSync(path.join(getDataDir(), 'config.json'), 'utf-8'));
+      const oc = raw.opencode || {};
+      _ocConfig = {
+        serverUrl: oc.serverUrl || 'http://localhost:4096',
+        defaultModel: oc.defaultModel || { providerID: 'anthropic', modelID: 'claude-sonnet-4-5' },
+      };
+    } catch {
+      _ocConfig = {
+        serverUrl: 'http://localhost:4096',
+        defaultModel: { providerID: 'anthropic', modelID: 'claude-sonnet-4-5' },
+      };
+    }
+  }
+  return _ocConfig;
+}
+
+const OC_SERVER_URL = () => getOcConfig().serverUrl;
+const OC_DEFAULT_MODEL = () => getOcConfig().defaultModel;
+
+// ================================================================
+// OpenCode Server HTTP 客户端
+// ================================================================
+
+interface OcMessagePart {
+  type: string;
+  text?: string;
+  tool_call?: { name: string; arguments: Record<string, any> };
+  tool_result?: { content: string };
+}
+
+interface OcMessage {
+  info: { id: string; role: string; model?: string };
+  parts: OcMessagePart[];
+}
+
+async function ocCreateSession(title: string): Promise<string> {
+  const ac = new AbortController();
+  const timer = setTimeout(() => ac.abort(), 300_000);
+  const res = await fetch(`${OC_SERVER_URL()}/session`, {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify({ title }),
+    signal: ac.signal,
+  }).finally(() => clearTimeout(timer));
+  if (!res.ok) throw new Error(`oc create session: ${res.status} ${await res.text()}`);
+  const data = await res.json();
+  return data.id;
+}
+
+async function ocSendPrompt(
+  sessionId: string,
+  initialText: string,
+  system?: string,
+  onTool?: (name: string, args: Record<string, any>) => void
+): Promise<{ response: string }> {
+  const MAX_TURNS = 50;
+  const TURN_TIMEOUT = 300_000;
+  const startTime = Date.now();
+  const MAX_DURATION = 600_000; // 10 分钟总超时
+
+  let promptText = initialText;
+  let accumulatedResponse = '';
+  let turn = 0;
+
+  while (turn < MAX_TURNS) {
+    if (Date.now() - startTime > MAX_DURATION) {
+      console.error('[OpenCode] 任务总超时 (10min)');
+      break;
+    }
+    turn++;
+
+    const body: any = {
+      model: OC_DEFAULT_MODEL(),
+      parts: [{ type: 'text', text: promptText }],
+    };
+    if (turn === 1 && system) body.system = system;
+
+    const ac = new AbortController();
+    const timer = setTimeout(() => ac.abort(), TURN_TIMEOUT);
+    const res = await fetch(`${OC_SERVER_URL()}/session/${sessionId}/message`, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify(body),
+      signal: ac.signal,
+    }).finally(() => clearTimeout(timer));
+    if (!res.ok) throw new Error(`oc send prompt: ${res.status} ${await res.text()}`);
+
+    const data: OcMessage = await res.json();
+    let hasToolCall = false;
+
+    for (const part of data.parts || []) {
+      if (part.type === 'text' && part.text) {
+        accumulatedResponse += (accumulatedResponse ? '\n' : '') + part.text;
+      } else if (part.type === 'tool_call' && part.tool_call) {
+        hasToolCall = true;
+        if (onTool) onTool(part.tool_call.name, part.tool_call.arguments);
+      }
+    }
+
+    // 纯文本回复或没有 tool_call → 任务完成
+    if (!hasToolCall) break;
+
+    // 有 tool_call，继续推进（空 prompt）
+    promptText = '继续';
+  }
+
+  return { response: accumulatedResponse };
+}
+
+async function ocDeleteSession(sessionId: string): Promise<void> {
+  await fetch(`${OC_SERVER_URL()}/session/${sessionId}`, { method: 'DELETE' }).catch(() => {});
+}
+
+async function ocHealthCheck(): Promise<boolean> {
+  try {
+    const res = await fetch(`${OC_SERVER_URL()}/global/health`);
+    return res.ok;
+  } catch {
+    return false;
+  }
+}
+
+// ================================================================
+// Agent 模块类
+// ================================================================
+export class OpenCodeAgentModule {
+  private ctx: AgentContext;
+
+  constructor(ctx: AgentContext) {
+    this.ctx = ctx;
+  }
+
+  async handleMessage(chatId: string, text: string, session: SessionData) {
+    const ctx = this.ctx;
+    console.log(`[${ctx.name}] OpenCode chat=${chatId.slice(-8)}`);
+
+    // 工具回调
+    const onTool = (name: string, args: Record<string, any>) => {
+      const summary = args.command || args.cmd || args.query || JSON.stringify(args).slice(0, 80);
+      ctx.addToolLog(chatId, { name, summary });
+    };
+
+    try {
+      // ① Plan 模式处理
+      let effectiveText = text;
+      if (session.codexMode === 'plan') {
+        effectiveText = `[模式: 先计划后执行] 请先制定一个清晰的计划，等我确认后再执行。用户请求: ${text}`;
+      }
+
+      // ② 清理标记
+      const shouldClear = session.startFresh;
+      session.startFresh = false;
+
+      // ③ 获取或创建 OpenCode session
+      if (shouldClear || !session.ocSessionId) {
+        if (session.ocSessionId) {
+          await ocDeleteSession(session.ocSessionId);
+          console.log(`[${ctx.name}] 已清除 oc session=${session.ocSessionId.slice(-8)}`);
+        }
+        session.ocSessionId = await ocCreateSession(chatId);
+        console.log(`[${ctx.name}] 新建 oc session=${session.ocSessionId.slice(-8)}`);
+      }
+
+      // ④ 发送进度提示
+      await ctx.sendProgress(chatId, '💭 思考中...');
+
+      // ④.⑤ 构建系统提示词
+      const systemPrompt = buildSystemPrompt({
+        imModule: ctx.imModule || null,
+        botName: ctx.name,
+      });
+      console.log(`[${ctx.name}] 📝 system prompt built (${systemPrompt.length} chars, bot=${ctx.name})`);
+
+      // ⑤ 发送 prompt（多轮循环）
+      const { response } = await ocSendPrompt(
+        session.ocSessionId,
+        effectiveText,
+        systemPrompt,
+        onTool,
+      );
+
+      // ⑥ 刷新工具日志
+      ctx.flushToolLog(chatId);
+
+      // ⑦ 输出
+      if (response) {
+        await ctx.sendFormattedReply(chatId, response);
+      } else {
+        await ctx.reply(chatId, '✅ 已完成');
+      }
+
+      // ⑧ 统计
+      const { sharedState } = await import('../proxy/anthropic-proxy');
+      const lastUsage = (sharedState as any).lastCallUsage;
+      if (lastUsage && (lastUsage.inputTokens > 0 || lastUsage.outputTokens > 0)) {
+        const cost = calculateCost(ctx.activeModel, lastUsage.inputTokens, lastUsage.outputTokens);
+        ctx.accumulateStats(session, { ...lastUsage, costUSD: cost });
+        await ctx.sendProgress(chatId,
+          `输入 ${lastUsage.inputTokens.toLocaleString()} Token\n输出 ${lastUsage.outputTokens.toLocaleString()} Token\n费用 $${cost.toFixed(4)}`);
+      }
+
+      // ⑨ 持久化会话
+      ctx.persistSession(chatId, session);
+
+    } catch (err: any) {
+      console.error(`[${ctx.name}] OpenCode 错误: ${err.message}`);
+      await ctx.reply(chatId, `⚠️ OpenCode 出错：${err.message}`);
+    }
+  }
+
+  /** 健康检查 */
+  async healthCheck(): Promise<boolean> {
+    return ocHealthCheck();
+  }
+}

package/modules/bot-context.ts

@@ -0,0 +1,26 @@
+// Bot 上下文 — 网关与代理之间的动态上下文传递
+// 同一进程内共享，网关在 spawn CLI 前设置当前 bot，代理在处理请求时读取
+
+import type { IMCapabilities } from './types';
+import type { ModelAliases } from './proxy/anthropic-proxy';
+
+export type { ModelAliases };
+
+export interface BotContextData {
+  botName: string;
+  caps: IMCapabilities | null;
+  /** Bot 级别的模型别名（/model 命令修改后传入，优先级高于全局 config.json） */
+  modelAliases?: ModelAliases;
+}
+
+let _currentBot: BotContextData | null = null;
+
+/** 网关调用：在 handleMessage 前设置当前 bot */
+export function setCurrentBot(ctx: BotContextData | null) {
+  _currentBot = ctx;
+}
+
+/** 代理调用：读取当前 bot 上下文 */
+export function getCurrentBot(): BotContextData | null {
+  return _currentBot;
+}

package/modules/capabilities.ts

@@ -0,0 +1,189 @@
+// IM 能力 → Agent Prompt + 输出解析
+// Agent 产出文本 → 网关解析为结构化块 → IM 原生渲染
+
+import type { IMCapabilities } from './types';
+
+export type UnifiedBlock =
+  | { type: 'text'; content: string }
+  | { type: 'code_block'; code: string; language: string; title?: string }
+  | { type: 'image'; url: string; alt?: string }
+  | { type: 'card'; title: string; content: string; color?: string; buttons?: { label: string; url?: string }[] }
+  | { type: 'table'; headers: string[]; rows: string[][]; caption?: string }
+  | { type: 'file'; url: string; filename: string }
+  | { type: 'audio'; url: string; filename: string; duration?: number }
+  | { type: 'divider' };
+
+// ================================================================
+// System Prompt：告诉 Agent 可用的输出格式
+// ================================================================
+// 设计原则：
+//   只告诉 Agent 它能通过 markdown 语法表达的能力。
+//   IMCapabilities 里 capability=true 但 parseToBlocks 没有对应语法
+//   → 不生成提示词，避免 Agent 误以为能输出。
+// ================================================================
+
+export function buildCapabilityPrompt(caps: IMCapabilities): string {
+  const lines: string[] = [];
+
+  // 总览
+  lines.push('## IM 客户端环境');
+  lines.push('你通过飞书（Lark）即时通讯与用户对话。你的回复会被网关解析为飞书原生消息格式。');
+  lines.push('');
+
+  // 文本限制
+  lines.push(`**文本限制**：单条消息最多 ${caps.maxTextLength} 字符，超长会自动截断。`);
+
+  // ========== 只能生成 parseToBlocks 支持的能力 ==========
+
+  // 代码 — ``` 语法
+  if (caps.codeBlock) {
+    lines.push('**代码输出**：当输出代码时，使用标准 markdown 代码块（\\```语言\\n代码\\n\\```）。');
+    lines.push('⚠️ 注意：飞书对代码块的渲染有限，长代码建议使用折叠面板或分段输出，避免单条消息过长。');
+  }
+
+  // 图片 — ![]() 语法
+  if (caps.imageSend) {
+    lines.push('**图片**：可以使用 markdown 图片语法 `![描述](URL)` 发送图片。支持本地 file:// 路径（如图表截图）和远程 URL。网关会自动渲染，无需额外上传步骤。');
+  }
+
+  // 表格 + 卡片 — | 语法（需要 cardMessage 容器来渲染）
+  if (caps.cardMessage) {
+    lines.push('**表格**：可以使用标准 markdown 表格语法来展示结构化数据。');
+    lines.push('```');
+    lines.push('| 列A | 列B |');
+    lines.push('| --- | --- |');
+    lines.push('| 数据1 | 数据2 |');
+    lines.push('```');
+    lines.push('**卡片消息**：可以使用富文本卡片（多块内容会自动组合为一张卡片消息）。');
+  }
+
+  // 文件发送 — fileSend + 本地路径语法
+  if (caps.fileSend) {
+    lines.push('**文件发送**：如果你生成了文件（如图表、CSV、代码文件等），在回复中直接使用以下语法即可发送，网关会自动完成上传和投递，你不需要调用任何额外工具：');
+    lines.push('`📎 [文件名](file:///本地绝对路径)`');
+    lines.push('例如：`📎 [分析结果.csv](file:///tmp/result.csv)`');
+  }
+
+
+  // 语音发送 — audioSend + 本地路径语法
+  if (caps.audioSend) {
+    lines.push('**语音/音频**：如果你生成了音频文件（如语音合成、录音等），在回复中直接使用以下语法即可发送，网关会自动处理：');
+    lines.push('`🎙️ [文件名](file:///本地绝对路径)`');
+    lines.push('例如：`🎙️ [语音播报.mp3](file:///tmp/tts-output.mp3)`');
+  }
+
+  // 注：buttonAction 有 IM 能力但无 markdown 语法，不生成提示词
+
+  lines.push('');
+  lines.push('### 行为规则');
+    lines.push('- 不要在回复中提及或尝试调用 lark-cli、feishu 等第三方上传工具——网关会自动解析 📎 和 ![图片]() 语法并完成发送');
+  lines.push('- **每次修改/创建/删除文件后，必须简要汇报结果**（如"已修改 xxx.ts，修复了 YYY 问题"），不要默默完成就结束');
+  lines.push('- 任务完成后用一两句话总结做了什么');
+  lines.push('');
+  lines.push('### 格式转换规则');
+  lines.push('- 你的回复会被按 markdown 格式解析为多个块（文本、代码、图片、卡片等）');
+  lines.push('- 每个块会被渲染为对应的飞书原生元素');
+  lines.push('- 不要提及这些技术细节，直接使用对应格式即可');
+
+  return lines.join('\n');
+}
+
+// ================================================================
+// 输出解析：Agent 文本 → UnifiedBlock[]
+// ================================================================
+
+export function parseToBlocks(text: string, caps: IMCapabilities): UnifiedBlock[] {
+  const blocks: UnifiedBlock[] = [];
+
+  // 构建所有匹配模式
+  type MatchDef = { regex: RegExp; make: (m: RegExpExecArray) => UnifiedBlock };
+  const patterns: MatchDef[] = [];
+  if (caps.codeBlock) {
+    patterns.push({
+      regex: /```(\w*)\n([\s\S]*?)```/g,
+      make: (m) => ({ type: 'code_block', code: m[2].trim(), language: m[1] || '' }),
+    });
+  }
+  if (caps.audioSend) {
+    patterns.push({
+      regex: /🎙️\s*\[([^\]]*)\]\((file:\/\/[^)]+)\)/g,
+      make: (m) => ({ type: 'audio', url: m[2], filename: m[1] }),
+    });
+  }
+  if (caps.imageSend) {
+    patterns.push({
+      regex: /!\[([^\]]*)\]\(([^)]+)\)/g,
+      make: (m) => ({ type: 'image', alt: m[1], url: m[2] }),
+    });
+  }
+  if (caps.fileSend) {
+    patterns.push({
+      regex: /📎\s*\[([^\]]*)\]\((file:\/\/[^)]+)\)/g,
+      make: (m) => ({ type: 'file', url: m[2], filename: m[1] }),
+    });
+  }
+
+  // 表格：仅在有 cardMessage 能力时解析
+  // 匹配 markdown 表格：表头行 | 分隔行 | 数据行...
+  const tableRegex = caps.cardMessage
+    ? /(?:^[^\n]*\n)?\|[^|\n]+\|[^|\n]*\|\n\|[-: |]+\|\n(?:\|[^|\n]+\|[^|\n]*\|\n?)+/gm
+    : null;
+
+  // 第一遍：收集表格匹配位置（标记为"占位"，避免被其他正则误匹配）
+  type TableMatch = { index: number; end: number; block: UnifiedBlock };
+  const tableMatches: TableMatch[] = [];
+  const tableMasked = text; // 保存原文本用于表格内容提取
+  if (tableRegex) {
+    tableRegex.lastIndex = 0;
+    let tm: RegExpExecArray | null;
+    while ((tm = tableRegex.exec(text)) !== null) {
+      const lines = tm[0].split('\n').filter(l => l.startsWith('|'));
+      if (lines.length < 3) continue; // 至少需要 header + separator + 1 row
+      const headerCells = lines[0].split('|').map(c => c.trim()).filter(Boolean);
+      const rows = lines.slice(2).map(l =>
+        l.split('|').map(c => c.trim()).filter(Boolean)
+      );
+      if (headerCells.length === 0 || rows.length === 0) continue;
+      tableMatches.push({
+        index: tm.index,
+        end: tm.index + tm[0].length,
+        block: { type: 'table', headers: headerCells, rows },
+      });
+    }
+  }
+
+  if (patterns.length === 0 && tableMatches.length === 0) return [{ type: 'text', content: text }];
+
+  // 收集所有匹配，按位置排序
+  const hits: { index: number; end: number; block: UnifiedBlock }[] = [];
+  for (const p of patterns) {
+    p.regex.lastIndex = 0;
+    let m: RegExpExecArray | null;
+    while ((m = p.regex.exec(text)) !== null) {
+      hits.push({ index: m.index, end: m.index + m[0].length, block: p.make(m) });
+    }
+  }
+  hits.push(...tableMatches);
+  hits.sort((a, b) => a.index - b.index);
+
+  // 去重（重叠匹配只保留第一个，表格优先因为可能更长）
+  const deduped: typeof hits = [];
+  for (const h of hits) {
+    if (deduped.length > 0 && h.index < deduped[deduped.length - 1].end) continue;
+    deduped.push(h);
+  }
+
+  // 按位置切分
+  let lastIndex = 0;
+  for (const h of deduped) {
+    const before = text.slice(lastIndex, h.index).trim();
+    if (before) blocks.push({ type: 'text', content: before });
+    blocks.push(h.block);
+    lastIndex = h.end;
+  }
+  const after = text.slice(lastIndex).trim();
+  if (after) blocks.push({ type: 'text', content: after });
+  if (blocks.length === 0) blocks.push({ type: 'text', content: text });
+
+  return blocks;
+}