mocode-ai 0.1.3 → 0.1.5

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 (48) hide show
  1. package/README.md +196 -161
  2. package/dist/agent/core.js +302 -0
  3. package/dist/agent/index.js +85 -202
  4. package/dist/agent/mode.js +46 -0
  5. package/dist/agent/spawn.js +123 -0
  6. package/dist/commands/config.js +8 -26
  7. package/dist/config/file.js +48 -0
  8. package/dist/config/index.js +76 -22
  9. package/dist/llm/index.js +11 -9
  10. package/dist/memory/discover.js +49 -0
  11. package/dist/memory/index.js +45 -0
  12. package/dist/memory/reflect.js +265 -0
  13. package/dist/memory/store.js +339 -0
  14. package/dist/repl/index.js +334 -87
  15. package/dist/session/compact.js +51 -3
  16. package/dist/skills/index.js +2 -2
  17. package/dist/tools/builtins/ask-human.js +50 -0
  18. package/dist/tools/builtins/codegraph.js +113 -0
  19. package/dist/tools/builtins/edit-file.js +3 -3
  20. package/dist/tools/builtins/glob.js +3 -2
  21. package/dist/tools/builtins/grep.js +4 -3
  22. package/dist/tools/builtins/index.js +19 -1
  23. package/dist/tools/builtins/memory-forget.js +32 -0
  24. package/dist/tools/builtins/memory-list.js +34 -0
  25. package/dist/tools/builtins/memory-save.js +51 -0
  26. package/dist/tools/builtins/memory-search.js +42 -0
  27. package/dist/tools/builtins/memory-update.js +41 -0
  28. package/dist/tools/builtins/read-file.js +5 -4
  29. package/dist/tools/builtins/run-command.js +40 -7
  30. package/dist/tools/builtins/switch-mode.js +46 -0
  31. package/dist/tools/builtins/task.js +62 -0
  32. package/dist/tools/builtins/use-skill.js +2 -2
  33. package/dist/tools/builtins/web-fetch.js +16 -3
  34. package/dist/tools/builtins/web-search.js +6 -6
  35. package/dist/tools/builtins/write-file.js +3 -3
  36. package/dist/tools/constants.js +29 -0
  37. package/dist/tools/registry.js +11 -4
  38. package/dist/ui/content.js +25 -17
  39. package/dist/ui/diff.js +28 -26
  40. package/dist/ui/intervention.js +318 -0
  41. package/dist/ui/layout.js +402 -103
  42. package/dist/ui/markdown.js +607 -0
  43. package/dist/ui/mouse.js +93 -0
  44. package/dist/ui/prompt.js +469 -27
  45. package/dist/ui/render.js +19 -1
  46. package/dist/ui/spinner.js +15 -0
  47. package/dist/ui/theme.js +147 -13
  48. package/package.json +2 -2
@@ -1,17 +1,22 @@
1
1
  import readline from 'node:readline/promises';
2
2
  import { emitKeypressEvents } from 'node:readline';
3
3
  import { stdin, stdout } from 'node:process';
4
- import { config } from '../config/index.js';
4
+ import { config, PLAN_MODE_SUFFIX } from '../config/index.js';
5
+ import { updateConfigKey } from '../config/file.js';
5
6
  import { runAgent } from '../agent/index.js';
6
- import { ui } from '../ui/theme.js';
7
+ import { getAgentMode, setAgentMode, onModeChange } from '../agent/mode.js';
8
+ import { ui, setTheme, getTheme, listThemes, themeExists } from '../ui/theme.js';
7
9
  import { bannerString, displayWidth, padEndDisplay, summarizeToolCall, summarizeToolResult } from '../ui/render.js';
8
10
  import * as layout from '../ui/layout.js';
9
- import { promptWithSlashMenu, promptTurnPicker } from '../ui/prompt.js';
11
+ import * as mouse from '../ui/mouse.js';
12
+ import { promptWithSlashMenu, promptTurnPicker, promptSessionPicker, promptThemePicker, promptRevertChoice, } from '../ui/prompt.js';
13
+ import { promptIntervention } from '../ui/intervention.js';
10
14
  import { tools } from '../tools/registry.js';
11
15
  import { estimateMessagesTokens, estimateToolSchemaTokens, } from '../llm/index.js';
12
16
  import { compactHistory, contextState, newSessionId, saveSession, loadSession, listSessions, } from '../session/index.js';
13
17
  import { listTurns, planRollback, applyRollback, persistSnapshots, loadSnapshots, rebuildFromHistory, resetState, } from '../rollback/index.js';
14
18
  import { listSkills, effectiveSystemPrompt } from '../skills/index.js';
19
+ import { buildMemorySection, buildMemoryIndexSection, kickoffReflection, drainMemoryBackground, getLastReflectResult, clearLastReflectResult, snapshotTranscript, formatReflectResult, loadAll, } from '../memory/index.js';
15
20
  /**
16
21
  * readline 的 prompt 必须是纯文本(无 ANSI):readline 按字符数算光标位置,
17
22
  * 颜色码会让光标错位、编辑时漂移。颜色只用在直接 stdout.write 的横幅 / 工具行 / 回复。
@@ -25,9 +30,51 @@ const SLASH_COMMANDS = [
25
30
  { name: '/skills', desc: '列出已发现的 skill' },
26
31
  { name: '/compact', desc: '压缩历史(可带焦点 /compact …)' },
27
32
  { name: '/resume', desc: '续接已保存的会话' },
28
- { name: '/think', desc: '展开折叠思考段(/think N)' },
29
33
  { name: '/rollback', desc: '菜单选轮次回滚(↑↓·Enter)' },
34
+ { name: '/memory', desc: '记忆库:条目计数与近期索引' },
35
+ { name: '/reflect', desc: '手动触发后台记忆反思 pass' },
36
+ { name: '/init', desc: '扫描项目生成 MOCODE.md 项目记忆' },
37
+ { name: '/theme', desc: '切换颜色主题(↑↓·Enter)' },
38
+ { name: '/plan', desc: '切到 plan 模式(只读探查+产出计划)' },
39
+ { name: '/auto', desc: '切回 auto 模式(全工具执行)' },
30
40
  ];
41
+ /** 主题名 → 一句描述(供 /theme 菜单 / 列表显示)。新增主题时在 src/ui/theme.ts THEMES 加键后于此补一句。 */
42
+ const THEME_DESCRIPTIONS = {
43
+ default: '16 色原版(深底)',
44
+ light: '浅底终端',
45
+ solarized: 'Solarized 强调色',
46
+ gruvbox: 'Gruvbox 暖色',
47
+ nord: 'Nord 冷色',
48
+ };
49
+ /**
50
+ * /init 指令:发给 agent 扫描项目并生成 MOCODE.md(对标 Claude Code /init 生成 CLAUDE.md,
51
+ * 但 mocode 读 MOCODE.md)。已存在则让 agent 读后更新(不丢失事实)。写完供 memory 子系统下轮加载。
52
+ */
53
+ const INIT_PROMPT = `分析当前项目(process.cwd()),生成 MOCODE.md 项目记忆文件,供 mocode 后续会话自动加载——目标是让后续会话无需重新摸索就能上手。
54
+
55
+ 先探查(尽量少调用拿全貌):
56
+ - 若有 .codegraph/:用 codegraph 工具(explore "<架构或入口符号>")一次拿相关源码+调用路径,别逐文件读!!!
57
+ - read_file package.json(或 Cargo.toml/pyproject.toml/go.mod 等):scripts、依赖、入口、模块类型。
58
+ - glob 顶层目录;read_file 入口文件 + 各子系统 index.ts/README。
59
+ - 若 MOCODE.md 已存在:read_file 读它,在其基础上更新(补缺、修正过时),不丢已有准确事实。
60
+
61
+ MOCODE.md 按以下结构写(每节简短,只写稳定、非显然的事实):
62
+ ## 项目
63
+ 一两句:是什么、技术栈、运行环境。
64
+ ## 命令
65
+ install / dev / build / test / typecheck / lint 等——从 package.json scripts 提炼,写原样命令行(如 \`npm run typecheck\`);没有的注明"无测试"/"无 lint"。
66
+ ## 目录结构
67
+ 顶层各目录与子系统职责,一句话/个;不逐文件列。
68
+ ## 约定
69
+ 从代码与现有文档提炼的硬约定:模块系统(ESM?)、命名、错误处理、工具/函数契约、易踩坑点。只写非显然、会让人踩坑的;不写"保持简洁"这种正确废话。
70
+ ## 扩展点
71
+ 加工具/命令/provider/模块的接缝(改哪个文件、加在哪)。
72
+
73
+ 硬要求:
74
+ - 从实际代码提炼,引用具体文件名/命令/符号;不编造、不泛泛。
75
+ - 总长 ≤ 1500 字;只写后续会话有用的稳定事实,不写易变项(当前 bug、临时文件、未决 TODO)。
76
+ - 用 write_file 写入项目根 MOCODE.md。
77
+ - 写完简述:写了哪几节 + 从代码里发现的 2-3 条非显然关键约定(供用户校验)。`;
31
78
  /** 临时 readline 读一行(cooked,用于子提问;主输入走 promptWithSlashMenu)。 */
32
79
  async function askLine(prompt) {
33
80
  const rl = readline.createInterface({ input: stdin, output: stdout });
@@ -47,7 +94,7 @@ function renderContextBar(history) {
47
94
  const pct = Math.min(1, est / win);
48
95
  const W = 10;
49
96
  const filled = Math.round(pct * W);
50
- const bar = '█'.repeat(filled) + '░'.repeat(W - filled);
97
+ const bar = filled === 0 ? ' '.repeat(W) : '█'.repeat(filled) + '░'.repeat(W - filled);
51
98
  const src = contextState.lastUsage ? '实测' : '估算';
52
99
  const k = (n) => `${Math.round(n / 1000)}k`;
53
100
  const pctCol = pct >= config.compactThreshold ? ui.yellow : ui.cyan;
@@ -62,17 +109,18 @@ function renderContextBarInline(history) {
62
109
  const pct = Math.min(1, est / win);
63
110
  const W = 10;
64
111
  const filled = Math.round(pct * W);
65
- const bar = '█'.repeat(filled) + '░'.repeat(W - filled);
112
+ const bar = filled === 0 ? ' '.repeat(W) : '█'.repeat(filled) + '░'.repeat(W - filled);
66
113
  const k = (n) => `${Math.round(n / 1000)}k`;
67
114
  const pctCol = pct >= config.compactThreshold ? ui.yellow : ui.cyan;
68
115
  return `${ui.gray}[${pctCol}${bar}${ui.reset}] ${pctCol}${Math.round(pct * 100)}%${ui.reset} ${ui.dim}${k(est)}/${k(win)}${ui.reset}`;
69
116
  }
70
- /** 状态行基线:模型 / context / cwd。repl 在轮次边界调,刷新 context 用量。 */
117
+ /** 状态行基线:模型 / context / cwd / 模式标识。repl 在轮次边界与切模式时调,刷新 context 用量与 mode chip。 */
71
118
  function refreshStatusBase(history) {
72
119
  layout.setStatusBase({
73
120
  model: config.model,
74
121
  contextBar: renderContextBarInline(history),
75
122
  cwd: process.cwd(),
123
+ modeTag: getAgentMode() === 'plan' ? 'plan' : 'auto',
76
124
  });
77
125
  }
78
126
  /** 命令 → 运行态状态文字 + 底栏 dim 占位。 */
@@ -84,10 +132,21 @@ function runningStateFor(cmd) {
84
132
  return { status: '续接', placeholder: '选择会话…' };
85
133
  case '/rollback':
86
134
  return { status: '回滚', placeholder: '选择轮次…' };
135
+ case '/init':
136
+ return { status: '初始化', placeholder: '生成 MOCODE.md…' };
137
+ case '/plan':
138
+ return { status: '切 plan', placeholder: '…' };
139
+ case '/auto':
140
+ return { status: '切 auto', placeholder: '…' };
87
141
  case '/clear':
88
142
  return { status: '清空', placeholder: '…' };
143
+ case '/theme':
144
+ return { status: '切主题', placeholder: '选择主题…' };
89
145
  default:
90
- return { status: '处理', placeholder: '思考中… Ctrl+C 中断' };
146
+ // 输入框留空(运行中可 typeahead 打字,dim 回显);运行状态由内联 spinner 承载(思考中/执行…),
147
+ // 状态行只显走时——故常态 status 留空,不塞「处理」这种与内联重复的泛标签。
148
+ // 不把「思考中… Ctrl+C 中断」塞进输入框占位——避免看起来像已有输入、妨碍正常输入。
149
+ return { status: '', placeholder: '' };
91
150
  }
92
151
  }
93
152
  const emitter = stdin;
@@ -97,11 +156,20 @@ let runningInput = ''; // 运行中已打字缓冲(单行;agent 结束后预填
97
156
  let runningPlaceholder = '';
98
157
  let currentAbort = null;
99
158
  let pendingPrefill = null; // /rollback 选中后预填的 user 输入(下轮 INPUT 态消费)
159
+ // agent 模式状态已提到 src/agent/mode.ts(共享叶子:switch_mode 工具可写、agent 每步读、repl 注册 onModeChange 监听器)。
100
160
  /** 运行态按键:滚动优先,再 Ctrl+C 中断,再 typeahead 编辑(单行,Enter=无操作)。 */
101
161
  function onRunningKey(_str, key) {
102
162
  if (!key)
103
163
  return;
104
- // 滚动回看键(优先;不触发回尾):PgUp/PgDn 翻页,↑/↓ 单行(含鼠标滚轮——alt 屏滚轮转发↑↓)。
164
+ // SGR 鼠标滚轮:readline 把 \x1B[<btn;col;rowM 拆成多 fragment,经 mouse.consumeMouse 重组;
165
+ // 滚轮 → scrollBy(±5)(不调 setUserActive——与键盘滚动一致;流式 flushTimer 在 scrollOffset>0 时不重画,安全)。
166
+ const _m = mouse.consumeMouse(key.sequence ?? '');
167
+ if (_m.suppress) {
168
+ if (_m.wheel)
169
+ layout.scrollBy(_m.wheel * 5);
170
+ return;
171
+ }
172
+ // 滚动回看键(优先;不触发回尾):PgUp/PgDn 翻页,↑/↓ 每次 5 行(键盘;鼠标滚轮已由上方守卫处理)。
105
173
  // 运行态无输入光标,↑/↓ 无其他用途,直接作滚动。
106
174
  if (key.name === 'pageup' ||
107
175
  key.name === 'pagedown' ||
@@ -113,17 +181,27 @@ function onRunningKey(_str, key) {
113
181
  else if (key.name === 'pagedown')
114
182
  layout.scrollBy(-pageH);
115
183
  else if (key.name === 'up')
116
- layout.scrollBy(1);
184
+ layout.scrollBy(5);
117
185
  else
118
- layout.scrollBy(-1);
186
+ layout.scrollBy(-5);
119
187
  return;
120
188
  }
189
+ // 用户在交互(非滚动键)→ 暂停流式物理写,避免光标去 contentRow 扰动 IME 候选窗(停手后自动 flush)
190
+ layout.setUserActive();
121
191
  // 其他键:若处于滚动回看,先回尾再处理(打字即回底)
122
192
  if (layout.isScrolled())
123
193
  layout.resetScroll();
124
- // Ctrl+C 中断当前 agent 轮次(不退进程;raw 模式下 Ctrl+C 是按键,不触发 SIGINT)
194
+ // Ctrl+C 4 层语义(RUNNING 态):有 typeahead → 清空(层 1,不中断);空 → abort(层 2,中断 agent)。
195
+ // 两次 Ctrl+C 才中断(先清 typeahead 再 abort),与 INPUT 态 onCtrlC 的 fish 式一致。
196
+ // raw 模式下 Ctrl+C 是按键不触发 SIGINT;signal 经 executeTool 串进工具,run_command/web_fetch 即时被杀。
125
197
  if (key.ctrl && key.name === 'c') {
126
- currentAbort?.abort();
198
+ if (runningInput.length > 0) {
199
+ runningInput = '';
200
+ layout.paintRunningInputEcho(runningInput, runningPlaceholder);
201
+ }
202
+ else {
203
+ currentAbort?.abort();
204
+ }
127
205
  return;
128
206
  }
129
207
  const s = key.sequence ?? '';
@@ -177,6 +255,7 @@ function stopRunningListener() {
177
255
  * 把用户消息格式化为带满宽背景色的文本(上滑时易辨认用户消息)。
178
256
  * 每行用 padEndDisplay 填充到终端宽度(含 ❯ / 缩进),背景色 SGR 包裹整行 + 行末 reset。
179
257
  * 满宽 pad 使终端背景色覆盖整行(含行尾空单元格),上滑滚动时用户消息呈连续色块、与 assistant 正文区分。
258
+ * 末尾多留一空行(\n\n 收尾):用户消息与后续(agent 流式输出 / 下条消息)之间空一行,仿 Claude Code。
180
259
  */
181
260
  function formatUserMessage(lines) {
182
261
  const cols = layout.getGeo().cols;
@@ -190,7 +269,7 @@ function formatUserMessage(lines) {
190
269
  const padded = padEndDisplay(full, cols);
191
270
  return `${userBg}${padded}${reset}`;
192
271
  })
193
- .join('\n') + '\n');
272
+ .join('\n') + '\n\n');
194
273
  }
195
274
  /** 把多行提交输入回显进内容区(❯ 首行,续行按 prompt 宽度缩进)。仅 TUI 态回显(非 TTY 由 readline 自带回显)。 */
196
275
  function echoInput(lines) {
@@ -230,7 +309,7 @@ export function renderHistory(history) {
230
309
  if (m.role === 'assistant') {
231
310
  const text = textOf(m.content);
232
311
  if (text) {
233
- layout.contentWrite(text);
312
+ layout.contentWriteMdOnce(text);
234
313
  if (!text.endsWith('\n'))
235
314
  layout.contentWrite('\n');
236
315
  }
@@ -263,16 +342,23 @@ export function renderHistory(history) {
263
342
  * agent 只读取并追加(+ 经 session/ 压缩)。每轮成功结束后自动落盘,退出后可用 --resume / /resume 续接。
264
343
  */
265
344
  export async function startRepl(initialHistory, sessionId, updateNotice = null) {
345
+ // 模式重置:agentMode 不落盘,每个 REPL 会话从 auto 开始(/resume / --resume 亦重置)。
346
+ setAgentMode('auto');
347
+ // 构造系统提示:auto 用 base;plan 在 config.systemPrompt 后追加 PLAN_MODE_SUFFIX。
348
+ // 切模式时 applyMode 重算 history[0](history[0] 恒 system,compaction 保它,不破坏)。
349
+ const buildSystemMessage = (planMode) => effectiveSystemPrompt(config.systemPrompt +
350
+ (planMode ? PLAN_MODE_SUFFIX : '') +
351
+ buildMemorySection() +
352
+ buildMemoryIndexSection());
266
353
  // 有预加载(--resume)则用它,并把 history[0] 刷成当前 system prompt(config 可能已变);
267
- // 否则新会话只塞 system 提示。
268
- const systemPrompt = effectiveSystemPrompt(config.systemPrompt);
354
+ // 否则新会话只塞 system 提示(默认 auto)。
269
355
  const history = initialHistory && initialHistory.length
270
356
  ? initialHistory
271
- : [{ role: 'system', content: systemPrompt }];
357
+ : [{ role: 'system', content: buildSystemMessage(false) }];
272
358
  if (initialHistory &&
273
359
  initialHistory.length &&
274
360
  history[0]?.role === 'system') {
275
- history[0] = { role: 'system', content: systemPrompt };
361
+ history[0] = { role: 'system', content: buildSystemMessage(false) };
276
362
  }
277
363
  // --resume:读回该会话的轮次/快照;无文件则从 history 重建 turns(无快照→旧轮次文件改动不可撤销)
278
364
  if (sessionId && initialHistory && initialHistory.length) {
@@ -280,8 +366,8 @@ export async function startRepl(initialHistory, sessionId, updateNotice = null)
280
366
  rebuildFromHistory(history);
281
367
  }
282
368
  let currentSessionId = sessionId;
283
- // 本会话累积的折叠思考段,供 /think N 重打原文。
284
- const collapsedThinkings = [];
369
+ // 反思 cadence 计数:每 reflectEveryN 轮 fire-and-forget 一次后台反思 pass。
370
+ let turnCount = 0;
285
371
  const toolsLine = tools.map((t) => t.name).join(' · ');
286
372
  const banner = () => ({
287
373
  model: config.model,
@@ -289,7 +375,9 @@ export async function startRepl(initialHistory, sessionId, updateNotice = null)
289
375
  cwd: process.cwd(),
290
376
  tools: toolsLine,
291
377
  });
292
- // 开场:进 alt screen + 状态基线 + 清内容区。--resume 有历史则渲染对话(仿 Claude Code),否则横幅。
378
+ // 开场:按 config.theme 切主题(横幅 / 状态行 / 后续渲染皆用新色),再进 alt screen + 状态基线 + 清内容区。
379
+ // --resume 有历史则渲染对话(仿 Claude Code),否则横幅。
380
+ setTheme(config.theme);
293
381
  layout.enterAltScreen();
294
382
  refreshStatusBase(history);
295
383
  layout.clearContent();
@@ -304,10 +392,29 @@ export async function startRepl(initialHistory, sessionId, updateNotice = null)
304
392
  // 自更新提示:开场静态段(进 INPUT 态前),dim 一行,不与流式 / 输入争用。
305
393
  layout.contentWrite(` ${ui.gray}↳ ${updateNotice}${ui.reset}\n`);
306
394
  }
395
+ layout.contentWrite(`${ui.dim} /plan · /auto · Shift+Tab 切换模式(plan:只读探查 + 产出计划,审批后切 auto 执行)${ui.reset}\n`);
396
+ /**
397
+ * 切换 agent 模式(Shift+Tab 触发,经 prompt.ts 的 onCycleMode 回调)。
398
+ * cycleMode 翻 agentMode + applyMode 重写 history[0] + refreshStatusBase 设状态行 modeTag;
399
+ * 随后 prompt.ts 自调 redraw() 经 paintInput 把 chip 画出来(光标留输入框,不调 drawStatusBar)。
400
+ */
401
+ const applyMode = (planMode) => {
402
+ history[0] = { role: 'system', content: buildSystemMessage(planMode) };
403
+ };
404
+ const cycleMode = () => {
405
+ setAgentMode(getAgentMode() === 'plan' ? 'auto' : 'plan'); // listener 接手 applyMode + refreshStatusBase
406
+ };
407
+ // 注册模式变更监听器:switch_mode 工具 / cycleMode / /plan / /auto / runTurn 调 setAgentMode 时同步触发,
408
+ // 重写 history[0] 系统提示(切 plan 追加 PLAN_MODE_SUFFIX)+ 刷状态行 modeTag。
409
+ // 不调 drawStatusBar:INPUT 态靠 prompt.ts redraw 画 chip;RUNNING 态(switch_mode 中途切)靠 200ms turnTimer 兜底。
410
+ onModeChange((m) => {
411
+ applyMode(m === 'plan');
412
+ refreshStatusBase(history);
413
+ });
307
414
  /**
308
415
  * 回滚子流程(由 /rollback 触发):菜单(↑/↓)选轮次 → 选中第 X 轮 = 删第 X 轮及之后 + 预填第 X 轮 user 输入
309
- * (仿 Claude Code rewind,Enter 重新跑该轮);被删轮次的文件改动仍逐个「保留/撤销」询问(cooked readline)。
310
- * 选轮菜单走 promptTurnPicker(raw mode);文件询问走 askLine(cooked)。预填经 pendingPrefill 注入下轮 INPUT。
416
+ * (仿 Claude Code rewind,Enter 重新跑该轮);被删轮次的文件改动走二选一菜单(promptRevertChoice:
417
+ * 撤销文件 / 只撤销消息)。选轮 + 方式菜单均走 raw mode。预填经 pendingPrefill 注入下轮 INPUT。
311
418
  */
312
419
  const rollbackFlow = async () => {
313
420
  const turnList = listTurns();
@@ -345,24 +452,25 @@ export async function startRepl(initialHistory, sessionId, updateNotice = null)
345
452
  dim: true,
346
453
  });
347
454
  const revertPaths = new Set();
348
- for (const c of plan.changes) {
349
- layout.contentWrite(` ${ui.cyan}${c.path}${ui.reset} ${ui.dim}(${c.ops.join(', ')})${ui.reset}\n`);
350
- let ans = '';
455
+ // 文件撤销:二选一菜单(1=撤销文件改动 / 2=只撤销消息保留文件)
456
+ // 返回 true=撤销文件(把 changes 里所有可撤销的路径加入 revertPaths);
457
+ // false/null(取消/非 TTY/Ctrl+C)=只撤销消息,保留文件改动。
458
+ const changes = plan.changes;
459
+ const revertable = changes.filter((c) => c.snapshotAvailable);
460
+ let revertFiles = false;
461
+ if (changes.length > 0) {
351
462
  try {
352
- ans = (await askLine(' 保留/撤销 [k/u](回车=保留): ')).trim();
463
+ const choice = await promptRevertChoice(revertable.length);
464
+ revertFiles = choice === true;
353
465
  }
354
466
  catch {
355
- continue;
356
- }
357
- if (ans.startsWith('u') || ans.startsWith('U')) {
358
- if (c.snapshotAvailable) {
359
- revertPaths.add(c.path);
360
- }
361
- else {
362
- layout.contentWrite(`${ui.dim} (无快照,无法撤销——保留)${ui.reset}\n`);
363
- }
467
+ revertFiles = false; // Ctrl+C(SIGINT)→ 保留文件,只撤销消息
364
468
  }
365
469
  }
470
+ if (revertFiles) {
471
+ for (const c of revertable)
472
+ revertPaths.add(c.path);
473
+ }
366
474
  applyRollback(plan, history, revertPaths);
367
475
  if (!currentSessionId)
368
476
  currentSessionId = newSessionId();
@@ -378,15 +486,69 @@ export async function startRepl(initialHistory, sessionId, updateNotice = null)
378
486
  renderHistory(history);
379
487
  pendingPrefill = prefillText.split('\n');
380
488
  };
489
+ /**
490
+ * 跑一轮 agent(enterRunningMode 已由调用方完成):startRunningListener → runAgent → autosave / reflect。
491
+ * plan 模式传 planMode=true(runAgent 用 planChatTools 只读子集)。返 ok=正常结束(未中断 / 未抛错),
492
+ * 供调用方决定是否弹审批面板。execute 轮的合成输入也走这里。
493
+ */
494
+ const runTurn = async (input, planMode, placeholder) => {
495
+ let ok = false;
496
+ try {
497
+ const signal = startRunningListener(placeholder);
498
+ // 入口设定本轮初始模式(合成执行轮传 false→auto;用户轮传当前 mode)。
499
+ // setAgentMode 触发 listener 重写 history[0];LLM 可在轮中调 switch_mode 切模式,runAgent 每步读实时值。
500
+ setAgentMode(planMode ? 'plan' : 'auto');
501
+ // 运行中每步 chat() 返回后刷新状态行 context 用量条(用 fresh lastUsage / 估算),
502
+ // 否则整轮冻结在轮首 refreshStatusBase 的值,「执行 grep」时 2k/1000k 不动。
503
+ await runAgent(history, input, signal, () => {
504
+ refreshStatusBase(history);
505
+ layout.drawStatusBar();
506
+ });
507
+ ok = !signal.aborted; // 中断(Ctrl+C)→ runAgent 已还原 history,ok=false 不弹审批
508
+ // 成功轮次自动落盘(崩溃也保住上一轮);新会话首轮分配 id
509
+ if (!currentSessionId)
510
+ currentSessionId = newSessionId();
511
+ try {
512
+ saveSession(history, currentSessionId);
513
+ }
514
+ catch {
515
+ // 落盘失败不阻断 REPL
516
+ }
517
+ persistSnapshots(currentSessionId); // 随会话落盘回滚快照(/resume 后仍可撤销)
518
+ // 后台反思:每 reflectEveryN 轮 fire-and-forget 一次(与下一轮 agent 并发,不阻塞)。
519
+ // 已有在飞任务 / autoReflect 关 → kickoff 内部自守卫。快照同步取(避免下一轮 mutate history 竞态)。
520
+ turnCount++;
521
+ if (turnCount % config.reflectEveryN === 0) {
522
+ kickoffReflection(snapshotTranscript(history, 20));
523
+ }
524
+ }
525
+ catch (e) {
526
+ ok = false;
527
+ layout.contentWrite(`${ui.red}[错误]${ui.reset} ${e instanceof Error ? e.message : String(e)}\n`);
528
+ }
529
+ finally {
530
+ stopRunningListener();
531
+ }
532
+ layout.contentWrite('\n'); // 轮次之间空行
533
+ return ok;
534
+ };
381
535
  while (true) {
382
536
  // INPUT 态:画底栏输入框 + 状态行,光标入输入框
383
537
  refreshStatusBase(history);
538
+ // 后台反思若已完成(上轮 fire-and-forget),在安全点 flush 一行 dim 摘要:
539
+ // 不在 RUNNING 态写——防与正在跑的 agent 争屏(contentWrite / 状态行)。
540
+ const reflectRes = getLastReflectResult();
541
+ if (reflectRes) {
542
+ layout.contentWrite(` ${ui.gray}↳ ${formatReflectResult(reflectRes)}${ui.reset}\n`);
543
+ clearLastReflectResult();
544
+ }
384
545
  layout.enterInputMode('空闲');
385
546
  let input = null;
386
547
  try {
387
548
  input = await promptWithSlashMenu({
388
549
  prompt: PROMPT,
389
550
  commands: SLASH_COMMANDS,
551
+ onCycleMode: cycleMode,
390
552
  // /rollback 预填优先;否则上一轮运行中 typeahead 打的字 → 预填进输入框,用户可改可发
391
553
  ...(pendingPrefill
392
554
  ? { initialLines: pendingPrefill }
@@ -402,7 +564,7 @@ export async function startRepl(initialHistory, sessionId, updateNotice = null)
402
564
  runningInput = ''; // 预填已消费,清空(下轮运行态从空开始)
403
565
  if (input === null)
404
566
  break; // 空 prompt Ctrl+D
405
- const joined = input.join('\n');
567
+ let joined = input.join('\n');
406
568
  const line = joined.trim();
407
569
  if (!line)
408
570
  continue;
@@ -414,11 +576,37 @@ export async function startRepl(initialHistory, sessionId, updateNotice = null)
414
576
  const { status, placeholder } = runningStateFor(cmd);
415
577
  refreshStatusBase(history);
416
578
  layout.enterRunningMode(status, placeholder);
579
+ if (line === '/init') {
580
+ // /init:把 init 指令当 user 输入发给 agent(扫描项目 + 生成 MOCODE.md),fall through 走 runAgent
581
+ joined = INIT_PROMPT;
582
+ }
583
+ if (line === '/plan') {
584
+ // /plan:切到 plan 模式(只读探查 + 产出计划)。Shift+Tab 的可靠后备——
585
+ // 中文 IME(微软拼音用 Shift 切中英)与部分终端会吞掉 Shift+Tab 的 \x1b[Z,命令不受影响。
586
+ if (getAgentMode() === 'plan') {
587
+ layout.contentWrite(`${ui.dim}(已在 plan 模式)${ui.reset}\n`);
588
+ }
589
+ else {
590
+ setAgentMode('plan'); // listener 接手 applyMode + refreshStatusBase
591
+ layout.contentWrite(`${ui.dim}(已切到 plan 模式:只读探查 + 产出计划,审批后切 auto 执行。/auto 或 Shift+Tab 切回)${ui.reset}\n`);
592
+ }
593
+ continue;
594
+ }
595
+ if (line === '/auto') {
596
+ if (getAgentMode() === 'auto') {
597
+ layout.contentWrite(`${ui.dim}(已在 auto 模式)${ui.reset}\n`);
598
+ }
599
+ else {
600
+ setAgentMode('auto'); // listener 接手 applyMode + refreshStatusBase
601
+ layout.contentWrite(`${ui.dim}(已切回 auto 模式:全工具执行)${ui.reset}\n`);
602
+ }
603
+ continue;
604
+ }
417
605
  if (line === '/clear') {
418
606
  history.length = 1; // 保留 system 提示
419
- collapsedThinkings.length = 0; // 同步清空折叠的思考段
420
607
  resetState(); // 同步清空回滚轮次/快照
421
608
  currentSessionId = undefined; // 下轮起新会话文件
609
+ turnCount = 0; // 反思 cadence 重新计数
422
610
  contextState.lastUsage = undefined;
423
611
  layout.clearContent();
424
612
  layout.contentWrite(bannerString(banner()));
@@ -429,6 +617,35 @@ export async function startRepl(initialHistory, sessionId, updateNotice = null)
429
617
  layout.contentWrite(` ${renderContextBar(history)}\n`);
430
618
  continue;
431
619
  }
620
+ if (line === '/memory') {
621
+ // 记忆库概览:计数 + 近期 active 索引(详情用 memory_search)。
622
+ const all = loadAll();
623
+ const active = all.filter((e) => e.status === 'active');
624
+ const archived = all.filter((e) => e.status === 'archived').length;
625
+ const byType = {};
626
+ for (const e of active)
627
+ byType[e.type] = (byType[e.type] || 0) + 1;
628
+ layout.contentWrite(`${ui.dim}记忆库:active ${active.length}${archived ? ` · archived ${archived}` : ''}${ui.reset}\n`);
629
+ if (Object.keys(byType).length) {
630
+ layout.contentWrite(`${ui.dim}按类:${Object.entries(byType).map(([t, n]) => `${t} ${n}`).join(' ')}${ui.reset}\n`);
631
+ }
632
+ const recent = [...active]
633
+ .sort((a, b) => (b.updatedAt || '').localeCompare(a.updatedAt || ''))
634
+ .slice(0, 10);
635
+ for (const e of recent) {
636
+ layout.contentWrite(` ${ui.cyan}${e.id}${ui.reset} ${ui.dim}${e.name} — ${e.summary}${ui.reset}\n`);
637
+ }
638
+ if (active.length === 0)
639
+ layout.contentWrite(`${ui.dim}(无 active 记忆;用 memory_save 存,或 /init 生成 MOCODE.md)${ui.reset}\n`);
640
+ layout.contentWrite(`${ui.dim}(详情用 memory_search;启动索引已注入 systemPrompt)${ui.reset}\n`);
641
+ continue;
642
+ }
643
+ if (line === '/reflect') {
644
+ // 手动触发后台反思 pass(不等;完成后下次 INPUT 态显摘要)。
645
+ kickoffReflection(snapshotTranscript(history, 20));
646
+ layout.contentWrite(`${ui.dim}(反思已触发,后台进行;完成后下次输入态显示摘要。日志见 .mocode/memory.log)${ui.reset}\n`);
647
+ continue;
648
+ }
432
649
  if (line === '/skills') {
433
650
  const skills = listSkills();
434
651
  if (skills.length === 0) {
@@ -458,61 +675,97 @@ export async function startRepl(initialHistory, sessionId, updateNotice = null)
458
675
  continue;
459
676
  }
460
677
  if (line === '/resume') {
678
+ // /resume:打开会话菜单(↑/↓ 选,Enter 续接,Esc 取消)。默认仅最近 10 条,按 a 展开全部。
679
+ // 仿 /rollback 菜单化(promptSessionPicker);选中项 cyan+bold + ▸ 高亮。
461
680
  const sessions = listSessions();
462
681
  if (sessions.length === 0) {
463
682
  layout.contentWrite(`${ui.dim}(没有已保存的会话)${ui.reset}\n`);
464
683
  continue;
465
684
  }
466
- sessions.forEach((s, i) => {
467
- layout.contentWrite(` ${ui.dim}${i + 1}${ui.reset} ${s.id} ${ui.cyan}${s.firstUser || '(无)'}${ui.reset} ${ui.dim}${s.model}${ui.reset}\n`);
468
- });
469
- let pick = '';
685
+ const items = sessions.map((s) => ({
686
+ id: s.id,
687
+ title: s.firstUser || '(无)',
688
+ subtitle: `${s.id} ${s.model}`,
689
+ }));
690
+ let pick;
470
691
  try {
471
- pick = (await askLine('序号(回车取消): ')).trim();
692
+ pick = await promptSessionPicker(items);
472
693
  }
473
694
  catch {
474
- continue;
695
+ continue; // Ctrl+C(SIGINT)→ 取消
475
696
  }
476
- const idx = Number(pick);
477
- if (!pick || !Number.isInteger(idx) || idx < 1 || idx > sessions.length)
478
- continue;
479
- const loaded = loadSession(sessions[idx - 1].id);
697
+ if (!pick)
698
+ continue; // Esc / Ctrl+D 取消
699
+ const loaded = loadSession(pick.id);
480
700
  if (!loaded || !loaded.history.length) {
481
701
  layout.contentWrite(`${ui.yellow}(加载失败)${ui.reset}\n`);
482
702
  continue;
483
703
  }
484
704
  if (loaded.history[0]?.role === 'system') {
485
- loaded.history[0] = { role: 'system', content: systemPrompt };
705
+ loaded.history[0] = { role: 'system', content: buildSystemMessage(false) };
486
706
  }
487
707
  history.length = 0;
488
708
  history.push(...loaded.history);
709
+ setAgentMode('auto'); // /resume 重置为 auto(mode 不落盘;listener 重写 history[0] 回 auto,与 loaded 幂等)
489
710
  currentSessionId = loaded.id;
490
711
  // 读回该会话的轮次/快照;无文件则从 history 重建 turns(无快照→旧轮次文件改动不可撤销)
491
712
  if (!loadSnapshots(loaded.id))
492
713
  rebuildFromHistory(history);
493
714
  contextState.lastUsage = undefined;
494
- collapsedThinkings.length = 0;
495
715
  layout.clearContent();
496
716
  renderHistory(history);
497
717
  layout.contentWrite(`${ui.dim}(已续接会话 ${loaded.id})${ui.reset}\n`);
498
718
  continue;
499
719
  }
500
- if (line === '/think' || line.startsWith('/think ')) {
501
- const arg = line.split(/\s+/)[1];
502
- if (!arg) {
503
- layout.contentWrite(`${ui.dim}折叠思考段: ${collapsedThinkings.length} 段 · 用法: /think N (展开第 N 段)${ui.reset}\n`);
504
- continue;
720
+ if (line === '/theme' || line.startsWith('/theme ')) {
721
+ // /theme:无参开菜单(↑↓ 选,Enter 切换,Esc 取消);/theme <name> 直切;/theme list 或未知名 → 列出。
722
+ const arg = line.startsWith('/theme ') ? line.slice('/theme '.length).trim() : '';
723
+ let name;
724
+ if (arg === '') {
725
+ // 无参:菜单选(仿 /resume /rollback 的 picker)
726
+ const items = listThemes().map((t) => ({
727
+ id: t,
728
+ title: t,
729
+ subtitle: THEME_DESCRIPTIONS[t] ?? '',
730
+ }));
731
+ let pick;
732
+ try {
733
+ pick = await promptThemePicker(items);
734
+ }
735
+ catch {
736
+ continue; // Ctrl+C(SIGINT)→ 取消
737
+ }
738
+ name = pick?.id ?? null;
505
739
  }
506
- const idx = Number(arg);
507
- if (!Number.isInteger(idx) || idx < 1 || idx > collapsedThinkings.length) {
508
- layout.contentWrite(`${ui.yellow}无第 ${arg} 段(共 ${collapsedThinkings.length})${ui.reset}\n`);
740
+ else if (arg === 'list' || !themeExists(arg)) {
741
+ layout.contentWrite(`${ui.dim}可用主题:${ui.reset}\n`);
742
+ for (const t of listThemes()) {
743
+ layout.contentWrite(` ${ui.cyan}${t}${ui.reset} ${ui.dim}${THEME_DESCRIPTIONS[t] ?? ''}${ui.reset}\n`);
744
+ }
745
+ layout.contentWrite(`${ui.dim}(当前:${getTheme()};用 /theme <名称> 切换)${ui.reset}\n`);
509
746
  continue;
510
747
  }
511
- const content = collapsedThinkings[idx - 1];
512
- layout.contentWrite(`${ui.dim}▎ 思考 ▾ (第 ${idx} 段)${ui.reset}\n`);
513
- layout.contentWrite(`${ui.dim}${content}${ui.reset}\n`);
514
- if (!content.endsWith('\n'))
515
- layout.contentWrite('\n');
748
+ else {
749
+ name = arg;
750
+ }
751
+ if (name === null)
752
+ continue; // Esc / Ctrl+D 取消
753
+ // 切:setTheme → 重算状态行(新色)→ 清内容重绘(历史 / 横幅,镜像启动 + /resume)→ 确认 → 持久化。
754
+ // markdown MEMO 按 themeVersion 自动失效,故 renderHistory 取新色;状态栏 / 输入框由 continue 回 INPUT 态时读 getter 刷。
755
+ setTheme(name);
756
+ refreshStatusBase(history);
757
+ layout.clearContent();
758
+ if (history.some((m) => m.role === 'user')) {
759
+ renderHistory(history);
760
+ }
761
+ else {
762
+ layout.contentWrite(bannerString(banner()));
763
+ }
764
+ layout.contentWrite(`${ui.dim}(已切换主题 ${name})${ui.reset}\n`);
765
+ updateConfigKey('MOCODE_THEME', name);
766
+ if (config.themeFromShell) {
767
+ layout.contentWrite(`${ui.dim}(shell 环境变量 MOCODE_THEME 已设,文件写入下次启动被其覆盖;取消该 shell 设置后生效)${ui.reset}\n`);
768
+ }
516
769
  continue;
517
770
  }
518
771
  if (line === '/rollback' || line.startsWith('/rollback ')) {
@@ -521,32 +774,26 @@ export async function startRepl(initialHistory, sessionId, updateNotice = null)
521
774
  await rollbackFlow();
522
775
  continue;
523
776
  }
524
- try {
525
- const signal = startRunningListener(placeholder);
526
- // 运行中每步 chat() 返回后刷新状态行 context 用量条(用 fresh lastUsage / 估算),
527
- // 否则整轮冻结在轮首 refreshStatusBase 的值,「执行 grep」时 2k/1000k 不动。
528
- await runAgent(history, joined, collapsedThinkings, signal, () => {
529
- refreshStatusBase(history);
530
- layout.drawStatusBar();
777
+ const initialPlan = getAgentMode() === 'plan'; // 轮首模式(在 runTurn 之前读)
778
+ const ok = await runTurn(joined, initialPlan, placeholder);
779
+ // plan 轮正常结束(未中断 / 未抛错)→ 看轮末模式决定:
780
+ // - 仍 plan:LLM 没自切(只产计划就 STOP)→ 弹审批面板(原行为)。
781
+ // - 已 auto:LLM 调了 switch_mode('auto') 在同轮自主执行了 → 跳过审批,不重复打扰。
782
+ if (initialPlan && ok && getAgentMode() === 'plan') {
783
+ const res = await promptIntervention({
784
+ type: 'choice',
785
+ title: '计划已就绪',
786
+ detail: '切换到 auto 模式按上述计划执行?(plan 模式只读探查,执行需切 auto)',
787
+ options: ['切 auto 执行', '留 plan 细化'],
531
788
  });
532
- // 成功轮次自动落盘(崩溃也保住上一轮);新会话首轮分配 id
533
- if (!currentSessionId)
534
- currentSessionId = newSessionId();
535
- try {
536
- saveSession(history, currentSessionId);
537
- }
538
- catch {
539
- // 落盘失败不阻断 REPL
789
+ if (res.action === 'selected' && res.value === '切 auto 执行') {
790
+ // setAgentMode('auto') 由 runTurn 入口做(listener 重写 history[0] 回 auto);这里只切运行态 + 合成执行轮。
791
+ layout.enterRunningMode('执行', '按计划执行…');
792
+ await runTurn('请按上述计划执行。', false, '按计划执行…');
540
793
  }
541
- persistSnapshots(currentSessionId); // 随会话落盘回滚快照(/resume 后仍可撤销)
542
- }
543
- catch (e) {
544
- layout.contentWrite(`${ui.red}[错误]${ui.reset} ${e instanceof Error ? e.message : String(e)}\n`);
545
- }
546
- finally {
547
- stopRunningListener();
548
794
  }
549
- layout.contentWrite('\n'); // 轮次之间空行
550
795
  }
796
+ // 退出前等在飞反思收尾(Ctrl+C 走 SIGINT 直退不等;fire-and-forget 不承诺中断时完成)。
797
+ await drainMemoryBackground();
551
798
  layout.exitAltScreen();
552
799
  }