mocode-ai 0.1.0

package/dist/tools/builtins/web-search.js

@@ -0,0 +1,132 @@
+import { MAX_OUTPUT } from '../constants.js';
+import { config } from '../../config/index.js';
+const SEARCH_TIMEOUT_MS = 30000;
+/** 每条结果的 content 截断长度(字符),避免单条正文占满整个输出。 */
+const MAX_CONTENT_CHARS = 800;
+// ---------- web_search ----------
+export const webSearchTool = {
+    name: 'web_search',
+    description: '联网搜索(AnySearch)。返回每条结果的标题/URL/摘要/正文,用于查模型训练数据之外的最新信息(新版本发布、新闻、实时数据、最新 API 文档等)。可选 tag 切换子域:general.general(默认通用网页)/ code.doc(开发文档,须配 params.library)/ code.snippet(代码片段)。',
+    parameters: {
+        type: 'object',
+        properties: {
+            query: { type: 'string', description: '搜索查询词' },
+            max_results: { type: 'integer', description: '返回条数,1-20,默认 10' },
+            tag: {
+                type: 'string',
+                description: '子域能力标签,如 general.general / code.doc / code.snippet;不传走通用搜索',
+            },
+            language: { type: 'string', description: '偏好语言,如 zh-CN / en,默认 zh-CN' },
+            params: {
+                type: 'object',
+                description: '特定 tag 的扩展参数,如 code.doc 的 {"library":"golang"};通用搜索不需要',
+            },
+        },
+        required: ['query'],
+    },
+    async execute(args) {
+        const query = String(args.query ?? '').trim();
+        if (!query)
+            return '错误:query 不能为空。';
+        const body = { query, format: 'json', language: 'zh-CN' };
+        const maxResults = Number(args.max_results);
+        if (Number.isFinite(maxResults)) {
+            body.max_results = Math.min(20, Math.max(1, Math.trunc(maxResults)));
+        }
+        if (args.tag)
+            body.tag = String(args.tag);
+        if (args.language)
+            body.language = String(args.language);
+        if (args.params &&
+            typeof args.params === 'object' &&
+            !Array.isArray(args.params)) {
+            body.params = args.params;
+        }
+        const base = config.searchBaseUrl.replace(/\/+$/, '');
+        const url = `${base}/v1/search`;
+        const headers = {
+            'Content-Type': 'application/json',
+        };
+        // 无 key 走匿名(按 IP 限流 + 每日免费额度);有 key 用付费额度。
+        if (config.searchApiKey) {
+            headers['Authorization'] = `Bearer ${config.searchApiKey}`;
+        }
+        const ctrl = new AbortController();
+        const timer = setTimeout(() => ctrl.abort(), SEARCH_TIMEOUT_MS);
+        try {
+            const resp = await fetch(url, {
+                method: 'POST',
+                headers,
+                body: JSON.stringify(body),
+                signal: ctrl.signal,
+            });
+            const text = await resp.text();
+            let data;
+            try {
+                data = JSON.parse(text);
+            }
+            catch {
+                return `错误:搜索返回非 JSON(HTTP ${resp.status}): ${text.slice(0, 500)}`;
+            }
+            // AnySearch 成功返回 code===0;否则把 message/request_id 喂回 LLM。
+            if (!resp.ok || data?.code !== 0) {
+                const code = data?.code ?? resp.status;
+                const message = data?.message ?? resp.statusText ?? '未知错误';
+                const rid = data?.request_id ? ` (request_id=${data.request_id})` : '';
+                let hint = '';
+                if (resp.status === 401 || resp.status === 403) {
+                    hint = ' API Key 非法/过期;可在 .env 去掉 ANYSEARCH_API_KEY 走匿名免费额度。';
+                }
+                else if (resp.status === 402) {
+                    hint = ' 额度用尽,可次日重试或去掉 key 走匿名额度。';
+                }
+                else if (resp.status === 429) {
+                    hint = ' 触发限流,请稍后重试。';
+                }
+                return `错误:搜索失败 [${code}] ${message}${rid}${hint}`;
+            }
+            const results = data?.data?.results;
+            if (!Array.isArray(results) || results.length === 0) {
+                return `无搜索结果(query="${query}")。`;
+            }
+            const lines = [];
+            results.forEach((r, i) => {
+                const title = String(r?.title ?? '').trim();
+                const u = String(r?.url ?? '').trim();
+                const snippet = String(r?.snippet ?? '').trim();
+                const content = String(r?.content ?? '').trim();
+                lines.push(`[${i + 1}] ${title}`);
+                if (u)
+                    lines.push(`    ${u}`);
+                if (snippet)
+                    lines.push(`    ${snippet}`);
+                if (content) {
+                    const c = content.length > MAX_CONTENT_CHARS
+                        ? content.slice(0, MAX_CONTENT_CHARS) +
+                            ` …(共 ${content.length} 字符)`
+                        : content;
+                    lines.push(`    ${c}`);
+                }
+            });
+            const meta = data?.data?.metadata;
+            if (meta) {
+                lines.push(`(共 ${meta.total_results ?? results.length} 条,耗时 ${meta.search_time_ms ?? '?'}ms)`);
+            }
+            let out = lines.join('\n');
+            if (out.length > MAX_OUTPUT) {
+                out = out.slice(0, MAX_OUTPUT) + '\n...(结果已截断)';
+            }
+            return out;
+        }
+        catch (e) {
+            if (ctrl.signal.aborted) {
+                return `错误:搜索超时(${SEARCH_TIMEOUT_MS}ms)。`;
+            }
+            const msg = e instanceof Error ? e.message : String(e);
+            return `错误:联网搜索请求失败: ${msg}`;
+        }
+        finally {
+            clearTimeout(timer);
+        }
+    },
+};

package/dist/tools/builtins/write-file.js

@@ -0,0 +1,23 @@
+import { writeFile, mkdir } from 'node:fs/promises';
+import { resolve, dirname } from 'node:path';
+// ---------- write_file ----------
+export const writeFileTool = {
+    name: 'write_file',
+    description: '创建或覆盖文件,自动创建父目录。',
+    parameters: {
+        type: 'object',
+        properties: {
+            path: { type: 'string', description: '文件路径' },
+            content: { type: 'string', description: '完整文件内容' },
+        },
+        required: ['path', 'content'],
+    },
+    async execute(args) {
+        const path = String(args.path);
+        const content = String(args.content);
+        const full = resolve(path);
+        await mkdir(dirname(full), { recursive: true });
+        await writeFile(full, content, 'utf8');
+        return `已写入 ${path} (${content.length} 字符)`;
+    },
+};

package/dist/tools/constants.js

@@ -0,0 +1,11 @@
+/** 工具共享的截断 / 上限 / 忽略规则。 */
+export const MAX_FILE_LINES = 2000;
+export const MAX_OUTPUT = 20000;
+export const MAX_RESULTS = 100;
+/** 进 history 的单条工具结果上限(字符)。push-time 第一层裁剪,保 head + 标记 + tail。 */
+export const MAX_HISTORY_RESULT = 8000;
+/** use_skill 结果(SKILL.md 正文)的放宽上限:指令须完整,中截会破坏语义。 */
+export const MAX_SKILL_RESULT = 64000;
+/** 微压缩时旧工具结果截到的存根长度(字符)。 */
+export const MAX_OLD_TOOL_STUB = 600;
+export const IGNORE = ['**/node_modules/**', '**/.git/**'];

package/dist/tools/registry.js

@@ -0,0 +1,32 @@
+import { builtinTools } from './builtins/index.js';
+import { recordMutation } from '../rollback/index.js';
+/**
+ * 工具注册表。当前 = 内置工具;
+ * 未来可在此合并 MCP 工具、用户自定义工具等(见 src/mcp/)。
+ */
+export const tools = builtinTools;
+/** 按名调度工具,统一 try/catch + JSON 解析,返回字符串而非抛错。 */
+export async function executeTool(name, argsRaw) {
+    const tool = tools.find((t) => t.name === name);
+    if (!tool)
+        return `错误:未知工具 "${name}"`;
+    let args;
+    try {
+        args = argsRaw.trim() ? JSON.parse(argsRaw) : {};
+    }
+    catch {
+        return `错误:工具 ${name} 的 arguments 不是合法 JSON: ${argsRaw}`;
+    }
+    try {
+        // 撤销回滚用:write_file/edit_file 改动前记 before 快照(回滚时恢复到轮末状态)。
+        if ((name === 'write_file' || name === 'edit_file') &&
+            typeof args.path === 'string' &&
+            args.path) {
+            recordMutation(args.path);
+        }
+        return await tool.execute(args);
+    }
+    catch (e) {
+        return `错误:工具 ${name} 执行失败: ${e instanceof Error ? e.message : String(e)}`;
+    }
+}

package/dist/tools/types.js

@@ -0,0 +1 @@
+export {};

package/dist/ui/content.js

@@ -0,0 +1,98 @@
+/**
+ * 内容物理行缓冲(Phase 2 滚动回滚用)。
+ *
+ * 存折行后的物理行,每行自洽带色,供 layout.repaintViewport 单行直出、无需继承 SGR 状态。
+ * contentWrite 逐字符喂入(feedChar / feedSgr / breakRow);eraseSegmentBack 用 popRows 同步删段。
+ *
+ * **SGR 自洽模型**:每行存 `rowStartSgr + curRaw + \x1B[0m`:
+ *  - rowStartSgr:本行起始时继承的活跃 SGR 前缀(来自上一行末状态);
+ *  - curRaw:本行原始字节——可见字符 + 行内出现的 SGR 码(保留行内开合,如 `${dim}ab${reset}cd`);
+ *  - 末尾 reset:使单行渲染不依赖后续。
+ * 这样行内 SGR 变化与跨行继承都正确——简单"前缀+可见文本"模型会丢行内开合的色。
+ *
+ * **当前行**:任何 feedChar/feedSgr/breakRow 后 hasCurrent=true;breakRow(\n/折行)提交本行并开新空行
+ * (hasCurrent 仍 true,代表光标所在的新空行)。这让 offset=0 的 viewport 重画与实时屏严格一致
+ * (含 \n 后的光标空行),避免回尾时内容错位。
+ *
+ * 主题(theme.ts)只用单参数 SGR 码,故 curSgr 状态机简单:遇 \x1B[0m / \x1B[m 清空,其余累加。
+ * 上限 20000 行防无限增长(超限丢头部,批量 trim)。
+ */
+const MAX_ROWS = 20000;
+let rows = [];
+let curSgr = ''; // 当前活跃 SGR 状态(行内演进;行末值 = 下行继承值)
+let rowStartSgr = ''; // 当前行起始继承的 SGR 前缀
+let curRaw = ''; // 当前行原始字节(可见 + 行内 SGR 码)
+let hasCurrent = false; // 是否有当前行(feed/breakRow 后;新空行也算)
+let segMark = null; // 段起点快照(供 eraseSegment 恢复)
+export function reset() {
+    rows = [];
+    curSgr = '';
+    rowStartSgr = '';
+    curRaw = '';
+    hasCurrent = false;
+    segMark = null;
+}
+/** 喂一个 SGR 码(\\x1B[...m):入 curRaw,并更新活跃状态(遇 reset 清空)。 */
+export function feedSgr(code) {
+    curRaw += code;
+    if (code === '\x1B[0m' || code === '\x1B[m')
+        curSgr = '';
+    else
+        curSgr += code;
+    hasCurrent = true;
+}
+/** 喂一个可见字符(已按码点取出)。 */
+export function feedChar(ch) {
+    curRaw += ch;
+    hasCurrent = true;
+}
+/** 提交当前行为自洽物理行,开新空行(SGR 状态继承到下行)。在 \\n / 折行 / tab 换行时调。 */
+export function breakRow() {
+    rows.push(rowStartSgr + curRaw + '\x1B[0m');
+    if (rows.length > MAX_ROWS + 512)
+        rows.splice(0, rows.length - MAX_ROWS);
+    curRaw = '';
+    rowStartSgr = curSgr; // 下行继承本行末状态
+    hasCurrent = true; // 新空行即当前行
+}
+/** 标记段(思考段)起点:快照当前缓冲状态,供 eraseSegment 恢复。 */
+export function beginSegment() {
+    segMark = { rowIdx: rows.length, rowStartSgr, curSgr, curRaw, hasCurrent };
+}
+/**
+ * 删段并恢复到 beginSegment 时的状态;返回段物理行数(= segLines + 末行部分,与屏幕擦除行数一致)。
+ * 段足迹 = 自 mark 后提交的行(committedErased)+ 当前行若有内容(currentErased)。
+ * 恢复后当前行 = 段起点的那一行(折叠标题将写入此处),保证滚动回看只看到标题、不看到原文。
+ */
+export function eraseSegment() {
+    if (!segMark)
+        return 0;
+    const committedErased = rows.length - segMark.rowIdx;
+    const currentErased = hasCurrent && curRaw.length > 0 ? 1 : 0;
+    rows.length = segMark.rowIdx;
+    rowStartSgr = segMark.rowStartSgr;
+    curSgr = segMark.curSgr;
+    curRaw = segMark.curRaw;
+    hasCurrent = segMark.hasCurrent;
+    segMark = null;
+    return committedErased + currentErased;
+}
+/** 快照(已提交行 + 当前行若有)。viewport 取行窗用。 */
+function snapshot() {
+    return hasCurrent
+        ? [...rows, rowStartSgr + curRaw + '\x1B[0m']
+        : rows;
+}
+/** 距尾 offset 行、取 count 行的窗口(映射到屏 1..count)。offset=0 即尾窗。 */
+export function sliceFromEnd(offset, count) {
+    const all = snapshot();
+    const end = all.length - offset;
+    if (end <= 0)
+        return [];
+    const start = Math.max(0, end - count);
+    return all.slice(start, end);
+}
+/** 快照总行数(含当前行),供 scrollBy 钳位。 */
+export function totalRows() {
+    return rows.length + (hasCurrent ? 1 : 0);
+}