codecane 1.0.123

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 (156) hide show
  1. package/README.md +87 -0
  2. package/dist/__tests__/tool-handlers.test.d.ts +1 -0
  3. package/dist/__tests__/tool-handlers.test.js +32 -0
  4. package/dist/__tests__/tool-handlers.test.js.map +1 -0
  5. package/dist/chat-storage.d.ts +29 -0
  6. package/dist/chat-storage.js +125 -0
  7. package/dist/chat-storage.js.map +1 -0
  8. package/dist/cli.d.ts +38 -0
  9. package/dist/cli.js +453 -0
  10. package/dist/cli.js.map +1 -0
  11. package/dist/client.d.ts +71 -0
  12. package/dist/client.js +458 -0
  13. package/dist/client.js.map +1 -0
  14. package/dist/code-map/languages.d.ts +12 -0
  15. package/dist/code-map/languages.js +129 -0
  16. package/dist/code-map/languages.js.map +1 -0
  17. package/dist/code-map/parse.d.ts +11 -0
  18. package/dist/code-map/parse.js +127 -0
  19. package/dist/code-map/parse.js.map +1 -0
  20. package/dist/code-map/test-langs/test.d.ts +11 -0
  21. package/dist/code-map/test-langs/test.js +23 -0
  22. package/dist/code-map/test-langs/test.js.map +1 -0
  23. package/dist/code-map/tree-sitter-queries/readme.md +23 -0
  24. package/dist/code-map/tree-sitter-queries/tree-sitter-c-tags.scm +11 -0
  25. package/dist/code-map/tree-sitter-queries/tree-sitter-c_sharp-tags.scm +13 -0
  26. package/dist/code-map/tree-sitter-queries/tree-sitter-cpp-tags.scm +17 -0
  27. package/dist/code-map/tree-sitter-queries/tree-sitter-go-tags.scm +26 -0
  28. package/dist/code-map/tree-sitter-queries/tree-sitter-java-tags.scm +19 -0
  29. package/dist/code-map/tree-sitter-queries/tree-sitter-javascript-tags.scm +15 -0
  30. package/dist/code-map/tree-sitter-queries/tree-sitter-php-tags.scm +26 -0
  31. package/dist/code-map/tree-sitter-queries/tree-sitter-python-tags.scm +9 -0
  32. package/dist/code-map/tree-sitter-queries/tree-sitter-ruby-tags.scm +58 -0
  33. package/dist/code-map/tree-sitter-queries/tree-sitter-rust-tags.scm +26 -0
  34. package/dist/code-map/tree-sitter-queries/tree-sitter-typescript-tags.scm +15 -0
  35. package/dist/code-map/tsconfig.tsbuildinfo +1 -0
  36. package/dist/common/actions.d.ts +1405 -0
  37. package/dist/common/actions.js +193 -0
  38. package/dist/common/actions.js.map +1 -0
  39. package/dist/common/billing/quota-manager.d.ts +29 -0
  40. package/dist/common/billing/quota-manager.js +213 -0
  41. package/dist/common/billing/quota-manager.js.map +1 -0
  42. package/dist/common/constants.d.ts +41 -0
  43. package/dist/common/constants.js +126 -0
  44. package/dist/common/constants.js.map +1 -0
  45. package/dist/common/db/drizzle.config.d.ts +2 -0
  46. package/dist/common/db/drizzle.config.js +17 -0
  47. package/dist/common/db/drizzle.config.js.map +1 -0
  48. package/dist/common/db/env.d.mts +1 -0
  49. package/dist/common/db/env.mjs +26 -0
  50. package/dist/common/db/env.mjs.map +1 -0
  51. package/dist/common/db/index.d.ts +6 -0
  52. package/dist/common/db/index.js +35 -0
  53. package/dist/common/db/index.js.map +1 -0
  54. package/dist/common/db/schema.d.ts +1089 -0
  55. package/dist/common/db/schema.js +114 -0
  56. package/dist/common/db/schema.js.map +1 -0
  57. package/dist/common/env.d.mts +1 -0
  58. package/dist/common/env.mjs +34 -0
  59. package/dist/common/env.mjs.map +1 -0
  60. package/dist/common/project-file-tree.d.ts +9 -0
  61. package/dist/common/project-file-tree.js +166 -0
  62. package/dist/common/project-file-tree.js.map +1 -0
  63. package/dist/common/types/referral.d.ts +2 -0
  64. package/dist/common/types/referral.js +5 -0
  65. package/dist/common/types/referral.js.map +1 -0
  66. package/dist/common/types/usage.d.ts +24 -0
  67. package/dist/common/types/usage.js +13 -0
  68. package/dist/common/types/usage.js.map +1 -0
  69. package/dist/common/util/__tests__/string.test.d.ts +1 -0
  70. package/dist/common/util/__tests__/string.test.js +83 -0
  71. package/dist/common/util/__tests__/string.test.js.map +1 -0
  72. package/dist/common/util/array.d.ts +6 -0
  73. package/dist/common/util/array.js +32 -0
  74. package/dist/common/util/array.js.map +1 -0
  75. package/dist/common/util/changes.d.ts +6 -0
  76. package/dist/common/util/changes.js +72 -0
  77. package/dist/common/util/changes.js.map +1 -0
  78. package/dist/common/util/credentials.d.ts +25 -0
  79. package/dist/common/util/credentials.js +24 -0
  80. package/dist/common/util/credentials.js.map +1 -0
  81. package/dist/common/util/dates.d.ts +1 -0
  82. package/dist/common/util/dates.js +13 -0
  83. package/dist/common/util/dates.js.map +1 -0
  84. package/dist/common/util/file.d.ts +151 -0
  85. package/dist/common/util/file.js +180 -0
  86. package/dist/common/util/file.js.map +1 -0
  87. package/dist/common/util/git.d.ts +6 -0
  88. package/dist/common/util/git.js +81 -0
  89. package/dist/common/util/git.js.map +1 -0
  90. package/dist/common/util/helpers.d.ts +1 -0
  91. package/dist/common/util/helpers.js +6 -0
  92. package/dist/common/util/helpers.js.map +1 -0
  93. package/dist/common/util/object.d.ts +18 -0
  94. package/dist/common/util/object.js +91 -0
  95. package/dist/common/util/object.js.map +1 -0
  96. package/dist/common/util/patch.d.ts +1 -0
  97. package/dist/common/util/patch.js +215 -0
  98. package/dist/common/util/patch.js.map +1 -0
  99. package/dist/common/util/promise.d.ts +1 -0
  100. package/dist/common/util/promise.js +33 -0
  101. package/dist/common/util/promise.js.map +1 -0
  102. package/dist/common/util/referral.d.ts +1 -0
  103. package/dist/common/util/referral.js +6 -0
  104. package/dist/common/util/referral.js.map +1 -0
  105. package/dist/common/util/server/referral.d.ts +14 -0
  106. package/dist/common/util/server/referral.js +85 -0
  107. package/dist/common/util/server/referral.js.map +1 -0
  108. package/dist/common/util/string.d.ts +9 -0
  109. package/dist/common/util/string.js +90 -0
  110. package/dist/common/util/string.js.map +1 -0
  111. package/dist/common/util/stripe.d.ts +2 -0
  112. package/dist/common/util/stripe.js +19 -0
  113. package/dist/common/util/stripe.js.map +1 -0
  114. package/dist/common/util/tools.d.ts +2 -0
  115. package/dist/common/util/tools.js +13 -0
  116. package/dist/common/util/tools.js.map +1 -0
  117. package/dist/common/websockets/websocket-client.d.ts +40 -0
  118. package/dist/common/websockets/websocket-client.js +187 -0
  119. package/dist/common/websockets/websocket-client.js.map +1 -0
  120. package/dist/common/websockets/websocket-schema.d.ts +3058 -0
  121. package/dist/common/websockets/websocket-schema.js +55 -0
  122. package/dist/common/websockets/websocket-schema.js.map +1 -0
  123. package/dist/config.d.ts +2 -0
  124. package/dist/config.js +8 -0
  125. package/dist/config.js.map +1 -0
  126. package/dist/credentials.d.ts +3 -0
  127. package/dist/credentials.js +33 -0
  128. package/dist/credentials.js.map +1 -0
  129. package/dist/fingerprint.d.ts +1 -0
  130. package/dist/fingerprint.js +43 -0
  131. package/dist/fingerprint.js.map +1 -0
  132. package/dist/index.d.ts +2 -0
  133. package/dist/index.js +75 -0
  134. package/dist/index.js.map +1 -0
  135. package/dist/menu.d.ts +1 -0
  136. package/dist/menu.js +94 -0
  137. package/dist/menu.js.map +1 -0
  138. package/dist/project-files.d.ts +64 -0
  139. package/dist/project-files.js +338 -0
  140. package/dist/project-files.js.map +1 -0
  141. package/dist/tool-handlers.d.ts +11 -0
  142. package/dist/tool-handlers.js +215 -0
  143. package/dist/tool-handlers.js.map +1 -0
  144. package/dist/types.d.ts +7 -0
  145. package/dist/types.js +3 -0
  146. package/dist/types.js.map +1 -0
  147. package/dist/update-codebuff.d.ts +1 -0
  148. package/dist/update-codebuff.js +156 -0
  149. package/dist/update-codebuff.js.map +1 -0
  150. package/dist/web-scraper.d.ts +3 -0
  151. package/dist/web-scraper.js +57 -0
  152. package/dist/web-scraper.js.map +1 -0
  153. package/dist/worker-script-project-context.d.ts +1 -0
  154. package/dist/worker-script-project-context.js +13 -0
  155. package/dist/worker-script-project-context.js.map +1 -0
  156. package/package.json +71 -0
package/dist/cli.js ADDED
@@ -0,0 +1,453 @@
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 lodash_1 = require("lodash");
28
+ const project_file_tree_1 = require("./common/project-file-tree");
29
+ const changes_1 = require("./common/util/changes");
30
+ const readline = __importStar(require("readline"));
31
+ const picocolors_1 = require("picocolors");
32
+ const path_1 = require("path");
33
+ const config_1 = require("./config");
34
+ const chat_storage_1 = require("./chat-storage");
35
+ const client_1 = require("./client");
36
+ const menu_1 = require("./menu");
37
+ const project_files_1 = require("./project-files");
38
+ const tool_handlers_1 = require("./tool-handlers");
39
+ const constants_1 = require("./common/constants");
40
+ const file_1 = require("./common/util/file");
41
+ const web_scraper_1 = require("./web-scraper");
42
+ const git_1 = require("./common/util/git");
43
+ const string_1 = require("./common/util/string");
44
+ class CLI {
45
+ client;
46
+ chatStorage;
47
+ readyPromise;
48
+ git;
49
+ costMode;
50
+ rl;
51
+ isReceivingResponse = false;
52
+ stopResponse = null;
53
+ loadingInterval = null;
54
+ lastChanges = [];
55
+ lastSigintTime = 0;
56
+ lastInputTime = 0;
57
+ consecutiveFastInputs = 0;
58
+ pastedContent = '';
59
+ isPasting = false;
60
+ constructor(readyPromise, { git, costMode }) {
61
+ this.git = git;
62
+ this.costMode = costMode;
63
+ this.chatStorage = new chat_storage_1.ChatStorage();
64
+ this.rl = readline.createInterface({
65
+ input: process.stdin,
66
+ output: process.stdout,
67
+ historySize: 1000,
68
+ terminal: true,
69
+ completer: (line) => {
70
+ if (!this.client.fileContext?.fileTree)
71
+ return [[], line];
72
+ const tokenNames = Object.values(this.client.fileContext.fileTokenScores).flatMap((o) => Object.keys(o));
73
+ const paths = (0, project_file_tree_1.getAllFilePaths)(this.client.fileContext.fileTree);
74
+ const lastWord = line.split(' ').pop() || '';
75
+ const matchingTokens = [...tokenNames, ...paths].filter((token) => token.startsWith(lastWord) || token.includes('/' + lastWord));
76
+ return [matchingTokens, lastWord];
77
+ },
78
+ });
79
+ this.client = new client_1.Client(config_1.websocketUrl, this.chatStorage, this.onWebSocketError.bind(this), () => {
80
+ this.rl.prompt();
81
+ this.isReceivingResponse = false;
82
+ if (this.stopResponse) {
83
+ this.stopResponse();
84
+ }
85
+ this.stopLoadingAnimation();
86
+ }, this.costMode);
87
+ this.readyPromise = Promise.all([
88
+ readyPromise.then((results) => {
89
+ const [_, fileContext] = results;
90
+ this.client.initFileVersions(fileContext);
91
+ return this.client.warmContextCache();
92
+ }),
93
+ this.client.connect(),
94
+ ]);
95
+ this.setPrompt();
96
+ this.rl.on('line', (line) => {
97
+ this.handleInput(line);
98
+ });
99
+ this.rl.on('SIGINT', () => {
100
+ if (this.isReceivingResponse) {
101
+ this.handleStopResponse();
102
+ }
103
+ else {
104
+ const now = Date.now();
105
+ if (now - this.lastSigintTime < 3000) {
106
+ // 3 second window
107
+ this.handleExit();
108
+ }
109
+ else {
110
+ this.lastSigintTime = now;
111
+ console.log('\nPress Ctrl-C again to exit');
112
+ this.rl.prompt();
113
+ }
114
+ }
115
+ });
116
+ this.rl.on('close', () => {
117
+ this.handleExit();
118
+ });
119
+ process.on('SIGTSTP', () => {
120
+ // Exit on Ctrl+Z
121
+ this.handleExit();
122
+ });
123
+ process.stdin.on('keypress', (_, key) => {
124
+ if (key.name === 'escape') {
125
+ this.handleEscKey();
126
+ }
127
+ this.detectPasting();
128
+ });
129
+ }
130
+ onWebSocketError() {
131
+ this.stopLoadingAnimation();
132
+ this.isReceivingResponse = false;
133
+ if (this.stopResponse) {
134
+ this.stopResponse();
135
+ this.stopResponse = null;
136
+ }
137
+ console.error((0, picocolors_1.yellow)('\nCould not connect. Retrying...'));
138
+ }
139
+ detectPasting() {
140
+ const currentTime = Date.now();
141
+ const timeDiff = currentTime - this.lastInputTime;
142
+ if (timeDiff < 10) {
143
+ this.consecutiveFastInputs++;
144
+ if (this.consecutiveFastInputs >= 2) {
145
+ this.isPasting = true;
146
+ }
147
+ }
148
+ else {
149
+ this.consecutiveFastInputs = 0;
150
+ if (this.isPasting) {
151
+ this.isPasting = false;
152
+ }
153
+ }
154
+ this.lastInputTime = currentTime;
155
+ }
156
+ handleInput(line) {
157
+ this.detectPasting();
158
+ if (this.isPasting) {
159
+ this.pastedContent += line + '\n';
160
+ }
161
+ else if (!this.isReceivingResponse) {
162
+ if (this.pastedContent) {
163
+ this.handleUserInput((this.pastedContent + line).trim());
164
+ this.pastedContent = '';
165
+ }
166
+ else {
167
+ this.handleUserInput(line.trim());
168
+ }
169
+ }
170
+ }
171
+ setPrompt() {
172
+ this.rl.setPrompt((0, picocolors_1.green)(`${(0, path_1.parse)((0, project_files_1.getProjectRoot)()).base} > `));
173
+ }
174
+ printInitialPrompt(initialInput) {
175
+ if (this.client.user) {
176
+ console.log(`\nWelcome back ${this.client.user.name}! What would you like to do?`);
177
+ }
178
+ else {
179
+ console.log(`What would you like to do?\n`);
180
+ }
181
+ this.rl.prompt();
182
+ if (initialInput) {
183
+ process.stdout.write(initialInput + '\n');
184
+ this.handleUserInput(initialInput);
185
+ }
186
+ }
187
+ handleUndo() {
188
+ this.navigateFileVersion('undo');
189
+ this.rl.prompt();
190
+ }
191
+ handleRedo() {
192
+ this.navigateFileVersion('redo');
193
+ this.rl.prompt();
194
+ }
195
+ navigateFileVersion(direction) {
196
+ const currentVersion = this.chatStorage.getCurrentVersion();
197
+ const filePaths = Object.keys(currentVersion ? currentVersion.files : {});
198
+ const currentFiles = (0, project_files_1.getExistingFiles)(filePaths);
199
+ this.chatStorage.saveCurrentFileState(currentFiles);
200
+ const navigated = this.chatStorage.navigateVersion(direction);
201
+ if (navigated) {
202
+ console.log(direction === 'undo'
203
+ ? (0, picocolors_1.green)('Undo last change')
204
+ : (0, picocolors_1.green)('Redo last change'));
205
+ const files = this.applyAndDisplayCurrentFileVersion();
206
+ console.log((0, picocolors_1.green)('Loaded files:'), (0, picocolors_1.green)(Object.keys(files).join(', ')));
207
+ }
208
+ else {
209
+ console.log((0, picocolors_1.green)(`No more ${direction === 'undo' ? 'undo' : 'redo'}s`));
210
+ }
211
+ }
212
+ handleStopResponse() {
213
+ console.log((0, picocolors_1.yellow)('\n[Response stopped by user]'));
214
+ this.isReceivingResponse = false;
215
+ if (this.stopResponse) {
216
+ this.stopResponse();
217
+ }
218
+ this.stopLoadingAnimation();
219
+ }
220
+ handleExit() {
221
+ console.log('\n\n');
222
+ console.log(`${(0, string_1.pluralize)(this.client.sessionCreditsUsed, 'credit')} used this session.`);
223
+ if (!!this.client.limit &&
224
+ !!this.client.usage &&
225
+ !!this.client.nextQuotaReset) {
226
+ const daysUntilReset = Math.max(0, Math.floor((this.client.nextQuotaReset.getTime() - Date.now()) /
227
+ (1000 * 60 * 60 * 24)));
228
+ console.log(`${Math.max(0, this.client.limit - this.client.usage)} / ${this.client.limit} credits remaining. Renews in ${(0, string_1.pluralize)(daysUntilReset, 'day')}.`);
229
+ }
230
+ console.log((0, picocolors_1.green)('Codebuff out!'));
231
+ process.exit(0);
232
+ }
233
+ handleEscKey() {
234
+ if (this.isReceivingResponse) {
235
+ this.handleStopResponse();
236
+ }
237
+ }
238
+ applyAndDisplayCurrentFileVersion() {
239
+ const currentVersion = this.chatStorage.getCurrentVersion();
240
+ if (currentVersion) {
241
+ (0, project_files_1.setFiles)(currentVersion.files);
242
+ return currentVersion.files;
243
+ }
244
+ return {};
245
+ }
246
+ startLoadingAnimation() {
247
+ const chars = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'];
248
+ let i = 0;
249
+ this.loadingInterval = setInterval(() => {
250
+ process.stdout.clearLine(0);
251
+ process.stdout.cursorTo(0);
252
+ process.stdout.write((0, picocolors_1.green)(`${chars[i]} Thinking...`));
253
+ i = (i + 1) % chars.length;
254
+ }, 100);
255
+ }
256
+ stopLoadingAnimation() {
257
+ if (this.loadingInterval) {
258
+ clearInterval(this.loadingInterval);
259
+ this.loadingInterval = null;
260
+ process.stdout.clearLine(0);
261
+ process.stdout.cursorTo(0);
262
+ }
263
+ }
264
+ async autoCommitChanges() {
265
+ if ((0, git_1.hasStagedChanges)()) {
266
+ const stagedChanges = (0, git_1.getStagedChanges)();
267
+ if (!stagedChanges)
268
+ return;
269
+ const commitMessage = await this.client.generateCommitMessage(stagedChanges);
270
+ (0, git_1.commitChanges)(commitMessage);
271
+ return commitMessage;
272
+ }
273
+ return undefined;
274
+ }
275
+ handleDiff() {
276
+ if (this.lastChanges.length === 0) {
277
+ console.log((0, picocolors_1.yellow)('No changes found in the last assistant response.'));
278
+ return;
279
+ }
280
+ this.lastChanges.forEach((change) => {
281
+ console.log('-', change.filePath);
282
+ const lines = change.content
283
+ .split('\n')
284
+ .map((line) => (change.type === 'file' ? '+' + line : line));
285
+ lines.forEach((line) => {
286
+ if (line.startsWith('+')) {
287
+ console.log((0, picocolors_1.green)(line));
288
+ }
289
+ else if (line.startsWith('-')) {
290
+ console.log((0, picocolors_1.red)(line));
291
+ }
292
+ else {
293
+ console.log(line);
294
+ }
295
+ });
296
+ });
297
+ }
298
+ async handleUserInput(userInput) {
299
+ if (!userInput)
300
+ return;
301
+ userInput = userInput.trim();
302
+ // Handle commands
303
+ if (userInput === 'help' || userInput === 'h') {
304
+ (0, menu_1.displayMenu)();
305
+ this.rl.prompt();
306
+ return;
307
+ }
308
+ if (userInput === 'login' || userInput === 'signin') {
309
+ await this.client.login();
310
+ return;
311
+ }
312
+ else if (userInput === 'logout' || userInput === 'signout') {
313
+ await this.client.logout();
314
+ this.rl.prompt();
315
+ return;
316
+ }
317
+ else if (userInput.startsWith('ref-')) {
318
+ await this.client.handleReferralCode(userInput.trim());
319
+ return;
320
+ }
321
+ else if (userInput === 'usage' || userInput === 'credits') {
322
+ this.client.getUsage();
323
+ return;
324
+ }
325
+ else if (userInput === 'undo' || userInput === 'u') {
326
+ this.handleUndo();
327
+ return;
328
+ }
329
+ else if (userInput === 'redo' || userInput === 'r') {
330
+ this.handleRedo();
331
+ return;
332
+ }
333
+ else if (userInput === 'quit' ||
334
+ userInput === 'exit' ||
335
+ userInput === 'q') {
336
+ this.handleExit();
337
+ return;
338
+ }
339
+ else if (userInput === 'diff' ||
340
+ userInput === 'doff' ||
341
+ userInput === 'dif' ||
342
+ userInput === 'iff' ||
343
+ userInput === 'd') {
344
+ this.handleDiff();
345
+ this.rl.prompt();
346
+ return;
347
+ }
348
+ const runPrefix = '/run ';
349
+ const hasRunPrefix = userInput.startsWith(runPrefix);
350
+ if (hasRunPrefix ||
351
+ (!constants_1.SKIPPED_TERMINAL_COMMANDS.some((command) => userInput.toLowerCase().startsWith(command)) &&
352
+ !userInput.includes('error ') &&
353
+ userInput.split(' ').length <= 5)) {
354
+ const withoutRunPrefix = userInput.replace(runPrefix, '');
355
+ const { result, stdout } = await (0, tool_handlers_1.handleRunTerminalCommand)({ command: withoutRunPrefix }, 'user', 'user');
356
+ if (result !== 'command not found') {
357
+ this.setPrompt();
358
+ this.rl.prompt();
359
+ return;
360
+ }
361
+ else if (hasRunPrefix) {
362
+ process.stdout.write(stdout);
363
+ this.setPrompt();
364
+ this.rl.prompt();
365
+ return;
366
+ }
367
+ }
368
+ this.startLoadingAnimation();
369
+ await this.readyPromise;
370
+ const currentChat = this.chatStorage.getCurrentChat();
371
+ const { fileVersions } = currentChat;
372
+ const currentFileVersion = fileVersions[fileVersions.length - 1]?.files ?? {};
373
+ const changesSinceLastFileVersion = (0, project_files_1.getChangesSinceLastFileVersion)(currentFileVersion);
374
+ const changesFileBlocks = Object.entries(changesSinceLastFileVersion)
375
+ .map(([filePath, patch]) => [
376
+ filePath,
377
+ patch.length < 8_000
378
+ ? patch
379
+ : '[LARGE_FILE_CHANGE_TOO_LONG_TO_REPRODUCE]',
380
+ ])
381
+ .map(([filePath, patch]) => (0, file_1.createFileBlock)(filePath, patch));
382
+ const changesMessage = changesFileBlocks.length > 0
383
+ ? `<user_edits_since_last_chat>\n${changesFileBlocks.join('\n')}\n</user_edits_since_last_chat>\n\n`
384
+ : '';
385
+ const urls = (0, web_scraper_1.parseUrlsFromContent)(userInput);
386
+ const scrapedBlocks = await (0, web_scraper_1.getScrapedContentBlocks)(urls);
387
+ const scrapedContent = scrapedBlocks.length > 0 ? scrapedBlocks.join('\n\n') + '\n\n' : '';
388
+ const newMessage = {
389
+ role: 'user',
390
+ content: `${changesMessage}${scrapedContent}${userInput}`,
391
+ };
392
+ this.chatStorage.addMessage(currentChat, newMessage);
393
+ this.isReceivingResponse = true;
394
+ const { response, changes, changesAlreadyApplied } = await this.sendUserInputAndAwaitResponse();
395
+ this.isReceivingResponse = false;
396
+ this.stopLoadingAnimation();
397
+ const allChanges = [...changesAlreadyApplied, ...changes];
398
+ const filesChanged = (0, lodash_1.uniq)(allChanges.map((change) => change.filePath));
399
+ const allFilesChanged = this.chatStorage.saveFilesChanged(filesChanged);
400
+ // Stage previous changes if flag was set
401
+ if (this.git === 'stage' &&
402
+ this.lastChanges.length > 0 &&
403
+ changes.length > 0) {
404
+ const didStage = (0, git_1.stagePatches)((0, project_files_1.getProjectRoot)(), this.lastChanges);
405
+ if (didStage) {
406
+ console.log((0, picocolors_1.green)('\nStaged previous changes'));
407
+ }
408
+ }
409
+ const { created, modified } = (0, changes_1.applyChanges)((0, project_files_1.getProjectRoot)(), changes);
410
+ if (created.length > 0 || modified.length > 0) {
411
+ console.log();
412
+ }
413
+ for (const file of created) {
414
+ console.log((0, picocolors_1.green)(`- Created ${file}`));
415
+ }
416
+ for (const file of modified) {
417
+ console.log((0, picocolors_1.green)(`- Updated ${file}`));
418
+ }
419
+ if (created.length > 0 || modified.length > 0) {
420
+ if (this.client.lastRequestCredits > constants_1.REQUEST_CREDIT_SHOW_THRESHOLD) {
421
+ console.log(`\n${(0, string_1.pluralize)(this.client.lastRequestCredits, 'credit')} used for this request.`);
422
+ }
423
+ console.log('Complete! Type "diff" to see the changes made.');
424
+ this.client.showUsageWarning();
425
+ }
426
+ console.log();
427
+ this.lastChanges = allChanges;
428
+ const assistantMessage = {
429
+ role: 'assistant',
430
+ content: response,
431
+ };
432
+ this.chatStorage.addMessage(this.chatStorage.getCurrentChat(), assistantMessage);
433
+ const updatedFiles = (0, project_files_1.getExistingFiles)(allFilesChanged);
434
+ this.chatStorage.addNewFileState(updatedFiles);
435
+ this.rl.prompt();
436
+ }
437
+ async sendUserInputAndAwaitResponse() {
438
+ const userInputId = `mc-input-` + Math.random().toString(36).substring(2, 15);
439
+ const { responsePromise, stopResponse } = this.client.subscribeToResponse((chunk) => {
440
+ process.stdout.write(chunk);
441
+ }, userInputId, () => {
442
+ this.stopLoadingAnimation();
443
+ process.stdout.write((0, picocolors_1.green)((0, picocolors_1.underline)('\nCodebuff') + ':') + ' ');
444
+ });
445
+ this.stopResponse = stopResponse;
446
+ this.client.sendUserInput([], userInputId);
447
+ const result = await responsePromise;
448
+ this.stopResponse = null;
449
+ return result;
450
+ }
451
+ }
452
+ exports.CLI = CLI;
453
+ //# sourceMappingURL=cli.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"cli.js","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,mCAA6B;AAC7B,gEAA0D;AAC1D,iDAAkD;AAClD,mDAAoC;AACpC,2CAA0D;AAC1D,+BAA4B;AAE5B,qCAAuC;AACvC,iDAA4C;AAC5C,qCAAiC;AAEjC,iCAAoC;AACpC,mDAKwB;AACxB,mDAA0D;AAC1D,gDAIyB;AACzB,2CAAsE;AACtE,+CAA6E;AAE7E,yCAMwB;AACxB,+CAA8C;AAG9C,MAAa,GAAG;IACN,MAAM,CAAQ;IACd,WAAW,CAAa;IACxB,YAAY,CAAc;IAC1B,GAAG,CAAY;IACf,QAAQ,CAAU;IAClB,EAAE,CAAoB;IACtB,mBAAmB,GAAY,KAAK,CAAA;IACpC,YAAY,GAAwB,IAAI,CAAA;IACxC,eAAe,GAA0B,IAAI,CAAA;IAC7C,WAAW,GAAgB,EAAE,CAAA;IAC7B,cAAc,GAAW,CAAC,CAAA;IAE1B,aAAa,GAAW,CAAC,CAAA;IACzB,qBAAqB,GAAW,CAAC,CAAA;IACjC,aAAa,GAAW,EAAE,CAAA;IAC1B,SAAS,GAAY,KAAK,CAAA;IAElC,YACE,YAAiD,EACjD,EAAE,GAAG,EAAE,QAAQ,EAAc;QAE7B,IAAI,CAAC,GAAG,GAAG,GAAG,CAAA;QACd,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAA;QACxB,IAAI,CAAC,WAAW,GAAG,IAAI,0BAAW,EAAE,CAAA;QACpC,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,CAAC,IAAY,EAAE,EAAE;gBAC1B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,QAAQ;oBAAE,OAAO,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;gBAEzD,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAC9B,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,eAAe,CACxC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAA;gBAChC,MAAM,KAAK,GAAG,IAAA,mCAAe,EAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAA;gBAE/D,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,CAAA;gBAE5C,MAAM,cAAc,GAAG,CAAC,GAAG,UAAU,EAAE,GAAG,KAAK,CAAC,CAAC,MAAM,CACrD,CAAC,KAAK,EAAE,EAAE,CACR,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,GAAG,QAAQ,CAAC,CAC/D,CAAA;gBACD,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAA;YACnC,CAAC;SACF,CAAC,CAAA;QACF,IAAI,CAAC,MAAM,GAAG,IAAI,eAAM,CACtB,qBAAY,EACZ,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,EAChC,GAAG,EAAE;YACH,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAA;YAChB,IAAI,CAAC,mBAAmB,GAAG,KAAK,CAAA;YAChC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACtB,IAAI,CAAC,YAAY,EAAE,CAAA;YACrB,CAAC;YACD,IAAI,CAAC,oBAAoB,EAAE,CAAA;QAC7B,CAAC,EACD,IAAI,CAAC,QAAQ,CACd,CAAA;QAED,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC;YAC9B,YAAY,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC5B,MAAM,CAAC,CAAC,EAAE,WAAW,CAAC,GAAG,OAAO,CAAA;gBAChC,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAA;gBACzC,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,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YAC1B,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAA;QACxB,CAAC,CAAC,CAAA;QAEF,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE;YACxB,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC7B,IAAI,CAAC,kBAAkB,EAAE,CAAA;YAC3B,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;gBACtB,IAAI,GAAG,GAAG,IAAI,CAAC,cAAc,GAAG,IAAI,EAAE,CAAC;oBACrC,kBAAkB;oBAClB,IAAI,CAAC,UAAU,EAAE,CAAA;gBACnB,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,cAAc,GAAG,GAAG,CAAA;oBACzB,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAA;oBAC3C,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAA;gBAClB,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAA;QAEF,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACvB,IAAI,CAAC,UAAU,EAAE,CAAA;QACnB,CAAC,CAAC,CAAA;QAEF,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;YACzB,iBAAiB;YACjB,IAAI,CAAC,UAAU,EAAE,CAAA;QACnB,CAAC,CAAC,CAAA;QAEF,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE;YACtC,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC1B,IAAI,CAAC,YAAY,EAAE,CAAA;YACrB,CAAC;YACD,IAAI,CAAC,aAAa,EAAE,CAAA;QACtB,CAAC,CAAC,CAAA;IACJ,CAAC;IAEO,gBAAgB;QACtB,IAAI,CAAC,oBAAoB,EAAE,CAAA;QAC3B,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,aAAa;QACnB,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QAC9B,MAAM,QAAQ,GAAG,WAAW,GAAG,IAAI,CAAC,aAAa,CAAA;QAEjD,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;QAED,IAAI,CAAC,aAAa,GAAG,WAAW,CAAA;IAClC,CAAC;IAEO,WAAW,CAAC,IAAY;QAC9B,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,IAAI,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,CAAA;gBACxD,IAAI,CAAC,aAAa,GAAG,EAAE,CAAA;YACzB,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAA;YACnC,CAAC;QACH,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;IAEM,kBAAkB,CAAC,YAAqB;QAC7C,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACrB,OAAO,CAAC,GAAG,CACT,kBAAkB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,8BAA8B,CACtE,CAAA;QACH,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAA;QAC7C,CAAC;QACD,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAA;QAChB,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC,CAAA;YACzC,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,CAAA;QACpC,CAAC;IACH,CAAC;IAEO,UAAU;QAChB,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAA;QAChC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAA;IAClB,CAAC;IAEO,UAAU;QAChB,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAA;QAChC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAA;IAClB,CAAC;IAEO,mBAAmB,CAAC,SAA0B;QACpD,MAAM,cAAc,GAAG,IAAI,CAAC,WAAW,CAAC,iBAAiB,EAAE,CAAA;QAC3D,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;QACzE,MAAM,YAAY,GAAG,IAAA,gCAAgB,EAAC,SAAS,CAAC,CAAA;QAChD,IAAI,CAAC,WAAW,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAA;QAEnD,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,SAAS,CAAC,CAAA;QAE7D,IAAI,SAAS,EAAE,CAAC;YACd,OAAO,CAAC,GAAG,CACT,SAAS,KAAK,MAAM;gBAClB,CAAC,CAAC,IAAA,kBAAK,EAAC,kBAAkB,CAAC;gBAC3B,CAAC,CAAC,IAAA,kBAAK,EAAC,kBAAkB,CAAC,CAC9B,CAAA;YACD,MAAM,KAAK,GAAG,IAAI,CAAC,iCAAiC,EAAE,CAAA;YACtD,OAAO,CAAC,GAAG,CAAC,IAAA,kBAAK,EAAC,eAAe,CAAC,EAAE,IAAA,kBAAK,EAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC3E,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,IAAA,kBAAK,EAAC,WAAW,SAAS,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;QAC1E,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,IAAI,CAAC,oBAAoB,EAAE,CAAA;IAC7B,CAAC;IAEO,UAAU;QAChB,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;QACnB,OAAO,CAAC,GAAG,CACT,GAAG,IAAA,kBAAS,EAAC,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,QAAQ,CAAC,qBAAqB,CAC5E,CAAA;QACD,IACE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK;YACnB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK;YACnB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,EAC5B,CAAC;YACD,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAC7B,CAAC,EACD,IAAI,CAAC,KAAK,CACR,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACjD,CAAC,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CACxB,CACF,CAAA;YACD,OAAO,CAAC,GAAG,CACT,GAAG,IAAI,CAAC,GAAG,CACT,CAAC,EACD,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CACtC,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,iCAAiC,IAAA,kBAAS,EAAC,cAAc,EAAE,KAAK,CAAC,GAAG,CAC7F,CAAA;QACH,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,IAAA,kBAAK,EAAC,eAAe,CAAC,CAAC,CAAA;QACnC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACjB,CAAC;IAEO,YAAY;QAClB,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC7B,IAAI,CAAC,kBAAkB,EAAE,CAAA;QAC3B,CAAC;IACH,CAAC;IAEO,iCAAiC;QACvC,MAAM,cAAc,GAAG,IAAI,CAAC,WAAW,CAAC,iBAAiB,EAAE,CAAA;QAC3D,IAAI,cAAc,EAAE,CAAC;YACnB,IAAA,wBAAQ,EAAC,cAAc,CAAC,KAAK,CAAC,CAAA;YAC9B,OAAO,cAAc,CAAC,KAAK,CAAA;QAC7B,CAAC;QACD,OAAO,EAAE,CAAA;IACX,CAAC;IAEO,qBAAqB;QAC3B,MAAM,KAAK,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAA;QAChE,IAAI,CAAC,GAAG,CAAC,CAAA;QACT,IAAI,CAAC,eAAe,GAAG,WAAW,CAAC,GAAG,EAAE;YACtC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAA;YAC3B,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAA;YAC1B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAA,kBAAK,EAAC,GAAG,KAAK,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAA;YACtD,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,MAAM,CAAA;QAC5B,CAAC,EAAE,GAAG,CAAC,CAAA;IACT,CAAC;IAEO,oBAAoB;QAC1B,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;YACnC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAA;YAC3B,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAA;YAC3B,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAA;QAC5B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,iBAAiB;QAC7B,IAAI,IAAA,sBAAgB,GAAE,EAAE,CAAC;YACvB,MAAM,aAAa,GAAG,IAAA,sBAAgB,GAAE,CAAA;YACxC,IAAI,CAAC,aAAa;gBAAE,OAAM;YAE1B,MAAM,aAAa,GACjB,MAAM,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC,aAAa,CAAC,CAAA;YACxD,IAAA,mBAAa,EAAC,aAAa,CAAC,CAAA;YAC5B,OAAO,aAAa,CAAA;QACtB,CAAC;QACD,OAAO,SAAS,CAAA;IAClB,CAAC;IAEO,UAAU;QAChB,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClC,OAAO,CAAC,GAAG,CAAC,IAAA,mBAAM,EAAC,kDAAkD,CAAC,CAAC,CAAA;YACvE,OAAM;QACR,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YAClC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAA;YACjC,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO;iBACzB,KAAK,CAAC,IAAI,CAAC;iBACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAA;YAE9D,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBACrB,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;oBACzB,OAAO,CAAC,GAAG,CAAC,IAAA,kBAAK,EAAC,IAAI,CAAC,CAAC,CAAA;gBAC1B,CAAC;qBAAM,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;oBAChC,OAAO,CAAC,GAAG,CAAC,IAAA,gBAAG,EAAC,IAAI,CAAC,CAAC,CAAA;gBACxB,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;gBACnB,CAAC;YACH,CAAC,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;IACJ,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,SAAiB;QAC7C,IAAI,CAAC,SAAS;YAAE,OAAM;QACtB,SAAS,GAAG,SAAS,CAAC,IAAI,EAAE,CAAA;QAE5B,kBAAkB;QAClB,IAAI,SAAS,KAAK,MAAM,IAAI,SAAS,KAAK,GAAG,EAAE,CAAC;YAC9C,IAAA,kBAAW,GAAE,CAAA;YACb,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAA;YAChB,OAAM;QACR,CAAC;QACD,IAAI,SAAS,KAAK,OAAO,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;YACpD,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAA;YACzB,OAAM;QACR,CAAC;aAAM,IAAI,SAAS,KAAK,QAAQ,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC7D,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAA;YAC1B,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAA;YAChB,OAAM;QACR,CAAC;aAAM,IAAI,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YACxC,MAAM,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAA;YACtD,OAAM;QACR,CAAC;aAAM,IAAI,SAAS,KAAK,OAAO,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC5D,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAA;YACtB,OAAM;QACR,CAAC;aAAM,IAAI,SAAS,KAAK,MAAM,IAAI,SAAS,KAAK,GAAG,EAAE,CAAC;YACrD,IAAI,CAAC,UAAU,EAAE,CAAA;YACjB,OAAM;QACR,CAAC;aAAM,IAAI,SAAS,KAAK,MAAM,IAAI,SAAS,KAAK,GAAG,EAAE,CAAC;YACrD,IAAI,CAAC,UAAU,EAAE,CAAA;YACjB,OAAM;QACR,CAAC;aAAM,IACL,SAAS,KAAK,MAAM;YACpB,SAAS,KAAK,MAAM;YACpB,SAAS,KAAK,GAAG,EACjB,CAAC;YACD,IAAI,CAAC,UAAU,EAAE,CAAA;YACjB,OAAM;QACR,CAAC;aAAM,IACL,SAAS,KAAK,MAAM;YACpB,SAAS,KAAK,MAAM;YACpB,SAAS,KAAK,KAAK;YACnB,SAAS,KAAK,KAAK;YACnB,SAAS,KAAK,GAAG,EACjB,CAAC;YACD,IAAI,CAAC,UAAU,EAAE,CAAA;YACjB,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAA;YAChB,OAAM;QACR,CAAC;QAED,MAAM,SAAS,GAAG,OAAO,CAAA;QACzB,MAAM,YAAY,GAAG,SAAS,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;QACpD,IACE,YAAY;YACZ,CAAC,CAAC,qCAAyB,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAC3C,SAAS,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,CAC5C;gBACC,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBAC7B,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,EACnC,CAAC;YACD,MAAM,gBAAgB,GAAG,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAA;YACzD,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,IAAA,wCAAwB,EACvD,EAAE,OAAO,EAAE,gBAAgB,EAAE,EAC7B,MAAM,EACN,MAAM,CACP,CAAA;YACD,IAAI,MAAM,KAAK,mBAAmB,EAAE,CAAC;gBACnC,IAAI,CAAC,SAAS,EAAE,CAAA;gBAChB,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAA;gBAChB,OAAM;YACR,CAAC;iBAAM,IAAI,YAAY,EAAE,CAAC;gBACxB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;gBAC5B,IAAI,CAAC,SAAS,EAAE,CAAA;gBAChB,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAA;gBAChB,OAAM;YACR,CAAC;QACH,CAAC;QAED,IAAI,CAAC,qBAAqB,EAAE,CAAA;QAC5B,MAAM,IAAI,CAAC,YAAY,CAAA;QAEvB,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,cAAc,EAAE,CAAA;QACrD,MAAM,EAAE,YAAY,EAAE,GAAG,WAAW,CAAA;QACpC,MAAM,kBAAkB,GACtB,YAAY,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,KAAK,IAAI,EAAE,CAAA;QACpD,MAAM,2BAA2B,GAC/B,IAAA,8CAA8B,EAAC,kBAAkB,CAAC,CAAA;QACpD,MAAM,iBAAiB,GAAG,MAAM,CAAC,OAAO,CAAC,2BAA2B,CAAC;aAClE,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC;YAC1B,QAAQ;YACR,KAAK,CAAC,MAAM,GAAG,KAAK;gBAClB,CAAC,CAAC,KAAK;gBACP,CAAC,CAAC,2CAA2C;SAChD,CAAC;aACD,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,IAAA,sBAAe,EAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAA;QAC/D,MAAM,cAAc,GAClB,iBAAiB,CAAC,MAAM,GAAG,CAAC;YAC1B,CAAC,CAAC,iCAAiC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,qCAAqC;YACpG,CAAC,CAAC,EAAE,CAAA;QAER,MAAM,IAAI,GAAG,IAAA,kCAAoB,EAAC,SAAS,CAAC,CAAA;QAC5C,MAAM,aAAa,GAAG,MAAM,IAAA,qCAAuB,EAAC,IAAI,CAAC,CAAA;QACzD,MAAM,cAAc,GAClB,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,CAAA;QAErE,MAAM,UAAU,GAAY;YAC1B,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,GAAG,cAAc,GAAG,cAAc,GAAG,SAAS,EAAE;SAC1D,CAAA;QACD,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,WAAW,EAAE,UAAU,CAAC,CAAA;QAEpD,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAA;QAC/B,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,qBAAqB,EAAE,GAChD,MAAM,IAAI,CAAC,6BAA6B,EAAE,CAAA;QAC5C,IAAI,CAAC,mBAAmB,GAAG,KAAK,CAAA;QAEhC,IAAI,CAAC,oBAAoB,EAAE,CAAA;QAE3B,MAAM,UAAU,GAAG,CAAC,GAAG,qBAAqB,EAAE,GAAG,OAAO,CAAC,CAAA;QACzD,MAAM,YAAY,GAAG,IAAA,aAAI,EAAC,UAAU,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAA;QACtE,MAAM,eAAe,GAAG,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAA;QAEvE,yCAAyC;QACzC,IACE,IAAI,CAAC,GAAG,KAAK,OAAO;YACpB,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC;YAC3B,OAAO,CAAC,MAAM,GAAG,CAAC,EAClB,CAAC;YACD,MAAM,QAAQ,GAAG,IAAA,kBAAY,EAAC,IAAA,8BAAc,GAAE,EAAE,IAAI,CAAC,WAAW,CAAC,CAAA;YACjE,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO,CAAC,GAAG,CAAC,IAAA,kBAAK,EAAC,2BAA2B,CAAC,CAAC,CAAA;YACjD,CAAC;QACH,CAAC;QAED,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,IAAA,sBAAY,EAAC,IAAA,8BAAc,GAAE,EAAE,OAAO,CAAC,CAAA;QACrE,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9C,OAAO,CAAC,GAAG,EAAE,CAAA;QACf,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;YAC3B,OAAO,CAAC,GAAG,CAAC,IAAA,kBAAK,EAAC,aAAa,IAAI,EAAE,CAAC,CAAC,CAAA;QACzC,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;YAC5B,OAAO,CAAC,GAAG,CAAC,IAAA,kBAAK,EAAC,aAAa,IAAI,EAAE,CAAC,CAAC,CAAA;QACzC,CAAC;QACD,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9C,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,GAAG,yCAA6B,EAAE,CAAC;gBACnE,OAAO,CAAC,GAAG,CACT,KAAK,IAAA,kBAAS,EAAC,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,QAAQ,CAAC,yBAAyB,CAClF,CAAA;YACH,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,gDAAgD,CAAC,CAAA;YAC7D,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAA;QAChC,CAAC;QACD,OAAO,CAAC,GAAG,EAAE,CAAA;QAEb,IAAI,CAAC,WAAW,GAAG,UAAU,CAAA;QAE7B,MAAM,gBAAgB,GAAY;YAChC,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,QAAQ;SAClB,CAAA;QACD,IAAI,CAAC,WAAW,CAAC,UAAU,CACzB,IAAI,CAAC,WAAW,CAAC,cAAc,EAAE,EACjC,gBAAgB,CACjB,CAAA;QAED,MAAM,YAAY,GAAG,IAAA,gCAAgB,EAAC,eAAe,CAAC,CAAA;QACtD,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,YAAY,CAAC,CAAA;QAE9C,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAA;IAClB,CAAC;IAEO,KAAK,CAAC,6BAA6B;QACzC,MAAM,WAAW,GACf,WAAW,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QAE3D,MAAM,EAAE,eAAe,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,mBAAmB,CACvE,CAAC,KAAK,EAAE,EAAE;YACR,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;QAC7B,CAAC,EACD,WAAW,EACX,GAAG,EAAE;YACH,IAAI,CAAC,oBAAoB,EAAE,CAAA;YAC3B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAA,kBAAK,EAAC,IAAA,sBAAS,EAAC,YAAY,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC,CAAA;QAClE,CAAC,CACF,CAAA;QAED,IAAI,CAAC,YAAY,GAAG,YAAY,CAAA;QAChC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,EAAE,WAAW,CAAC,CAAA;QAE1C,MAAM,MAAM,GAAG,MAAM,eAAe,CAAA;QACpC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;QACxB,OAAO,MAAM,CAAA;IACf,CAAC;CACF;AAxfD,kBAwfC"}
@@ -0,0 +1,71 @@
1
+ import { User } from './common/util/credentials';
2
+ import { ChatStorage } from './chat-storage';
3
+ import { FileChanges, UsageResponse } from './common/actions';
4
+ import { type CostMode } from './common/constants';
5
+ import { FileVersion, ProjectFileContext } from './common/util/file';
6
+ export declare class Client {
7
+ private webSocket;
8
+ private chatStorage;
9
+ private currentUserInputId;
10
+ private returnControlToUser;
11
+ private fingerprintId;
12
+ private costMode;
13
+ fileVersions: FileVersion[][];
14
+ fileContext: ProjectFileContext | undefined;
15
+ user: User | undefined;
16
+ lastWarnedPct: number;
17
+ usage: number;
18
+ limit: number;
19
+ subscription_active: boolean;
20
+ lastRequestCredits: number;
21
+ sessionCreditsUsed: number;
22
+ nextQuotaReset: Date | null;
23
+ constructor(websocketUrl: string, chatStorage: ChatStorage, onWebSocketError: () => void, returnControlToUser: () => void, costMode: CostMode);
24
+ initFileVersions(projectFileContext: ProjectFileContext): void;
25
+ private getFingerprintId;
26
+ private getUser;
27
+ connect(): Promise<void>;
28
+ handleReferralCode(referralCode: string): Promise<void>;
29
+ logout(): Promise<void>;
30
+ login(referralCode?: string): Promise<void>;
31
+ setUsage({ usage, limit, subscription_active, next_quota_reset, referralLink, session_credits_used, }: Omit<UsageResponse, 'type'>): void;
32
+ private setupSubscriptions;
33
+ showUsageWarning(referralLink?: string): void;
34
+ generateCommitMessage(stagedChanges: string): Promise<string>;
35
+ sendUserInput(previousChanges: FileChanges, userInputId: string): Promise<void>;
36
+ subscribeToResponse(onChunk: (chunk: string) => void, userInputId: string, onStreamStart: () => void): {
37
+ responsePromise: Promise<{
38
+ type: "response-complete";
39
+ response: string;
40
+ userInputId: string;
41
+ changesAlreadyApplied: {
42
+ content: string;
43
+ type: "file" | "patch";
44
+ filePath: string;
45
+ }[];
46
+ changes: {
47
+ content: string;
48
+ type: "file" | "patch";
49
+ filePath: string;
50
+ }[];
51
+ addedFileVersions: {
52
+ path: string;
53
+ content: string;
54
+ }[];
55
+ resetFileVersions: boolean;
56
+ next_quota_reset?: Date | undefined;
57
+ subscription_active?: boolean | undefined;
58
+ limit?: number | undefined;
59
+ session_credits_used?: number | undefined;
60
+ usage?: number | undefined;
61
+ referralLink?: string | undefined;
62
+ } & {
63
+ type: "response-complete";
64
+ } & {
65
+ wasStoppedByUser: boolean;
66
+ }>;
67
+ stopResponse: () => void;
68
+ };
69
+ getUsage(): Promise<void>;
70
+ warmContextCache(): Promise<void>;
71
+ }