summaryrefslogtreecommitdiffhomepage
path: root/packages/core/src
diff options
context:
space:
mode:
authorAdam Malczewski <[email protected]>2026-06-02 16:48:12 +0900
committerAdam Malczewski <[email protected]>2026-06-02 16:48:12 +0900
commit644cfb097b845384c4a470a97328922e5d6dd129 (patch)
tree75d43b9a272a412c3524ba762f177e2c8a614c21 /packages/core/src
parent0f99a0c92707b44aef7c627012c4711bad5a8efd (diff)
downloaddispatch-644cfb097b845384c4a470a97328922e5d6dd129.tar.gz
dispatch-644cfb097b845384c4a470a97328922e5d6dd129.zip
fix(search_code): surface cs failures + harden query/path handling
Address findings from an independent code review of the search_code tool: - Critical: cs failures (non-zero exit, or SIGTERM from the spawn timeout) were swallowed and reported to the model as 'No matches found', discarding stderr. Now capture exit code + signal from 'close' and return a real Error: (timeout message for SIGTERM, exit-code + stderr otherwise). cs exits 0 on a genuine no-match, so that path still reports correctly. - High: a query beginning with '-' (e.g. '-foo') was parsed by cs as a (usually invalid) flag. Insert a '--' separator before the query so it is always treated as the positional search term. - Low: relative-path display fallback now matches the workdir only at a path boundary, so a sibling dir sharing the prefix (e.g. /app vs /app-secrets) isn't rendered as a '../app-secrets/...' path. Adds tests for the non-zero-exit (stderr surfaced, not 'No matches') and dash-leading-query cases. All tests (598), biome, and tsc pass.
Diffstat (limited to 'packages/core/src')
-rw-r--r--packages/core/src/tools/search-code.ts48
1 files changed, 40 insertions, 8 deletions
diff --git a/packages/core/src/tools/search-code.ts b/packages/core/src/tools/search-code.ts
index 39cbbdd..c9ad086 100644
--- a/packages/core/src/tools/search-code.ts
+++ b/packages/core/src/tools/search-code.ts
@@ -1,5 +1,5 @@
import { spawn } from "node:child_process";
-import { relative } from "node:path";
+import { relative, sep } from "node:path";
import { z } from "zod";
import type { ToolDefinition } from "../types/index.js";
import { canonicalize } from "./path-utils.js";
@@ -134,11 +134,19 @@ export function createSearchCodeTool(workingDirectory: string): ToolDefinition {
}
const flags = buildFlags(args, searchDir);
- const spawnArgs = [...flags, query];
+ // `--` terminates cs flag parsing so a query that begins with "-"
+ // (e.g. "-hello" or "--foo") is treated as the positional search term
+ // rather than parsed as a (possibly invalid) cs flag.
+ const spawnArgs = [...flags, "--", query];
let stdout = "";
let stderr = "";
- const result = await new Promise<{ error?: string; errorCode?: string }>((resolve) => {
+ const result = await new Promise<{
+ code: number | null;
+ signal: NodeJS.Signals | null;
+ error?: string;
+ errorCode?: string;
+ }>((resolve) => {
const child = spawn(resolveCsBin(), spawnArgs, {
cwd: workingDirectory,
env: process.env,
@@ -151,9 +159,14 @@ export function createSearchCodeTool(workingDirectory: string): ToolDefinition {
child.stderr?.on("data", (d: Buffer) => {
stderr += d.toString();
});
- child.on("close", () => resolve({}));
+ child.on("close", (code, signal) => resolve({ code, signal }));
child.on("error", (err) =>
- resolve({ error: err.message, errorCode: (err as NodeJS.ErrnoException).code }),
+ resolve({
+ code: null,
+ signal: null,
+ error: err.message,
+ errorCode: (err as NodeJS.ErrnoException).code,
+ }),
);
});
@@ -165,6 +178,22 @@ export function createSearchCodeTool(workingDirectory: string): ToolDefinition {
return `Error: failed to run cs: ${result.error}`;
}
+ // A signal kill (e.g. SIGTERM from the spawn timeout) or a non-zero
+ // exit means cs failed — surface it (with stderr) instead of silently
+ // reporting "No matches found". cs exits 0 even when there are no
+ // matches, so a clean exit always falls through to the parsing below.
+ if (result.signal) {
+ const detail = stderr.trim() ? `\n${stderr.trim()}` : "";
+ if (result.signal === "SIGTERM") {
+ return `Error: cs search timed out after ${TIMEOUT_MS / 1000}s. Try a narrower query or a smaller path.${detail}`;
+ }
+ return `Error: cs was terminated by signal ${result.signal}.${detail}`;
+ }
+ if (result.code !== 0) {
+ const detail = stderr.trim() ? `\n${stderr.trim()}` : "";
+ return `Error: cs exited with code ${result.code}.${detail}`;
+ }
+
// cs prints `null` (and exit 0) when there are no matches.
const trimmed = stdout.trim();
if (trimmed === "" || trimmed === "null") {
@@ -226,11 +255,14 @@ function buildFlags(args: Record<string, unknown>, searchDir: string): string[]
/** Render cs JSON results into compact, readable per-file blocks. */
function formatResults(results: CsResult[], absoluteWorkDir: string): string {
const blocks: string[] = [];
+ // Match the workdir only at a path boundary so a sibling dir that merely
+ // shares the prefix (e.g. workdir "/app" vs "/app-secrets") isn't treated
+ // as nested and rendered as a "../app-secrets/..." relative path.
+ const workdirPrefix = absoluteWorkDir.endsWith(sep) ? absoluteWorkDir : absoluteWorkDir + sep;
for (const r of results) {
// Present paths relative to the workdir so output is portable and compact.
- const rel = r.location.startsWith(absoluteWorkDir)
- ? relative(absoluteWorkDir, r.location) || r.filename
- : r.location;
+ const insideWorkdir = r.location === absoluteWorkDir || r.location.startsWith(workdirPrefix);
+ const rel = insideWorkdir ? relative(absoluteWorkDir, r.location) || r.filename : r.location;
const lang = r.language ? ` [${r.language}]` : "";
const score = typeof r.score === "number" ? ` (score ${r.score.toFixed(2)})` : "";
const header = `${rel}${lang}${score}`;