ace-tool-windows 0.1.0 → 0.1.3

package/README.md

@@ -1,115 +1,260 @@
-# rust-win32 ace-tool（Win32 GUI + MCP stdio）
-
-本目录是 ace-tool 的 Rust + 纯 Win32 重写版本。目标是与 Node 版本功能 1:1 对齐，同时保持 MCP stdio 兼容，供 Codex 客户端调用。
-
-## 对齐清单（必须与 Node 一致）
-
-### MCP 协议
-- [ ] 基于 stdio 的 JSON-RPC
-- [ ] ListTools 返回 search_context + enhance_prompt schema
-- [ ] CallTool 按名称分发并返回 content[].text
-- [ ] MCP logging 通道推送（level + data）
-
-### CLI / 配置
-- [ ] 必填参数：--base-url、--token
-- [ ] 可选参数：--enable-log
-- [ ] base_url 自动规范化（必须 https，去掉末尾 /）
-
-### 项目数据
-- [ ] 项目根目录创建 .ace-tool/
-- [ ] index.json 写入 .ace-tool/
-- [ ] --enable-log 时写入 ace-tool.log
-- [ ] .ace-tool 自动加入 .gitignore
-
-### search_context 工具
-- [ ] project_root_path / query 输入校验
-- [ ] 路径统一为正斜杠
-- [ ] 目录存在性与类型检查
-- [ ] 检索前自动索引
-- [ ] POST {baseUrl}/agents/codebase-retrieval
-- [ ] 返回 formatted_retrieval 或友好错误信息
-
-### 索引行为
-- [ ] 默认文本后缀与排除规则与 Node 一致
-- [ ] 读取编码兜底（utf-8、gbk、gb2312、latin1）
-- [ ] 二进制内容检测并跳过
-- [ ] 清洗控制字符
-- [ ] 按 maxLinesPerBlob 分片（默认 800）
-- [ ] Blob 名称 hash：SHA-256(path + content)
-- [ ] MAX_BLOB_SIZE：单 blob 500KB
-- [ ] MAX_BATCH_SIZE：单批 5MB
-- [ ] 基于 index.json 的增量索引
-- [ ] 按 blob 数量自适应上传策略（batch size + 并发）
-- [ ] 指数退避重试 + 友好错误映射
-
-### enhance_prompt 工具
-- [ ] POST {baseUrl}/prompt-enhancer（payload 含 nodes/chat_history/blobs）
-- [ ] 语言检测（中文输入 -> 中文输出；英文输入 -> 英文输出）
-- [ ] 工具名映射：codebase-retrieval -> search_context
-- [ ] 8 分钟超时并回退到原始 prompt
-
-### Win32 UI（替代浏览器 UI）
-- [ ] 四个动作：发送增强 / 使用原始 / 继续增强 / 结束对话
-- [ ] Session 状态：pending / completed / timeout
-- [ ] UI 与增强流程通过通道协作
-
-### 错误提示
-- [ ] Token 无效（401）/ 访问被拒绝（403）
-- [ ] SSL 错误 / 非 https / DNS / 超时 / 连接被拒绝
-- [ ] 路径不存在 / 非目录 / 空索引
-
-### 日志格式
-- [ ] MCP logging 推送（level + data）
-- [ ] 文件日志行格式："YYYY-MM-DD HH:MM:SS | LEVEL | message"
-- [ ] 新日志流写入 Session 分隔符
-
-## 目录结构（规划）
-
-```
-rust-win32/
-- Cargo.toml
-- README.md
-- src/
-  - main.rs
-  - mcp/
-  - index/
-  - enhancer/
-  - ui/
-  - utils/
-- tests/
-```
+# ace-tool-windows
+
+Rust + Win32 的 MCP Server（stdio），提供 `search_context` 与 `enhance_prompt`，并用原生 Win32 GUI 替代浏览器交互。
+
+> 原项目：https://github.com/eastxiaodong/ace-tool
+
+## 目录
+
+- 项目简介
+- 功能特性
+- 环境要求
+- 快速开始
+- 配置为 MCP Server
+- 从源码运行
+- MCP 工具说明
+- 配置说明
+- npm 打包与发布
+- 项目结构
+- 对齐清单（Node 版本）
+- 测试
+- License
+- 致谢
+
+## 项目简介
+
+本仓库是 ace-tool 的 Rust + Win32 重写版本，目标与 Node 版本功能 1:1 对齐，同时保持 MCP stdio 兼容，供 Codex 客户端调用。
+
+## 功能特性
+
+- MCP JSON-RPC over stdio，支持工具列举与调用
+- `search_context`：索引 + 检索代码库上下文
+- `enhance_prompt`：调用远端增强服务，支持中文/英文自动判断
+- Win32 GUI 交互窗口，提供继续增强 / 使用原始 / 结束对话
+- 本地索引与日志输出（`.ace-tool/`）
+
+## 环境要求
+
+- Windows 10/11（Win32 GUI）
+- Rust 工具链（建议稳定版）
+- Node.js + npm（用于发布到 npm）
+
+## 快速开始
+
+全局安装后使用：
+
+```powershell
+npm i -g ace-tool-windows
+ace-tool-win --base-url <URL> --token <TOKEN> [--enable-log]
+```
+
+## 配置为 MCP Server
+
+本项目是 MCP stdio Server。配置时只需要声明 command + args，不需要额外端口配置。
+
+说明：不同 MCP 客户端字段名可能是 `mcpServers` 或 `servers`，以客户端文档为准。下面给出两种常见 JSON 格式。
+
+### 方式一：mcpServers 格式（常见）
+
+使用 npm 全局命令：
+
+```json
+{
+  "mcpServers": {
+    "ace-tool-windows": {
+      "command": "ace-tool-win",
+      "args": ["--base-url", "<URL>", "--token", "<TOKEN>", "--enable-log"]
+    }
+  }
+}
+```
+
+使用本地 exe 路径：
+
+```json
+{
+  "mcpServers": {
+    "ace-tool-windows": {
+      "command": "C:\\\\path\\\\to\\\\ace-tool-win.exe",
+      "args": ["--base-url", "<URL>", "--token", "<TOKEN>"]
+    }
+  }
+}
+```
+
+### 方式二：JSON 模式（servers 格式）
+
+```json
+{
+  "servers": {
+    "ace-tool-windows": {
+      "command": "ace-tool-win",
+      "args": ["--base-url", "<URL>", "--token", "<TOKEN>"]
+    }
+  }
+}
+```
+
+### 参数说明
+
+- `--base-url` 必填，服务地址；未写协议会自动补 `https://`
+- `--token` 必填，ACE 服务 Token
+- `--enable-log` 可选，写入 `.ace-tool/ace-tool.log`
+
+> 提示：`enhance_prompt` 会弹 Win32 窗口确认；如果只想后台检索，可只调用 `search_context`。
 
-## 使用方式
+## 从源码运行
 
-```bash
-# 在 rust-win32/ 目录下
+```powershell
+# 在仓库根目录
 cargo run -- --base-url <URL> --token <TOKEN> [--enable-log]
 ```
 
-## npm 发布与安装（Win32 x64）
+## MCP 工具说明
+
+### search_context
+
+- 传入项目根路径与检索 query
+- 自动进行索引（`.ace-tool/index.json`）
+- 调用 `/agents/codebase-retrieval` 返回 formatted_retrieval
+
+### enhance_prompt
+
+- 调用 `/prompt-enhancer`
+- 语言检测：中文输入 -> 中文输出；英文输入 -> 英文输出
+- 工具名映射：`codebase-retrieval` -> `search_context`
+- 8 分钟超时回退到原始 prompt
+
+## 配置说明
+
+必填参数：
+- `--base-url`
+- `--token`
+
+可选参数：
+- `--enable-log`（写入 `.ace-tool/ace-tool.log`）
+
+### 认证 URL 与 Token 获取
+
+认证服务的 URL 与 Token 获取方式以原项目说明为准，建议先阅读原项目文档与相关讨论：
+
+- 原项目仓库：https://github.com/eastxiaodong/ace-tool
+- Issue #6: baseUrl 和 Token 有办法获取吗？
+
+注意：Issue #6 中有用户提到 baseUrl 直接指向 augmentcode.com，但未给出 Token 的官方获取流程，请以服务提供方的最新说明为准。
+
+## npm 打包与发布
+
+构建 + 拷贝 exe：
 
 ```powershell
-# 1) 构建 exe
-cargo build --release
-New-Item -Force -ItemType Directory bin | Out-Null
-Copy-Item target\release\rust-win32.exe bin\ace-tool-win.exe
+npm run build:bin
+```
 
-# 2) 本地打包验证
-npm pack
+本地打包验证：
 
-# 3) 发布到 npm（首次需 npm login）
+```powershell
+npm run pack:local
+```
+
+发布到 npm（首次需登录）：
+
+```powershell
 npm publish --access public
 ```
 
-全局安装后使用：
+## 项目结构
+
+```
+ace-tool-windows/
+- Cargo.toml
+- package.json
+- README.md
+- src/
+  - main.rs
+  - mcp/
+  - index/
+  - enhancer/
+  - ui/
+  - utils/
+- tests/
+```
+
+## 对齐清单（Node 版本）
+
+### MCP 协议
+- [ ] 基于 stdio 的 JSON-RPC
+- [ ] ListTools 返回 search_context + enhance_prompt schema
+- [ ] CallTool 按名称分发并返回 content[].text
+- [ ] MCP logging 通道推送（level + data）
+
+### CLI / 配置
+- [ ] 必填参数：--base-url、--token
+- [ ] 可选参数：--enable-log
+- [ ] base_url 自动规范化（必须 https，去掉末尾 /）
+
+### 项目数据
+- [ ] 项目根目录创建 .ace-tool/
+- [ ] index.json 写入 .ace-tool/
+- [ ] --enable-log 时写入 ace-tool.log
+- [ ] .ace-tool 自动加入 .gitignore
+
+### search_context 工具
+- [ ] project_root_path / query 输入校验
+- [ ] 路径统一为正斜杠
+- [ ] 目录存在性与类型检查
+- [ ] 检索前自动索引
+- [ ] POST {baseUrl}/agents/codebase-retrieval
+- [ ] 返回 formatted_retrieval 或友好错误信息
+
+### 索引行为
+- [ ] 默认文本后缀与排除规则与 Node 一致
+- [ ] 读取编码兜底（utf-8、gbk、gb2312、latin1）
+- [ ] 二进制内容检测并跳过
+- [ ] 清洗控制字符
+- [ ] 按 maxLinesPerBlob 分片（默认 800）
+- [ ] Blob 名称 hash：SHA-256(path + content)
+- [ ] MAX_BLOB_SIZE：单 blob 500KB
+- [ ] MAX_BATCH_SIZE：单批 5MB
+- [ ] 基于 index.json 的增量索引
+- [ ] 按 blob 数量自适应上传策略（batch size + 并发）
+- [ ] 指数退避重试 + 友好错误映射
+
+### enhance_prompt 工具
+- [ ] POST {baseUrl}/prompt-enhancer（payload 含 nodes/chat_history/blobs）
+- [ ] 语言检测（中文输入 -> 中文输出；英文输入 -> 英文输出）
+- [ ] 工具名映射：codebase-retrieval -> search_context
+- [ ] 8 分钟超时并回退到原始 prompt
+
+### Win32 UI（替代浏览器 UI）
+- [ ] 四个动作：发送增强 / 使用原始 / 继续增强 / 结束对话
+- [ ] Session 状态：pending / completed / timeout
+- [ ] UI 与增强流程通过通道协作
+
+### 错误提示
+- [ ] Token 无效（401）/ 访问被拒绝（403）
+- [ ] SSL 错误 / 非 https / DNS / 超时 / 连接被拒绝
+- [ ] 路径不存在 / 非目录 / 空索引
+
+### 日志格式
+- [ ] MCP logging 推送（level + data）
+- [ ] 文件日志行格式："YYYY-MM-DD HH:MM:SS | LEVEL | message"
+- [ ] 新日志流写入 Session 分隔符
+
+## 测试
 
 ```powershell
-npm i -g ace-tool-windows
-ace-tool-win --base-url <URL> --token <TOKEN> [--enable-log]
+cargo test
 ```
 
-## 备注
+## License
+
+Apache-2.0
 
-- 这是 Win32 GUI 应用，非 Windows 平台不支持运行。
-- MCP 通信使用 stdio；Codex 需要指向该二进制作为 MCP Server。
-- 本重写不会修改现有 Node 版本实现。
+## 致谢
+
+- 原项目：ace-tool（https://github.com/eastxiaodong/ace-tool）
+
+
+
+

package/bin/ace-tool-win.js

@@ -0,0 +1,237 @@
+#!/usr/bin/env node
+'use strict';
+
+const fs = require('fs');
+const path = require('path');
+const { spawn } = require('child_process');
+
+const DEFAULT_PROTOCOL_VERSION = '2024-11-05';
+const SUPPORTED_PROTOCOL_VERSIONS = new Set([
+  'DRAFT-2026-v1',
+  '2025-06-18',
+  '2025-03-26',
+  '2024-11-05',
+  '2024-10-07',
+]);
+
+function logError(message) {
+  process.stderr.write(`${message}\n`);
+}
+
+const backendExe = process.env.ACE_TOOL_WIN_EXE || path.join(__dirname, 'ace-tool-win.exe');
+if (!fs.existsSync(backendExe)) {
+  logError(`ace-tool-win backend not found: ${backendExe}`);
+  process.exit(1);
+}
+
+const backendArgs = process.argv.slice(2);
+const backend = spawn(backendExe, backendArgs, { stdio: ['pipe', 'pipe', 'pipe'] });
+backend.stderr.on('data', (chunk) => process.stderr.write(chunk));
+
+let backendClosed = false;
+backend.on('exit', (code) => {
+  backendClosed = true;
+  logError(`ace-tool-win backend exited (${code ?? 'unknown'})`);
+  process.exit(code ?? 1);
+});
+backend.on('error', (error) => {
+  backendClosed = true;
+  logError(`ace-tool-win backend failed: ${error.message}`);
+  process.exit(1);
+});
+
+let backendRequestId = 1;
+const backendPending = new Map();
+
+function sendBackend(message) {
+  if (backendClosed) {
+    return Promise.reject(new Error('Backend closed'));
+  }
+  const payload = JSON.stringify(message);
+  const header = `Content-Length: ${Buffer.byteLength(payload)}\r\n\r\n`;
+  backend.stdin.write(header + payload);
+  return new Promise((resolve, reject) => {
+    backendPending.set(message.id, { resolve, reject });
+  });
+}
+
+function createParser(onMessage) {
+  let buffer = Buffer.alloc(0);
+  return (chunk) => {
+    buffer = Buffer.concat([buffer, chunk]);
+    while (true) {
+      const headerEnd = buffer.indexOf('\r\n\r\n');
+      if (headerEnd === -1) {
+        return;
+      }
+      const headerText = buffer.slice(0, headerEnd).toString('utf8');
+      const match = headerText.match(/content-length:\s*(\d+)/i);
+      if (!match) {
+        buffer = buffer.slice(headerEnd + 4);
+        continue;
+      }
+      const contentLength = Number(match[1]);
+      const totalLength = headerEnd + 4 + contentLength;
+      if (buffer.length < totalLength) {
+        return;
+      }
+      const body = buffer.slice(headerEnd + 4, totalLength).toString('utf8');
+      buffer = buffer.slice(totalLength);
+      try {
+        const message = JSON.parse(body);
+        onMessage(message);
+      } catch (error) {
+        logError(`Failed to parse backend message: ${error.message}`);
+      }
+    }
+  };
+}
+
+backend.stdout.on('data', createParser((message) => {
+  if (!message || message.id === undefined || message.id === null) {
+    return;
+  }
+  const pending = backendPending.get(message.id);
+  if (!pending) {
+    return;
+  }
+  backendPending.delete(message.id);
+  if (message.error) {
+    pending.reject(message.error);
+  } else {
+    pending.resolve(message);
+  }
+}));
+
+const backendReady = (async () => {
+  const initMessage = {
+    jsonrpc: '2.0',
+    id: backendRequestId++,
+    method: 'initialize',
+    params: {
+      protocolVersion: DEFAULT_PROTOCOL_VERSION,
+      capabilities: {},
+      clientInfo: { name: 'ace-tool-win-proxy', version: '0.1.3' },
+    },
+  };
+
+  await sendBackend(initMessage);
+})().catch((error) => {
+  logError(`Backend initialize failed: ${error?.message || error}`);
+  process.exit(1);
+});
+
+function sendRouter(message) {
+  const payload = JSON.stringify(message);
+  const header = `Content-Length: ${Buffer.byteLength(payload)}\r\n\r\n`;
+  process.stdout.write(header + payload);
+}
+
+function normalizeMethod(method) {
+  const value = (method || '').toLowerCase();
+  if (value === 'tools.list' || value === 'tools/list' || value === 'list_tools' || value === 'listtools') {
+    return 'tools/list';
+  }
+  if (value === 'tools.call' || value === 'tools/call' || value === 'call_tool' || value === 'calltool') {
+    return 'tools/call';
+  }
+  if (value === 'initialize') {
+    return 'initialize';
+  }
+  if (value === 'notifications/initialized' || value === 'initialized') {
+    return 'initialized';
+  }
+  return method || '';
+}
+
+process.stdin.on('data', createParser(async (message) => {
+  if (!message || typeof message !== 'object') {
+    return;
+  }
+
+  const method = normalizeMethod(message.method);
+  const id = message.id;
+
+  if (method === 'initialized') {
+    return;
+  }
+
+  if (method === 'initialize') {
+    const requested = message.params?.protocolVersion;
+    const protocolVersion = SUPPORTED_PROTOCOL_VERSIONS.has(requested) ? requested : DEFAULT_PROTOCOL_VERSION;
+    sendRouter({
+      jsonrpc: '2.0',
+      id,
+      result: {
+        protocolVersion,
+        capabilities: { tools: {}, logging: {} },
+        serverInfo: { name: 'ace-tool-windows', version: '0.1.3' },
+      },
+    });
+    return;
+  }
+
+  if (method === 'tools/list') {
+    await backendReady;
+    const backendMessage = {
+      jsonrpc: '2.0',
+      id: backendRequestId++,
+      method: 'tools/list',
+      params: message.params || {},
+    };
+    try {
+      const response = await sendBackend(backendMessage);
+      if (response.error) {
+        sendRouter({ jsonrpc: '2.0', id, error: response.error });
+      } else {
+        sendRouter({ jsonrpc: '2.0', id, result: response.result });
+      }
+    } catch (error) {
+      sendRouter({
+        jsonrpc: '2.0',
+        id,
+        error: { code: -32000, message: error?.message || 'Backend error' },
+      });
+    }
+    return;
+  }
+
+  if (method === 'tools/call') {
+    await backendReady;
+    const backendMessage = {
+      jsonrpc: '2.0',
+      id: backendRequestId++,
+      method: 'tools/call',
+      params: message.params || {},
+    };
+    try {
+      const response = await sendBackend(backendMessage);
+      if (response.error) {
+        sendRouter({ jsonrpc: '2.0', id, error: response.error });
+      } else {
+        sendRouter({ jsonrpc: '2.0', id, result: response.result });
+      }
+    } catch (error) {
+      sendRouter({
+        jsonrpc: '2.0',
+        id,
+        error: { code: -32000, message: error?.message || 'Backend error' },
+      });
+    }
+    return;
+  }
+
+  if (id !== undefined && id !== null) {
+    sendRouter({
+      jsonrpc: '2.0',
+      id,
+      error: { code: -32601, message: 'Method not found' },
+    });
+  }
+}));
+
+process.stdin.on('end', () => {
+  if (!backendClosed) {
+    backend.kill();
+  }
+});

package/package.json

@@ -1,9 +1,9 @@
 {
   "name": "ace-tool-windows",
-  "version": "0.1.0",
+  "version": "0.1.3",
   "description": "ACE MCP server for Windows (Win32)",
   "bin": {
-    "ace-tool-win": "bin/ace-tool-win.exe"
+    "ace-tool-win": "bin/ace-tool-win.js"
   },
   "os": [
     "win32"
@@ -12,6 +12,7 @@
     "x64"
   ],
   "files": [
+    "bin/ace-tool-win.js",
     "bin/ace-tool-win.exe",
     "README.md",
     "LICENSE"
@@ -19,7 +20,11 @@
   "license": "Apache-2.0",
   "scripts": {
     "build:bin": "powershell -NoProfile -ExecutionPolicy Bypass -File scripts/build-bin.ps1",
-    "pack:local": "npm run build:bin && npm pack"
+    "pack:local": "npm run build:bin && npm pack",
+    "publish:public": "npm run build:bin && npm publish --access public"
+  },
+  "engines": {
+    "node": ">=18.0.0"
   },
   "repository": {
     "type": "git",