@zhongqian97-code/ecode 0.0.6 → 0.0.7
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.
- package/dist/index.js +663 -137
- package/package.json +5 -2
package/dist/index.js
CHANGED
|
@@ -3,6 +3,8 @@ const _ew=process.emitWarning.bind(process);process.emitWarning=function(w,...a)
|
|
|
3
3
|
|
|
4
4
|
// src/index.ts
|
|
5
5
|
import { createRequire } from "module";
|
|
6
|
+
import React4 from "react";
|
|
7
|
+
import { render } from "ink";
|
|
6
8
|
|
|
7
9
|
// src/config.ts
|
|
8
10
|
import { existsSync, readFileSync } from "fs";
|
|
@@ -11,7 +13,25 @@ import { join } from "path";
|
|
|
11
13
|
var DEFAULTS = {
|
|
12
14
|
baseUrl: "https://api.openai.com/v1",
|
|
13
15
|
apiKey: "",
|
|
14
|
-
model: "gpt-4o"
|
|
16
|
+
model: "gpt-4o",
|
|
17
|
+
dangerousPatterns: [
|
|
18
|
+
"rm -rf",
|
|
19
|
+
"sudo",
|
|
20
|
+
"chmod",
|
|
21
|
+
"chown",
|
|
22
|
+
"mkfs",
|
|
23
|
+
"dd",
|
|
24
|
+
"fdisk",
|
|
25
|
+
"kill",
|
|
26
|
+
"pkill",
|
|
27
|
+
"killall",
|
|
28
|
+
"reboot",
|
|
29
|
+
"shutdown",
|
|
30
|
+
"halt",
|
|
31
|
+
"curl -X DELETE",
|
|
32
|
+
"wget --delete-after"
|
|
33
|
+
],
|
|
34
|
+
logDir: void 0
|
|
15
35
|
};
|
|
16
36
|
function loadConfig() {
|
|
17
37
|
const configPath = join(homedir(), ".ecode", "config.json");
|
|
@@ -26,58 +46,17 @@ function loadConfig() {
|
|
|
26
46
|
return {
|
|
27
47
|
baseUrl: process.env.ECODE_BASE_URL ?? fileConfig.baseUrl ?? DEFAULTS.baseUrl,
|
|
28
48
|
apiKey: process.env.ECODE_API_KEY ?? fileConfig.apiKey ?? DEFAULTS.apiKey,
|
|
29
|
-
model: process.env.ECODE_MODEL ?? fileConfig.model ?? DEFAULTS.model
|
|
49
|
+
model: process.env.ECODE_MODEL ?? fileConfig.model ?? DEFAULTS.model,
|
|
50
|
+
// dangerousPatterns: 文件配置覆盖默认值,不支持环境变量(命令数组不适合通过环境变量传递)
|
|
51
|
+
dangerousPatterns: fileConfig.dangerousPatterns ?? DEFAULTS.dangerousPatterns,
|
|
52
|
+
// logDir: ECODE_LOG_DIR > 文件配置 > undefined
|
|
53
|
+
logDir: process.env.ECODE_LOG_DIR ?? fileConfig.logDir ?? DEFAULTS.logDir
|
|
30
54
|
};
|
|
31
55
|
}
|
|
32
56
|
|
|
33
|
-
// src/
|
|
34
|
-
import
|
|
35
|
-
|
|
36
|
-
// src/safety.ts
|
|
37
|
-
var ALLOWLIST = [
|
|
38
|
-
"ls",
|
|
39
|
-
"cat",
|
|
40
|
-
"pwd",
|
|
41
|
-
"echo",
|
|
42
|
-
"head",
|
|
43
|
-
"tail",
|
|
44
|
-
"wc",
|
|
45
|
-
"date",
|
|
46
|
-
"whoami",
|
|
47
|
-
"which",
|
|
48
|
-
"env",
|
|
49
|
-
"printenv"
|
|
50
|
-
];
|
|
51
|
-
var DANGER_LIST = [
|
|
52
|
-
"rm -rf",
|
|
53
|
-
"sudo",
|
|
54
|
-
"chmod",
|
|
55
|
-
"chown",
|
|
56
|
-
"mkfs",
|
|
57
|
-
"dd",
|
|
58
|
-
"fdisk",
|
|
59
|
-
"kill",
|
|
60
|
-
"pkill",
|
|
61
|
-
"killall",
|
|
62
|
-
"reboot",
|
|
63
|
-
"shutdown",
|
|
64
|
-
"halt",
|
|
65
|
-
"curl -X DELETE",
|
|
66
|
-
"wget --delete-after"
|
|
67
|
-
];
|
|
68
|
-
function matchesEntry(cmd, entry) {
|
|
69
|
-
return cmd === entry || cmd.startsWith(entry + " ");
|
|
70
|
-
}
|
|
71
|
-
function classifyCommand(cmd) {
|
|
72
|
-
const trimmed = cmd.trim();
|
|
73
|
-
for (const entry of DANGER_LIST) {
|
|
74
|
-
if (matchesEntry(trimmed, entry)) return "danger";
|
|
75
|
-
}
|
|
76
|
-
for (const entry of ALLOWLIST) {
|
|
77
|
-
if (matchesEntry(trimmed, entry)) return "allow";
|
|
78
|
-
}
|
|
79
|
-
return "normal";
|
|
80
|
-
}
|
|
57
|
+
// src/ui/App.tsx
|
|
58
|
+
import { useState as useState3, useCallback, useRef, useEffect as useEffect3 } from "react";
|
|
59
|
+
import { Box as Box4, useInput as useInput2 } from "ink";
|
|
81
60
|
|
|
82
61
|
// src/llm.ts
|
|
83
62
|
import OpenAI from "openai";
|
|
@@ -93,12 +72,15 @@ function createLLMClient(config2) {
|
|
|
93
72
|
const requestParams = {
|
|
94
73
|
model: config2.model,
|
|
95
74
|
messages,
|
|
96
|
-
stream: true
|
|
75
|
+
stream: true,
|
|
76
|
+
stream_options: { include_usage: true }
|
|
97
77
|
};
|
|
98
78
|
if (tools && tools.length > 0) {
|
|
99
79
|
requestParams.tools = tools;
|
|
100
80
|
}
|
|
101
|
-
const response = await openai.chat.completions.create(
|
|
81
|
+
const response = await openai.chat.completions.create(
|
|
82
|
+
requestParams
|
|
83
|
+
);
|
|
102
84
|
const tcAccumulator = /* @__PURE__ */ new Map();
|
|
103
85
|
let reasoningAccumulator = "";
|
|
104
86
|
for await (const chunk of response) {
|
|
@@ -125,12 +107,18 @@ function createLLMClient(config2) {
|
|
|
125
107
|
}
|
|
126
108
|
const isLast = choice.finish_reason !== null;
|
|
127
109
|
if (isLast) {
|
|
110
|
+
const rawUsage = chunk.usage;
|
|
128
111
|
yield {
|
|
129
112
|
text: delta.content ?? "",
|
|
130
113
|
done: true,
|
|
131
114
|
finishReason: choice.finish_reason,
|
|
132
115
|
toolCalls: tcAccumulator.size > 0 ? Array.from(tcAccumulator.values()) : void 0,
|
|
133
|
-
reasoning: reasoningAccumulator || void 0
|
|
116
|
+
reasoning: reasoningAccumulator || void 0,
|
|
117
|
+
usage: rawUsage ? {
|
|
118
|
+
promptTokens: rawUsage.prompt_tokens,
|
|
119
|
+
completionTokens: rawUsage.completion_tokens,
|
|
120
|
+
totalTokens: rawUsage.total_tokens
|
|
121
|
+
} : void 0
|
|
134
122
|
};
|
|
135
123
|
} else {
|
|
136
124
|
yield {
|
|
@@ -146,6 +134,56 @@ function createLLMClient(config2) {
|
|
|
146
134
|
};
|
|
147
135
|
}
|
|
148
136
|
|
|
137
|
+
// src/repl.ts
|
|
138
|
+
import * as readline from "readline/promises";
|
|
139
|
+
|
|
140
|
+
// src/safety.ts
|
|
141
|
+
var ALLOWLIST = [
|
|
142
|
+
"ls",
|
|
143
|
+
"cat",
|
|
144
|
+
"pwd",
|
|
145
|
+
"echo",
|
|
146
|
+
"head",
|
|
147
|
+
"tail",
|
|
148
|
+
"wc",
|
|
149
|
+
"date",
|
|
150
|
+
"whoami",
|
|
151
|
+
"which",
|
|
152
|
+
"env",
|
|
153
|
+
"printenv"
|
|
154
|
+
];
|
|
155
|
+
var DEFAULT_DANGER_LIST = [
|
|
156
|
+
"rm -rf",
|
|
157
|
+
"sudo",
|
|
158
|
+
"chmod",
|
|
159
|
+
"chown",
|
|
160
|
+
"mkfs",
|
|
161
|
+
"dd",
|
|
162
|
+
"fdisk",
|
|
163
|
+
"kill",
|
|
164
|
+
"pkill",
|
|
165
|
+
"killall",
|
|
166
|
+
"reboot",
|
|
167
|
+
"shutdown",
|
|
168
|
+
"halt",
|
|
169
|
+
"curl -X DELETE",
|
|
170
|
+
"wget --delete-after"
|
|
171
|
+
];
|
|
172
|
+
function matchesEntry(cmd, entry) {
|
|
173
|
+
return cmd === entry || cmd.startsWith(entry + " ");
|
|
174
|
+
}
|
|
175
|
+
function classifyCommand(cmd, dangerPatterns) {
|
|
176
|
+
const trimmed = cmd.trim();
|
|
177
|
+
const patterns = dangerPatterns ?? DEFAULT_DANGER_LIST;
|
|
178
|
+
for (const entry of patterns) {
|
|
179
|
+
if (matchesEntry(trimmed, entry)) return "danger";
|
|
180
|
+
}
|
|
181
|
+
for (const entry of ALLOWLIST) {
|
|
182
|
+
if (matchesEntry(trimmed, entry)) return "allow";
|
|
183
|
+
}
|
|
184
|
+
return "normal";
|
|
185
|
+
}
|
|
186
|
+
|
|
149
187
|
// src/tools/bash.ts
|
|
150
188
|
import { exec } from "child_process";
|
|
151
189
|
var DEFAULT_TIMEOUT_MS = 3e4;
|
|
@@ -179,12 +217,14 @@ var BASH_TOOL = {
|
|
|
179
217
|
}
|
|
180
218
|
};
|
|
181
219
|
async function handleBashTool(command, deps) {
|
|
182
|
-
const { confirm, print } = deps;
|
|
183
|
-
const cls = classifyCommand(command);
|
|
220
|
+
const { confirm, print, dangerousPatterns, autoApproveNormal } = deps;
|
|
221
|
+
const cls = classifyCommand(command, dangerousPatterns);
|
|
184
222
|
if (cls === "normal") {
|
|
185
|
-
|
|
223
|
+
if (!autoApproveNormal) {
|
|
224
|
+
const ok = await confirm(`Execute command: ${command}
|
|
186
225
|
Proceed? (y/n) `);
|
|
187
|
-
|
|
226
|
+
if (!ok) return SKIP_MESSAGE;
|
|
227
|
+
}
|
|
188
228
|
} else if (cls === "danger") {
|
|
189
229
|
print(`\u26A0\uFE0F DANGEROUS COMMAND: ${command}`);
|
|
190
230
|
const first = await confirm("Are you sure? (y/n) ");
|
|
@@ -202,107 +242,593 @@ Proceed? (y/n) `);
|
|
|
202
242
|
[exit code: ${result.exitCode}]`;
|
|
203
243
|
return output || "(no output)";
|
|
204
244
|
}
|
|
205
|
-
|
|
206
|
-
|
|
207
|
-
|
|
208
|
-
|
|
209
|
-
|
|
210
|
-
|
|
211
|
-
const
|
|
212
|
-
const
|
|
213
|
-
|
|
214
|
-
|
|
215
|
-
|
|
245
|
+
|
|
246
|
+
// src/logger.ts
|
|
247
|
+
import * as fs from "fs";
|
|
248
|
+
import * as path from "path";
|
|
249
|
+
function createLogger(logDir, sessionStart) {
|
|
250
|
+
fs.mkdirSync(logDir, { recursive: true });
|
|
251
|
+
const filename = sessionStart.toISOString().replace(/:/g, "-").replace(/\..+/, "") + ".jsonl";
|
|
252
|
+
const filePath = path.join(logDir, filename);
|
|
253
|
+
return {
|
|
254
|
+
filePath,
|
|
255
|
+
append(entry) {
|
|
256
|
+
try {
|
|
257
|
+
fs.appendFileSync(filePath, JSON.stringify(entry) + "\n");
|
|
258
|
+
} catch (err) {
|
|
259
|
+
process.stderr.write(`[logger] Failed to write log entry: ${err}
|
|
260
|
+
`);
|
|
261
|
+
}
|
|
262
|
+
}
|
|
216
263
|
};
|
|
217
|
-
|
|
218
|
-
|
|
219
|
-
|
|
220
|
-
|
|
221
|
-
|
|
222
|
-
|
|
223
|
-
|
|
224
|
-
|
|
225
|
-
|
|
226
|
-
|
|
227
|
-
|
|
228
|
-
|
|
229
|
-
|
|
230
|
-
|
|
231
|
-
|
|
232
|
-
|
|
233
|
-
|
|
234
|
-
|
|
235
|
-
|
|
236
|
-
|
|
237
|
-
|
|
238
|
-
|
|
239
|
-
|
|
264
|
+
}
|
|
265
|
+
|
|
266
|
+
// src/ui/StatusBar.tsx
|
|
267
|
+
import { useEffect, useState } from "react";
|
|
268
|
+
import { Box, Text } from "ink";
|
|
269
|
+
import { Fragment, jsx, jsxs } from "react/jsx-runtime";
|
|
270
|
+
var SPINNER_FRAMES = ["\u280B", "\u2819", "\u2839", "\u2838", "\u283C", "\u2834", "\u2826", "\u2827", "\u2807", "\u280F"];
|
|
271
|
+
var SPINNER_INTERVAL_MS = 80;
|
|
272
|
+
function formatTokenCount(n) {
|
|
273
|
+
if (n < 1e3) return String(n);
|
|
274
|
+
return (n / 1e3).toFixed(1) + "k";
|
|
275
|
+
}
|
|
276
|
+
function buildStatusLabel(status, toolName, confirmPrompt) {
|
|
277
|
+
if (confirmPrompt) {
|
|
278
|
+
return `[y/n] ${confirmPrompt}`;
|
|
279
|
+
}
|
|
280
|
+
switch (status) {
|
|
281
|
+
case "thinking":
|
|
282
|
+
return "\u601D\u8003\u4E2D";
|
|
283
|
+
case "tool_calling":
|
|
284
|
+
return `\u8C03\u7528\u5DE5\u5177: ${toolName ?? ""}`;
|
|
285
|
+
case "awaiting_confirm":
|
|
286
|
+
return "\u7B49\u5F85\u786E\u8BA4";
|
|
287
|
+
case "idle":
|
|
288
|
+
default:
|
|
289
|
+
return "\u7B49\u5F85\u8F93\u5165";
|
|
290
|
+
}
|
|
291
|
+
}
|
|
292
|
+
function statusIcon(status, confirmPrompt) {
|
|
293
|
+
if (confirmPrompt) return "";
|
|
294
|
+
switch (status) {
|
|
295
|
+
case "thinking":
|
|
296
|
+
return "\u27F3";
|
|
297
|
+
case "tool_calling":
|
|
298
|
+
return "\u2699";
|
|
299
|
+
case "awaiting_confirm":
|
|
300
|
+
case "idle":
|
|
301
|
+
default:
|
|
302
|
+
return "\u25CE";
|
|
303
|
+
}
|
|
304
|
+
}
|
|
305
|
+
function StatusBar({
|
|
306
|
+
status,
|
|
307
|
+
toolName,
|
|
308
|
+
confirmPrompt,
|
|
309
|
+
version: version2,
|
|
310
|
+
tokenUsage
|
|
311
|
+
}) {
|
|
312
|
+
const [spinnerIdx, setSpinnerIdx] = useState(0);
|
|
313
|
+
const isAnimating = !confirmPrompt && (status === "thinking" || status === "tool_calling");
|
|
314
|
+
useEffect(() => {
|
|
315
|
+
if (!isAnimating) return;
|
|
316
|
+
const timer = setInterval(() => {
|
|
317
|
+
setSpinnerIdx((prev) => (prev + 1) % SPINNER_FRAMES.length);
|
|
318
|
+
}, SPINNER_INTERVAL_MS);
|
|
319
|
+
return () => clearInterval(timer);
|
|
320
|
+
}, [isAnimating]);
|
|
321
|
+
const label = buildStatusLabel(status, toolName, confirmPrompt);
|
|
322
|
+
const icon = statusIcon(status, confirmPrompt);
|
|
323
|
+
const spinner = isAnimating ? SPINNER_FRAMES[spinnerIdx] : " ";
|
|
324
|
+
const usedStr = formatTokenCount(tokenUsage.used);
|
|
325
|
+
const limitStr = formatTokenCount(tokenUsage.limit);
|
|
326
|
+
const ctxStr = `ctx: ${tokenUsage.estimated ? "~" : ""}${usedStr}/${limitStr}`;
|
|
327
|
+
const versionStr = `v${version2}`;
|
|
328
|
+
return (
|
|
329
|
+
// justifyContent="space-between" 让 Ink 自动撑开左右,
|
|
330
|
+
// 避免用 string.length 手动计算填充(CJK 双宽字符会导致计算偏差)
|
|
331
|
+
/* @__PURE__ */ jsxs(Box, { flexDirection: "row", justifyContent: "space-between", children: [
|
|
332
|
+
/* @__PURE__ */ jsxs(Box, { flexDirection: "row", children: [
|
|
333
|
+
/* @__PURE__ */ jsxs(Text, { color: "gray", children: [
|
|
334
|
+
spinner,
|
|
335
|
+
" "
|
|
336
|
+
] }),
|
|
337
|
+
confirmPrompt ? (
|
|
338
|
+
// 确认模式:整体用黄色显示
|
|
339
|
+
/* @__PURE__ */ jsxs(Text, { color: "yellow", children: [
|
|
340
|
+
icon,
|
|
341
|
+
label
|
|
342
|
+
] })
|
|
343
|
+
) : status === "thinking" || status === "tool_calling" ? (
|
|
344
|
+
// 进行中状态:icon 用绿色,文字用白色
|
|
345
|
+
/* @__PURE__ */ jsxs(Fragment, { children: [
|
|
346
|
+
/* @__PURE__ */ jsxs(Text, { color: "green", children: [
|
|
347
|
+
icon,
|
|
348
|
+
" "
|
|
349
|
+
] }),
|
|
350
|
+
/* @__PURE__ */ jsx(Text, { color: "white", children: label })
|
|
351
|
+
] })
|
|
352
|
+
) : (
|
|
353
|
+
// 空闲 / 等待确认:淡灰
|
|
354
|
+
/* @__PURE__ */ jsxs(Fragment, { children: [
|
|
355
|
+
/* @__PURE__ */ jsxs(Text, { color: "gray", children: [
|
|
356
|
+
icon,
|
|
357
|
+
" "
|
|
358
|
+
] }),
|
|
359
|
+
/* @__PURE__ */ jsx(Text, { color: "gray", children: label })
|
|
360
|
+
] })
|
|
361
|
+
)
|
|
362
|
+
] }),
|
|
363
|
+
/* @__PURE__ */ jsxs(Box, { flexDirection: "row", children: [
|
|
364
|
+
/* @__PURE__ */ jsxs(Text, { color: "gray", children: [
|
|
365
|
+
versionStr,
|
|
366
|
+
" "
|
|
367
|
+
] }),
|
|
368
|
+
/* @__PURE__ */ jsx(Text, { color: tokenUsage.estimated ? "yellow" : "cyan", children: ctxStr })
|
|
369
|
+
] })
|
|
370
|
+
] })
|
|
371
|
+
);
|
|
372
|
+
}
|
|
373
|
+
|
|
374
|
+
// src/ui/ConversationHistory.tsx
|
|
375
|
+
import { Box as Box2, Text as Text2 } from "ink";
|
|
376
|
+
import { jsx as jsx2, jsxs as jsxs2 } from "react/jsx-runtime";
|
|
377
|
+
var TOOL_RESULT_MAX_LINES = 3;
|
|
378
|
+
function truncateLines(text, maxLines) {
|
|
379
|
+
const lines = text.split("\n");
|
|
380
|
+
if (lines.length <= maxLines) return text;
|
|
381
|
+
return lines.slice(0, maxLines).join("\n") + "\n\u2026";
|
|
382
|
+
}
|
|
383
|
+
function UserMessage({
|
|
384
|
+
content
|
|
385
|
+
}) {
|
|
386
|
+
return (
|
|
387
|
+
// 用户消息使用 ">" 前缀,白色,与助手消息对比
|
|
388
|
+
/* @__PURE__ */ jsx2(Box2, { flexDirection: "row", marginBottom: 0, children: /* @__PURE__ */ jsxs2(Text2, { color: "white", children: [
|
|
389
|
+
"> ",
|
|
390
|
+
content
|
|
391
|
+
] }) })
|
|
392
|
+
);
|
|
393
|
+
}
|
|
394
|
+
function AssistantMessage({
|
|
395
|
+
content,
|
|
396
|
+
tool_calls,
|
|
397
|
+
reasoning_content,
|
|
398
|
+
expandTools
|
|
399
|
+
}) {
|
|
400
|
+
return /* @__PURE__ */ jsxs2(Box2, { flexDirection: "column", marginBottom: 0, children: [
|
|
401
|
+
reasoning_content && reasoning_content.length > 0 && (expandTools ? /* @__PURE__ */ jsxs2(Box2, { flexDirection: "column", children: [
|
|
402
|
+
/* @__PURE__ */ jsx2(Text2, { color: "gray", children: "<thinking>" }),
|
|
403
|
+
/* @__PURE__ */ jsx2(Text2, { dimColor: true, children: reasoning_content })
|
|
404
|
+
] }) : /* @__PURE__ */ jsx2(Text2, { color: "gray", children: "[+ thinking]" })),
|
|
405
|
+
content && content.trim().length > 0 && /* @__PURE__ */ jsx2(Text2, { color: "cyan", children: content }),
|
|
406
|
+
tool_calls && tool_calls.length > 0 && (expandTools ? tool_calls.map((tc, idx) => {
|
|
407
|
+
let argsDisplay = tc.function.arguments;
|
|
408
|
+
try {
|
|
409
|
+
const parsed = JSON.parse(tc.function.arguments);
|
|
410
|
+
if (typeof parsed === "object" && parsed !== null) {
|
|
411
|
+
argsDisplay = Object.values(parsed).map(String).join(", ");
|
|
240
412
|
}
|
|
413
|
+
} catch {
|
|
414
|
+
}
|
|
415
|
+
return /* @__PURE__ */ jsxs2(Text2, { color: "yellow", children: [
|
|
416
|
+
"\u2699 \u8C03\u7528\u5DE5\u5177: ",
|
|
417
|
+
tc.function.name,
|
|
418
|
+
"(",
|
|
419
|
+
argsDisplay,
|
|
420
|
+
")"
|
|
421
|
+
] }, idx);
|
|
422
|
+
}) : /* @__PURE__ */ jsxs2(Text2, { color: "gray", children: [
|
|
423
|
+
"[+ ",
|
|
424
|
+
tool_calls.length,
|
|
425
|
+
" \u4E2A\u5DE5\u5177\u8C03\u7528]"
|
|
426
|
+
] }))
|
|
427
|
+
] });
|
|
428
|
+
}
|
|
429
|
+
function ToolMessage({
|
|
430
|
+
content,
|
|
431
|
+
expandTools
|
|
432
|
+
}) {
|
|
433
|
+
if (!expandTools) {
|
|
434
|
+
return /* @__PURE__ */ jsx2(Box2, { flexDirection: "row", marginBottom: 0, children: /* @__PURE__ */ jsx2(Text2, { color: "gray", children: "[+ \u5DE5\u5177\u7ED3\u679C]" }) });
|
|
435
|
+
}
|
|
436
|
+
const truncated = truncateLines(content, TOOL_RESULT_MAX_LINES);
|
|
437
|
+
return /* @__PURE__ */ jsx2(Box2, { flexDirection: "row", marginBottom: 0, children: /* @__PURE__ */ jsxs2(Text2, { color: "gray", children: [
|
|
438
|
+
"[\u5DE5\u5177\u7ED3\u679C] ",
|
|
439
|
+
truncated
|
|
440
|
+
] }) });
|
|
441
|
+
}
|
|
442
|
+
function ConversationHistory({
|
|
443
|
+
messages,
|
|
444
|
+
maxHeight = 20,
|
|
445
|
+
expandTools = false
|
|
446
|
+
}) {
|
|
447
|
+
const visible = messages.filter(
|
|
448
|
+
(m) => m.role !== "system"
|
|
449
|
+
);
|
|
450
|
+
function estimateLines(msg) {
|
|
451
|
+
if (msg.role === "user") {
|
|
452
|
+
return Math.max(1, msg.content.split("\n").length);
|
|
453
|
+
}
|
|
454
|
+
if (msg.role === "assistant") {
|
|
455
|
+
const contentLines = msg.content ? msg.content.split("\n").length : 0;
|
|
456
|
+
const toolLines = msg.tool_calls?.length ?? 0;
|
|
457
|
+
return Math.max(1, contentLines + toolLines);
|
|
458
|
+
}
|
|
459
|
+
if (msg.role === "tool") {
|
|
460
|
+
return Math.min(TOOL_RESULT_MAX_LINES, msg.content.split("\n").length) + 1;
|
|
461
|
+
}
|
|
462
|
+
return 1;
|
|
463
|
+
}
|
|
464
|
+
let totalLines = 0;
|
|
465
|
+
let startIdx = visible.length;
|
|
466
|
+
for (let i = visible.length - 1; i >= 0; i--) {
|
|
467
|
+
const lines = estimateLines(visible[i]);
|
|
468
|
+
if (totalLines + lines > maxHeight) break;
|
|
469
|
+
totalLines += lines;
|
|
470
|
+
startIdx = i;
|
|
471
|
+
}
|
|
472
|
+
const displayMessages = visible.slice(startIdx);
|
|
473
|
+
return /* @__PURE__ */ jsx2(Box2, { flexDirection: "column", children: displayMessages.map((msg, idx) => {
|
|
474
|
+
if (msg.role === "user") {
|
|
475
|
+
return /* @__PURE__ */ jsx2(UserMessage, { content: msg.content }, idx);
|
|
476
|
+
}
|
|
477
|
+
if (msg.role === "assistant") {
|
|
478
|
+
const assistantMsg = msg;
|
|
479
|
+
return /* @__PURE__ */ jsx2(
|
|
480
|
+
AssistantMessage,
|
|
481
|
+
{
|
|
482
|
+
content: assistantMsg.content,
|
|
483
|
+
tool_calls: assistantMsg.tool_calls,
|
|
484
|
+
reasoning_content: assistantMsg.reasoning_content,
|
|
485
|
+
expandTools
|
|
486
|
+
},
|
|
487
|
+
idx
|
|
488
|
+
);
|
|
489
|
+
}
|
|
490
|
+
if (msg.role === "tool") {
|
|
491
|
+
return /* @__PURE__ */ jsx2(ToolMessage, { content: msg.content, expandTools }, idx);
|
|
492
|
+
}
|
|
493
|
+
return null;
|
|
494
|
+
}) });
|
|
495
|
+
}
|
|
496
|
+
|
|
497
|
+
// src/ui/Input.tsx
|
|
498
|
+
import { useState as useState2, useEffect as useEffect2 } from "react";
|
|
499
|
+
import { Box as Box3, Text as Text3, useInput } from "ink";
|
|
500
|
+
import { jsx as jsx3, jsxs as jsxs3 } from "react/jsx-runtime";
|
|
501
|
+
var CURSOR_CHAR = "\u258C";
|
|
502
|
+
var BLINK_INTERVAL_MS = 530;
|
|
503
|
+
function Input({ isActive, onSubmit, placeholder }) {
|
|
504
|
+
const [lines, setLines] = useState2([""]);
|
|
505
|
+
const [cursorVisible, setCursorVisible] = useState2(true);
|
|
506
|
+
useEffect2(() => {
|
|
507
|
+
if (!isActive) {
|
|
508
|
+
setCursorVisible(true);
|
|
509
|
+
return;
|
|
510
|
+
}
|
|
511
|
+
const timer = setInterval(() => {
|
|
512
|
+
setCursorVisible((prev) => !prev);
|
|
513
|
+
}, BLINK_INTERVAL_MS);
|
|
514
|
+
return () => {
|
|
515
|
+
clearInterval(timer);
|
|
516
|
+
};
|
|
517
|
+
}, [isActive]);
|
|
518
|
+
useInput(
|
|
519
|
+
(input, key) => {
|
|
520
|
+
if (key.return && key.shift) {
|
|
521
|
+
setLines((prev) => [...prev, ""]);
|
|
522
|
+
return;
|
|
241
523
|
}
|
|
242
|
-
if (
|
|
243
|
-
|
|
244
|
-
|
|
245
|
-
|
|
246
|
-
|
|
247
|
-
|
|
248
|
-
|
|
249
|
-
|
|
250
|
-
|
|
251
|
-
|
|
524
|
+
if (key.return) {
|
|
525
|
+
const text = lines.join("\n");
|
|
526
|
+
onSubmit(text);
|
|
527
|
+
setLines([""]);
|
|
528
|
+
return;
|
|
529
|
+
}
|
|
530
|
+
if (key.backspace || key.delete) {
|
|
531
|
+
setLines((prev) => {
|
|
532
|
+
const next = [...prev];
|
|
533
|
+
const lastIdx = next.length - 1;
|
|
534
|
+
const lastLine = next[lastIdx];
|
|
535
|
+
if (lastLine.length > 0) {
|
|
536
|
+
next[lastIdx] = lastLine.slice(0, -1);
|
|
537
|
+
return next;
|
|
538
|
+
}
|
|
539
|
+
if (next.length > 1) {
|
|
540
|
+
return next.slice(0, -1);
|
|
541
|
+
}
|
|
542
|
+
return next;
|
|
252
543
|
});
|
|
253
|
-
|
|
254
|
-
|
|
255
|
-
|
|
256
|
-
|
|
257
|
-
|
|
258
|
-
|
|
259
|
-
|
|
260
|
-
|
|
261
|
-
|
|
262
|
-
|
|
263
|
-
|
|
264
|
-
|
|
265
|
-
|
|
266
|
-
|
|
267
|
-
|
|
268
|
-
|
|
269
|
-
|
|
544
|
+
return;
|
|
545
|
+
}
|
|
546
|
+
if (key.ctrl || key.escape || key.upArrow || key.downArrow || key.leftArrow || key.rightArrow || key.tab || key.pageUp || key.pageDown) {
|
|
547
|
+
return;
|
|
548
|
+
}
|
|
549
|
+
if (input.length > 0) {
|
|
550
|
+
setLines((prev) => {
|
|
551
|
+
const next = [...prev];
|
|
552
|
+
next[next.length - 1] = next[next.length - 1] + input;
|
|
553
|
+
return next;
|
|
554
|
+
});
|
|
555
|
+
}
|
|
556
|
+
},
|
|
557
|
+
{ isActive }
|
|
558
|
+
);
|
|
559
|
+
const isEmpty = lines.every((line) => line === "");
|
|
560
|
+
const renderLines = () => {
|
|
561
|
+
if (isEmpty && placeholder) {
|
|
562
|
+
return /* @__PURE__ */ jsxs3(Box3, { children: [
|
|
563
|
+
/* @__PURE__ */ jsx3(Text3, { color: "cyan", children: "> " }),
|
|
564
|
+
/* @__PURE__ */ jsx3(Text3, { dimColor: true, children: placeholder }),
|
|
565
|
+
isActive && cursorVisible && /* @__PURE__ */ jsx3(Text3, { color: "cyan", children: CURSOR_CHAR })
|
|
566
|
+
] });
|
|
567
|
+
}
|
|
568
|
+
return /* @__PURE__ */ jsx3(Box3, { flexDirection: "column", children: lines.map((line, idx) => {
|
|
569
|
+
const isLastLine = idx === lines.length - 1;
|
|
570
|
+
const prefix = idx === 0 ? "> " : " ";
|
|
571
|
+
return /* @__PURE__ */ jsxs3(Box3, { children: [
|
|
572
|
+
/* @__PURE__ */ jsx3(Text3, { color: "cyan", children: prefix }),
|
|
573
|
+
/* @__PURE__ */ jsx3(Text3, { children: line }),
|
|
574
|
+
isActive && isLastLine && cursorVisible && /* @__PURE__ */ jsx3(Text3, { color: "cyan", children: CURSOR_CHAR })
|
|
575
|
+
] }, idx);
|
|
576
|
+
}) });
|
|
577
|
+
};
|
|
578
|
+
return /* @__PURE__ */ jsx3(Box3, { children: isActive ? renderLines() : /* @__PURE__ */ jsxs3(Box3, { children: [
|
|
579
|
+
/* @__PURE__ */ jsx3(Text3, { dimColor: true, children: "> " }),
|
|
580
|
+
/* @__PURE__ */ jsx3(Text3, { dimColor: true, children: isEmpty ? placeholder ?? "" : lines.join(" ") })
|
|
581
|
+
] }) });
|
|
582
|
+
}
|
|
583
|
+
|
|
584
|
+
// src/ui/App.tsx
|
|
585
|
+
import { jsx as jsx4, jsxs as jsxs4 } from "react/jsx-runtime";
|
|
586
|
+
function App({ config: config2, version: version2, autoMode: autoMode2 = false }) {
|
|
587
|
+
const [messages, setMessages] = useState3([]);
|
|
588
|
+
const [status, setStatus] = useState3("idle");
|
|
589
|
+
const [tokenUsage, setTokenUsage] = useState3({
|
|
590
|
+
used: 0,
|
|
591
|
+
estimated: true,
|
|
592
|
+
limit: 128e3
|
|
593
|
+
});
|
|
594
|
+
const [toolName, setToolName] = useState3(void 0);
|
|
595
|
+
const [confirmPrompt, setConfirmPrompt] = useState3(void 0);
|
|
596
|
+
const [expandTools, setExpandTools] = useState3(false);
|
|
597
|
+
const pendingConfirmRef = useRef(null);
|
|
598
|
+
const llmRef = useRef(createLLMClient(config2));
|
|
599
|
+
const loggerRef = useRef(null);
|
|
600
|
+
const loggedCountRef = useRef(0);
|
|
601
|
+
useEffect3(() => {
|
|
602
|
+
if (config2.logDir) {
|
|
603
|
+
loggerRef.current = createLogger(config2.logDir, /* @__PURE__ */ new Date());
|
|
604
|
+
}
|
|
605
|
+
}, []);
|
|
606
|
+
useEffect3(() => {
|
|
607
|
+
if (!loggerRef.current) return;
|
|
608
|
+
for (let i = loggedCountRef.current; i < messages.length; i++) {
|
|
609
|
+
const msg = messages[i];
|
|
610
|
+
loggerRef.current.append({
|
|
611
|
+
ts: (/* @__PURE__ */ new Date()).toISOString(),
|
|
612
|
+
role: msg.role,
|
|
613
|
+
content: typeof msg.content === "string" ? msg.content : null,
|
|
614
|
+
tool_call_id: "tool_call_id" in msg ? msg.tool_call_id : void 0,
|
|
615
|
+
tool_calls: "tool_calls" in msg ? msg.tool_calls : void 0
|
|
616
|
+
});
|
|
617
|
+
}
|
|
618
|
+
loggedCountRef.current = messages.length;
|
|
619
|
+
}, [messages]);
|
|
620
|
+
useInput2((_input, key) => {
|
|
621
|
+
if (key.tab) {
|
|
622
|
+
setExpandTools((prev) => !prev);
|
|
623
|
+
}
|
|
624
|
+
});
|
|
625
|
+
const confirm = useCallback((prompt) => {
|
|
626
|
+
return new Promise((resolve) => {
|
|
627
|
+
setStatus("awaiting_confirm");
|
|
628
|
+
setConfirmPrompt(prompt);
|
|
629
|
+
pendingConfirmRef.current = { resolve };
|
|
630
|
+
});
|
|
631
|
+
}, []);
|
|
632
|
+
const runLlmLoop = useCallback(
|
|
633
|
+
async (history) => {
|
|
634
|
+
const print = (_text) => {
|
|
635
|
+
};
|
|
636
|
+
const deps = {
|
|
637
|
+
executeBash,
|
|
638
|
+
confirm,
|
|
639
|
+
print,
|
|
640
|
+
dangerousPatterns: config2.dangerousPatterns,
|
|
641
|
+
autoApproveNormal: autoMode2
|
|
642
|
+
};
|
|
643
|
+
let currentMessages = history;
|
|
644
|
+
let continueLoop = true;
|
|
645
|
+
while (continueLoop) {
|
|
646
|
+
continueLoop = false;
|
|
647
|
+
setStatus("thinking");
|
|
648
|
+
let assistantText = "";
|
|
649
|
+
let assistantReasoning;
|
|
650
|
+
const toolCalls = [];
|
|
651
|
+
for await (const chunk of llmRef.current.stream(currentMessages, [BASH_TOOL])) {
|
|
652
|
+
if (chunk.text) {
|
|
653
|
+
assistantText += chunk.text;
|
|
654
|
+
setMessages((prev) => {
|
|
655
|
+
const last = prev[prev.length - 1];
|
|
656
|
+
if (last?.role === "assistant" && !last.tool_calls) {
|
|
657
|
+
return [...prev.slice(0, -1), { ...last, content: assistantText }];
|
|
658
|
+
}
|
|
659
|
+
return [...prev, { role: "assistant", content: assistantText }];
|
|
270
660
|
});
|
|
271
661
|
}
|
|
662
|
+
if (chunk.done) {
|
|
663
|
+
if (chunk.toolCalls) toolCalls.push(...chunk.toolCalls);
|
|
664
|
+
if (chunk.reasoning) assistantReasoning = chunk.reasoning;
|
|
665
|
+
if (chunk.usage) {
|
|
666
|
+
setTokenUsage({
|
|
667
|
+
used: chunk.usage.totalTokens,
|
|
668
|
+
estimated: false,
|
|
669
|
+
limit: 128e3
|
|
670
|
+
});
|
|
671
|
+
}
|
|
672
|
+
} else {
|
|
673
|
+
const estimatedUsed = Math.floor(
|
|
674
|
+
currentMessages.reduce(
|
|
675
|
+
(acc, m) => acc + (typeof m.content === "string" ? m.content.length : 0),
|
|
676
|
+
0
|
|
677
|
+
) * 0.25 + assistantText.length * 0.25
|
|
678
|
+
);
|
|
679
|
+
setTokenUsage((prev) => ({ ...prev, used: estimatedUsed, estimated: true }));
|
|
680
|
+
}
|
|
272
681
|
}
|
|
273
|
-
|
|
274
|
-
|
|
275
|
-
process.stdout.write("\n");
|
|
276
|
-
if (assistantText) {
|
|
277
|
-
messages.push({
|
|
682
|
+
if (toolCalls.length > 0) {
|
|
683
|
+
const assistantMsg = {
|
|
278
684
|
role: "assistant",
|
|
279
|
-
content: assistantText,
|
|
685
|
+
content: assistantText || null,
|
|
686
|
+
tool_calls: toolCalls.map((tc) => ({
|
|
687
|
+
id: tc.id,
|
|
688
|
+
type: "function",
|
|
689
|
+
function: { name: tc.name, arguments: tc.arguments }
|
|
690
|
+
})),
|
|
280
691
|
...assistantReasoning ? { reasoning_content: assistantReasoning } : {}
|
|
692
|
+
};
|
|
693
|
+
setMessages((prev) => {
|
|
694
|
+
const last = prev[prev.length - 1];
|
|
695
|
+
const withoutStreaming = last?.role === "assistant" && !last.tool_calls ? prev.slice(0, -1) : prev;
|
|
696
|
+
return [...withoutStreaming, assistantMsg];
|
|
281
697
|
});
|
|
698
|
+
currentMessages = [...currentMessages, assistantMsg];
|
|
699
|
+
setStatus("tool_calling");
|
|
700
|
+
for (const tc of toolCalls) {
|
|
701
|
+
if (tc.name === "bash") {
|
|
702
|
+
let args;
|
|
703
|
+
try {
|
|
704
|
+
args = JSON.parse(tc.arguments);
|
|
705
|
+
} catch {
|
|
706
|
+
args = { command: "" };
|
|
707
|
+
}
|
|
708
|
+
setToolName(tc.name);
|
|
709
|
+
const toolResult = await handleBashTool(args.command, deps);
|
|
710
|
+
const toolMsg = {
|
|
711
|
+
role: "tool",
|
|
712
|
+
tool_call_id: tc.id,
|
|
713
|
+
content: toolResult
|
|
714
|
+
};
|
|
715
|
+
setMessages((prev) => [...prev, toolMsg]);
|
|
716
|
+
currentMessages = [...currentMessages, toolMsg];
|
|
717
|
+
}
|
|
718
|
+
}
|
|
719
|
+
continueLoop = true;
|
|
720
|
+
} else {
|
|
721
|
+
setMessages((prev) => {
|
|
722
|
+
const last = prev[prev.length - 1];
|
|
723
|
+
if (last?.role === "assistant" && !last.tool_calls) {
|
|
724
|
+
const updated = {
|
|
725
|
+
role: "assistant",
|
|
726
|
+
content: assistantText,
|
|
727
|
+
...assistantReasoning ? { reasoning_content: assistantReasoning } : {}
|
|
728
|
+
};
|
|
729
|
+
return [...prev.slice(0, -1), updated];
|
|
730
|
+
}
|
|
731
|
+
return prev;
|
|
732
|
+
});
|
|
733
|
+
if (assistantText) {
|
|
734
|
+
currentMessages = [
|
|
735
|
+
...currentMessages,
|
|
736
|
+
{
|
|
737
|
+
role: "assistant",
|
|
738
|
+
content: assistantText,
|
|
739
|
+
...assistantReasoning ? { reasoning_content: assistantReasoning } : {}
|
|
740
|
+
}
|
|
741
|
+
];
|
|
742
|
+
}
|
|
282
743
|
}
|
|
283
744
|
}
|
|
284
|
-
|
|
285
|
-
|
|
286
|
-
|
|
287
|
-
|
|
745
|
+
setStatus("idle");
|
|
746
|
+
setToolName(void 0);
|
|
747
|
+
},
|
|
748
|
+
[confirm, config2.dangerousPatterns, autoMode2]
|
|
749
|
+
);
|
|
750
|
+
const handleSubmit = useCallback(
|
|
751
|
+
(text) => {
|
|
752
|
+
const trimmed = text.trim();
|
|
753
|
+
if (status === "awaiting_confirm") {
|
|
754
|
+
const pending = pendingConfirmRef.current;
|
|
755
|
+
if (pending) {
|
|
756
|
+
pendingConfirmRef.current = null;
|
|
757
|
+
setConfirmPrompt(void 0);
|
|
758
|
+
setStatus("tool_calling");
|
|
759
|
+
pending.resolve(trimmed.toLowerCase() === "y");
|
|
760
|
+
}
|
|
761
|
+
return;
|
|
762
|
+
}
|
|
763
|
+
if (!trimmed) return;
|
|
764
|
+
const userMsg = { role: "user", content: trimmed };
|
|
765
|
+
const nextMessages = [...messages, userMsg];
|
|
766
|
+
setMessages(nextMessages);
|
|
767
|
+
runLlmLoop(nextMessages).catch((err) => {
|
|
768
|
+
setStatus("idle");
|
|
769
|
+
setToolName(void 0);
|
|
770
|
+
setMessages((prev) => [
|
|
771
|
+
...prev,
|
|
772
|
+
{ role: "assistant", content: `[error] ${String(err)}` }
|
|
773
|
+
]);
|
|
774
|
+
});
|
|
775
|
+
},
|
|
776
|
+
[status, messages, runLlmLoop]
|
|
777
|
+
);
|
|
778
|
+
const isInputActive = status === "idle" || status === "awaiting_confirm";
|
|
779
|
+
return /* @__PURE__ */ jsxs4(Box4, { flexDirection: "column", height: "100%", children: [
|
|
780
|
+
/* @__PURE__ */ jsx4(ConversationHistory, { messages, expandTools }),
|
|
781
|
+
/* @__PURE__ */ jsx4(
|
|
782
|
+
StatusBar,
|
|
783
|
+
{
|
|
784
|
+
status,
|
|
785
|
+
toolName,
|
|
786
|
+
confirmPrompt,
|
|
787
|
+
version: version2,
|
|
788
|
+
tokenUsage
|
|
789
|
+
}
|
|
790
|
+
),
|
|
791
|
+
/* @__PURE__ */ jsx4(
|
|
792
|
+
Input,
|
|
793
|
+
{
|
|
794
|
+
isActive: isInputActive,
|
|
795
|
+
onSubmit: handleSubmit,
|
|
796
|
+
placeholder: status === "awaiting_confirm" ? "y / n" : void 0
|
|
797
|
+
}
|
|
798
|
+
)
|
|
799
|
+
] });
|
|
288
800
|
}
|
|
289
801
|
|
|
290
802
|
// src/index.ts
|
|
291
803
|
var require2 = createRequire(import.meta.url);
|
|
292
804
|
var { version } = require2("../package.json");
|
|
293
|
-
var
|
|
294
|
-
|
|
295
|
-
|
|
296
|
-
|
|
805
|
+
var VERSION = version;
|
|
806
|
+
var rawArgs = process.argv.slice(2);
|
|
807
|
+
var autoMode = false;
|
|
808
|
+
var cliLogDir;
|
|
809
|
+
for (let i = 0; i < rawArgs.length; i++) {
|
|
810
|
+
const arg = rawArgs[i];
|
|
811
|
+
if (arg === "-v" || arg === "--version") {
|
|
812
|
+
console.log(VERSION);
|
|
813
|
+
process.exit(0);
|
|
814
|
+
}
|
|
815
|
+
if (arg === "--auto") {
|
|
816
|
+
autoMode = true;
|
|
817
|
+
}
|
|
818
|
+
if (arg === "--log-dir") {
|
|
819
|
+
const next = rawArgs[i + 1];
|
|
820
|
+
if (next && !next.startsWith("-")) {
|
|
821
|
+
cliLogDir = next;
|
|
822
|
+
i++;
|
|
823
|
+
}
|
|
824
|
+
}
|
|
297
825
|
}
|
|
298
826
|
var config = loadConfig();
|
|
299
|
-
|
|
827
|
+
var finalConfig = cliLogDir ? { ...config, logDir: cliLogDir } : config;
|
|
828
|
+
if (!finalConfig.apiKey) {
|
|
300
829
|
console.error(
|
|
301
830
|
"Error: no API key configured.\nSet ECODE_API_KEY or add apiKey to ~/.ecode/config.json"
|
|
302
831
|
);
|
|
303
832
|
process.exit(1);
|
|
304
833
|
}
|
|
305
|
-
|
|
306
|
-
console.error("Fatal error:", err);
|
|
307
|
-
process.exit(1);
|
|
308
|
-
});
|
|
834
|
+
render(React4.createElement(App, { config: finalConfig, version: VERSION, autoMode }));
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@zhongqian97-code/ecode",
|
|
3
|
-
"version": "0.0.
|
|
3
|
+
"version": "0.0.7",
|
|
4
4
|
"description": "A minimal Claude Code clone with REPL interface and bash tool calling",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"author": "zhongqian97-code",
|
|
@@ -39,7 +39,10 @@
|
|
|
39
39
|
"test:watch": "vitest"
|
|
40
40
|
},
|
|
41
41
|
"dependencies": {
|
|
42
|
-
"
|
|
42
|
+
"@types/react": "^19.2.14",
|
|
43
|
+
"ink": "^7.0.2",
|
|
44
|
+
"openai": "^4.0.0",
|
|
45
|
+
"react": "^19.2.5"
|
|
43
46
|
},
|
|
44
47
|
"devDependencies": {
|
|
45
48
|
"@types/node": "^18.0.0",
|