codebuff 1.0.246 → 1.0.247

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 (115) hide show
  1. package/dist/background-process-manager.d.ts +50 -0
  2. package/dist/background-process-manager.js +359 -0
  3. package/dist/background-process-manager.js.map +1 -0
  4. package/dist/browser-runner.d.ts +35 -0
  5. package/dist/browser-runner.js +680 -0
  6. package/dist/browser-runner.js.map +1 -0
  7. package/dist/chat-storage.d.ts +2 -0
  8. package/dist/chat-storage.js +93 -0
  9. package/dist/chat-storage.js.map +1 -0
  10. package/dist/checkpoints/checkpoint-manager.d.ts +94 -0
  11. package/dist/checkpoints/checkpoint-manager.js +280 -0
  12. package/dist/checkpoints/checkpoint-manager.js.map +1 -0
  13. package/dist/checkpoints/file-manager.d.ts +72 -0
  14. package/dist/checkpoints/file-manager.js +311 -0
  15. package/dist/checkpoints/file-manager.js.map +1 -0
  16. package/dist/cli-handlers/api-key.d.ts +25 -0
  17. package/dist/cli-handlers/api-key.js +66 -0
  18. package/dist/cli-handlers/api-key.js.map +1 -0
  19. package/dist/cli-handlers/checkpoint.d.ts +18 -0
  20. package/dist/cli-handlers/checkpoint.js +195 -0
  21. package/dist/cli-handlers/checkpoint.js.map +1 -0
  22. package/dist/cli-handlers/diff.d.ts +2 -0
  23. package/dist/cli-handlers/diff.js +31 -0
  24. package/dist/cli-handlers/diff.js.map +1 -0
  25. package/dist/cli-handlers/easter-egg.d.ts +1 -0
  26. package/dist/cli-handlers/easter-egg.js +126 -0
  27. package/dist/cli-handlers/easter-egg.js.map +1 -0
  28. package/dist/cli-handlers/inititalization-flow.d.ts +1 -0
  29. package/dist/cli-handlers/inititalization-flow.js +24 -0
  30. package/dist/cli-handlers/inititalization-flow.js.map +1 -0
  31. package/dist/cli.d.ts +44 -0
  32. package/dist/cli.js +478 -0
  33. package/dist/cli.js.map +1 -0
  34. package/dist/client.d.ts +157 -0
  35. package/dist/client.js +836 -0
  36. package/dist/client.js.map +1 -0
  37. package/dist/config.d.ts +4 -0
  38. package/dist/config.js +12 -0
  39. package/dist/config.js.map +1 -0
  40. package/dist/create-template-project.d.ts +1 -0
  41. package/dist/create-template-project.js +107 -0
  42. package/dist/create-template-project.js.map +1 -0
  43. package/dist/credentials.d.ts +4 -0
  44. package/dist/credentials.js +38 -0
  45. package/dist/credentials.js.map +1 -0
  46. package/dist/dev-process-manager.d.ts +10 -0
  47. package/dist/dev-process-manager.js +54 -0
  48. package/dist/dev-process-manager.js.map +1 -0
  49. package/dist/fingerprint.d.ts +1 -0
  50. package/dist/fingerprint.js +48 -0
  51. package/dist/fingerprint.js.map +1 -0
  52. package/dist/index.d.ts +2 -0
  53. package/dist/index.js +117 -0
  54. package/dist/index.js.map +1 -0
  55. package/dist/menu.d.ts +3 -0
  56. package/dist/menu.js +126 -0
  57. package/dist/menu.js.map +1 -0
  58. package/dist/project-files.d.ts +114 -0
  59. package/dist/project-files.js +513 -0
  60. package/dist/project-files.js.map +1 -0
  61. package/dist/startup-process-handler.d.ts +2 -0
  62. package/dist/startup-process-handler.js +21 -0
  63. package/dist/startup-process-handler.js.map +1 -0
  64. package/dist/tool-handlers.d.ts +28 -0
  65. package/dist/tool-handlers.js +240 -0
  66. package/dist/tool-handlers.js.map +1 -0
  67. package/dist/types.d.ts +15 -0
  68. package/dist/types.js +3 -0
  69. package/dist/types.js.map +1 -0
  70. package/dist/update-codebuff.d.ts +1 -0
  71. package/dist/update-codebuff.js +160 -0
  72. package/dist/update-codebuff.js.map +1 -0
  73. package/dist/utils/__tests__/background-process-manager.test.d.ts +1 -0
  74. package/dist/utils/__tests__/background-process-manager.test.js +289 -0
  75. package/dist/utils/__tests__/background-process-manager.test.js.map +1 -0
  76. package/dist/utils/__tests__/tool-renderers.test.d.ts +1 -0
  77. package/dist/utils/__tests__/tool-renderers.test.js +51 -0
  78. package/dist/utils/__tests__/tool-renderers.test.js.map +1 -0
  79. package/dist/utils/__tests__/xml-stream-parser.test.d.ts +1 -0
  80. package/dist/utils/__tests__/xml-stream-parser.test.js +229 -0
  81. package/dist/utils/__tests__/xml-stream-parser.test.js.map +1 -0
  82. package/dist/utils/analytics.d.ts +6 -0
  83. package/dist/utils/analytics.js +59 -0
  84. package/dist/utils/analytics.js.map +1 -0
  85. package/dist/utils/detect-shell.d.ts +1 -0
  86. package/dist/utils/detect-shell.js +60 -0
  87. package/dist/utils/detect-shell.js.map +1 -0
  88. package/dist/utils/logger.d.ts +21 -0
  89. package/dist/utils/logger.js +105 -0
  90. package/dist/utils/logger.js.map +1 -0
  91. package/dist/utils/spinner.d.ts +11 -0
  92. package/dist/utils/spinner.js +87 -0
  93. package/dist/utils/spinner.js.map +1 -0
  94. package/dist/utils/system-info.d.ts +8 -0
  95. package/dist/utils/system-info.js +22 -0
  96. package/dist/utils/system-info.js.map +1 -0
  97. package/dist/utils/terminal.d.ts +41 -0
  98. package/dist/utils/terminal.js +475 -0
  99. package/dist/utils/terminal.js.map +1 -0
  100. package/dist/utils/tool-renderers.d.ts +16 -0
  101. package/dist/utils/tool-renderers.js +145 -0
  102. package/dist/utils/tool-renderers.js.map +1 -0
  103. package/dist/utils/xml-stream-parser.d.ts +9 -0
  104. package/dist/utils/xml-stream-parser.js +128 -0
  105. package/dist/utils/xml-stream-parser.js.map +1 -0
  106. package/dist/web-scraper.d.ts +3 -0
  107. package/dist/web-scraper.js +57 -0
  108. package/dist/web-scraper.js.map +1 -0
  109. package/dist/workers/checkpoint-worker.d.ts +1 -0
  110. package/dist/workers/checkpoint-worker.js +48 -0
  111. package/dist/workers/checkpoint-worker.js.map +1 -0
  112. package/dist/workers/project-context.d.ts +1 -0
  113. package/dist/workers/project-context.js +17 -0
  114. package/dist/workers/project-context.js.map +1 -0
  115. package/package.json +1 -1
package/dist/cli.js ADDED
@@ -0,0 +1,478 @@
1
+ "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
14
+ Object.defineProperty(o, "default", { enumerable: true, value: v });
15
+ }) : function(o, v) {
16
+ o["default"] = v;
17
+ });
18
+ var __importStar = (this && this.__importStar) || function (mod) {
19
+ if (mod && mod.__esModule) return mod;
20
+ var result = {};
21
+ if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
22
+ __setModuleDefault(result, mod);
23
+ return result;
24
+ };
25
+ Object.defineProperty(exports, "__esModule", { value: true });
26
+ exports.CLI = void 0;
27
+ const fs_1 = require("fs");
28
+ const os_1 = require("os");
29
+ const path_1 = __importStar(require("path"));
30
+ const readline = __importStar(require("readline"));
31
+ const string_1 = require("./common/util/string");
32
+ const picocolors_1 = require("picocolors");
33
+ const background_process_manager_1 = require("./background-process-manager");
34
+ const chat_storage_1 = require("./chat-storage");
35
+ const checkpoint_manager_1 = require("./checkpoints/checkpoint-manager");
36
+ const api_key_1 = require("./cli-handlers/api-key");
37
+ const checkpoint_1 = require("./cli-handlers/checkpoint");
38
+ const diff_1 = require("./cli-handlers/diff");
39
+ const easter_egg_1 = require("./cli-handlers/easter-egg");
40
+ const inititalization_flow_1 = require("./cli-handlers/inititalization-flow");
41
+ const client_1 = require("./client");
42
+ const config_1 = require("./config");
43
+ const menu_1 = require("./menu");
44
+ const project_files_1 = require("./project-files");
45
+ const analytics_1 = require("./utils/analytics");
46
+ const spinner_1 = require("./utils/spinner");
47
+ const terminal_1 = require("./utils/terminal");
48
+ class CLI {
49
+ client;
50
+ readyPromise;
51
+ git;
52
+ costMode;
53
+ rl;
54
+ isReceivingResponse = false;
55
+ stopResponse = null;
56
+ lastSigintTime = 0;
57
+ lastInputTime = 0;
58
+ consecutiveFastInputs = 0;
59
+ pastedContent = '';
60
+ isPasting = false;
61
+ shouldReconnectWhenIdle = false;
62
+ constructor(readyPromise, { git, costMode, model }) {
63
+ this.git = git;
64
+ this.costMode = costMode;
65
+ this.setupSignalHandlers();
66
+ this.initReadlineInterface();
67
+ this.client = new client_1.Client({
68
+ websocketUrl: config_1.websocketUrl,
69
+ onWebSocketError: this.onWebSocketError.bind(this),
70
+ onWebSocketReconnect: this.onWebSocketReconnect.bind(this),
71
+ freshPrompt: this.freshPrompt.bind(this),
72
+ reconnectWhenNextIdle: this.reconnectWhenNextIdle.bind(this),
73
+ costMode: this.costMode,
74
+ git: this.git,
75
+ rl: this.rl,
76
+ model,
77
+ });
78
+ this.readyPromise = Promise.all([
79
+ readyPromise.then((results) => {
80
+ const [fileContext, ,] = results;
81
+ this.client.initAgentState(fileContext);
82
+ return this.client.warmContextCache();
83
+ }),
84
+ this.client.connect(),
85
+ ]);
86
+ this.setPrompt();
87
+ process.on('unhandledRejection', (reason, promise) => {
88
+ console.error('\nUnhandled Rejection at:', promise, 'reason:', reason);
89
+ this.freshPrompt();
90
+ });
91
+ process.on('uncaughtException', (err, origin) => {
92
+ console.error(`\nCaught exception: ${err}\n` + `Exception origin: ${origin}`);
93
+ console.error(err.stack);
94
+ this.freshPrompt();
95
+ });
96
+ }
97
+ setupSignalHandlers() {
98
+ process.on('exit', () => {
99
+ spinner_1.Spinner.get().restoreCursor();
100
+ // Kill the persistent PTY process first
101
+ if (terminal_1.persistentProcess?.type === 'pty') {
102
+ terminal_1.persistentProcess.pty.kill();
103
+ }
104
+ (0, background_process_manager_1.sendKillSignalToAllBackgroundProcesses)();
105
+ console.log((0, picocolors_1.green)('Codebuff out!'));
106
+ });
107
+ for (const signal of ['SIGTERM', 'SIGHUP']) {
108
+ process.on(signal, async () => {
109
+ process.removeAllListeners('unhandledRejection');
110
+ process.removeAllListeners('uncaughtException');
111
+ spinner_1.Spinner.get().restoreCursor();
112
+ await (0, background_process_manager_1.killAllBackgroundProcesses)();
113
+ await (0, analytics_1.flushAnalytics)();
114
+ process.exit(0);
115
+ });
116
+ }
117
+ process.on('SIGTSTP', async () => await this.handleExit());
118
+ // Doesn't catch SIGKILL (e.g. `kill -9`)
119
+ }
120
+ initReadlineInterface() {
121
+ this.rl = readline.createInterface({
122
+ input: process.stdin,
123
+ output: process.stdout,
124
+ historySize: 1000,
125
+ terminal: true,
126
+ completer: this.filePathCompleter.bind(this),
127
+ });
128
+ this.rl.on('line', (line) => this.handleLine(line));
129
+ this.rl.on('SIGINT', async () => await this.handleSigint());
130
+ this.rl.on('close', async () => await this.handleExit());
131
+ process.stdin.on('keypress', (str, key) => this.handleKeyPress(str, key));
132
+ }
133
+ filePathCompleter(line) {
134
+ const lastWord = line.split(' ').pop() || '';
135
+ const input = lastWord.startsWith('~')
136
+ ? (0, os_1.homedir)() + lastWord.slice(1)
137
+ : lastWord;
138
+ const directorySuffix = process.platform === 'win32' ? '\\' : '/';
139
+ const dir = input.endsWith(directorySuffix)
140
+ ? input.slice(0, input.length - 1)
141
+ : (0, path_1.dirname)(input);
142
+ const partial = input.endsWith(directorySuffix) ? '' : (0, path_1.basename)(input);
143
+ let baseDir = (0, path_1.isAbsolute)(dir) ? dir : path_1.default.join((0, project_files_1.getProjectRoot)(), dir);
144
+ try {
145
+ const files = (0, fs_1.readdirSync)(baseDir);
146
+ const matches = files
147
+ .filter((file) => file.startsWith(partial))
148
+ .map((file) => file + ((0, project_files_1.isDir)(path_1.default.join(baseDir, file)) ? directorySuffix : ''));
149
+ return [matches, partial];
150
+ }
151
+ catch {
152
+ return [[], line];
153
+ }
154
+ }
155
+ setPrompt() {
156
+ this.rl.setPrompt((0, picocolors_1.green)(`${(0, path_1.parse)((0, project_files_1.getProjectRoot)()).base} > `));
157
+ }
158
+ /**
159
+ * Prompts the user with a clean prompt state
160
+ */
161
+ freshPrompt(userInput = '') {
162
+ spinner_1.Spinner.get().stop();
163
+ this.isReceivingResponse = false;
164
+ if (this.shouldReconnectWhenIdle) {
165
+ this.client.reconnect();
166
+ this.shouldReconnectWhenIdle = false;
167
+ }
168
+ readline.cursorTo(process.stdout, 0);
169
+ const rlAny = this.rl;
170
+ // Check for pending auto-topup message before showing prompt
171
+ if (this.client.pendingTopUpMessageAmount > 0) {
172
+ this.client.displayChunk('\n\n' +
173
+ (0, picocolors_1.green)(`Auto top-up successful! ${this.client.pendingTopUpMessageAmount.toLocaleString()} credits added.`) +
174
+ '\n\n');
175
+ this.client.pendingTopUpMessageAmount = 0;
176
+ }
177
+ // clear line first
178
+ rlAny.line = '';
179
+ this.setPrompt();
180
+ // then prompt
181
+ this.rl.prompt();
182
+ if (!userInput) {
183
+ return;
184
+ }
185
+ // then rewrite new prompt
186
+ this.rl.write(' '.repeat(userInput.length)); // hacky way to move cursor
187
+ rlAny.line = userInput;
188
+ rlAny._refreshLine();
189
+ }
190
+ async printInitialPrompt({ initialInput, runInitFlow, }) {
191
+ if (this.client.user) {
192
+ (0, menu_1.displayGreeting)(this.costMode, this.client.user.name);
193
+ }
194
+ else {
195
+ console.log(`Welcome to Codebuff! Give us a sec to get your account set up...`);
196
+ await this.client.login();
197
+ return;
198
+ }
199
+ this.freshPrompt();
200
+ if (runInitFlow) {
201
+ process.stdout.write('init\n');
202
+ await this.handleUserInput('init');
203
+ }
204
+ if (initialInput) {
205
+ process.stdout.write(initialInput + '\n');
206
+ await this.handleUserInput(initialInput);
207
+ }
208
+ }
209
+ async printDiff() {
210
+ (0, diff_1.handleDiff)(this.client.lastChanges);
211
+ this.freshPrompt();
212
+ }
213
+ async handleLine(line) {
214
+ this.detectPasting();
215
+ if (this.isPasting) {
216
+ this.pastedContent += line + '\n';
217
+ }
218
+ else if (!this.isReceivingResponse) {
219
+ if (this.pastedContent) {
220
+ await this.handleUserInput((this.pastedContent + line).trim());
221
+ this.pastedContent = '';
222
+ }
223
+ else {
224
+ await this.handleUserInput(line.trim());
225
+ }
226
+ }
227
+ }
228
+ async handleUserInput(userInput) {
229
+ this.rl.setPrompt('');
230
+ if (!userInput) {
231
+ this.freshPrompt();
232
+ return;
233
+ }
234
+ userInput = userInput.trim();
235
+ if (await this.processCommand(userInput)) {
236
+ return;
237
+ }
238
+ await this.forwardUserInput(userInput);
239
+ }
240
+ async processCommand(userInput) {
241
+ if (userInput === 'help' || userInput === 'h' || userInput === '/help') {
242
+ (0, menu_1.displayMenu)();
243
+ this.freshPrompt();
244
+ return true;
245
+ }
246
+ if (userInput === 'login' || userInput === 'signin') {
247
+ await this.client.login();
248
+ checkpoint_manager_1.checkpointManager.clearCheckpoints();
249
+ return true;
250
+ }
251
+ if (userInput === 'logout' || userInput === 'signout') {
252
+ await this.client.logout();
253
+ this.freshPrompt();
254
+ return true;
255
+ }
256
+ if (userInput.startsWith('ref-')) {
257
+ await this.client.handleReferralCode(userInput.trim());
258
+ return true;
259
+ }
260
+ // Detect potential API key input first
261
+ const detectionResult = (0, api_key_1.detectApiKey)(userInput);
262
+ if (detectionResult.status !== 'not_found') {
263
+ // If something resembling an API key was detected (valid or just prefix), handle it
264
+ await (0, api_key_1.handleApiKeyInput)(this.client, detectionResult, this.readyPromise, this.freshPrompt.bind(this));
265
+ return true; // Indicate command was handled
266
+ }
267
+ // Continue with other commands if no API key input was detected/handled
268
+ if (userInput === 'usage' || userInput === 'credits') {
269
+ await this.client.getUsage();
270
+ return true;
271
+ }
272
+ if (userInput === 'quit' || userInput === 'exit' || userInput === 'q') {
273
+ await this.handleExit();
274
+ return true;
275
+ }
276
+ if (['diff', 'doff', 'dif', 'iff', 'd'].includes(userInput)) {
277
+ (0, diff_1.handleDiff)(this.client.lastChanges);
278
+ this.freshPrompt();
279
+ return true;
280
+ }
281
+ if (userInput === 'uuddlrlrba' ||
282
+ userInput === 'konami' ||
283
+ userInput === 'codebuffy') {
284
+ (0, easter_egg_1.showEasterEgg)(this.freshPrompt.bind(this));
285
+ return true;
286
+ }
287
+ // Checkpoint commands
288
+ if ((0, checkpoint_1.isCheckpointCommand)(userInput)) {
289
+ if ((0, checkpoint_1.isCheckpointCommand)(userInput, 'undo')) {
290
+ await (0, checkpoint_1.saveCheckpoint)(userInput, this.client, this.readyPromise);
291
+ const toRestore = await (0, checkpoint_1.handleUndo)(this.client, this.rl);
292
+ this.freshPrompt(toRestore);
293
+ return true;
294
+ }
295
+ if ((0, checkpoint_1.isCheckpointCommand)(userInput, 'redo')) {
296
+ await (0, checkpoint_1.saveCheckpoint)(userInput, this.client, this.readyPromise);
297
+ const toRestore = await (0, checkpoint_1.handleRedo)(this.client, this.rl);
298
+ this.freshPrompt(toRestore);
299
+ return true;
300
+ }
301
+ if ((0, checkpoint_1.isCheckpointCommand)(userInput, 'list')) {
302
+ await (0, checkpoint_1.saveCheckpoint)(userInput, this.client, this.readyPromise);
303
+ await (0, checkpoint_1.listCheckpoints)();
304
+ this.freshPrompt();
305
+ return true;
306
+ }
307
+ const restoreMatch = (0, checkpoint_1.isCheckpointCommand)(userInput, 'restore');
308
+ if (restoreMatch) {
309
+ const id = parseInt(restoreMatch[1], 10);
310
+ await (0, checkpoint_1.saveCheckpoint)(userInput, this.client, this.readyPromise);
311
+ const toRestore = await (0, checkpoint_1.handleRestoreCheckpoint)(id, this.client, this.rl);
312
+ this.freshPrompt(toRestore);
313
+ return true;
314
+ }
315
+ if ((0, checkpoint_1.isCheckpointCommand)(userInput, 'clear')) {
316
+ (0, checkpoint_1.handleClearCheckpoints)();
317
+ this.freshPrompt();
318
+ return true;
319
+ }
320
+ if ((0, checkpoint_1.isCheckpointCommand)(userInput, 'save')) {
321
+ await (0, checkpoint_1.saveCheckpoint)(userInput, this.client, this.readyPromise, true);
322
+ (0, checkpoint_1.displayCheckpointMenu)();
323
+ this.freshPrompt();
324
+ return true;
325
+ }
326
+ (0, checkpoint_1.displayCheckpointMenu)();
327
+ this.freshPrompt();
328
+ return true;
329
+ }
330
+ if (userInput === 'init') {
331
+ // no need to save checkpoint, since we are passing request to backend
332
+ (0, inititalization_flow_1.handleInitializationFlowLocally)();
333
+ // Also forward user input to the backend
334
+ return false;
335
+ }
336
+ return false;
337
+ }
338
+ async forwardUserInput(userInput) {
339
+ await (0, checkpoint_1.saveCheckpoint)(userInput, this.client, this.readyPromise);
340
+ spinner_1.Spinner.get().start();
341
+ this.client.lastChanges = [];
342
+ const newMessage = {
343
+ role: 'user',
344
+ content: userInput,
345
+ };
346
+ if (this.client.agentState) {
347
+ (0, chat_storage_1.setMessages)([...this.client.agentState.messageHistory, newMessage]);
348
+ }
349
+ this.isReceivingResponse = true;
350
+ const { responsePromise, stopResponse } = await this.client.sendUserInput(userInput);
351
+ this.stopResponse = stopResponse;
352
+ await responsePromise;
353
+ this.stopResponse = null;
354
+ this.isReceivingResponse = false;
355
+ spinner_1.Spinner.get().stop();
356
+ this.freshPrompt();
357
+ }
358
+ reconnectWhenNextIdle() {
359
+ if (!this.isReceivingResponse) {
360
+ this.client.reconnect();
361
+ }
362
+ else {
363
+ this.shouldReconnectWhenIdle = true;
364
+ }
365
+ }
366
+ onWebSocketError() {
367
+ spinner_1.Spinner.get().stop();
368
+ this.isReceivingResponse = false;
369
+ if (this.stopResponse) {
370
+ this.stopResponse();
371
+ this.stopResponse = null;
372
+ }
373
+ console.error((0, picocolors_1.yellow)('\nCould not connect. Retrying...'));
374
+ }
375
+ onWebSocketReconnect() {
376
+ console.log((0, picocolors_1.green)('\nReconnected!'));
377
+ this.freshPrompt();
378
+ }
379
+ handleKeyPress(str, key) {
380
+ if (key.name === 'escape') {
381
+ this.handleEscKey();
382
+ }
383
+ if (!this.isPasting &&
384
+ str === ' ' &&
385
+ '_refreshLine' in this.rl &&
386
+ 'line' in this.rl &&
387
+ 'cursor' in this.rl) {
388
+ const rlAny = this.rl;
389
+ const { cursor, line } = rlAny;
390
+ const prevTwoChars = cursor > 1 ? line.slice(cursor - 2, cursor) : '';
391
+ if (prevTwoChars === ' ') {
392
+ rlAny.line = line.slice(0, cursor - 2) + '\n\n' + line.slice(cursor);
393
+ rlAny._refreshLine();
394
+ }
395
+ }
396
+ this.detectPasting();
397
+ }
398
+ async handleSigint() {
399
+ if ((0, terminal_1.isCommandRunning)()) {
400
+ (0, terminal_1.resetShell)((0, project_files_1.getProjectRoot)());
401
+ }
402
+ if ('line' in this.rl) {
403
+ ;
404
+ this.rl.line = '';
405
+ }
406
+ if (this.isReceivingResponse) {
407
+ this.handleStopResponse();
408
+ }
409
+ else {
410
+ const now = Date.now();
411
+ if (now - this.lastSigintTime < 5000) {
412
+ await this.handleExit();
413
+ }
414
+ else {
415
+ this.lastSigintTime = now;
416
+ console.log('\nPress Ctrl-C again to exit');
417
+ this.freshPrompt();
418
+ }
419
+ }
420
+ }
421
+ handleEscKey() {
422
+ if (this.isReceivingResponse) {
423
+ this.handleStopResponse();
424
+ }
425
+ }
426
+ handleStopResponse() {
427
+ console.log((0, picocolors_1.yellow)('\n[Response stopped by user]'));
428
+ this.isReceivingResponse = false;
429
+ if (this.stopResponse) {
430
+ this.stopResponse();
431
+ }
432
+ spinner_1.Spinner.get().stop();
433
+ }
434
+ async handleExit() {
435
+ spinner_1.Spinner.get().restoreCursor();
436
+ process.removeAllListeners('unhandledRejection');
437
+ process.removeAllListeners('uncaughtException');
438
+ console.log('\n');
439
+ // Kill the persistent PTY process first
440
+ (0, terminal_1.killAndResetPersistentProcess)();
441
+ await (0, background_process_manager_1.killAllBackgroundProcesses)();
442
+ const logMessages = [];
443
+ const totalCreditsUsedThisSession = Object.values(this.client.creditsByPromptId)
444
+ .flat()
445
+ .reduce((sum, credits) => sum + credits, 0);
446
+ logMessages.push(`${(0, string_1.pluralize)(totalCreditsUsedThisSession, 'credit')} used this session${this.client.usageData.remainingBalance !== null
447
+ ? `, ${this.client.usageData.remainingBalance.toLocaleString()} credits left.`
448
+ : '.'}`);
449
+ if (this.client.usageData.next_quota_reset) {
450
+ const daysUntilReset = Math.ceil((new Date(this.client.usageData.next_quota_reset).getTime() -
451
+ Date.now()) /
452
+ (1000 * 60 * 60 * 24));
453
+ logMessages.push(`Your free credits will reset in ${(0, string_1.pluralize)(daysUntilReset, 'day')}.`);
454
+ }
455
+ console.log(logMessages.join(' '));
456
+ await (0, analytics_1.flushAnalytics)();
457
+ process.exit(0);
458
+ }
459
+ detectPasting() {
460
+ const currentTime = Date.now();
461
+ const timeDiff = currentTime - this.lastInputTime;
462
+ if (timeDiff < 10) {
463
+ this.consecutiveFastInputs++;
464
+ if (this.consecutiveFastInputs >= 2) {
465
+ this.isPasting = true;
466
+ }
467
+ }
468
+ else {
469
+ this.consecutiveFastInputs = 0;
470
+ if (this.isPasting) {
471
+ this.isPasting = false;
472
+ }
473
+ }
474
+ this.lastInputTime = currentTime;
475
+ }
476
+ }
477
+ exports.CLI = CLI;
478
+ //# sourceMappingURL=cli.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"cli.js","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,2BAAgC;AAChC,2BAA4B;AAC5B,6CAAiE;AACjE,mDAAoC;AAMpC,+CAA8C;AAC9C,2CAA0C;AAE1C,6EAGqC;AACrC,iDAA4C;AAC5C,yEAAoE;AACpE,oDAAwE;AACxE,0DASkC;AAClC,8CAAgD;AAChD,0DAAyD;AACzD,8EAAqF;AACrF,qCAAiC;AACjC,qCAAuC;AACvC,iCAAqD;AACrD,mDAAuD;AAEvD,iDAAkD;AAClD,6CAAyC;AACzC,+CAKyB;AAOzB,MAAa,GAAG;IACN,MAAM,CAAQ;IACd,YAAY,CAAc;IAC1B,GAAG,CAAY;IACf,QAAQ,CAAU;IAClB,EAAE,CAAqB;IACvB,mBAAmB,GAAY,KAAK,CAAA;IACpC,YAAY,GAAwB,IAAI,CAAA;IACxC,cAAc,GAAW,CAAC,CAAA;IAC1B,aAAa,GAAW,CAAC,CAAA;IACzB,qBAAqB,GAAW,CAAC,CAAA;IACjC,aAAa,GAAW,EAAE,CAAA;IAC1B,SAAS,GAAY,KAAK,CAAA;IAC1B,uBAAuB,GAAY,KAAK,CAAA;IAEhD,YACE,YAAuD,EACvD,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAc;QAEpC,IAAI,CAAC,GAAG,GAAG,GAAG,CAAA;QACd,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAA;QAExB,IAAI,CAAC,mBAAmB,EAAE,CAAA;QAC1B,IAAI,CAAC,qBAAqB,EAAE,CAAA;QAE5B,IAAI,CAAC,MAAM,GAAG,IAAI,eAAM,CAAC;YACvB,YAAY,EAAZ,qBAAY;YACZ,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC;YAClD,oBAAoB,EAAE,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC;YAC1D,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC;YACxC,qBAAqB,EAAE,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC;YAC5D,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,GAAG,EAAE,IAAI,CAAC,GAAG;YACb,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,KAAK;SACN,CAAC,CAAA;QAEF,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC;YAC9B,YAAY,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC5B,MAAM,CAAC,WAAW,EAAE,AAAD,EAAG,GAAG,OAAO,CAAA;gBAChC,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC,CAAA;gBACvC,OAAO,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAA;YACvC,CAAC,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE;SACtB,CAAC,CAAA;QAEF,IAAI,CAAC,SAAS,EAAE,CAAA;QAEhB,OAAO,CAAC,EAAE,CAAC,oBAAoB,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;YACnD,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,CAAA;YACtE,IAAI,CAAC,WAAW,EAAE,CAAA;QACpB,CAAC,CAAC,CAAA;QAEF,OAAO,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE;YAC9C,OAAO,CAAC,KAAK,CACX,uBAAuB,GAAG,IAAI,GAAG,qBAAqB,MAAM,EAAE,CAC/D,CAAA;YACD,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;YACxB,IAAI,CAAC,WAAW,EAAE,CAAA;QACpB,CAAC,CAAC,CAAA;IACJ,CAAC;IAEO,mBAAmB;QACzB,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;YACtB,iBAAO,CAAC,GAAG,EAAE,CAAC,aAAa,EAAE,CAAA;YAC7B,wCAAwC;YACxC,IAAI,4BAAiB,EAAE,IAAI,KAAK,KAAK,EAAE,CAAC;gBACtC,4BAAiB,CAAC,GAAG,CAAC,IAAI,EAAE,CAAA;YAC9B,CAAC;YACD,IAAA,mEAAsC,GAAE,CAAA;YACxC,OAAO,CAAC,GAAG,CAAC,IAAA,kBAAK,EAAC,eAAe,CAAC,CAAC,CAAA;QACrC,CAAC,CAAC,CAAA;QACF,KAAK,MAAM,MAAM,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,EAAE,CAAC;YAC3C,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE;gBAC5B,OAAO,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,CAAA;gBAChD,OAAO,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAA;gBAC/C,iBAAO,CAAC,GAAG,EAAE,CAAC,aAAa,EAAE,CAAA;gBAC7B,MAAM,IAAA,uDAA0B,GAAE,CAAA;gBAClC,MAAM,IAAA,0BAAc,GAAE,CAAA;gBACtB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;YACjB,CAAC,CAAC,CAAA;QACJ,CAAC;QACD,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC,CAAA;QAC1D,yCAAyC;IAC3C,CAAC;IAEO,qBAAqB;QAC3B,IAAI,CAAC,EAAE,GAAG,QAAQ,CAAC,eAAe,CAAC;YACjC,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,WAAW,EAAE,IAAI;YACjB,QAAQ,EAAE,IAAI;YACd,SAAS,EAAE,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC;SAC7C,CAAC,CAAA;QAEF,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAA;QACnD,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC,CAAA;QAC3D,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC,CAAA;QAExD,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,UAAU,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAA;IAC3E,CAAC;IAEO,iBAAiB,CAAC,IAAY;QACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,CAAA;QAC5C,MAAM,KAAK,GAAG,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC;YACpC,CAAC,CAAC,IAAA,YAAO,GAAE,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;YAC/B,CAAC,CAAC,QAAQ,CAAA;QAEZ,MAAM,eAAe,GAAG,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAA;QAEjE,MAAM,GAAG,GAAG,KAAK,CAAC,QAAQ,CAAC,eAAe,CAAC;YACzC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;YAClC,CAAC,CAAC,IAAA,cAAO,EAAC,KAAK,CAAC,CAAA;QAClB,MAAM,OAAO,GAAG,KAAK,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAA,eAAQ,EAAC,KAAK,CAAC,CAAA;QAEtE,IAAI,OAAO,GAAG,IAAA,iBAAU,EAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,cAAI,CAAC,IAAI,CAAC,IAAA,8BAAc,GAAE,EAAE,GAAG,CAAC,CAAA;QAEtE,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAA,gBAAW,EAAC,OAAO,CAAC,CAAA;YAClC,MAAM,OAAO,GAAG,KAAK;iBAClB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;iBAC1C,GAAG,CACF,CAAC,IAAI,EAAE,EAAE,CACP,IAAI,GAAG,CAAC,IAAA,qBAAK,EAAC,cAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,CAClE,CAAA;YACH,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;QAC3B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;QACnB,CAAC;IACH,CAAC;IAEO,SAAS;QACf,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,IAAA,kBAAK,EAAC,GAAG,IAAA,YAAK,EAAC,IAAA,8BAAc,GAAE,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAA;IAChE,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,YAAoB,EAAE;QACxC,iBAAO,CAAC,GAAG,EAAE,CAAC,IAAI,EAAE,CAAA;QACpB,IAAI,CAAC,mBAAmB,GAAG,KAAK,CAAA;QAEhC,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAA;YACvB,IAAI,CAAC,uBAAuB,GAAG,KAAK,CAAA;QACtC,CAAC;QAED,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACpC,MAAM,KAAK,GAAG,IAAI,CAAC,EAAS,CAAA;QAE5B,6DAA6D;QAC7D,IAAI,IAAI,CAAC,MAAM,CAAC,yBAAyB,GAAG,CAAC,EAAE,CAAC;YAC9C,IAAI,CAAC,MAAM,CAAC,YAAY,CACtB,MAAM;gBACJ,IAAA,kBAAK,EACH,2BAA2B,IAAI,CAAC,MAAM,CAAC,yBAAyB,CAAC,cAAc,EAAE,iBAAiB,CACnG;gBACD,MAAM,CACT,CAAA;YACD,IAAI,CAAC,MAAM,CAAC,yBAAyB,GAAG,CAAC,CAAA;QAC3C,CAAC;QAED,mBAAmB;QACnB,KAAK,CAAC,IAAI,GAAG,EAAE,CAAA;QACf,IAAI,CAAC,SAAS,EAAE,CAAA;QAEhB,cAAc;QACd,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAA;QAEhB,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAM;QACR,CAAC;QAED,0BAA0B;QAC1B,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAA,CAAC,2BAA2B;QACvE,KAAK,CAAC,IAAI,GAAG,SAAS,CAAA;QACtB,KAAK,CAAC,YAAY,EAAE,CAAA;IACtB,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,EAC9B,YAAY,EACZ,WAAW,GAIZ;QACC,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACrB,IAAA,sBAAe,EAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACvD,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CACT,kEAAkE,CACnE,CAAA;YACD,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAA;YACzB,OAAM;QACR,CAAC;QACD,IAAI,CAAC,WAAW,EAAE,CAAA;QAClB,IAAI,WAAW,EAAE,CAAC;YAChB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;YAC9B,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAA;QACpC,CAAC;QACD,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC,CAAA;YACzC,MAAM,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,CAAA;QAC1C,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,SAAS;QACpB,IAAA,iBAAU,EAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAA;QACnC,IAAI,CAAC,WAAW,EAAE,CAAA;IACpB,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,IAAY;QACnC,IAAI,CAAC,aAAa,EAAE,CAAA;QACpB,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC,aAAa,IAAI,IAAI,GAAG,IAAI,CAAA;QACnC,CAAC;aAAM,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACrC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;gBACvB,MAAM,IAAI,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,CAAA;gBAC9D,IAAI,CAAC,aAAa,GAAG,EAAE,CAAA;YACzB,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAA;YACzC,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,SAAiB;QAC7C,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAA;QACrB,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,IAAI,CAAC,WAAW,EAAE,CAAA;YAClB,OAAM;QACR,CAAC;QACD,SAAS,GAAG,SAAS,CAAC,IAAI,EAAE,CAAA;QAC5B,IAAI,MAAM,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,EAAE,CAAC;YACzC,OAAM;QACR,CAAC;QACD,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAA;IACxC,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,SAAiB;QAC5C,IAAI,SAAS,KAAK,MAAM,IAAI,SAAS,KAAK,GAAG,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;YACvE,IAAA,kBAAW,GAAE,CAAA;YACb,IAAI,CAAC,WAAW,EAAE,CAAA;YAClB,OAAO,IAAI,CAAA;QACb,CAAC;QACD,IAAI,SAAS,KAAK,OAAO,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;YACpD,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAA;YACzB,sCAAiB,CAAC,gBAAgB,EAAE,CAAA;YACpC,OAAO,IAAI,CAAA;QACb,CAAC;QACD,IAAI,SAAS,KAAK,QAAQ,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YACtD,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAA;YAC1B,IAAI,CAAC,WAAW,EAAE,CAAA;YAClB,OAAO,IAAI,CAAA;QACb,CAAC;QACD,IAAI,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAA;YACtD,OAAO,IAAI,CAAA;QACb,CAAC;QAED,uCAAuC;QACvC,MAAM,eAAe,GAAG,IAAA,sBAAY,EAAC,SAAS,CAAC,CAAA;QAC/C,IAAI,eAAe,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YAC3C,oFAAoF;YACpF,MAAM,IAAA,2BAAiB,EACrB,IAAI,CAAC,MAAM,EACX,eAAe,EACf,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAC5B,CAAA;YACD,OAAO,IAAI,CAAA,CAAC,+BAA+B;QAC7C,CAAC;QAED,wEAAwE;QACxE,IAAI,SAAS,KAAK,OAAO,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YACrD,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAA;YAC5B,OAAO,IAAI,CAAA;QACb,CAAC;QACD,IAAI,SAAS,KAAK,MAAM,IAAI,SAAS,KAAK,MAAM,IAAI,SAAS,KAAK,GAAG,EAAE,CAAC;YACtE,MAAM,IAAI,CAAC,UAAU,EAAE,CAAA;YACvB,OAAO,IAAI,CAAA;QACb,CAAC;QACD,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC5D,IAAA,iBAAU,EAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAA;YACnC,IAAI,CAAC,WAAW,EAAE,CAAA;YAClB,OAAO,IAAI,CAAA;QACb,CAAC;QACD,IACE,SAAS,KAAK,YAAY;YAC1B,SAAS,KAAK,QAAQ;YACtB,SAAS,KAAK,WAAW,EACzB,CAAC;YACD,IAAA,0BAAa,EAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;YAC1C,OAAO,IAAI,CAAA;QACb,CAAC;QAED,sBAAsB;QACtB,IAAI,IAAA,gCAAmB,EAAC,SAAS,CAAC,EAAE,CAAC;YACnC,IAAI,IAAA,gCAAmB,EAAC,SAAS,EAAE,MAAM,CAAC,EAAE,CAAC;gBAC3C,MAAM,IAAA,2BAAc,EAAC,SAAS,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,CAAC,CAAA;gBAC/D,MAAM,SAAS,GAAG,MAAM,IAAA,uBAAU,EAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE,CAAC,CAAA;gBACxD,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAA;gBAC3B,OAAO,IAAI,CAAA;YACb,CAAC;YACD,IAAI,IAAA,gCAAmB,EAAC,SAAS,EAAE,MAAM,CAAC,EAAE,CAAC;gBAC3C,MAAM,IAAA,2BAAc,EAAC,SAAS,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,CAAC,CAAA;gBAC/D,MAAM,SAAS,GAAG,MAAM,IAAA,uBAAU,EAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE,CAAC,CAAA;gBACxD,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAA;gBAC3B,OAAO,IAAI,CAAA;YACb,CAAC;YACD,IAAI,IAAA,gCAAmB,EAAC,SAAS,EAAE,MAAM,CAAC,EAAE,CAAC;gBAC3C,MAAM,IAAA,2BAAc,EAAC,SAAS,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,CAAC,CAAA;gBAC/D,MAAM,IAAA,4BAAe,GAAE,CAAA;gBACvB,IAAI,CAAC,WAAW,EAAE,CAAA;gBAClB,OAAO,IAAI,CAAA;YACb,CAAC;YACD,MAAM,YAAY,GAAG,IAAA,gCAAmB,EAAC,SAAS,EAAE,SAAS,CAAC,CAAA;YAC9D,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,EAAE,GAAG,QAAQ,CAAE,YAAiC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;gBAC9D,MAAM,IAAA,2BAAc,EAAC,SAAS,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,CAAC,CAAA;gBAC/D,MAAM,SAAS,GAAG,MAAM,IAAA,oCAAuB,EAC7C,EAAE,EACF,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,EAAE,CACR,CAAA;gBACD,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAA;gBAC3B,OAAO,IAAI,CAAA;YACb,CAAC;YACD,IAAI,IAAA,gCAAmB,EAAC,SAAS,EAAE,OAAO,CAAC,EAAE,CAAC;gBAC5C,IAAA,mCAAsB,GAAE,CAAA;gBACxB,IAAI,CAAC,WAAW,EAAE,CAAA;gBAClB,OAAO,IAAI,CAAA;YACb,CAAC;YACD,IAAI,IAAA,gCAAmB,EAAC,SAAS,EAAE,MAAM,CAAC,EAAE,CAAC;gBAC3C,MAAM,IAAA,2BAAc,EAAC,SAAS,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,CAAA;gBACrE,IAAA,kCAAqB,GAAE,CAAA;gBACvB,IAAI,CAAC,WAAW,EAAE,CAAA;gBAClB,OAAO,IAAI,CAAA;YACb,CAAC;YACD,IAAA,kCAAqB,GAAE,CAAA;YACvB,IAAI,CAAC,WAAW,EAAE,CAAA;YAClB,OAAO,IAAI,CAAA;QACb,CAAC;QAED,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;YACzB,sEAAsE;YACtE,IAAA,sDAA+B,GAAE,CAAA;YACjC,yCAAyC;YACzC,OAAO,KAAK,CAAA;QACd,CAAC;QAED,OAAO,KAAK,CAAA;IACd,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,SAAiB;QAC9C,MAAM,IAAA,2BAAc,EAAC,SAAS,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,CAAC,CAAA;QAC/D,iBAAO,CAAC,GAAG,EAAE,CAAC,KAAK,EAAE,CAAA;QAErB,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,EAAE,CAAA;QAE5B,MAAM,UAAU,GAAY;YAC1B,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,SAAS;SACnB,CAAA;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YAC3B,IAAA,0BAAW,EAAC,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC,CAAA;QACrE,CAAC;QAED,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAA;QAC/B,MAAM,EAAE,eAAe,EAAE,YAAY,EAAE,GACrC,MAAM,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,CAAA;QAE5C,IAAI,CAAC,YAAY,GAAG,YAAY,CAAA;QAChC,MAAM,eAAe,CAAA;QACrB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;QAExB,IAAI,CAAC,mBAAmB,GAAG,KAAK,CAAA;QAEhC,iBAAO,CAAC,GAAG,EAAE,CAAC,IAAI,EAAE,CAAA;QAEpB,IAAI,CAAC,WAAW,EAAE,CAAA;IACpB,CAAC;IAEO,qBAAqB;QAC3B,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC9B,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAA;QACzB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,uBAAuB,GAAG,IAAI,CAAA;QACrC,CAAC;IACH,CAAC;IAEO,gBAAgB;QACtB,iBAAO,CAAC,GAAG,EAAE,CAAC,IAAI,EAAE,CAAA;QACpB,IAAI,CAAC,mBAAmB,GAAG,KAAK,CAAA;QAChC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,IAAI,CAAC,YAAY,EAAE,CAAA;YACnB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;QAC1B,CAAC;QACD,OAAO,CAAC,KAAK,CAAC,IAAA,mBAAM,EAAC,kCAAkC,CAAC,CAAC,CAAA;IAC3D,CAAC;IAEO,oBAAoB;QAC1B,OAAO,CAAC,GAAG,CAAC,IAAA,kBAAK,EAAC,gBAAgB,CAAC,CAAC,CAAA;QACpC,IAAI,CAAC,WAAW,EAAE,CAAA;IACpB,CAAC;IAEO,cAAc,CAAC,GAAW,EAAE,GAAQ;QAC1C,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,YAAY,EAAE,CAAA;QACrB,CAAC;QAED,IACE,CAAC,IAAI,CAAC,SAAS;YACf,GAAG,KAAK,GAAG;YACX,cAAc,IAAI,IAAI,CAAC,EAAE;YACzB,MAAM,IAAI,IAAI,CAAC,EAAE;YACjB,QAAQ,IAAI,IAAI,CAAC,EAAE,EACnB,CAAC;YACD,MAAM,KAAK,GAAG,IAAI,CAAC,EAAS,CAAA;YAC5B,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,KAAK,CAAA;YAC9B,MAAM,YAAY,GAAG,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAA;YACrE,IAAI,YAAY,KAAK,IAAI,EAAE,CAAC;gBAC1B,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,GAAG,CAAC,CAAC,GAAG,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;gBACpE,KAAK,CAAC,YAAY,EAAE,CAAA;YACtB,CAAC;QACH,CAAC;QACD,IAAI,CAAC,aAAa,EAAE,CAAA;IACtB,CAAC;IAEO,KAAK,CAAC,YAAY;QACxB,IAAI,IAAA,2BAAgB,GAAE,EAAE,CAAC;YACvB,IAAA,qBAAU,EAAC,IAAA,8BAAc,GAAE,CAAC,CAAA;QAC9B,CAAC;QAED,IAAI,MAAM,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACtB,CAAC;YAAC,IAAI,CAAC,EAAU,CAAC,IAAI,GAAG,EAAE,CAAA;QAC7B,CAAC;QAED,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC7B,IAAI,CAAC,kBAAkB,EAAE,CAAA;QAC3B,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;YACtB,IAAI,GAAG,GAAG,IAAI,CAAC,cAAc,GAAG,IAAI,EAAE,CAAC;gBACrC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAA;YACzB,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,cAAc,GAAG,GAAG,CAAA;gBACzB,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAA;gBAC3C,IAAI,CAAC,WAAW,EAAE,CAAA;YACpB,CAAC;QACH,CAAC;IACH,CAAC;IAEO,YAAY;QAClB,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC7B,IAAI,CAAC,kBAAkB,EAAE,CAAA;QAC3B,CAAC;IACH,CAAC;IAEO,kBAAkB;QACxB,OAAO,CAAC,GAAG,CAAC,IAAA,mBAAM,EAAC,8BAA8B,CAAC,CAAC,CAAA;QACnD,IAAI,CAAC,mBAAmB,GAAG,KAAK,CAAA;QAChC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,IAAI,CAAC,YAAY,EAAE,CAAA;QACrB,CAAC;QACD,iBAAO,CAAC,GAAG,EAAE,CAAC,IAAI,EAAE,CAAA;IACtB,CAAC;IAEO,KAAK,CAAC,UAAU;QACtB,iBAAO,CAAC,GAAG,EAAE,CAAC,aAAa,EAAE,CAAA;QAC7B,OAAO,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,CAAA;QAChD,OAAO,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAA;QAC/C,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;QAEjB,wCAAwC;QACxC,IAAA,wCAA6B,GAAE,CAAA;QAE/B,MAAM,IAAA,uDAA0B,GAAE,CAAA;QAElC,MAAM,WAAW,GAAG,EAAE,CAAA;QACtB,MAAM,2BAA2B,GAAG,MAAM,CAAC,MAAM,CAC/C,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAC9B;aACE,IAAI,EAAE;aACN,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,GAAG,GAAG,OAAO,EAAE,CAAC,CAAC,CAAA;QAE7C,WAAW,CAAC,IAAI,CACd,GAAG,IAAA,kBAAS,EAAC,2BAA2B,EAAE,QAAQ,CAAC,qBACjD,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,KAAK,IAAI;YAC7C,CAAC,CAAC,KAAK,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,cAAc,EAAE,gBAAgB;YAC9E,CAAC,CAAC,GACN,EAAE,CACH,CAAA;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC;YAC3C,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAC9B,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC,OAAO,EAAE;gBACzD,IAAI,CAAC,GAAG,EAAE,CAAC;gBACX,CAAC,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CACxB,CAAA;YACD,WAAW,CAAC,IAAI,CACd,mCAAmC,IAAA,kBAAS,EAAC,cAAc,EAAE,KAAK,CAAC,GAAG,CACvE,CAAA;QACH,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAA;QAClC,MAAM,IAAA,0BAAc,GAAE,CAAA;QAEtB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACjB,CAAC;IAEO,aAAa;QACnB,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QAC9B,MAAM,QAAQ,GAAG,WAAW,GAAG,IAAI,CAAC,aAAa,CAAA;QACjD,IAAI,QAAQ,GAAG,EAAE,EAAE,CAAC;YAClB,IAAI,CAAC,qBAAqB,EAAE,CAAA;YAC5B,IAAI,IAAI,CAAC,qBAAqB,IAAI,CAAC,EAAE,CAAC;gBACpC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAA;YACvB,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,qBAAqB,GAAG,CAAC,CAAA;YAC9B,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAA;YACxB,CAAC;QACH,CAAC;QACD,IAAI,CAAC,aAAa,GAAG,WAAW,CAAA;IAClC,CAAC;CACF;AA9gBD,kBA8gBC"}
@@ -0,0 +1,157 @@
1
+ import { Interface } from 'readline';
2
+ import { FileChanges, UsageResponse } from './common/actions';
3
+ import { ApiKeyType } from './common/api-keys/constants';
4
+ import { CostMode } from './common/constants';
5
+ import { AgentState, ToolResult } from './common/types/agent-state';
6
+ import { User } from './common/util/credentials';
7
+ import { ProjectFileContext } from './common/util/file';
8
+ import { GitCommand, MakeNullable } from './types';
9
+ type UsageData = Omit<MakeNullable<UsageResponse, 'remainingBalance'>, 'type'>;
10
+ export declare class Client {
11
+ private webSocket;
12
+ private freshPrompt;
13
+ private reconnectWhenNextIdle;
14
+ private fingerprintId;
15
+ private costMode;
16
+ private hadFileChanges;
17
+ private git;
18
+ private rl;
19
+ private responseComplete;
20
+ private responseBuffer;
21
+ private oneTimeFlags;
22
+ usageData: UsageData;
23
+ pendingTopUpMessageAmount: number;
24
+ fileContext: ProjectFileContext | undefined;
25
+ lastChanges: FileChanges;
26
+ agentState: AgentState | undefined;
27
+ originalFileVersions: Record<string, string | null>;
28
+ creditsByPromptId: Record<string, number[]>;
29
+ user: User | undefined;
30
+ lastWarnedPct: number;
31
+ storedApiKeyTypes: ApiKeyType[];
32
+ lastToolResults: ToolResult[];
33
+ model: string | undefined;
34
+ constructor({ websocketUrl, onWebSocketError, onWebSocketReconnect, freshPrompt, reconnectWhenNextIdle, costMode, git, rl, model, }: {
35
+ websocketUrl: string;
36
+ onWebSocketError: () => void;
37
+ onWebSocketReconnect: () => void;
38
+ freshPrompt: () => void;
39
+ reconnectWhenNextIdle: () => void;
40
+ costMode: CostMode;
41
+ git: GitCommand;
42
+ rl: Interface;
43
+ model: string | undefined;
44
+ });
45
+ exit(): Promise<void>;
46
+ initAgentState(projectFileContext: ProjectFileContext): void;
47
+ private initFingerprintId;
48
+ private getUser;
49
+ connect(): Promise<void>;
50
+ fetchStoredApiKeyTypes(): Promise<void>;
51
+ handleAddApiKey(keyType: ApiKeyType, apiKey: string): Promise<void>;
52
+ handleReferralCode(referralCode: string): Promise<void>;
53
+ logout(): Promise<void>;
54
+ login(referralCode?: string): Promise<void>;
55
+ setUsage(usageData: Omit<UsageResponse, 'type'>): void;
56
+ reconnect(): void;
57
+ private setupSubscriptions;
58
+ private showUsageWarning;
59
+ generateCommitMessage(stagedChanges: string): Promise<string>;
60
+ sendUserInput(prompt: string): Promise<{
61
+ responsePromise: Promise<{
62
+ type: "prompt-response";
63
+ promptId: string;
64
+ agentState: {
65
+ agentContext: string;
66
+ fileContext: {
67
+ currentWorkingDirectory: string;
68
+ fileTree: import("./common/util/file").FileTreeNode[];
69
+ fileTokenScores: Record<string, Record<string, number>>;
70
+ knowledgeFiles: Record<string, string>;
71
+ gitChanges: {
72
+ status: string;
73
+ diff: string;
74
+ diffCached: string;
75
+ lastCommitMessages: string;
76
+ };
77
+ changesSinceLastChat: Record<string, string>;
78
+ shellConfigFiles: Record<string, string>;
79
+ systemInfo: {
80
+ platform: string;
81
+ shell: string;
82
+ nodeVersion: string;
83
+ arch: string;
84
+ homedir: string;
85
+ cpus: number;
86
+ };
87
+ userKnowledgeFiles?: Record<string, string> | undefined;
88
+ fileVersions?: {
89
+ path: string;
90
+ content: string;
91
+ }[][] | undefined;
92
+ };
93
+ messageHistory: {
94
+ content: string | ({
95
+ type: "text";
96
+ text: string;
97
+ cache_control?: {
98
+ type: "ephemeral";
99
+ } | undefined;
100
+ } | {
101
+ type: "tool_use";
102
+ name: string;
103
+ id: string;
104
+ input: Record<string, any>;
105
+ cache_control?: {
106
+ type: "ephemeral";
107
+ } | undefined;
108
+ } | {
109
+ type: "tool_result";
110
+ content: string;
111
+ tool_use_id: string;
112
+ cache_control?: {
113
+ type: "ephemeral";
114
+ } | undefined;
115
+ } | {
116
+ type: "image";
117
+ source: {
118
+ type: "base64";
119
+ media_type: "image/jpeg";
120
+ data: string;
121
+ };
122
+ cache_control?: {
123
+ type: "ephemeral";
124
+ } | undefined;
125
+ })[];
126
+ role: "user" | "assistant";
127
+ }[];
128
+ consecutiveAssistantMessages?: number | undefined;
129
+ };
130
+ toolResults: {
131
+ name: string;
132
+ id: string;
133
+ result: string;
134
+ }[];
135
+ toolCalls: {
136
+ name: string;
137
+ id: string;
138
+ parameters: Record<string, string>;
139
+ }[];
140
+ } & {
141
+ type: "prompt-response";
142
+ } & {
143
+ wasStoppedByUser: boolean;
144
+ }>;
145
+ stopResponse: () => void;
146
+ }>;
147
+ /**
148
+ * Shrinks all instances of more than 2 newlines in a row.
149
+ * Note: don't start or end colored text with newlines
150
+ * @param chunk chunk to display
151
+ */
152
+ displayChunk(chunk: string): void;
153
+ private subscribeToResponse;
154
+ getUsage(): Promise<void>;
155
+ warmContextCache(): Promise<void>;
156
+ }
157
+ export {};