@zqagent/core 0.1.30 → 0.2.5

package/dist/tools/search.cjs

@@ -1,7 +1,9 @@
 "use strict";
+var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __export = (target, all) => {
   for (var name in all)
@@ -15,6 +17,14 @@ var __copyProps = (to, from, except, desc) => {
   }
   return to;
 };
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // src/tools/search.ts
@@ -23,46 +33,516 @@ __export(search_exports, {
   registerSearchTool: () => registerSearchTool
 });
 module.exports = __toCommonJS(search_exports);
-var import_child_process = require("child_process");
+
+// src/tools/execAsync.ts
+var import_node_child_process2 = require("child_process");
+
+// src/tools/killTree.ts
+var import_node_child_process = require("child_process");
+function killTree(pid) {
+  if (!pid) return;
+  if (process.platform === "win32") {
+    try {
+      (0, import_node_child_process.spawnSync)("taskkill", ["/pid", String(pid), "/T", "/F"], { windowsHide: true });
+    } catch {
+    }
+    return;
+  }
+  try {
+    process.kill(-pid, "SIGKILL");
+  } catch {
+    try {
+      process.kill(pid, "SIGKILL");
+    } catch {
+    }
+  }
+}
+
+// src/tools/execAsync.ts
+var liveChildren = /* @__PURE__ */ new Set();
+var exitCleanupInstalled = false;
+function trackChild(pid) {
+  if (!pid) return;
+  liveChildren.add(pid);
+  if (exitCleanupInstalled) return;
+  exitCleanupInstalled = true;
+  process.once("exit", () => {
+    for (const p of liveChildren) killTree(p);
+  });
+}
+var DEFAULT_MAX_BUFFER = 1024 * 1024 * 5;
+var KILL_GRACE_MS = 1e3;
+function makeSink(maxBuffer) {
+  const chunks = [];
+  let len = 0;
+  let over = false;
+  return {
+    push(chunk) {
+      if (len >= maxBuffer) {
+        over = true;
+        return;
+      }
+      const room = maxBuffer - len;
+      if (chunk.length > room) {
+        chunks.push(chunk.subarray(0, room));
+        len = maxBuffer;
+        over = true;
+        return;
+      }
+      chunks.push(chunk);
+      len += chunk.length;
+    },
+    done: () => Buffer.concat(chunks),
+    over: () => over
+  };
+}
+function execAsync(file, argv, opts = {}) {
+  return new Promise((resolve2) => {
+    const maxBuffer = opts.maxBuffer ?? DEFAULT_MAX_BUFFER;
+    const out = makeSink(maxBuffer);
+    const err = makeSink(maxBuffer);
+    if (opts.signal?.aborted) {
+      resolve2({
+        status: null,
+        stdout: Buffer.alloc(0),
+        stderr: Buffer.alloc(0),
+        timedOut: false,
+        truncated: false,
+        aborted: true
+      });
+      return;
+    }
+    let child;
+    try {
+      child = (0, import_node_child_process2.spawn)(file, argv, {
+        ...opts.cwd !== void 0 ? { cwd: opts.cwd } : {},
+        ...opts.env !== void 0 ? { env: opts.env } : {},
+        // 自成进程组（POSIX）：超时/中断时用 killTree 一次收掉整棵进程树
+        detached: process.platform !== "win32",
+        stdio: [opts.input === void 0 ? "ignore" : "pipe", "pipe", "pipe"],
+        windowsHide: true
+      });
+    } catch (e) {
+      resolve2({
+        status: null,
+        stdout: Buffer.alloc(0),
+        stderr: Buffer.alloc(0),
+        error: e,
+        timedOut: false,
+        truncated: false
+      });
+      return;
+    }
+    trackChild(child.pid);
+    let settled = false;
+    let timedOut = false;
+    let aborted = false;
+    let spawnError;
+    let timer = null;
+    let graceTimer = null;
+    const onAbort = () => {
+      aborted = true;
+      killAndSettle();
+    };
+    const finish = (status) => {
+      if (settled) return;
+      settled = true;
+      if (timer) clearTimeout(timer);
+      if (graceTimer) clearTimeout(graceTimer);
+      if (child.pid !== void 0) liveChildren.delete(child.pid);
+      opts.signal?.removeEventListener("abort", onAbort);
+      const result = {
+        status,
+        stdout: out.done(),
+        stderr: err.done(),
+        timedOut,
+        truncated: out.over() || err.over()
+      };
+      if (aborted) result.aborted = true;
+      if (spawnError) result.error = spawnError;
+      resolve2(result);
+    };
+    const killAndSettle = () => {
+      killTree(child.pid);
+      if (settled || graceTimer) return;
+      graceTimer = setTimeout(() => {
+        child.stdout?.destroy();
+        child.stderr?.destroy();
+        finish(null);
+      }, KILL_GRACE_MS);
+    };
+    if (opts.timeoutMs !== void 0 && opts.timeoutMs > 0) {
+      timer = setTimeout(() => {
+        timedOut = true;
+        killAndSettle();
+      }, opts.timeoutMs);
+    }
+    opts.signal?.addEventListener("abort", onAbort, { once: true });
+    child.stdout?.on("data", (c) => out.push(c));
+    child.stderr?.on("data", (c) => err.push(c));
+    child.on("error", (e) => {
+      spawnError = e;
+      if (child.pid === void 0) finish(null);
+    });
+    if (opts.input !== void 0 && child.stdin) {
+      child.stdin.on("error", () => {
+      });
+      child.stdin.end(opts.input);
+    }
+    child.on("close", (code) => finish(code));
+  });
+}
+
+// src/tools/shellRuntime.ts
+var path = __toESM(require("path"), 1);
+var WINDOWS_SUFFIXES = [".exe", ".cmd", ".bat", ".com", ""];
+function pathApi(platform) {
+  return platform === "win32" ? path.win32 : path.posix;
+}
+function joinPath(platform, ...parts) {
+  return pathApi(platform).join(...parts);
+}
+function findExecutable(name, ctx) {
+  if (name.includes("/") || name.includes("\\")) {
+    return ctx.fileExists(name) ? name : null;
+  }
+  const suffixes = ctx.platform === "win32" ? WINDOWS_SUFFIXES : [""];
+  const delimiter = ctx.platform === "win32" ? ";" : ":";
+  const dirs = (ctx.env.PATH ?? "").split(delimiter).filter((d) => d.length > 0);
+  for (const dir of dirs) {
+    for (const suffix of suffixes) {
+      const candidate = joinPath(ctx.platform, dir, name + suffix);
+      if (ctx.fileExists(candidate)) return candidate;
+    }
+  }
+  return null;
+}
+
+// src/tools/shellBackground.ts
+var import_node_child_process3 = require("child_process");
+var DEFAULT_BUFFER_LIMIT = 64 * 1024;
+var RingBuffer = class {
+  constructor(limit = DEFAULT_BUFFER_LIMIT) {
+    this.limit = limit;
+  }
+  limit;
+  chunks = [];
+  size = 0;
+  add(chunk) {
+    if (!chunk) return;
+    this.chunks.push(chunk);
+    this.size += chunk.length;
+    while (this.size > this.limit && this.chunks.length > 1) {
+      this.size -= this.chunks.shift().length;
+    }
+    if (this.size > this.limit) {
+      const joined = this.chunks.join("");
+      this.chunks = [joined.slice(-this.limit)];
+      this.size = this.chunks[0].length;
+    }
+  }
+  get() {
+    const joined = this.chunks.join("");
+    return joined.length > this.limit ? joined.slice(-this.limit) : joined;
+  }
+};
+var ShellTaskManager = class {
+  tasks = /* @__PURE__ */ new Map();
+  seq = 0;
+  cleanupInstalled = false;
+  /** 启动一个后台任务，立即返回快照（不等退出） */
+  start(opts) {
+    this.installCleanup();
+    const id = `shell-${++this.seq}-${Date.now().toString(36)}`;
+    const child = (0, import_node_child_process3.spawn)(opts.file, opts.argv, {
+      cwd: opts.cwd,
+      env: opts.env,
+      detached: process.platform !== "win32",
+      windowsHide: true,
+      stdio: ["ignore", "pipe", "pipe"]
+    });
+    const task = {
+      id,
+      command: opts.command,
+      scope: opts.scope ?? null,
+      child,
+      stdout: new RingBuffer(),
+      stderr: new RingBuffer(),
+      status: "running",
+      exitCode: null,
+      signal: null,
+      startedAt: Date.now(),
+      endedAt: null
+    };
+    this.tasks.set(id, task);
+    child.stdout?.on("data", (d) => task.stdout.add(d.toString("utf-8")));
+    child.stderr?.on("data", (d) => task.stderr.add(d.toString("utf-8")));
+    child.on("error", (err) => task.stderr.add(`[spawn error] ${err.message}
+`));
+    child.on("exit", (code, signal) => {
+      task.exitCode = code;
+      task.signal = signal;
+      task.endedAt = Date.now();
+      if (task.status !== "killed") task.status = "exited";
+    });
+    child.unref();
+    return this.snapshot(task);
+  }
+  /** 读取任务输出与状态 */
+  output(taskId) {
+    const task = this.tasks.get(taskId);
+    if (!task) return { error: `\u672A\u77E5\u7684\u540E\u53F0\u4EFB\u52A1: ${taskId}` };
+    return this.snapshot(task);
+  }
+  /** 终止任务（进程树）并返回快照；传 scope 时只允许终止该归属下的任务 */
+  kill(taskId, scope) {
+    const task = this.tasks.get(taskId);
+    if (!task) return { error: `\u672A\u77E5\u7684\u540E\u53F0\u4EFB\u52A1: ${taskId}` };
+    if (scope !== void 0 && task.scope !== scope) {
+      return { error: `\u672A\u77E5\u7684\u540E\u53F0\u4EFB\u52A1: ${taskId}` };
+    }
+    if (task.status === "running") {
+      task.status = "killed";
+      task.endedAt = Date.now();
+      killTree(task.child.pid);
+    }
+    return this.snapshot(task);
+  }
+  /** 列出任务快照（调试/展示用）；传 scope 时只返回该归属下的任务 */
+  list(scope) {
+    const all = [...this.tasks.values()];
+    return (scope === void 0 ? all : all.filter((t) => t.scope === scope)).map(
+      (t) => this.snapshot(t)
+    );
+  }
+  /** 终止全部任务（进程退出时调用） */
+  disposeAll() {
+    for (const task of this.tasks.values()) {
+      if (task.status === "running") {
+        task.status = "killed";
+        killTree(task.child.pid);
+      }
+    }
+  }
+  snapshot(task) {
+    return {
+      task_id: task.id,
+      command: task.command,
+      status: task.status,
+      running: task.status === "running",
+      exit_code: task.exitCode,
+      signal: task.signal,
+      stdout: task.stdout.get(),
+      stderr: task.stderr.get(),
+      duration_ms: (task.endedAt ?? Date.now()) - task.startedAt
+    };
+  }
+  installCleanup() {
+    if (this.cleanupInstalled) return;
+    this.cleanupInstalled = true;
+    process.once("exit", () => this.disposeAll());
+  }
+};
+
+// src/tools/shell.ts
+var MAX_BUFFER = 1024 * 1024 * 5;
+function decodeOutput(buf) {
+  if (!buf || buf.length === 0) return "";
+  try {
+    return new TextDecoder("utf-8", { fatal: true }).decode(buf);
+  } catch {
+    for (const enc of ["gbk", "big5", "shift_jis", "latin1"]) {
+      try {
+        return new TextDecoder(enc).decode(buf);
+      } catch {
+      }
+    }
+    return buf.toString("latin1");
+  }
+}
+var shellTasks = new ShellTaskManager();
+
+// src/tools/nodeGrep.ts
+var fs = __toESM(require("fs"), 1);
+var path2 = __toESM(require("path"), 1);
+var SKIP_DIRS = /* @__PURE__ */ new Set([
+  ".git",
+  "node_modules",
+  ".svn",
+  ".hg",
+  "dist",
+  "build",
+  ".next",
+  ".cache"
+]);
+function looksBinary(buf) {
+  const n = Math.min(buf.length, 8192);
+  for (let i = 0; i < n; i++) if (buf[i] === 0) return true;
+  return false;
+}
+function nodeGrep(opts) {
+  const maxMatches = opts.maxMatches ?? 50;
+  const maxFileBytes = opts.maxFileBytes ?? 1024 * 1024;
+  let regex;
+  try {
+    regex = new RegExp(opts.pattern);
+  } catch (err) {
+    throw new Error(`\u65E0\u6548\u7684\u6B63\u5219\u8868\u8FBE\u5F0F: ${opts.pattern}\uFF08${String(err)}\uFF09`);
+  }
+  const target = path2.resolve(opts.workspace, opts.target);
+  if (!fs.existsSync(target)) {
+    throw new Error(`\u641C\u7D22\u8DEF\u5F84\u4E0D\u5B58\u5728: ${opts.target}`);
+  }
+  const matches = [];
+  const scanFile = (file) => {
+    let stat2;
+    try {
+      stat2 = fs.statSync(file);
+    } catch {
+      return false;
+    }
+    if (!stat2.isFile() || stat2.size > maxFileBytes) return false;
+    let buf;
+    try {
+      buf = fs.readFileSync(file);
+    } catch {
+      return false;
+    }
+    if (looksBinary(buf)) return false;
+    const rel = path2.relative(opts.workspace, file) || path2.basename(file);
+    const lines = buf.toString("utf-8").split("\n");
+    for (let i = 0; i < lines.length; i++) {
+      const raw = lines[i].replace(/\r$/, "");
+      if (regex.test(raw)) {
+        matches.push({ file: rel, line: i + 1, text: raw.trim().slice(0, 200) });
+        if (matches.length >= maxMatches) return true;
+      }
+    }
+    return false;
+  };
+  const stat = fs.statSync(target);
+  if (stat.isFile()) {
+    scanFile(target);
+    return matches;
+  }
+  const stack = [target];
+  while (stack.length > 0) {
+    const dir = stack.pop();
+    let entries;
+    try {
+      entries = fs.readdirSync(dir, { withFileTypes: true });
+    } catch {
+      continue;
+    }
+    for (const entry of entries) {
+      const full = path2.join(dir, entry.name);
+      if (entry.isDirectory()) {
+        if (!SKIP_DIRS.has(entry.name)) stack.push(full);
+      } else if (entry.isFile()) {
+        if (scanFile(full)) return matches;
+      }
+    }
+  }
+  return matches;
+}
+
+// src/tools/ripgrep.ts
+var fs2 = __toESM(require("fs"), 1);
+var os = __toESM(require("os"), 1);
+var path3 = __toESM(require("path"), 1);
+var import_node_module = require("module");
+var cached;
+function extractBundledRipgrep() {
+  try {
+    const req = typeof require === "function" ? require : (0, import_node_module.createRequire)(path3.join(process.cwd(), "__zqagent__.js"));
+    const sea = req("node:sea");
+    if (!sea?.isSea?.() || typeof sea.getAsset !== "function") return null;
+    const asset = sea.getAsset("rg");
+    if (!asset) return null;
+    const dir = fs2.mkdtempSync(path3.join(os.tmpdir(), "zqagent-rg-"));
+    const file = path3.join(dir, process.platform === "win32" ? "rg.exe" : "rg");
+    fs2.writeFileSync(file, Buffer.from(asset));
+    fs2.chmodSync(file, 493);
+    return file;
+  } catch {
+    return null;
+  }
+}
+function resolveRipgrep() {
+  if (cached !== void 0) return cached;
+  const explicit = (process.env.ZQAGENT_RG ?? "").trim();
+  if (explicit && fs2.existsSync(explicit)) {
+    cached = explicit;
+    return cached;
+  }
+  const bundled = extractBundledRipgrep();
+  if (bundled) {
+    cached = bundled;
+    return cached;
+  }
+  cached = findExecutable("rg", {
+    platform: process.platform,
+    env: process.env,
+    fileExists: (p) => {
+      try {
+        return fs2.statSync(p).isFile();
+      } catch {
+        return false;
+      }
+    }
+  });
+  return cached;
+}
+
+// src/tools/search.ts
+function parseRgOutput(result) {
+  return result.split("\n").map((l) => l.replace(/\r$/, "")).filter((l) => l.length > 0).slice(0, 50).map((line) => {
+    const m = /^(.+?):(\d+):(.*)$/.exec(line);
+    if (!m) return { file: line.trim(), line: 0, text: "" };
+    return { file: m[1], line: Number(m[2]), text: m[3].trim().slice(0, 200) };
+  });
+}
 function makeGrep(workspace) {
-  return function grep(args) {
+  return async function grep(args) {
     const pattern = String(args.pattern ?? "");
     const filePath = String(args.path ?? ".");
     if (!pattern.trim()) {
       return { error: "\u641C\u7D22\u6A21\u5F0F\u4E0D\u80FD\u4E3A\u7A7A" };
     }
-    const execOpts = {
-      cwd: workspace,
-      encoding: "utf-8",
-      timeout: 1e4,
-      maxBuffer: 1024 * 1024 * 2
-    };
-    try {
-      let result;
-      try {
-        result = (0, import_child_process.execFileSync)(
-          "rg",
-          ["--line-number", "--no-heading", "--max-count", "50", pattern, filePath],
-          execOpts
-        ).toString();
-      } catch {
-        result = (0, import_child_process.execFileSync)(
-          "grep",
-          ["-rn", "--max-count", "50", pattern, filePath],
-          execOpts
-        ).toString();
+    const rg = resolveRipgrep();
+    if (rg) {
+      const res = await execAsync(
+        rg,
+        [
+          // --with-filename：单文件模式下 rg 默认不输出文件名前缀，会导致
+          // "行号:内容" 被误当成 "文件:行号:内容" 解析而错位；显式要求带文件名。
+          "--with-filename",
+          "--line-number",
+          "--no-heading",
+          "--max-count",
+          "50",
+          pattern,
+          filePath
+        ],
+        {
+          cwd: workspace,
+          timeoutMs: 1e4,
+          maxBuffer: 1024 * 1024 * 2
+        }
+      );
+      if (res.status === 0) {
+        const matches = parseRgOutput(decodeOutput(res.stdout));
+        return { query: pattern, engine: "ripgrep", matches, count: matches.length };
       }
-      const lines = result.trim().split("\n").slice(0, 50);
-      const matches = lines.map((line) => {
-        const [file, lineno, ...rest] = line.split(":");
-        return { file, line: Number(lineno), text: rest.join(":").trim().slice(0, 200) };
-      });
-      return { query: pattern, matches, count: matches.length };
-    } catch (exc) {
-      if (exc instanceof Error && "status" in exc && exc.status === 1) {
-        return { query: pattern, matches: [], count: 0 };
+      if (res.status === 1) {
+        return { query: pattern, engine: "ripgrep", matches: [], count: 0 };
       }
-      return { error: `\u641C\u7D22\u5931\u8D25: ${String(exc)}` };
+    }
+    try {
+      const matches = nodeGrep({ pattern, target: filePath, workspace });
+      return { query: pattern, engine: "node", matches, count: matches.length };
+    } catch (exc) {
+      return { error: `\u641C\u7D22\u5931\u8D25: ${String(exc instanceof Error ? exc.message : exc)}` };
     }
   };
 }
@@ -75,7 +555,7 @@ function registerSearchTool(registry, workspace) {
       type: "function",
       function: {
         name: "grep",
-        description: "\u5728\u6587\u4EF6\u4E2D\u641C\u7D22\u6587\u672C\u6A21\u5F0F\u3002\u8FD4\u56DE\u5339\u914D\u7684\u6587\u4EF6\u540D\u3001\u884C\u53F7\u548C\u5185\u5BB9\u3002\u4F18\u5148\u4F7F\u7528 ripgrep (rg) \u4EE5\u83B7\u5F97\u66F4\u597D\u7684\u6027\u80FD\u3002",
+        description: "\u5728\u6587\u4EF6\u4E2D\u641C\u7D22\u6587\u672C\u6A21\u5F0F\uFF08\u652F\u6301\u5B8C\u6574 JS \u6B63\u5219\u8BED\u6CD5\uFF09\u3002\u8FD4\u56DE\u5339\u914D\u7684\u6587\u4EF6\u540D\u3001\u884C\u53F7\u548C\u5185\u5BB9\uFF0C\u6700\u591A 50 \u6761\u3002\u4F18\u5148\u4F7F\u7528 ripgrep (rg) \u4EE5\u83B7\u5F97\u66F4\u597D\u7684\u6027\u80FD\uFF1Brg \u4E0D\u53EF\u7528\u65F6\u81EA\u52A8\u56DE\u9000\u5230\u5185\u7F6E\u626B\u63CF\u5668\u3002",
         parameters: {
           type: "object",
           properties: {