tinker-agent 1.3.0 → 1.5.0

package/src/tui/workspace-file-search.ts

@@ -1,90 +1,97 @@
 import { execFile } from "node:child_process";
 import { RIPGREP_MISSING_ERROR, findRipgrepCommand } from "../tools/ripgrep";
-
-const FILE_SEARCH_TIMEOUT_MS = 20_000;
-const FILE_SEARCH_MAX_BUFFER_BYTES = 20_000_000;
+import { DEFAULT_PUBLIC_TOOLING_CONFIG } from "../cli/public-config-contract";
 
 export type WorkspaceFileLister = (
   workspaceRoot: string,
   signal: AbortSignal,
 ) => Promise<readonly string[]>;
 
-export const listWorkspaceFiles: WorkspaceFileLister = (workspaceRoot, signal) =>
-  new Promise((resolve, reject) => {
-    if (signal.aborted) {
-      reject(new Error("Workspace file search was cancelled."));
-      return;
-    }
+export type WorkspaceFileListerOptions = {
+  readonly command?: string;
+  readonly timeoutMs?: number;
+  readonly maxBufferBytes?: number;
+};
+
+export function createWorkspaceFileLister(
+  options: WorkspaceFileListerOptions = {},
+): WorkspaceFileLister {
+  const command = findRipgrepCommand(options.command);
+  const timeoutMs = options.timeoutMs ?? DEFAULT_PUBLIC_TOOLING_CONFIG.grepTimeoutMs;
+  const maxBufferBytes =
+    options.maxBufferBytes ?? DEFAULT_PUBLIC_TOOLING_CONFIG.grepMaxBufferBytes;
+
+  return (workspaceRoot, signal) =>
+    new Promise((resolve, reject) => {
+      if (signal.aborted) {
+        reject(new Error("Workspace file search was cancelled."));
+        return;
+      }
 
-    execFile(
-      findRipgrepCommand(),
-      [
-        "--files",
-        "--hidden",
-        "--glob",
-        "!**/node_modules/**",
-        "--glob",
-        "!**/.git/**",
-        "--glob",
-        "!**/.tinker/**",
-      ],
-      {
-        cwd: workspaceRoot,
-        encoding: "utf8",
-        maxBuffer: FILE_SEARCH_MAX_BUFFER_BYTES,
-        signal,
-        timeout: FILE_SEARCH_TIMEOUT_MS,
-      },
-      (error, stdout, stderr) => {
-        if (signal.aborted) {
-          reject(error ?? new Error("Workspace file search was cancelled."));
-          return;
-        }
+      execFile(
+        command,
+        [
+          "--files",
+          "--hidden",
+          "--glob",
+          "!**/node_modules/**",
+          "--glob",
+          "!**/.git/**",
+          "--glob",
+          "!**/.tinker/**",
+        ],
+        {
+          cwd: workspaceRoot,
+          encoding: "utf8",
+          maxBuffer: maxBufferBytes,
+          signal,
+          timeout: timeoutMs,
+        },
+        (error, stdout, stderr) => {
+          if (signal.aborted) {
+            reject(error ?? new Error("Workspace file search was cancelled."));
+            return;
+          }
 
-        if (error === null) {
-          resolve(splitPaths(stdout));
-          return;
-        }
+          if (error === null) {
+            resolve(splitPaths(stdout));
+            return;
+          }
 
-        const execError = error as Error & {
-          code?: number | string;
-          killed?: boolean;
-          signal?: string | null;
-        };
+          const execError = error as Error & {
+            code?: number | string;
+            killed?: boolean;
+            signal?: string | null;
+          };
 
-        if (execError.code === 1 && stderr.trim() === "") {
-          resolve(splitPaths(stdout));
-          return;
-        }
+          if (execError.code === 1 && stderr.trim() === "") {
+            resolve(splitPaths(stdout));
+            return;
+          }
 
-        if (execError.code === "ENOENT") {
-          reject(new Error(RIPGREP_MISSING_ERROR));
-          return;
-        }
+          if (execError.code === "ENOENT") {
+            reject(new Error(RIPGREP_MISSING_ERROR));
+            return;
+          }
 
-        if (execError.killed === true || typeof execError.signal === "string") {
-          reject(
-            new Error(
-              `Workspace file search timed out after ${FILE_SEARCH_TIMEOUT_MS}ms.`,
-            ),
-          );
-          return;
-        }
+          if (execError.killed === true || typeof execError.signal === "string") {
+            reject(new Error(`Workspace file search timed out after ${timeoutMs}ms.`));
+            return;
+          }
 
-        if (execError.message.includes("maxBuffer")) {
-          reject(
-            new Error(
-              `Workspace file list exceeded ${FILE_SEARCH_MAX_BUFFER_BYTES} bytes.`,
-            ),
-          );
-          return;
-        }
+          if (execError.message.includes("maxBuffer")) {
+            reject(new Error(`Workspace file list exceeded ${maxBufferBytes} bytes.`));
+            return;
+          }
+
+          const detail = stderr.trim() === "" ? execError.message : stderr.trim();
+          reject(new Error(`Workspace file search failed: ${detail}`));
+        },
+      );
+    });
+}
 
-        const detail = stderr.trim() === "" ? execError.message : stderr.trim();
-        reject(new Error(`Workspace file search failed: ${detail}`));
-      },
-    );
-  });
+export const listWorkspaceFiles = createWorkspaceFileLister();
 
 function splitPaths(stdout: string): string[] {
   return stdout