@thegitai/cli 1.0.0-beta.9 → 1.0.0-preview.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (55) hide show
  1. package/README.md +36 -2
  2. package/dist/bin/ai.js +134 -18
  3. package/dist/parsers/NOTICE +18 -0
  4. package/dist/src/agent-mode.js +5 -0
  5. package/dist/src/api/auth.js +3 -3
  6. package/dist/src/api/browser-login.js +0 -16
  7. package/dist/src/api/chat.js +57 -11
  8. package/dist/src/api/http.js +49 -1
  9. package/dist/src/api/models.js +26 -20
  10. package/dist/src/artifact-policy.js +3 -0
  11. package/dist/src/background-jobs.js +410 -0
  12. package/dist/src/cli-args.js +0 -5
  13. package/dist/src/client-environment.js +2 -0
  14. package/dist/src/colors.js +50 -0
  15. package/dist/src/core/clipboard.js +19 -0
  16. package/dist/src/core/image-path-extractor.js +144 -0
  17. package/dist/src/executor.js +48 -12
  18. package/dist/src/help-text.js +11 -6
  19. package/dist/src/markdown-renderer.js +1 -1
  20. package/dist/src/patcher.js +1 -3
  21. package/dist/src/scanner.js +50 -12
  22. package/dist/src/scratch-dir.js +57 -0
  23. package/dist/src/secret-preview.js +0 -10
  24. package/dist/src/session-safety.js +0 -19
  25. package/dist/src/session-store.js +0 -1
  26. package/dist/src/todo-list.js +106 -0
  27. package/dist/src/tool-executor.js +159 -18
  28. package/dist/src/tools/delete-file.js +1 -1
  29. package/dist/src/tools/index.js +6 -0
  30. package/dist/src/tools/patch-file.js +3 -2
  31. package/dist/src/tools/path-suggest.js +81 -8
  32. package/dist/src/tools/read-document.js +2 -2
  33. package/dist/src/tools/read-file.js +14 -7
  34. package/dist/src/tools/replace-document-text.js +3 -11
  35. package/dist/src/tools/restore-checkpoint.js +1 -1
  36. package/dist/src/tools/run-command.js +83 -16
  37. package/dist/src/tools/run-node-script.js +3 -1
  38. package/dist/src/tools/shell-job-kill.js +48 -0
  39. package/dist/src/tools/shell-job-output.js +51 -0
  40. package/dist/src/tools/str-replace.js +3 -2
  41. package/dist/src/tools/undo-edit.js +1 -1
  42. package/dist/src/tools/update-todos.js +27 -0
  43. package/dist/src/tools/write-file.js +1 -1
  44. package/dist/src/tree-sitter-runtime.js +8 -1
  45. package/dist/src/ui/repl.js +313 -23
  46. package/dist/src/ui/tui/bridge.js +0 -4
  47. package/dist/src/ui/tui/build-frame.js +220 -24
  48. package/dist/src/ui/tui/shell-input.js +33 -4
  49. package/dist/src/ui/tui/terminal-title.js +81 -0
  50. package/dist/src/version.js +0 -6
  51. package/dist/vendor/web-tree-sitter/LICENSE +21 -0
  52. package/dist/vendor/web-tree-sitter/NOTICE +13 -0
  53. package/dist/vendor/web-tree-sitter/web-tree-sitter.cjs +4063 -0
  54. package/dist/vendor/web-tree-sitter/web-tree-sitter.wasm +0 -0
  55. package/package.json +14 -15
@@ -1,8 +1,12 @@
1
+ import { drainBackgroundJobNotifications, getBackgroundJob, } from './background-jobs.js';
1
2
  import { canStoreEditSnapshot, isEditToolName, isGitWorkTree, MAX_EDIT_JOURNAL_RECORDS, operationFromSnapshots, readFileEditSnapshot, } from './edit-journal.js';
2
3
  import { clearEditFailure, collectCommandMutations, captureMutationBaseline, ensureActiveCheckpoint, recordEditFailure, recordSessionEdit, rememberCheckpointFiles, } from './session-safety.js';
3
4
  import { buildAgentModeToolBlockedResult, } from './agent-mode.js';
4
5
  import { dispatchTool } from './tools/index.js';
6
+ import { syncIndexFromDisk } from './project-index.js';
7
+ import { PATH_REPAIRING_EDIT_TOOLS, repairFilePath } from './tools/path-suggest.js';
5
8
  import { invalidateShellDiagnosticsCache, runShellDiagnostics, } from './tools/shell-diagnostics.js';
9
+ import { extractTodosArg } from './todo-list.js';
6
10
  const EDIT_FILE_PATH_ARG_ALIASES = [
7
11
  'filePath',
8
12
  'file_path',
@@ -11,6 +15,7 @@ const EDIT_FILE_PATH_ARG_ALIASES = [
11
15
  'file',
12
16
  'filename',
13
17
  ];
18
+ const backgroundCommandTrackers = new Map();
14
19
  function toolCallSummary(call) {
15
20
  const args = call.args && typeof call.args === 'object' ? call.args : {};
16
21
  if (call.name === 'run_command') {
@@ -19,6 +24,17 @@ function toolCallSummary(call) {
19
24
  if (call.name === 'run_node_script') {
20
25
  return String(args.script ?? '').trim().slice(0, 120);
21
26
  }
27
+ if (call.name === 'shell_job_output' ||
28
+ call.name === 'shell_job_kill') {
29
+ return String(args.job_id ?? '').trim();
30
+ }
31
+ if (call.name === 'update_todos') {
32
+ const raw = extractTodosArg(args);
33
+ const todos = Array.isArray(raw) ? raw : [];
34
+ if (todos.length === 0)
35
+ return '';
36
+ return `${todos.length} item${todos.length === 1 ? '' : 's'}`;
37
+ }
22
38
  const filePath = getEditToolFilePath(call);
23
39
  if (filePath)
24
40
  return filePath;
@@ -45,6 +61,94 @@ function getEditToolFilePath(call) {
45
61
  }
46
62
  return '';
47
63
  }
64
+ function editToolWritesSeparateOutput(call) {
65
+ if (call.name !== 'replace_document_text')
66
+ return false;
67
+ const args = call.args && typeof call.args === 'object' ? call.args : {};
68
+ const output = args.outputPath ?? args.output_path;
69
+ return typeof output === 'string' && output.trim().length > 0;
70
+ }
71
+ async function collectTrackedCommandMutations({ session, projectIndex, result, tracker, toolName, toolCallId, turnId, }) {
72
+ const checkpoint = ensureActiveCheckpoint(session.clientState.safety, turnId);
73
+ const records = collectCommandMutations({
74
+ state: session.clientState.safety,
75
+ rootDir: session.rootDir,
76
+ tracker,
77
+ toolName,
78
+ toolCallId,
79
+ turnId,
80
+ checkpointId: checkpoint.id,
81
+ });
82
+ if (!records.length)
83
+ return;
84
+ invalidateShellDiagnosticsCache(session.rootDir);
85
+ rememberCheckpointFiles(session.clientState.safety, session.rootDir, records.map((record) => record.filePath), turnId);
86
+ const priorSync = result.repoSync;
87
+ if (!(priorSync &&
88
+ (priorSync.added || priorSync.modified || priorSync.removed))) {
89
+ result.repoSync = await syncIndexFromDisk(projectIndex);
90
+ }
91
+ result.sessionEdits = records.map((record) => ({
92
+ id: record.id,
93
+ filePath: record.filePath,
94
+ operation: record.operation,
95
+ beforeHash: record.beforeHash,
96
+ afterHash: record.afterHash,
97
+ }));
98
+ result.diagnostics = runShellDiagnostics(session.rootDir);
99
+ }
100
+ export async function collectBackgroundJobUiKillMutations({ session, projectIndex, jobId, result, }) {
101
+ const normalizedJobId = String(jobId ?? result.snapshot?.id ?? '').trim();
102
+ if (!normalizedJobId || !result.ok)
103
+ return;
104
+ const tracked = backgroundCommandTrackers.get(normalizedJobId);
105
+ if (!tracked)
106
+ return;
107
+ const mutationResult = result;
108
+ await collectTrackedCommandMutations({
109
+ session,
110
+ projectIndex,
111
+ result: mutationResult,
112
+ tracker: tracked.tracker,
113
+ toolName: tracked.toolName,
114
+ toolCallId: tracked.toolCallId,
115
+ turnId: tracked.turnId,
116
+ });
117
+ if (result.snapshot?.status === 'running') {
118
+ tracked.tracker = captureMutationBaseline(session.rootDir);
119
+ }
120
+ else {
121
+ backgroundCommandTrackers.delete(normalizedJobId);
122
+ }
123
+ }
124
+ export async function collectBackgroundJobUiOutputMutations({ session, projectIndex, jobId, }) {
125
+ const normalizedJobId = String(jobId ?? '').trim();
126
+ if (!normalizedJobId)
127
+ return;
128
+ const tracked = backgroundCommandTrackers.get(normalizedJobId);
129
+ if (!tracked)
130
+ return;
131
+ const snapshot = getBackgroundJob(normalizedJobId, {
132
+ sessionId: session.sessionId,
133
+ });
134
+ if (!snapshot)
135
+ return;
136
+ await collectTrackedCommandMutations({
137
+ session,
138
+ projectIndex,
139
+ result: {},
140
+ tracker: tracked.tracker,
141
+ toolName: tracked.toolName,
142
+ toolCallId: tracked.toolCallId,
143
+ turnId: tracked.turnId,
144
+ });
145
+ if (snapshot.status === 'running') {
146
+ tracked.tracker = captureMutationBaseline(session.rootDir);
147
+ }
148
+ else {
149
+ backgroundCommandTrackers.delete(normalizedJobId);
150
+ }
151
+ }
48
152
  function recordAssistantEdit(session, call, result, before) {
49
153
  if (!before || !isEditToolName(call.name))
50
154
  return;
@@ -119,7 +223,14 @@ export async function executeLocalToolCall(toolContext, session, call) {
119
223
  session.onToolEvent?.({ call, result });
120
224
  return result;
121
225
  }
122
- const filePathBeforeEdit = isEditToolName(call.name) ? getEditToolFilePath(call) : '';
226
+ const rawEditFilePath = isEditToolName(call.name)
227
+ ? getEditToolFilePath(call)
228
+ : '';
229
+ const filePathBeforeEdit = rawEditFilePath &&
230
+ PATH_REPAIRING_EDIT_TOOLS.has(call.name) &&
231
+ !editToolWritesSeparateOutput(call)
232
+ ? repairFilePath(session.rootDir, rawEditFilePath)
233
+ : rawEditFilePath;
123
234
  if (filePathBeforeEdit) {
124
235
  rememberCheckpointFiles(session.clientState.safety, session.rootDir, [filePathBeforeEdit], session.turnState.id);
125
236
  }
@@ -131,6 +242,7 @@ export async function executeLocalToolCall(toolContext, session, call) {
131
242
  : null;
132
243
  const context = {
133
244
  rootDir: session.rootDir,
245
+ sessionId: session.sessionId,
134
246
  projectIndex: toolContext.projectIndex,
135
247
  autoYes: session.autoYes,
136
248
  confirmCommand: session.confirmCommand,
@@ -152,28 +264,57 @@ export async function executeLocalToolCall(toolContext, session, call) {
152
264
  };
153
265
  const result = await dispatchTool(context, call);
154
266
  recordAssistantEdit(session, call, result, beforeEditSnapshot);
155
- if (result && commandTracker && (call.name === 'run_command' || call.name === 'run_node_script')) {
156
- const checkpoint = ensureActiveCheckpoint(session.clientState.safety, session.turnState.id);
157
- const records = collectCommandMutations({
158
- state: session.clientState.safety,
159
- rootDir: session.rootDir,
267
+ if (result && typeof result === 'object' && commandTracker) {
268
+ await collectTrackedCommandMutations({
269
+ session,
270
+ projectIndex: toolContext.projectIndex,
271
+ result,
160
272
  tracker: commandTracker,
161
273
  toolName: call.name,
162
274
  toolCallId: call.id,
163
275
  turnId: session.turnState.id,
164
- checkpointId: checkpoint.id,
165
276
  });
166
- if (records.length) {
167
- invalidateShellDiagnosticsCache(session.rootDir);
168
- rememberCheckpointFiles(session.clientState.safety, session.rootDir, records.map((record) => record.filePath), session.turnState.id);
169
- result.sessionEdits = records.map((record) => ({
170
- id: record.id,
171
- filePath: record.filePath,
172
- operation: record.operation,
173
- beforeHash: record.beforeHash,
174
- afterHash: record.afterHash,
175
- }));
176
- result.diagnostics = runShellDiagnostics(session.rootDir);
277
+ if (call.name === 'run_command' &&
278
+ result.backgrounded === true &&
279
+ result.status === 'running' &&
280
+ result.jobId) {
281
+ backgroundCommandTrackers.set(String(result.jobId), {
282
+ tracker: captureMutationBaseline(session.rootDir),
283
+ toolName: call.name,
284
+ toolCallId: call.id,
285
+ turnId: session.turnState.id,
286
+ });
287
+ }
288
+ }
289
+ if (result &&
290
+ typeof result === 'object' &&
291
+ (call.name === 'shell_job_output' || call.name === 'shell_job_kill')) {
292
+ const jobId = String(result.jobId ?? call.args?.job_id ?? '').trim();
293
+ const tracked = backgroundCommandTrackers.get(jobId);
294
+ if (tracked) {
295
+ await collectTrackedCommandMutations({
296
+ session,
297
+ projectIndex: toolContext.projectIndex,
298
+ result,
299
+ tracker: tracked.tracker,
300
+ toolName: tracked.toolName,
301
+ toolCallId: tracked.toolCallId,
302
+ turnId: tracked.turnId,
303
+ });
304
+ if (result.status === 'running') {
305
+ tracked.tracker = captureMutationBaseline(session.rootDir);
306
+ }
307
+ else {
308
+ backgroundCommandTrackers.delete(jobId);
309
+ }
310
+ }
311
+ }
312
+ if (result && typeof result === 'object') {
313
+ const backgroundJobUpdate = drainBackgroundJobNotifications({
314
+ sessionId: session.sessionId,
315
+ });
316
+ if (backgroundJobUpdate) {
317
+ result.backgroundJobUpdate = backgroundJobUpdate;
177
318
  }
178
319
  }
179
320
  session.onToolEvent?.({ call, result });
@@ -1,4 +1,4 @@
1
- import chalk from 'chalk';
1
+ import chalk from '../colors.js';
2
2
  import { deleteProjectFile } from '../patcher.js';
3
3
  import { isTuiMode } from '../runtime-mode.js';
4
4
  import { removeIndexFile } from '../project-index.js';
@@ -1,3 +1,4 @@
1
+ import { shellJobOutput } from './shell-job-output.js';
1
2
  import { deleteFile } from './delete-file.js';
2
3
  import { findSymbol } from './find-symbol.js';
3
4
  import { getDiagnostics } from './get-diagnostics.js';
@@ -17,8 +18,10 @@ import { runNodeScript } from './run-node-script.js';
17
18
  import { restoreFilesToCheckpoint, restoreToCheckpoint, } from './restore-checkpoint.js';
18
19
  import { searchCode } from './search-code.js';
19
20
  import { getSignatureHelp } from './signature-help.js';
21
+ import { shellJobKill } from './shell-job-kill.js';
20
22
  import { strReplace } from './str-replace.js';
21
23
  import { undoEdit } from './undo-edit.js';
24
+ import { updateTodos } from './update-todos.js';
22
25
  import { writeFile } from './write-file.js';
23
26
  export const TOOL_MAP = {
24
27
  search_code: (context, args) => searchCode(context.projectIndex, args),
@@ -44,6 +47,9 @@ export const TOOL_MAP = {
44
47
  undo_edit: undoEdit,
45
48
  run_command: runShellCommand,
46
49
  run_node_script: runNodeScript,
50
+ shell_job_output: shellJobOutput,
51
+ shell_job_kill: shellJobKill,
52
+ update_todos: updateTodos,
47
53
  };
48
54
  function invalidToolCall(error) {
49
55
  return {
@@ -1,15 +1,16 @@
1
- import chalk from 'chalk';
1
+ import chalk from '../colors.js';
2
2
  import path from 'node:path';
3
3
  import { normalizeProjectRelativePath } from '../artifact-policy.js';
4
4
  import { applyUnifiedPatch, readProjectFile, renderDiffPreview, writeProjectFile, } from '../patcher.js';
5
5
  import { upsertIndexFile } from '../project-index.js';
6
+ import { repairFilePath } from './path-suggest.js';
6
7
  import { isTuiMode } from '../runtime-mode.js';
7
8
  import { getCurrentFileHash, resolveRedactionTokens } from '../session-safety.js';
8
9
  import { invalidateShellDiagnosticsCache, runShellDiagnostics, } from './shell-diagnostics.js';
9
10
  const DOCUMENT_EXTENSIONS = new Set(['.pdf', '.xlsx', '.docx']);
10
11
  export async function patchFile(context, args) {
11
12
  const { rootDir, projectIndex, autoYes, confirmPatch } = context;
12
- const filePath = String(args.filePath ?? '').trim();
13
+ const filePath = repairFilePath(rootDir, String(args.filePath ?? '').trim());
13
14
  let patch = typeof args.patch === 'string' ? args.patch : '';
14
15
  if (!filePath) {
15
16
  return { ok: false, error: 'filePath is required' };
@@ -1,9 +1,6 @@
1
- import { readdirSync } from 'node:fs';
1
+ import { existsSync, readdirSync } from 'node:fs';
2
2
  import path from 'node:path';
3
- import { isSensitiveProjectPath } from '../artifact-policy.js';
4
- // "File not found" recovery hint: when a model mistypes a filename (most
5
- // often Unicode punctuation — a straight ' for a curly ’ — or a small typo),
6
- // suggest the closest real file from the same directory.
3
+ import { isSensitiveProjectPath, shouldIgnoreArtifactPath, } from '../artifact-policy.js';
7
4
  function foldName(name) {
8
5
  return name
9
6
  .normalize('NFC')
@@ -46,9 +43,6 @@ export function suggestClosestPath(rootDir, missingPath) {
46
43
  let best = null;
47
44
  let bestDistance = Number.POSITIVE_INFINITY;
48
45
  for (const candidate of candidates) {
49
- // Never suggest a file the caller would refuse to read/write directly:
50
- // probing a near-miss like `.enx` or `credential.docx` must not leak the
51
- // existence of `.env`/credentials through the recovery hint.
52
46
  const candidateRelative = path.relative(rootDir, path.join(directory, candidate));
53
47
  if (isSensitiveProjectPath(candidateRelative))
54
48
  continue;
@@ -64,3 +58,82 @@ export function suggestClosestPath(rootDir, missingPath) {
64
58
  const relative = path.relative(rootDir, suggested);
65
59
  return relative && !relative.startsWith('..') ? relative : suggested;
66
60
  }
61
+ function foldPunctuation(name) {
62
+ return name
63
+ .normalize('NFC')
64
+ .replace(/[‘’ʼ]/g, "'")
65
+ .replace(/[“”]/g, '"')
66
+ .replace(/ /g, ' ');
67
+ }
68
+ function stripSurroundingQuotes(p) {
69
+ if (p.length >= 2) {
70
+ const first = p[0];
71
+ const last = p[p.length - 1];
72
+ if ((first === "'" && last === "'") || (first === '"' && last === '"')) {
73
+ return p.slice(1, -1);
74
+ }
75
+ }
76
+ return p;
77
+ }
78
+ function collapseDoubledBackslashes(p) {
79
+ return p.replace(/\\\\/g, '\\');
80
+ }
81
+ function resolveAgainst(rootDir, p) {
82
+ return path.isAbsolute(p) ? p : path.resolve(rootDir, p);
83
+ }
84
+ function existsAgainst(rootDir, p) {
85
+ try {
86
+ return existsSync(resolveAgainst(rootDir, p));
87
+ }
88
+ catch {
89
+ return false;
90
+ }
91
+ }
92
+ function isProtectedRepairTarget(rootDir, candidate) {
93
+ const rel = path.relative(rootDir, resolveAgainst(rootDir, candidate));
94
+ const projectPath = rel && !rel.startsWith('..') ? rel : candidate;
95
+ return (isSensitiveProjectPath(projectPath) ||
96
+ (rel !== '' && !rel.startsWith('..') && shouldIgnoreArtifactPath(rel)));
97
+ }
98
+ export const PATH_REPAIRING_EDIT_TOOLS = new Set([
99
+ 'str_replace',
100
+ 'patch_file',
101
+ 'replace_document_text',
102
+ ]);
103
+ export function repairFilePath(rootDir, raw) {
104
+ if (!raw || existsAgainst(rootDir, raw))
105
+ return raw;
106
+ const dequoted = stripSurroundingQuotes(raw);
107
+ for (const candidate of [
108
+ dequoted,
109
+ collapseDoubledBackslashes(raw),
110
+ collapseDoubledBackslashes(dequoted),
111
+ ]) {
112
+ if (candidate !== raw &&
113
+ existsAgainst(rootDir, candidate) &&
114
+ !isProtectedRepairTarget(rootDir, candidate)) {
115
+ return candidate;
116
+ }
117
+ }
118
+ const probe = resolveAgainst(rootDir, dequoted);
119
+ const directory = path.dirname(probe);
120
+ const wanted = foldPunctuation(path.basename(probe));
121
+ if (!wanted)
122
+ return raw;
123
+ let entries;
124
+ try {
125
+ entries = readdirSync(directory);
126
+ }
127
+ catch {
128
+ return raw;
129
+ }
130
+ const matches = entries.filter((entry) => foldPunctuation(entry) === wanted);
131
+ if (matches.length !== 1)
132
+ return raw;
133
+ const matchedAbs = path.join(directory, matches[0]);
134
+ const matchedRel = path.relative(rootDir, matchedAbs);
135
+ const matched = path.isAbsolute(dequoted) ? matchedAbs : matchedRel || matchedAbs;
136
+ if (isProtectedRepairTarget(rootDir, matched))
137
+ return raw;
138
+ return matched;
139
+ }
@@ -1,7 +1,7 @@
1
1
  import path from 'node:path';
2
2
  import { existsSync, readFileSync } from 'node:fs';
3
3
  import { isSensitiveProjectPath, normalizeProjectRelativePath, } from '../artifact-policy.js';
4
- import { suggestClosestPath } from './path-suggest.js';
4
+ import { repairFilePath, suggestClosestPath } from './path-suggest.js';
5
5
  import { readCliAuthConfig } from '../api/auth.js';
6
6
  export function normalizeDocumentText(raw) {
7
7
  const text = String(raw ?? '').replace(/\r\n?/g, '\n');
@@ -71,7 +71,7 @@ async function parseDocumentOnServer(config, fileName, fileData, ext, args) {
71
71
  return data;
72
72
  }
73
73
  export async function readDocument(rootDir, args, env) {
74
- const raw = String(args.filePath ?? '').trim();
74
+ const raw = repairFilePath(rootDir, String(args.filePath ?? '').trim());
75
75
  if (!raw) {
76
76
  return { ok: false, error: 'filePath is required' };
77
77
  }
@@ -3,6 +3,7 @@ import path from 'path';
3
3
  import { normalizeProjectRelativePath, shouldIgnoreArtifactPath, } from '../artifact-policy.js';
4
4
  import { buildSecretFilePreview, isDotenvLikePath, looksLikeEditableDotenv, shouldUseSecretFilePreview, } from '../secret-preview.js';
5
5
  import { readProjectFile } from '../patcher.js';
6
+ import { repairFilePath } from './path-suggest.js';
6
7
  import { dotenvFitsRedactionBudget, getCurrentFileHash, recordReadCoverage, redactContentWithStableTokens, redactDotenvWithStableTokens, } from '../session-safety.js';
7
8
  import { readFileRange, truncate } from '../utils.js';
8
9
  const MAX_FILE_READ_CHARS = 12000;
@@ -10,7 +11,7 @@ const DOCUMENT_EXTENSIONS = new Set(['.pdf', '.xlsx', '.docx']);
10
11
  export async function readFile(context, args) {
11
12
  const rootDir = typeof context === 'string' ? context : context.rootDir;
12
13
  const safety = typeof context === 'string' ? undefined : context.safety;
13
- const filePath = String(args.filePath ?? '').trim();
14
+ const filePath = repairFilePath(rootDir, String(args.filePath ?? '').trim());
14
15
  if (!filePath) {
15
16
  return { ok: false, error: 'filePath is required' };
16
17
  }
@@ -41,13 +42,23 @@ export async function readFile(context, args) {
41
42
  content = readProjectFile(rootDir, filePath);
42
43
  }
43
44
  catch (err) {
44
- return { ok: false, error: err.message };
45
+ const message = String(err?.message ?? err);
46
+ const notFound = /^File does not exist:/.test(message);
47
+ return {
48
+ ok: false,
49
+ error: message,
50
+ ...(notFound ? { failureCategory: 'not_found' } : {}),
51
+ };
45
52
  }
46
53
  }
47
54
  else {
48
55
  const absPath = path.resolve(filePath);
49
56
  if (!existsSync(absPath)) {
50
- return { ok: false, error: `File does not exist: ${filePath}` };
57
+ return {
58
+ ok: false,
59
+ error: `File does not exist: ${filePath}`,
60
+ failureCategory: 'not_found',
61
+ };
51
62
  }
52
63
  try {
53
64
  content = readFileSync(absPath, 'utf-8');
@@ -57,10 +68,6 @@ export async function readFile(context, args) {
57
68
  }
58
69
  }
59
70
  const previewPath = projectPath ?? filePath;
60
- // A clean dotenv file is shown with keys visible and values tokenized so the
61
- // agent can still edit it (str_replace/write_file round-trip the tokens) and
62
- // read coverage is recorded. Any other secret file — PEM, JSON credentials,
63
- // or a dotenv with a stray non-assignment line — keeps the opaque blackout.
64
71
  const editableDotenv = Boolean(projectPath) &&
65
72
  Boolean(safety) &&
66
73
  isDotenvLikePath(previewPath) &&
@@ -1,10 +1,10 @@
1
- import chalk from 'chalk';
1
+ import chalk from '../colors.js';
2
2
  import { existsSync, readFileSync } from 'node:fs';
3
3
  import path from 'node:path';
4
4
  import { isSensitiveProjectPath, normalizeProjectRelativePath, } from '../artifact-policy.js';
5
5
  import { readCliAuthConfig } from '../api/auth.js';
6
6
  import { resolveProjectPath, writeProjectFileBuffer } from '../patcher.js';
7
- import { suggestClosestPath } from './path-suggest.js';
7
+ import { repairFilePath, suggestClosestPath } from './path-suggest.js';
8
8
  import { isTuiMode } from '../runtime-mode.js';
9
9
  function normalizeReplacements(value) {
10
10
  if (!Array.isArray(value))
@@ -76,7 +76,7 @@ async function replaceDocumentTextOnServer(config, fileName, fileData, replaceme
76
76
  return data;
77
77
  }
78
78
  export async function replaceDocumentText(context, args) {
79
- const sourceRaw = String(args.filePath ?? args.file_path ?? '').trim();
79
+ const sourceRaw = repairFilePath(context.rootDir, String(args.filePath ?? args.file_path ?? '').trim());
80
80
  if (!sourceRaw) {
81
81
  return { ok: false, error: 'filePath is required' };
82
82
  }
@@ -149,9 +149,6 @@ export async function replaceDocumentText(context, args) {
149
149
  failureCategory: serverResult.failureCategory ?? 'external_service',
150
150
  };
151
151
  }
152
- // Validate-only: report per-replacement match info without touching the file.
153
- // changed:false marks it non-mutating so the agent loop does not count a
154
- // dry-run as an applied edit.
155
152
  if (validateOnly) {
156
153
  return {
157
154
  ok: true,
@@ -162,8 +159,6 @@ export async function replaceDocumentText(context, args) {
162
159
  results: serverResult.results,
163
160
  };
164
161
  }
165
- // No replacement matched: nothing was written. Surface per-item reasons so
166
- // the model can correct and resend only the failing entries.
167
162
  const replacementCount = Number(serverResult.replacementCount ?? 0);
168
163
  if (replacementCount === 0) {
169
164
  const failures = Array.isArray(serverResult.replacements)
@@ -220,9 +215,6 @@ export async function replaceDocumentText(context, args) {
220
215
  failedCount: serverResult.failedCount,
221
216
  replacements: serverResult.replacements,
222
217
  bytesWritten: nextData.length,
223
- // A partial batch still wrote the matched entries (changed:true above), but
224
- // the loop must reflect and repair the missed entries — needsRepair forces
225
- // that without losing credit for the applied edits.
226
218
  ...(failedCount > 0
227
219
  ? {
228
220
  needsRepair: true,
@@ -1,4 +1,4 @@
1
- import chalk from 'chalk';
1
+ import chalk from '../colors.js';
2
2
  import { isTuiMode } from '../runtime-mode.js';
3
3
  import { createPromptCheckpoint, restoreCheckpointFiles, } from '../session-safety.js';
4
4
  import { invalidateShellDiagnosticsCache, runShellDiagnostics, } from './shell-diagnostics.js';
@@ -1,4 +1,5 @@
1
- import chalk from 'chalk';
1
+ import chalk from '../colors.js';
2
+ import { startBackgroundJob } from '../background-jobs.js';
2
3
  import { getBlockedCommandReason, runCommand, } from '../executor.js';
3
4
  import { syncIndexFromDisk } from '../project-index.js';
4
5
  import { isTuiMode } from '../runtime-mode.js';
@@ -12,9 +13,10 @@ export async function runShellCommand(context, args) {
12
13
  if (!command) {
13
14
  return { ok: false, error: 'command is required' };
14
15
  }
16
+ const runInBackground = args.background === true;
15
17
  const hasTimeout = typeof args.timeout_ms === 'number' && args.timeout_ms > 0;
16
18
  const repoHint = buildNestedGitHint(rootDir, command);
17
- const blockedReason = getBlockedCommandReason(command, hasTimeout, rootDir);
19
+ const blockedReason = getBlockedCommandReason(command, hasTimeout || runInBackground, rootDir);
18
20
  if (blockedReason) {
19
21
  const error = blockedReason;
20
22
  if (!isTuiMode())
@@ -37,7 +39,9 @@ export async function runShellCommand(context, args) {
37
39
  error: 'confirmCommand is required when autoYes is false',
38
40
  };
39
41
  }
40
- const approved = await confirmCommand(command);
42
+ const approved = await confirmCommand(runInBackground
43
+ ? `${command}\n\nRuns as a managed background job until it exits or is killed.`
44
+ : command);
41
45
  if (!approved) {
42
46
  if (!isTuiMode())
43
47
  console.log(chalk.dim(` ⏭ Skipped: ${command}`));
@@ -45,10 +49,13 @@ export async function runShellCommand(context, args) {
45
49
  ok: false,
46
50
  skipped: true,
47
51
  command,
48
- error: 'User declined command execution',
52
+ error: 'User declined command execution. Do not rerun this command or try a broader variant of it; either continue without it or ask the user one specific question about how to proceed.',
49
53
  };
50
54
  }
51
55
  }
56
+ if (runInBackground) {
57
+ return runBackgroundCommand(context, command, args.timeout_ms, repoHint);
58
+ }
52
59
  const result = await runCommand(command, rootDir, {
53
60
  requestSudoPassword,
54
61
  timeout: typeof args.timeout_ms === 'number' && args.timeout_ms > 0 ? args.timeout_ms : undefined,
@@ -67,19 +74,9 @@ export async function runShellCommand(context, args) {
67
74
  if (repoSync.added || repoSync.modified || repoSync.removed) {
68
75
  onStatus(`Synced repo state after command (${repoSync.added} added, ${repoSync.modified} modified, ${repoSync.removed} removed).`);
69
76
  }
70
- // Redact connection-string passwords so shell output (e.g. `cat .env`,
71
- // `printenv`) cannot leak them into history or telemetry.
72
- let output = typeof result.output === 'string'
73
- ? redactConnectionStringCredentials(result.output)
77
+ const output = typeof result.output === 'string'
78
+ ? boundCommandOutput(redactConnectionStringCredentials(result.output))
74
79
  : result.output;
75
- if (typeof output === 'string' && output.length > MAX_OUTPUT_CHARS) {
76
- const headSize = Math.floor(MAX_OUTPUT_CHARS * 0.2);
77
- const tailSize = MAX_OUTPUT_CHARS - headSize;
78
- output =
79
- output.slice(0, headSize) +
80
- `\n\n... (${output.length - headSize - tailSize} chars truncated) ...\n\n` +
81
- output.slice(-tailSize);
82
- }
83
80
  return {
84
81
  ok: result.exitCode === 0,
85
82
  command,
@@ -92,3 +89,73 @@ export async function runShellCommand(context, args) {
92
89
  repoHint,
93
90
  };
94
91
  }
92
+ export function boundCommandOutput(output) {
93
+ if (output.length <= MAX_OUTPUT_CHARS)
94
+ return output;
95
+ const headSize = Math.floor(MAX_OUTPUT_CHARS * 0.2);
96
+ const tailSize = MAX_OUTPUT_CHARS - headSize;
97
+ return (output.slice(0, headSize) +
98
+ `\n\n... (${output.length - headSize - tailSize} chars truncated) ...\n\n` +
99
+ output.slice(-tailSize));
100
+ }
101
+ async function runBackgroundCommand(context, command, timeoutMs, repoHint) {
102
+ const { rootDir, projectIndex, onStatus } = context;
103
+ const started = await startBackgroundJob(command, rootDir, {
104
+ startupWaitMs: typeof timeoutMs === 'number' && timeoutMs > 0 ? timeoutMs : undefined,
105
+ sessionId: context.sessionId,
106
+ });
107
+ if (!started.ok || !started.snapshot) {
108
+ if (!isTuiMode())
109
+ console.log(chalk.red(`\n ✖ ${started.error}`));
110
+ return {
111
+ ok: false,
112
+ blocked: true,
113
+ command,
114
+ error: started.error ?? 'Background job failed to start.',
115
+ repoHint,
116
+ };
117
+ }
118
+ const snapshot = started.snapshot;
119
+ const repoSync = projectIndex.initialized
120
+ ? await syncIndexFromDisk(projectIndex)
121
+ : {
122
+ added: 0,
123
+ modified: 0,
124
+ removed: 0,
125
+ indexedChunks: 0,
126
+ retrievalTokensUsed: 0,
127
+ };
128
+ invalidateShellDiagnosticsCache(rootDir);
129
+ if (repoSync.added || repoSync.modified || repoSync.removed) {
130
+ onStatus(`Synced repo state after command (${repoSync.added} added, ${repoSync.modified} modified, ${repoSync.removed} removed).`);
131
+ }
132
+ const output = boundCommandOutput(redactConnectionStringCredentials(started.startupOutput ?? '').trim());
133
+ if (snapshot.status === 'running') {
134
+ return {
135
+ ok: true,
136
+ backgrounded: true,
137
+ command,
138
+ jobId: snapshot.id,
139
+ status: 'running',
140
+ pid: snapshot.pid,
141
+ output,
142
+ note: `Background job ${snapshot.id} is running. Use shell_job_output to poll status and new output, and shell_job_kill to stop it.`,
143
+ repoSync,
144
+ retrievalTokensUsed: repoSync.retrievalTokensUsed,
145
+ repoHint,
146
+ };
147
+ }
148
+ return {
149
+ ok: snapshot.exitCode === 0,
150
+ backgrounded: true,
151
+ command,
152
+ jobId: snapshot.id,
153
+ status: snapshot.status,
154
+ exitCode: snapshot.exitCode,
155
+ output,
156
+ note: `Background job ${snapshot.id} finished during the startup window.`,
157
+ repoSync,
158
+ retrievalTokensUsed: repoSync.retrievalTokensUsed,
159
+ repoHint,
160
+ };
161
+ }