1import { readFile } from "node:fs/promises";
 2import { expandHomePath } from "../../util/path.js";
 3import { applyConfigOverrides, loadConfig } from "../../config/loader.js";
 4import { createWorkspace } from "../../workspace/manager.js";
 5import { createGrepTool } from "../../agent/tools/grep.js";
 6import { createReadTool } from "../../agent/tools/read.js";
 7import { createLsTool } from "../../agent/tools/ls.js";
 8import { createFindTool } from "../../agent/tools/find.js";
 9import { createGitLogTool } from "../../agent/tools/git/log.js";
10import { createGitShowTool } from "../../agent/tools/git/show.js";
11import { createGitBlameTool } from "../../agent/tools/git/blame.js";
12import { createGitDiffTool } from "../../agent/tools/git/diff.js";
13import { createGitRefsTool } from "../../agent/tools/git/refs.js";
14import { createGitCheckoutTool } from "../../agent/tools/git/checkout.js";
15import { runAgent } from "../../agent/runner.js";
16import { REPO_SYSTEM_PROMPT } from "../../agent/prompts/repo.js";
17import { createEventLogger, printUsageSummary } from "../output.js";
18import { CloneError } from "../../util/errors.js";
19import simpleGit from "simple-git";
20
21export interface RepoCommandOptions {
22  query: string;
23  uri: string;
24  ref?: string;
25  full: boolean;
26  model?: string;
27  verbose: boolean;
28  cleanup: boolean;
29}
30
31export async function runRepoCommand(options: RepoCommandOptions): Promise<void> {
32  const { config } = await loadConfig();
33  const overrides = applyConfigOverrides(config, {
34    defaults: { model: options.model ?? config.defaults.model, cleanup: options.cleanup },
35    repo: { model: options.model ?? config.repo.model },
36  });
37
38  const workspace = await createWorkspace({ cleanup: overrides.defaults.cleanup });
39
40  try {
41    const logger = createEventLogger({ verbose: options.verbose });
42
43    let systemPrompt = REPO_SYSTEM_PROMPT;
44    const promptPath = overrides.repo.system_prompt_path;
45    if (promptPath) {
46      systemPrompt = await readFile(expandHomePath(promptPath), "utf8");
47    }
48
49    const git = simpleGit();
50    const cloneArgs: string[] = [];
51    if (!options.full) {
52      const depth = overrides.repo.default_depth ?? 1;
53      const blobLimit = overrides.repo.blob_limit ?? "5m";
54      cloneArgs.push("--depth", String(depth), `--filter=blob:limit=${blobLimit}`);
55    }
56
57    try {
58      await git.clone(options.uri, workspace.path, cloneArgs);
59    } catch (error: any) {
60      if (!overrides.defaults.cleanup) {
61        console.error(`Workspace preserved at ${workspace.path}`);
62      }
63      throw new CloneError(options.uri, error?.message ?? String(error));
64    }
65
66    const repoGit = simpleGit(workspace.path);
67    if (options.ref) {
68      await repoGit.checkout(options.ref);
69    }
70
71    const tools = [
72      createReadTool(workspace.path),
73      createGrepTool(workspace.path),
74      createLsTool(workspace.path),
75      createFindTool(workspace.path),
76      createGitLogTool(workspace.path),
77      createGitShowTool(workspace.path),
78      createGitBlameTool(workspace.path),
79      createGitDiffTool(workspace.path),
80      createGitRefsTool(workspace.path),
81      createGitCheckoutTool(workspace.path),
82    ];
83
84    const result = await runAgent(options.query, {
85      model: overrides.repo.model ?? overrides.defaults.model,
86      systemPrompt,
87      tools,
88      onEvent: logger,
89      config: overrides,
90    });
91
92    process.stdout.write(result.message + "\n");
93    printUsageSummary(result.usage as any, result.requestCount);
94  } finally {
95    await workspace.cleanup();
96  }
97}