codebuff 1.0.234 → 1.0.236

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 (111) hide show
  1. package/dist/background-process-manager.d.ts +50 -0
  2. package/dist/background-process-manager.js +320 -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 +45 -0
  32. package/dist/cli.js +455 -0
  33. package/dist/cli.js.map +1 -0
  34. package/dist/client.d.ts +139 -0
  35. package/dist/client.js +791 -0
  36. package/dist/client.js.map +1 -0
  37. package/dist/common/json-config/constants.js +2 -2
  38. package/dist/common/json-config/constants.js.map +1 -1
  39. package/dist/config.d.ts +4 -0
  40. package/dist/config.js +12 -0
  41. package/dist/config.js.map +1 -0
  42. package/dist/create-template-project.d.ts +1 -0
  43. package/dist/create-template-project.js +107 -0
  44. package/dist/create-template-project.js.map +1 -0
  45. package/dist/credentials.d.ts +4 -0
  46. package/dist/credentials.js +38 -0
  47. package/dist/credentials.js.map +1 -0
  48. package/dist/dev-process-manager.d.ts +10 -0
  49. package/dist/dev-process-manager.js +54 -0
  50. package/dist/dev-process-manager.js.map +1 -0
  51. package/dist/fingerprint.d.ts +1 -0
  52. package/dist/fingerprint.js +48 -0
  53. package/dist/fingerprint.js.map +1 -0
  54. package/dist/index.d.ts +2 -0
  55. package/dist/index.js +113 -0
  56. package/dist/index.js.map +1 -0
  57. package/dist/menu.d.ts +3 -0
  58. package/dist/menu.js +124 -0
  59. package/dist/menu.js.map +1 -0
  60. package/dist/project-files.d.ts +114 -0
  61. package/dist/project-files.js +513 -0
  62. package/dist/project-files.js.map +1 -0
  63. package/dist/startup-process-handler.d.ts +2 -0
  64. package/dist/startup-process-handler.js +21 -0
  65. package/dist/startup-process-handler.js.map +1 -0
  66. package/dist/tool-handlers.d.ts +28 -0
  67. package/dist/tool-handlers.js +240 -0
  68. package/dist/tool-handlers.js.map +1 -0
  69. package/dist/types.d.ts +9 -0
  70. package/dist/types.js +3 -0
  71. package/dist/types.js.map +1 -0
  72. package/dist/update-codebuff.d.ts +1 -0
  73. package/dist/update-codebuff.js +158 -0
  74. package/dist/update-codebuff.js.map +1 -0
  75. package/dist/utils/__tests__/background-process-manager.test.d.ts +1 -0
  76. package/dist/utils/__tests__/background-process-manager.test.js +289 -0
  77. package/dist/utils/__tests__/background-process-manager.test.js.map +1 -0
  78. package/dist/utils/__tests__/tool-renderers.test.d.ts +1 -0
  79. package/dist/utils/__tests__/tool-renderers.test.js +51 -0
  80. package/dist/utils/__tests__/tool-renderers.test.js.map +1 -0
  81. package/dist/utils/__tests__/xml-stream-parser.test.d.ts +1 -0
  82. package/dist/utils/__tests__/xml-stream-parser.test.js +229 -0
  83. package/dist/utils/__tests__/xml-stream-parser.test.js.map +1 -0
  84. package/dist/utils/detect-shell.d.ts +1 -0
  85. package/dist/utils/detect-shell.js +60 -0
  86. package/dist/utils/detect-shell.js.map +1 -0
  87. package/dist/utils/spinner.d.ts +11 -0
  88. package/dist/utils/spinner.js +87 -0
  89. package/dist/utils/spinner.js.map +1 -0
  90. package/dist/utils/system-info.d.ts +8 -0
  91. package/dist/utils/system-info.js +22 -0
  92. package/dist/utils/system-info.js.map +1 -0
  93. package/dist/utils/terminal.d.ts +38 -0
  94. package/dist/utils/terminal.js +460 -0
  95. package/dist/utils/terminal.js.map +1 -0
  96. package/dist/utils/tool-renderers.d.ts +16 -0
  97. package/dist/utils/tool-renderers.js +145 -0
  98. package/dist/utils/tool-renderers.js.map +1 -0
  99. package/dist/utils/xml-stream-parser.d.ts +9 -0
  100. package/dist/utils/xml-stream-parser.js +128 -0
  101. package/dist/utils/xml-stream-parser.js.map +1 -0
  102. package/dist/web-scraper.d.ts +3 -0
  103. package/dist/web-scraper.js +57 -0
  104. package/dist/web-scraper.js.map +1 -0
  105. package/dist/workers/checkpoint-worker.d.ts +1 -0
  106. package/dist/workers/checkpoint-worker.js +48 -0
  107. package/dist/workers/checkpoint-worker.js.map +1 -0
  108. package/dist/workers/project-context.d.ts +1 -0
  109. package/dist/workers/project-context.js +17 -0
  110. package/dist/workers/project-context.js.map +1 -0
  111. package/package.json +1 -1
package/dist/cli.js ADDED
@@ -0,0 +1,455 @@
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 spinner_1 = require("./utils/spinner");
46
+ const terminal_1 = require("./utils/terminal");
47
+ class CLI {
48
+ client;
49
+ readyPromise;
50
+ git;
51
+ costMode;
52
+ rl;
53
+ isReceivingResponse = false;
54
+ stopResponse = null;
55
+ lastSigintTime = 0;
56
+ lastInputTime = 0;
57
+ consecutiveFastInputs = 0;
58
+ pastedContent = '';
59
+ isPasting = false;
60
+ shouldReconnectWhenIdle = false;
61
+ constructor(readyPromise, { git, costMode, model }) {
62
+ this.git = git;
63
+ this.costMode = costMode;
64
+ this.setupSignalHandlers();
65
+ this.initReadlineInterface();
66
+ this.client = new client_1.Client(config_1.websocketUrl, this.onWebSocketError.bind(this), this.onWebSocketReconnect.bind(this), this.returnControlToUser.bind(this), this.reconnectWhenNextIdle.bind(this), this.costMode, this.git, this.rl, model);
67
+ this.readyPromise = Promise.all([
68
+ readyPromise.then((results) => {
69
+ const [fileContext, ,] = results;
70
+ this.client.initAgentState(fileContext);
71
+ return this.client.warmContextCache();
72
+ }),
73
+ this.client.connect(),
74
+ ]);
75
+ this.setPrompt();
76
+ process.on('unhandledRejection', (reason, promise) => {
77
+ console.error('\nUnhandled Rejection at:', promise, 'reason:', reason);
78
+ this.freshPrompt();
79
+ });
80
+ process.on('uncaughtException', (err, origin) => {
81
+ console.error(`\nCaught exception: ${err}\n` + `Exception origin: ${origin}`);
82
+ console.error(err.stack);
83
+ this.freshPrompt();
84
+ });
85
+ }
86
+ setupSignalHandlers() {
87
+ process.on('exit', () => {
88
+ spinner_1.Spinner.get().restoreCursor();
89
+ (0, background_process_manager_1.sendKillSignalToAllBackgroundProcesses)();
90
+ });
91
+ process.on('SIGTERM', async () => {
92
+ spinner_1.Spinner.get().restoreCursor();
93
+ await (0, background_process_manager_1.killAllBackgroundProcesses)();
94
+ process.exit(0);
95
+ });
96
+ process.on('SIGTSTP', async () => await this.handleExit());
97
+ process.on('SIGHUP', async () => {
98
+ spinner_1.Spinner.get().restoreCursor();
99
+ await (0, background_process_manager_1.killAllBackgroundProcesses)();
100
+ process.exit(0);
101
+ });
102
+ // Doesn't catch SIGKILL (e.g. `kill -9`)
103
+ }
104
+ initReadlineInterface() {
105
+ this.rl = readline.createInterface({
106
+ input: process.stdin,
107
+ output: process.stdout,
108
+ historySize: 1000,
109
+ terminal: true,
110
+ completer: this.filePathCompleter.bind(this),
111
+ });
112
+ this.rl.on('line', (line) => this.handleLine(line));
113
+ this.rl.on('SIGINT', async () => await this.handleSigint());
114
+ this.rl.on('close', async () => await this.handleExit());
115
+ process.stdin.on('keypress', (str, key) => this.handleKeyPress(str, key));
116
+ }
117
+ filePathCompleter(line) {
118
+ const lastWord = line.split(' ').pop() || '';
119
+ const input = lastWord.startsWith('~')
120
+ ? (0, os_1.homedir)() + lastWord.slice(1)
121
+ : lastWord;
122
+ const directorySuffix = process.platform === 'win32' ? '\\' : '/';
123
+ const dir = input.endsWith(directorySuffix)
124
+ ? input.slice(0, input.length - 1)
125
+ : (0, path_1.dirname)(input);
126
+ const partial = input.endsWith(directorySuffix) ? '' : (0, path_1.basename)(input);
127
+ let baseDir = (0, path_1.isAbsolute)(dir) ? dir : path_1.default.join((0, project_files_1.getProjectRoot)(), dir);
128
+ try {
129
+ const files = (0, fs_1.readdirSync)(baseDir);
130
+ const matches = files
131
+ .filter((file) => file.startsWith(partial))
132
+ .map((file) => file + ((0, project_files_1.isDir)(path_1.default.join(baseDir, file)) ? directorySuffix : ''));
133
+ return [matches, partial];
134
+ }
135
+ catch {
136
+ return [[], line];
137
+ }
138
+ }
139
+ setPrompt() {
140
+ this.rl.setPrompt((0, picocolors_1.green)(`${(0, path_1.parse)((0, project_files_1.getProjectRoot)()).base} > `));
141
+ }
142
+ /**
143
+ * Prompts the user with a clean prompt state
144
+ */
145
+ freshPrompt(userInput = '') {
146
+ spinner_1.Spinner.get().stop();
147
+ if (this.shouldReconnectWhenIdle) {
148
+ this.client.reconnect();
149
+ this.shouldReconnectWhenIdle = false;
150
+ }
151
+ readline.cursorTo(process.stdout, 0);
152
+ const rlAny = this.rl;
153
+ // clear line first
154
+ rlAny.line = '';
155
+ this.setPrompt();
156
+ // then prompt
157
+ this.rl.prompt();
158
+ if (!userInput) {
159
+ return;
160
+ }
161
+ // then rewrite new prompt
162
+ this.rl.write(' '.repeat(userInput.length)); // hacky way to move cursor
163
+ rlAny.line = userInput;
164
+ rlAny._refreshLine();
165
+ }
166
+ async printInitialPrompt({ initialInput, runInitFlow, }) {
167
+ if (this.client.user) {
168
+ (0, menu_1.displayGreeting)(this.costMode, this.client.user.name);
169
+ }
170
+ else {
171
+ console.log(`Welcome to Codebuff! Give us a sec to get your account set up...`);
172
+ await this.client.login();
173
+ return;
174
+ }
175
+ this.freshPrompt();
176
+ if (runInitFlow) {
177
+ process.stdout.write('init\n');
178
+ await this.handleUserInput('init');
179
+ }
180
+ if (initialInput) {
181
+ process.stdout.write(initialInput + '\n');
182
+ await this.handleUserInput(initialInput);
183
+ }
184
+ }
185
+ async printDiff() {
186
+ (0, diff_1.handleDiff)(this.client.lastChanges);
187
+ this.freshPrompt();
188
+ }
189
+ async handleLine(line) {
190
+ this.detectPasting();
191
+ if (this.isPasting) {
192
+ this.pastedContent += line + '\n';
193
+ }
194
+ else if (!this.isReceivingResponse) {
195
+ if (this.pastedContent) {
196
+ await this.handleUserInput((this.pastedContent + line).trim());
197
+ this.pastedContent = '';
198
+ }
199
+ else {
200
+ await this.handleUserInput(line.trim());
201
+ }
202
+ }
203
+ }
204
+ async handleUserInput(userInput) {
205
+ this.rl.setPrompt('');
206
+ if (!userInput) {
207
+ this.freshPrompt();
208
+ return;
209
+ }
210
+ userInput = userInput.trim();
211
+ if (await this.processCommand(userInput)) {
212
+ return;
213
+ }
214
+ await this.forwardUserInput(userInput);
215
+ }
216
+ async processCommand(userInput) {
217
+ if (userInput === 'help' || userInput === 'h' || userInput === '/help') {
218
+ (0, menu_1.displayMenu)();
219
+ this.freshPrompt();
220
+ return true;
221
+ }
222
+ if (userInput === 'login' || userInput === 'signin') {
223
+ await this.client.login();
224
+ checkpoint_manager_1.checkpointManager.clearCheckpoints();
225
+ return true;
226
+ }
227
+ if (userInput === 'logout' || userInput === 'signout') {
228
+ await this.client.logout();
229
+ this.freshPrompt();
230
+ return true;
231
+ }
232
+ if (userInput.startsWith('ref-')) {
233
+ await this.client.handleReferralCode(userInput.trim());
234
+ return true;
235
+ }
236
+ // Detect potential API key input first
237
+ const detectionResult = (0, api_key_1.detectApiKey)(userInput);
238
+ if (detectionResult.status !== 'not_found') {
239
+ // If something resembling an API key was detected (valid or just prefix), handle it
240
+ await (0, api_key_1.handleApiKeyInput)(this.client, detectionResult, this.readyPromise, this.returnControlToUser.bind(this));
241
+ return true; // Indicate command was handled
242
+ }
243
+ // Continue with other commands if no API key input was detected/handled
244
+ if (userInput === 'usage' || userInput === 'credits') {
245
+ await this.client.getUsage();
246
+ return true;
247
+ }
248
+ if (userInput === 'quit' || userInput === 'exit' || userInput === 'q') {
249
+ await this.handleExit();
250
+ return true;
251
+ }
252
+ if (['diff', 'doff', 'dif', 'iff', 'd'].includes(userInput)) {
253
+ (0, diff_1.handleDiff)(this.client.lastChanges);
254
+ this.freshPrompt();
255
+ return true;
256
+ }
257
+ if (userInput === 'uuddlrlrba' ||
258
+ userInput === 'konami' ||
259
+ userInput === 'codebuffy') {
260
+ (0, easter_egg_1.showEasterEgg)(this.returnControlToUser.bind(this));
261
+ return true;
262
+ }
263
+ // Checkpoint commands
264
+ if ((0, checkpoint_1.isCheckpointCommand)(userInput)) {
265
+ if ((0, checkpoint_1.isCheckpointCommand)(userInput, 'undo')) {
266
+ await (0, checkpoint_1.saveCheckpoint)(userInput, this.client, this.readyPromise);
267
+ const toRestore = await (0, checkpoint_1.handleUndo)(this.client, this.rl);
268
+ this.freshPrompt(toRestore);
269
+ return true;
270
+ }
271
+ if ((0, checkpoint_1.isCheckpointCommand)(userInput, 'redo')) {
272
+ await (0, checkpoint_1.saveCheckpoint)(userInput, this.client, this.readyPromise);
273
+ const toRestore = await (0, checkpoint_1.handleRedo)(this.client, this.rl);
274
+ this.freshPrompt(toRestore);
275
+ return true;
276
+ }
277
+ if ((0, checkpoint_1.isCheckpointCommand)(userInput, 'list')) {
278
+ await (0, checkpoint_1.saveCheckpoint)(userInput, this.client, this.readyPromise);
279
+ await (0, checkpoint_1.listCheckpoints)();
280
+ this.freshPrompt();
281
+ return true;
282
+ }
283
+ const restoreMatch = (0, checkpoint_1.isCheckpointCommand)(userInput, 'restore');
284
+ if (restoreMatch) {
285
+ const id = parseInt(restoreMatch[1], 10);
286
+ await (0, checkpoint_1.saveCheckpoint)(userInput, this.client, this.readyPromise);
287
+ const toRestore = await (0, checkpoint_1.handleRestoreCheckpoint)(id, this.client, this.rl);
288
+ this.freshPrompt(toRestore);
289
+ return true;
290
+ }
291
+ if ((0, checkpoint_1.isCheckpointCommand)(userInput, 'clear')) {
292
+ (0, checkpoint_1.handleClearCheckpoints)();
293
+ this.freshPrompt();
294
+ return true;
295
+ }
296
+ if ((0, checkpoint_1.isCheckpointCommand)(userInput, 'save')) {
297
+ await (0, checkpoint_1.saveCheckpoint)(userInput, this.client, this.readyPromise, true);
298
+ (0, checkpoint_1.displayCheckpointMenu)();
299
+ this.freshPrompt();
300
+ return true;
301
+ }
302
+ (0, checkpoint_1.displayCheckpointMenu)();
303
+ this.freshPrompt();
304
+ return true;
305
+ }
306
+ if (userInput === 'init') {
307
+ // no need to save checkpoint, since we are passing request to backend
308
+ (0, inititalization_flow_1.handleInitializationFlowLocally)();
309
+ // Also forward user input to the backend
310
+ return false;
311
+ }
312
+ return false;
313
+ }
314
+ async forwardUserInput(userInput) {
315
+ await (0, checkpoint_1.saveCheckpoint)(userInput, this.client, this.readyPromise);
316
+ spinner_1.Spinner.get().start();
317
+ this.client.lastChanges = [];
318
+ const newMessage = {
319
+ role: 'user',
320
+ content: userInput,
321
+ };
322
+ if (this.client.agentState) {
323
+ (0, chat_storage_1.setMessages)([...this.client.agentState.messageHistory, newMessage]);
324
+ }
325
+ this.isReceivingResponse = true;
326
+ const { responsePromise, stopResponse } = await this.client.sendUserInput(userInput);
327
+ this.stopResponse = stopResponse;
328
+ await responsePromise;
329
+ this.stopResponse = null;
330
+ this.isReceivingResponse = false;
331
+ spinner_1.Spinner.get().stop();
332
+ this.freshPrompt();
333
+ }
334
+ reconnectWhenNextIdle() {
335
+ if (!this.isReceivingResponse) {
336
+ this.client.reconnect();
337
+ }
338
+ else {
339
+ this.shouldReconnectWhenIdle = true;
340
+ }
341
+ }
342
+ returnControlToUser() {
343
+ this.freshPrompt();
344
+ this.isReceivingResponse = false;
345
+ if (this.stopResponse) {
346
+ this.stopResponse();
347
+ }
348
+ }
349
+ onWebSocketError() {
350
+ spinner_1.Spinner.get().stop();
351
+ this.isReceivingResponse = false;
352
+ if (this.stopResponse) {
353
+ this.stopResponse();
354
+ this.stopResponse = null;
355
+ }
356
+ console.error((0, picocolors_1.yellow)('\nCould not connect. Retrying...'));
357
+ }
358
+ onWebSocketReconnect() {
359
+ console.log((0, picocolors_1.green)('\nReconnected!'));
360
+ this.returnControlToUser();
361
+ }
362
+ handleKeyPress(str, key) {
363
+ if (key.name === 'escape') {
364
+ this.handleEscKey();
365
+ }
366
+ if (!this.isPasting &&
367
+ str === ' ' &&
368
+ '_refreshLine' in this.rl &&
369
+ 'line' in this.rl &&
370
+ 'cursor' in this.rl) {
371
+ const rlAny = this.rl;
372
+ const { cursor, line } = rlAny;
373
+ const prevTwoChars = cursor > 1 ? line.slice(cursor - 2, cursor) : '';
374
+ if (prevTwoChars === ' ') {
375
+ rlAny.line = line.slice(0, cursor - 2) + '\n\n' + line.slice(cursor);
376
+ rlAny._refreshLine();
377
+ }
378
+ }
379
+ this.detectPasting();
380
+ }
381
+ async handleSigint() {
382
+ if ((0, terminal_1.isCommandRunning)()) {
383
+ (0, terminal_1.resetShell)((0, project_files_1.getProjectRoot)());
384
+ }
385
+ if ('line' in this.rl) {
386
+ ;
387
+ this.rl.line = '';
388
+ }
389
+ if (this.isReceivingResponse) {
390
+ this.handleStopResponse();
391
+ }
392
+ else {
393
+ const now = Date.now();
394
+ if (now - this.lastSigintTime < 5000) {
395
+ await this.handleExit();
396
+ }
397
+ else {
398
+ this.lastSigintTime = now;
399
+ console.log('\nPress Ctrl-C again to exit');
400
+ this.freshPrompt();
401
+ }
402
+ }
403
+ }
404
+ handleEscKey() {
405
+ if (this.isReceivingResponse) {
406
+ this.handleStopResponse();
407
+ }
408
+ }
409
+ handleStopResponse() {
410
+ console.log((0, picocolors_1.yellow)('\n[Response stopped by user]'));
411
+ this.isReceivingResponse = false;
412
+ if (this.stopResponse) {
413
+ this.stopResponse();
414
+ }
415
+ spinner_1.Spinner.get().stop();
416
+ }
417
+ async handleExit() {
418
+ spinner_1.Spinner.get().restoreCursor();
419
+ console.log('\n');
420
+ await (0, background_process_manager_1.killAllBackgroundProcesses)();
421
+ const logMessages = [];
422
+ const totalCreditsUsedThisSession = Object.values(this.client.creditsByPromptId)
423
+ .flat()
424
+ .reduce((sum, credits) => sum + credits, 0);
425
+ logMessages.push(`${(0, string_1.pluralize)(totalCreditsUsedThisSession, 'credit')} used this session, ${this.client.usageData.remainingBalance.toLocaleString()} credits left.`);
426
+ if (this.client.usageData.next_quota_reset) {
427
+ const daysUntilReset = Math.ceil((new Date(this.client.usageData.next_quota_reset).getTime() -
428
+ Date.now()) /
429
+ (1000 * 60 * 60 * 24));
430
+ logMessages.push(`Your free credits will reset in ${(0, string_1.pluralize)(daysUntilReset, 'day')}.`);
431
+ }
432
+ console.log(logMessages.join(' '));
433
+ console.log((0, picocolors_1.green)('Codebuff out!'));
434
+ process.exit(0);
435
+ }
436
+ detectPasting() {
437
+ const currentTime = Date.now();
438
+ const timeDiff = currentTime - this.lastInputTime;
439
+ if (timeDiff < 10) {
440
+ this.consecutiveFastInputs++;
441
+ if (this.consecutiveFastInputs >= 2) {
442
+ this.isPasting = true;
443
+ }
444
+ }
445
+ else {
446
+ this.consecutiveFastInputs = 0;
447
+ if (this.isPasting) {
448
+ this.isPasting = false;
449
+ }
450
+ }
451
+ this.lastInputTime = currentTime;
452
+ }
453
+ }
454
+ exports.CLI = CLI;
455
+ //# 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,6CAAyC;AACzC,+CAA+D;AAQ/D,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,CACtB,qBAAY,EACZ,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,EAChC,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,EACpC,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,EACnC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,EACrC,IAAI,CAAC,QAAQ,EACb,IAAI,CAAC,GAAG,EACR,IAAI,CAAC,EAAE,EACP,KAAK,CACN,CAAA;QAED,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,IAAA,mEAAsC,GAAE,CAAA;QAC1C,CAAC,CAAC,CAAA;QACF,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,IAAI,EAAE;YAC/B,iBAAO,CAAC,GAAG,EAAE,CAAC,aAAa,EAAE,CAAA;YAC7B,MAAM,IAAA,uDAA0B,GAAE,CAAA;YAClC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACjB,CAAC,CAAC,CAAA;QACF,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC,CAAA;QAC1D,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;YAC9B,iBAAO,CAAC,GAAG,EAAE,CAAC,aAAa,EAAE,CAAA;YAC7B,MAAM,IAAA,uDAA0B,GAAE,CAAA;YAClC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACjB,CAAC,CAAC,CAAA;QACF,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;QAEpB,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,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,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,CACpC,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,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;YAClD,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,mBAAmB;QACzB,IAAI,CAAC,WAAW,EAAE,CAAA;QAClB,IAAI,CAAC,mBAAmB,GAAG,KAAK,CAAA;QAChC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,IAAI,CAAC,YAAY,EAAE,CAAA;QACrB,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,mBAAmB,EAAE,CAAA;IAC5B,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,GAAG,CAAC,IAAI,CAAC,CAAA;QAEjB,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,uBAAuB,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,cAAc,EAAE,gBAAgB,CAClJ,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,OAAO,CAAC,GAAG,CAAC,IAAA,kBAAK,EAAC,eAAe,CAAC,CAAC,CAAA;QACnC,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;AA1fD,kBA0fC"}
@@ -0,0 +1,139 @@
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 } from './types';
9
+ export declare class Client {
10
+ private webSocket;
11
+ private returnControlToUser;
12
+ private reconnectWhenNextIdle;
13
+ private fingerprintId;
14
+ private costMode;
15
+ private hadFileChanges;
16
+ private git;
17
+ private rl;
18
+ private responseComplete;
19
+ private pendingTopUpMessageAmount;
20
+ private oneTimeFlags;
21
+ usageData: Omit<UsageResponse, 'type'>;
22
+ fileContext: ProjectFileContext | undefined;
23
+ lastChanges: FileChanges;
24
+ agentState: AgentState | undefined;
25
+ originalFileVersions: Record<string, string | null>;
26
+ creditsByPromptId: Record<string, number[]>;
27
+ user: User | undefined;
28
+ lastWarnedPct: number;
29
+ nextMonthlyGrant: number;
30
+ storedApiKeyTypes: ApiKeyType[];
31
+ lastToolResults: ToolResult[];
32
+ model: string | undefined;
33
+ constructor(websocketUrl: string, onWebSocketError: () => void, onWebSocketReconnect: () => void, returnControlToUser: () => void, reconnectWhenNextIdle: () => void, costMode: CostMode, git: GitCommand, rl: Interface, model: string | undefined);
34
+ exit(): Promise<void>;
35
+ initAgentState(projectFileContext: ProjectFileContext): void;
36
+ private initFingerprintId;
37
+ private getUser;
38
+ connect(): Promise<void>;
39
+ fetchStoredApiKeyTypes(): Promise<void>;
40
+ handleAddApiKey(keyType: ApiKeyType, apiKey: string): Promise<void>;
41
+ handleReferralCode(referralCode: string): Promise<void>;
42
+ logout(): Promise<void>;
43
+ login(referralCode?: string): Promise<void>;
44
+ setUsage(usageData: Omit<UsageResponse, 'type'>): void;
45
+ reconnect(): void;
46
+ private setupSubscriptions;
47
+ private showUsageWarning;
48
+ generateCommitMessage(stagedChanges: string): Promise<string>;
49
+ sendUserInput(prompt: string): Promise<{
50
+ responsePromise: Promise<{
51
+ type: "prompt-response";
52
+ promptId: string;
53
+ agentState: {
54
+ agentContext: string;
55
+ fileContext: {
56
+ currentWorkingDirectory: string;
57
+ fileTree: import("./common/util/file").FileTreeNode[];
58
+ fileTokenScores: Record<string, Record<string, number>>;
59
+ knowledgeFiles: Record<string, string>;
60
+ gitChanges: {
61
+ status: string;
62
+ diff: string;
63
+ diffCached: string;
64
+ lastCommitMessages: string;
65
+ };
66
+ changesSinceLastChat: Record<string, string>;
67
+ shellConfigFiles: Record<string, string>;
68
+ systemInfo: {
69
+ platform: string;
70
+ shell: string;
71
+ nodeVersion: string;
72
+ arch: string;
73
+ homedir: string;
74
+ cpus: number;
75
+ };
76
+ userKnowledgeFiles?: Record<string, string> | undefined;
77
+ fileVersions?: {
78
+ path: string;
79
+ content: string;
80
+ }[][] | undefined;
81
+ };
82
+ messageHistory: {
83
+ content: string | ({
84
+ type: "text";
85
+ text: string;
86
+ cache_control?: {
87
+ type: "ephemeral";
88
+ } | undefined;
89
+ } | {
90
+ id: string;
91
+ type: "tool_use";
92
+ name: string;
93
+ input: Record<string, any>;
94
+ cache_control?: {
95
+ type: "ephemeral";
96
+ } | undefined;
97
+ } | {
98
+ type: "tool_result";
99
+ content: string;
100
+ tool_use_id: string;
101
+ cache_control?: {
102
+ type: "ephemeral";
103
+ } | undefined;
104
+ } | {
105
+ type: "image";
106
+ source: {
107
+ type: "base64";
108
+ media_type: "image/jpeg";
109
+ data: string;
110
+ };
111
+ cache_control?: {
112
+ type: "ephemeral";
113
+ } | undefined;
114
+ })[];
115
+ role: "user" | "assistant";
116
+ }[];
117
+ consecutiveAssistantMessages?: number | undefined;
118
+ };
119
+ toolResults: {
120
+ id: string;
121
+ name: string;
122
+ result: string;
123
+ }[];
124
+ toolCalls: {
125
+ id: string;
126
+ name: string;
127
+ parameters: Record<string, string>;
128
+ }[];
129
+ } & {
130
+ type: "prompt-response";
131
+ } & {
132
+ wasStoppedByUser: boolean;
133
+ }>;
134
+ stopResponse: () => void;
135
+ }>;
136
+ private subscribeToResponse;
137
+ getUsage(): Promise<void>;
138
+ warmContextCache(): Promise<void>;
139
+ }