kumo-cli 1.0.4 → 1.0.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.
@@ -0,0 +1,113 @@
1
+ import fs from 'node:fs';
2
+ import path from 'node:path';
3
+ import { readSessionHistory } from './sessionService.js';
4
+ import { safePath } from '../utils/safePath.js';
5
+ export function buildRollbackResult(sessionId, targetMessageId, history = readSessionHistory(sessionId)) {
6
+ const targetIndex = history.findIndex((m) => m.id === targetMessageId);
7
+ if (targetIndex < 0)
8
+ throw new Error('rollback target message not found');
9
+ const target = history[targetIndex];
10
+ if (target.role !== 'user')
11
+ throw new Error('rollback target must be a user message');
12
+ const cutoff = findTurnEndIndex(history, targetIndex);
13
+ return {
14
+ sessionId,
15
+ targetMessageId,
16
+ restoredPrompt: target.content,
17
+ messages: history.slice(0, cutoff + 1),
18
+ };
19
+ }
20
+ export function rollbackChatTurnOnDisk(sessionId, targetMessageId, history = readSessionHistory(sessionId)) {
21
+ const plan = buildRollbackPlan(sessionId, targetMessageId, history);
22
+ for (const operation of plan.operations)
23
+ operation();
24
+ return plan.result;
25
+ }
26
+ function buildRollbackPlan(sessionId, targetMessageId, history) {
27
+ const targetIndex = history.findIndex((m) => m.id === targetMessageId);
28
+ if (targetIndex < 0)
29
+ throw new Error('rollback target message not found');
30
+ const target = history[targetIndex];
31
+ if (target.role !== 'user')
32
+ throw new Error('rollback target must be a user message');
33
+ const cutoff = findTurnEndIndex(history, targetIndex);
34
+ const laterMessages = history.slice(cutoff + 1);
35
+ const operations = [];
36
+ const restoredFiles = new Set();
37
+ for (let i = laterMessages.length - 1; i >= 0; i--) {
38
+ const message = laterMessages[i];
39
+ for (const tool of [...(message.toolUses ?? [])].reverse()) {
40
+ addToolRollbackOperation(tool, operations, restoredFiles);
41
+ }
42
+ }
43
+ return {
44
+ result: {
45
+ sessionId,
46
+ targetMessageId,
47
+ restoredPrompt: target.content,
48
+ messages: history.slice(0, cutoff + 1),
49
+ restoredFiles: [...restoredFiles],
50
+ },
51
+ operations,
52
+ };
53
+ }
54
+ function findTurnEndIndex(history, userIndex) {
55
+ let end = history.length - 1;
56
+ for (let i = userIndex + 1; i < history.length; i++) {
57
+ if (history[i].role === 'user') {
58
+ end = i - 1;
59
+ break;
60
+ }
61
+ }
62
+ return end;
63
+ }
64
+ function addToolRollbackOperation(tool, operations, restoredFiles) {
65
+ const filePath = getToolPath(tool);
66
+ if (!filePath)
67
+ return;
68
+ switch (tool.toolName) {
69
+ case 'Write': {
70
+ restoredFiles.add(filePath);
71
+ const oldContent = tool.input['old_content'];
72
+ if (typeof oldContent === 'string')
73
+ operations.push(() => writeFile(filePath, oldContent));
74
+ else
75
+ operations.push(() => removePath(filePath));
76
+ return;
77
+ }
78
+ case 'Edit':
79
+ case 'SearchReplace': {
80
+ const oldStr = tool.input['old_str'] ?? tool.input['old_string'];
81
+ const newStr = tool.input['new_str'] ?? tool.input['new_string'];
82
+ if (typeof oldStr !== 'string' || typeof newStr !== 'string')
83
+ return;
84
+ restoredFiles.add(filePath);
85
+ operations.push(() => revertTextReplacement(filePath, oldStr, newStr));
86
+ return;
87
+ }
88
+ }
89
+ }
90
+ function getToolPath(tool) {
91
+ const value = tool.input['file_path'] ?? tool.input['path'];
92
+ return typeof value === 'string' ? value : '';
93
+ }
94
+ function writeFile(filePath, content) {
95
+ const resolved = safePath(filePath);
96
+ fs.mkdirSync(path.dirname(resolved), { recursive: true });
97
+ fs.writeFileSync(resolved, content, 'utf-8');
98
+ }
99
+ function removePath(targetPath) {
100
+ const resolved = safePath(targetPath);
101
+ if (fs.existsSync(resolved))
102
+ fs.rmSync(resolved, { recursive: true, force: false });
103
+ }
104
+ function revertTextReplacement(filePath, oldStr, newStr) {
105
+ const resolved = safePath(filePath);
106
+ const content = fs.readFileSync(resolved, 'utf-8');
107
+ const index = content.indexOf(newStr);
108
+ if (index < 0)
109
+ throw new Error(`rollback replacement not found for ${filePath}`);
110
+ const next = `${content.slice(0, index)}${oldStr}${content.slice(index + newStr.length)}`;
111
+ fs.writeFileSync(resolved, next, 'utf-8');
112
+ }
113
+ //# sourceMappingURL=checkpointService.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"checkpointService.js","sourceRoot":"","sources":["../../src/services/checkpointService.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,kBAAkB,EAAuB,MAAM,qBAAqB,CAAC;AAC9E,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAgBhD,MAAM,UAAU,mBAAmB,CACjC,SAAiB,EACjB,eAAuB,EACvB,UAA4B,kBAAkB,CAAC,SAAS,CAAC;IAEzD,MAAM,WAAW,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,eAAe,CAAC,CAAC;IACvE,IAAI,WAAW,GAAG,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IAC1E,MAAM,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IACpC,IAAI,MAAM,CAAC,IAAI,KAAK,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IACtF,MAAM,MAAM,GAAG,gBAAgB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IACtD,OAAO;QACL,SAAS;QACT,eAAe;QACf,cAAc,EAAE,MAAM,CAAC,OAAO;QAC9B,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,GAAG,CAAC,CAAC;KACvC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,SAAiB,EACjB,eAAuB,EACvB,UAA4B,kBAAkB,CAAC,SAAS,CAAC;IAEzD,MAAM,IAAI,GAAG,iBAAiB,CAAC,SAAS,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;IACpE,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,UAAU;QAAE,SAAS,EAAE,CAAC;IACrD,OAAO,IAAI,CAAC,MAAM,CAAC;AACrB,CAAC;AAED,SAAS,iBAAiB,CACxB,SAAiB,EACjB,eAAuB,EACvB,OAAyB;IAEzB,MAAM,WAAW,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,eAAe,CAAC,CAAC;IACvE,IAAI,WAAW,GAAG,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IAC1E,MAAM,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IACpC,IAAI,MAAM,CAAC,IAAI,KAAK,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IACtF,MAAM,MAAM,GAAG,gBAAgB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IACtD,MAAM,aAAa,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAChD,MAAM,UAAU,GAAsB,EAAE,CAAC;IACzC,MAAM,aAAa,GAAG,IAAI,GAAG,EAAU,CAAC;IACxC,KAAK,IAAI,CAAC,GAAG,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACnD,MAAM,OAAO,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;QACjC,KAAK,MAAM,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;YAC3D,wBAAwB,CAAC,IAAI,EAAE,UAAU,EAAE,aAAa,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IACD,OAAO;QACL,MAAM,EAAE;YACN,SAAS;YACT,eAAe;YACf,cAAc,EAAE,MAAM,CAAC,OAAO;YAC9B,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,GAAG,CAAC,CAAC;YACtC,aAAa,EAAE,CAAC,GAAG,aAAa,CAAC;SAClC;QACD,UAAU;KACX,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAyB,EAAE,SAAiB;IACpE,IAAI,GAAG,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;IAC7B,KAAK,IAAI,CAAC,GAAG,SAAS,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACpD,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC/B,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;YACZ,MAAM;QACR,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,wBAAwB,CAC/B,IAAiB,EACjB,UAA6B,EAC7B,aAA0B;IAE1B,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IACnC,IAAI,CAAC,QAAQ;QAAE,OAAO;IACtB,QAAQ,IAAI,CAAC,QAAQ,EAAE,CAAC;QACtB,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC5B,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAC7C,IAAI,OAAO,UAAU,KAAK,QAAQ;gBAAE,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC;;gBACtF,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;YACjD,OAAO;QACT,CAAC;QACD,KAAK,MAAM,CAAC;QACZ,KAAK,eAAe,CAAC,CAAC,CAAC;YACrB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YACjE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YACjE,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,OAAO,MAAM,KAAK,QAAQ;gBAAE,OAAO;YACrE,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC5B,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,qBAAqB,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;YACvE,OAAO;QACT,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,WAAW,CAAC,IAAiB;IACpC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC5D,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;AAChD,CAAC;AAED,SAAS,SAAS,CAAC,QAAgB,EAAE,OAAe;IAClD,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACpC,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1D,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,UAAU,CAAC,UAAkB;IACpC,MAAM,QAAQ,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;IACtC,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;QAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;AACtF,CAAC;AAED,SAAS,qBAAqB,CAAC,QAAgB,EAAE,MAAc,EAAE,MAAc;IAC7E,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACpC,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACnD,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACtC,IAAI,KAAK,GAAG,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,sCAAsC,QAAQ,EAAE,CAAC,CAAC;IACjF,MAAM,IAAI,GAAG,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,GAAG,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;IAC1F,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;AAC5C,CAAC"}
@@ -0,0 +1,61 @@
1
+ import test from 'node:test';
2
+ import assert from 'node:assert/strict';
3
+ import fs from 'node:fs';
4
+ import os from 'node:os';
5
+ import path from 'node:path';
6
+ import { buildRollbackResult, rollbackChatTurnOnDisk } from './checkpointService.js';
7
+ test('buildRollbackResult truncates history to selected user message', () => {
8
+ const result = buildRollbackResult('sess-1', 'u1', [
9
+ { id: 'u1', role: 'user', content: 'first' },
10
+ { id: 'a1', role: 'assistant', content: 'reply' },
11
+ { id: 'u2', role: 'user', content: 'second' },
12
+ { id: 'a2', role: 'assistant', content: 'reply2' },
13
+ ]);
14
+ assert.equal(result.sessionId, 'sess-1');
15
+ assert.equal(result.restoredPrompt, 'first');
16
+ assert.deepEqual(result.messages.map((m) => m.id), ['u1', 'a1']);
17
+ });
18
+ test('rollbackChatTurnOnDisk restores file changes from later turns', () => {
19
+ const tempRoot = fs.mkdtempSync(path.join(os.tmpdir(), 'kumo-rollback-'));
20
+ const prevCwd = process.cwd();
21
+ process.chdir(tempRoot);
22
+ try {
23
+ fs.mkdirSync(path.join(tempRoot, 'lib'), { recursive: true });
24
+ fs.writeFileSync(path.join(tempRoot, 'lib', 'demo.txt'), 'after second', 'utf-8');
25
+ const result = rollbackChatTurnOnDisk('sess-1', 'u1', [
26
+ { id: 'u1', role: 'user', content: 'first' },
27
+ {
28
+ id: 'a1',
29
+ role: 'assistant',
30
+ content: 'reply1',
31
+ toolUses: [
32
+ { toolName: 'Write', input: { file_path: 'lib/demo.txt', content: 'first version' } },
33
+ ],
34
+ },
35
+ { id: 'u2', role: 'user', content: 'second' },
36
+ {
37
+ id: 'a2',
38
+ role: 'assistant',
39
+ content: 'reply2',
40
+ toolUses: [
41
+ {
42
+ toolName: 'Write',
43
+ input: {
44
+ file_path: 'lib/demo.txt',
45
+ content: 'after second',
46
+ old_content: 'first version',
47
+ },
48
+ },
49
+ ],
50
+ },
51
+ ]);
52
+ assert.equal(fs.readFileSync(path.join(tempRoot, 'lib', 'demo.txt'), 'utf-8'), 'first version');
53
+ assert.deepEqual(result.restoredFiles, ['lib/demo.txt']);
54
+ assert.deepEqual(result.messages.map((m) => m.id), ['u1', 'a1']);
55
+ }
56
+ finally {
57
+ process.chdir(prevCwd);
58
+ fs.rmSync(tempRoot, { recursive: true, force: true });
59
+ }
60
+ });
61
+ //# sourceMappingURL=checkpointService.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"checkpointService.test.js","sourceRoot":"","sources":["../../src/services/checkpointService.test.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,wBAAwB,CAAC;AAErF,IAAI,CAAC,gEAAgE,EAAE,GAAG,EAAE;IAC1E,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,EAAE,IAAI,EAAE;QACjD,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE;QAC5C,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE;QACjD,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE;QAC7C,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE;KACnD,CAAC,CAAC;IACH,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IACzC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IAC7C,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;AACnE,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+DAA+D,EAAE,GAAG,EAAE;IACzE,MAAM,QAAQ,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,gBAAgB,CAAC,CAAC,CAAC;IAC1E,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC9B,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACxB,IAAI,CAAC;QACH,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9D,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,EAAE,UAAU,CAAC,EAAE,cAAc,EAAE,OAAO,CAAC,CAAC;QAClF,MAAM,MAAM,GAAG,sBAAsB,CAAC,QAAQ,EAAE,IAAI,EAAE;YACpD,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE;YAC5C;gBACE,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,QAAQ;gBACjB,QAAQ,EAAE;oBACR,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,cAAc,EAAE,OAAO,EAAE,eAAe,EAAE,EAAE;iBACtF;aACF;YACD,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE;YAC7C;gBACE,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,QAAQ;gBACjB,QAAQ,EAAE;oBACR;wBACE,QAAQ,EAAE,OAAO;wBACjB,KAAK,EAAE;4BACL,SAAS,EAAE,cAAc;4BACzB,OAAO,EAAE,cAAc;4BACvB,WAAW,EAAE,eAAe;yBAC7B;qBACF;iBACF;aACF;SACF,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,EAAE,UAAU,CAAC,EAAE,OAAO,CAAC,EAAE,eAAe,CAAC,CAAC;QAChG,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC,cAAc,CAAC,CAAC,CAAC;QACzD,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;IACnE,CAAC;YAAS,CAAC;QACT,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACvB,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACxD,CAAC;AACH,CAAC,CAAC,CAAC"}
@@ -1 +1 @@
1
- {"version":3,"file":"cliRuntime.js","sourceRoot":"","sources":["../../src/services/cliRuntime.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,mBAAmB,CAAC;AACxC,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAE7D,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAW1D,sFAAsF;AACtF,MAAM,UAAU,gBAAgB,CAAC,IAAc;IAC7C,OAAO,IAAI,CAAC,QAAQ,CAAC,gCAAgC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,EAAE,gCAAgC,CAAC,CAAC;AACnH,CAAC;AAED,4GAA4G;AAC5G,MAAM,UAAU,yBAAyB,CAAC,MAMzC;IACC,OAAO,CAAC,SAAiB,EAAE,EAAE;QAC3B,MAAM,CAAC,KAAK,CAAC,gBAAgB,GAAG,SAAS,CAAC;QAC1C,MAAM,CAAC,aAAa,EAAE,CAAC;QACvB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,SAAS,EAAE,CAAC,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC;QAC7B,MAAM,CAAC,KAAK,CAAC,OAAO,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,IAAI,cAAc,CAAC,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAmB,CAAC;QAC/K,MAAM,CAAC,KAAK,CAAC,iBAAiB,GAAG,KAAK,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE;YAC7C,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC;YAC7C,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;gBACtE,GAAG,CAAC,iDAAiD,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;gBAC3E,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAC7B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa;YAAE,OAAO;QACxC,MAAM,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC;QAChD,MAAM,CAAC,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC;QAClC,MAAM,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;IACtC,CAAC,CAAC;AACJ,CAAC;AAED,mGAAmG;AACnG,MAAM,UAAU,mBAAmB,CAAC,MAMnC;IACC,OAAO,KAAK,IAAI,EAAE;QAChB,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC;QACzC,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,UAAU;YAChB,UAAU,EAAE,KAAK;YACjB,WAAW,EAAE,WAAW,EAAE,WAAW;YACrC,UAAU,EAAE,WAAW,EAAE,UAAU;YACnC,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE;SACnB,CAAC,CAAC;QACH,MAAM,CAAC,SAAS,EAAE,CAAC;QACnB,MAAM,CAAC,cAAc,EAAE,CAAC;QACxB,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,SAAS,EAAE,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAC7E,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,CAAC,CAAC;IAClD,CAAC,CAAC;AACJ,CAAC"}
1
+ {"version":3,"file":"cliRuntime.js","sourceRoot":"","sources":["../../src/services/cliRuntime.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,mBAAmB,CAAC;AACxC,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAE7D,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAY1D,sFAAsF;AACtF,MAAM,UAAU,gBAAgB,CAAC,IAAc;IAC7C,OAAO,IAAI,CAAC,QAAQ,CAAC,gCAAgC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,EAAE,gCAAgC,CAAC,CAAC;AACnH,CAAC;AAED,4GAA4G;AAC5G,MAAM,UAAU,yBAAyB,CAAC,MAMzC;IACC,OAAO,CAAC,SAAiB,EAAE,EAAE;QAC3B,MAAM,CAAC,KAAK,CAAC,gBAAgB,GAAG,SAAS,CAAC;QAC1C,MAAM,CAAC,aAAa,EAAE,CAAC;QACvB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,SAAS,EAAE,CAAC,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC;QAC7B,MAAM,CAAC,KAAK,CAAC,OAAO,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,IAAI,cAAc,CAAC,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAmB,CAAC;QAC/K,MAAM,CAAC,KAAK,CAAC,iBAAiB,GAAG,KAAK,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE;YAC7C,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC;YAC7C,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;gBACtE,GAAG,CAAC,iDAAiD,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;gBAC3E,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAC7B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa;YAAE,OAAO;QACxC,MAAM,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC;QAChD,MAAM,CAAC,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC;QAClC,MAAM,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;IACtC,CAAC,CAAC;AACJ,CAAC;AAED,mGAAmG;AACnG,MAAM,UAAU,mBAAmB,CAAC,MAMnC;IACC,OAAO,KAAK,IAAI,EAAE;QAChB,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC;QACzC,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,UAAU;YAChB,UAAU,EAAE,KAAK;YACjB,WAAW,EAAE,WAAW,EAAE,WAAW;YACrC,UAAU,EAAE,WAAW,EAAE,UAAU;YACnC,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE;SACnB,CAAC,CAAC;QACH,MAAM,CAAC,SAAS,EAAE,CAAC;QACnB,MAAM,CAAC,cAAc,EAAE,CAAC;QACxB,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,SAAS,EAAE,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAC7E,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,CAAC,CAAC;IAClD,CAAC,CAAC;AACJ,CAAC"}
@@ -0,0 +1,195 @@
1
+ import fs from 'node:fs';
2
+ import path from 'node:path';
3
+ import { listFileHistoryEntries, readFileHistoryVersion } from '../claude/fileHistory.js';
4
+ import { backupSessionFile, getSessionFileInfo, readSessionHistory, readSessionJsonlLines, writeSessionJsonlLines } from './sessionService.js';
5
+ const FILE_TOOL_NAMES = new Set(['Write', 'Edit', 'SearchReplace', 'DeleteFile']);
6
+ function getTextContent(content) {
7
+ if (typeof content === 'string')
8
+ return content;
9
+ if (!Array.isArray(content))
10
+ return '';
11
+ return content
12
+ .filter((block) => block.type === 'text' && typeof block.text === 'string')
13
+ .map((block) => block.text ?? '')
14
+ .join('')
15
+ .trim();
16
+ }
17
+ function extractToolPaths(message) {
18
+ const content = message.message?.content;
19
+ if (!Array.isArray(content))
20
+ return [];
21
+ const filePaths = new Set();
22
+ for (const block of content) {
23
+ if (block.type !== 'tool_use' || !block.name || !FILE_TOOL_NAMES.has(block.name))
24
+ continue;
25
+ const input = block.input ?? {};
26
+ const paths = [input['file_path'], input['path']];
27
+ if (Array.isArray(input['file_paths'])) {
28
+ for (const value of input['file_paths'])
29
+ paths.push(value);
30
+ }
31
+ for (const value of paths) {
32
+ if (typeof value === 'string' && value.trim().length > 0) {
33
+ filePaths.add(value);
34
+ }
35
+ }
36
+ }
37
+ return [...filePaths];
38
+ }
39
+ function parseTurns(sessionId, cwdOverride) {
40
+ const lines = readSessionJsonlLines(sessionId, cwdOverride);
41
+ const turns = [];
42
+ const fileUsage = new Map();
43
+ let currentTurn = null;
44
+ let userMessageIndex = -1;
45
+ for (let lineIndex = 0; lineIndex < lines.length; lineIndex++) {
46
+ try {
47
+ const message = JSON.parse(lines[lineIndex]);
48
+ if (message.isMeta || message.sourceToolUseID || !message.message)
49
+ continue;
50
+ if (message.type === 'user' && message.message.role === 'user') {
51
+ const prompt = getTextContent(message.message.content);
52
+ if (!prompt || /<local-command-/i.test(prompt))
53
+ continue;
54
+ userMessageIndex += 1;
55
+ currentTurn = {
56
+ turnId: message.uuid ?? `turn-${userMessageIndex}`,
57
+ prompt,
58
+ messageIndex: userMessageIndex,
59
+ lineStart: lineIndex,
60
+ filePaths: new Set(),
61
+ };
62
+ turns.push(currentTurn);
63
+ continue;
64
+ }
65
+ if (!currentTurn)
66
+ continue;
67
+ for (const filePath of extractToolPaths(message)) {
68
+ currentTurn.filePaths.add(filePath);
69
+ const usages = fileUsage.get(filePath) ?? [];
70
+ usages.push({ turnIndex: currentTurn.messageIndex, lineIndex });
71
+ fileUsage.set(filePath, usages);
72
+ }
73
+ }
74
+ catch { }
75
+ }
76
+ return { lines, turns, fileUsage };
77
+ }
78
+ function resolveRollbackSnapshotVersion(sessionId, filePath, turnIndex, fileUsage) {
79
+ const usages = fileUsage.get(filePath) ?? [];
80
+ const firstUsageInTurn = usages.find((entry) => entry.turnIndex === turnIndex);
81
+ const entries = listFileHistoryEntries(sessionId, filePath);
82
+ if (entries.length === 0)
83
+ return null;
84
+ if (!firstUsageInTurn)
85
+ return entries[entries.length - 1]?.version ?? null;
86
+ const usageOrdinal = usages.filter((entry) => entry.lineIndex <= firstUsageInTurn.lineIndex).length;
87
+ const targetIndex = entries.length - usageOrdinal - 1;
88
+ if (targetIndex < 0)
89
+ return entries[0]?.version ?? null;
90
+ return entries[targetIndex]?.version ?? null;
91
+ }
92
+ function readCurrentFileContent(filePath) {
93
+ try {
94
+ const resolved = path.resolve(filePath);
95
+ if (!fs.existsSync(resolved))
96
+ return null;
97
+ return fs.readFileSync(resolved, 'utf-8');
98
+ }
99
+ catch {
100
+ return null;
101
+ }
102
+ }
103
+ function resolveTurn(turns, turnId, messageIndex) {
104
+ return turns.find((turn) => turn.turnId === turnId || (messageIndex != null && turn.messageIndex === messageIndex)) ?? null;
105
+ }
106
+ export function buildRollbackPreview(sessionId, turnId, cwdOverride, messageIndex) {
107
+ const { turns, fileUsage } = parseTurns(sessionId, cwdOverride);
108
+ const turn = resolveTurn(turns, turnId, messageIndex);
109
+ if (turn == null)
110
+ return null;
111
+ const turnIndex = turns.indexOf(turn);
112
+ if (turnIndex < 0)
113
+ return null;
114
+ const fileChanges = [...turn.filePaths].map((filePath) => {
115
+ const snapshotVersion = resolveRollbackSnapshotVersion(sessionId, filePath, turn.messageIndex, fileUsage);
116
+ const rollbackContent = snapshotVersion !== null ? readFileHistoryVersion(sessionId, filePath, snapshotVersion) : null;
117
+ const currentContent = readCurrentFileContent(filePath);
118
+ return {
119
+ path: filePath,
120
+ status: rollbackContent === null ? 'unsupported' : currentContent === null ? 'deleted' : 'modified',
121
+ currentContent,
122
+ rollbackContent,
123
+ snapshotVersion,
124
+ };
125
+ });
126
+ return {
127
+ sessionId,
128
+ turnId,
129
+ prompt: turn.prompt,
130
+ deleteCount: Math.max(0, turns.length - turnIndex),
131
+ deleteFromMessageIndex: turn.messageIndex,
132
+ startsNewSession: turnIndex === 0,
133
+ fileChanges,
134
+ };
135
+ }
136
+ export function applyRollback(sessionId, turnId, cwdOverride, messageIndex) {
137
+ const preview = buildRollbackPreview(sessionId, turnId, cwdOverride, messageIndex);
138
+ if (!preview)
139
+ return null;
140
+ const { lines, turns } = parseTurns(sessionId, cwdOverride);
141
+ const turn = resolveTurn(turns, turnId, messageIndex);
142
+ if (!turn)
143
+ return null;
144
+ const sessionBackup = backupSessionFile(sessionId, cwdOverride);
145
+ const fileBackup = new Map();
146
+ try {
147
+ for (const fileChange of preview.fileChanges) {
148
+ if (fileChange.rollbackContent === null)
149
+ continue;
150
+ const filePath = path.resolve(fileChange.path);
151
+ fileBackup.set(filePath, fs.existsSync(filePath) ? fs.readFileSync(filePath, 'utf-8') : null);
152
+ fs.mkdirSync(path.dirname(filePath), { recursive: true });
153
+ fs.writeFileSync(filePath, fileChange.rollbackContent, 'utf-8');
154
+ }
155
+ writeSessionJsonlLines(sessionId, lines.slice(0, turn.lineStart), cwdOverride);
156
+ return preview;
157
+ }
158
+ catch (error) {
159
+ for (const [filePath, content] of fileBackup.entries()) {
160
+ if (content === null) {
161
+ try {
162
+ if (fs.existsSync(filePath))
163
+ fs.unlinkSync(filePath);
164
+ }
165
+ catch { }
166
+ continue;
167
+ }
168
+ try {
169
+ fs.mkdirSync(path.dirname(filePath), { recursive: true });
170
+ fs.writeFileSync(filePath, content, 'utf-8');
171
+ }
172
+ catch { }
173
+ }
174
+ if (sessionBackup) {
175
+ const { sessionFile } = getSessionFileInfo(sessionId, cwdOverride);
176
+ try {
177
+ fs.copyFileSync(sessionBackup, sessionFile);
178
+ }
179
+ catch { }
180
+ }
181
+ throw error;
182
+ }
183
+ }
184
+ export function loadRolledBackHistory(sessionId, cwdOverride) {
185
+ const { cwd } = getSessionFileInfo(sessionId, cwdOverride);
186
+ const previousCwd = process.cwd();
187
+ try {
188
+ process.chdir(cwd);
189
+ return readSessionHistory(sessionId);
190
+ }
191
+ finally {
192
+ process.chdir(previousCwd);
193
+ }
194
+ }
195
+ //# sourceMappingURL=sessionRollbackService.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"sessionRollbackService.js","sourceRoot":"","sources":["../../src/services/sessionRollbackService.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,sBAAsB,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAE1F,OAAO,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,qBAAqB,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAE/I,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,YAAY,CAAC,CAAC,CAAC;AA0ClF,SAAS,cAAc,CAAC,OAAiD;IACvE,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,OAAO,CAAC;IAChD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;QAAE,OAAO,EAAE,CAAC;IACvC,OAAO,OAAO;SACX,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC;SAC1E,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC;SAChC,IAAI,CAAC,EAAE,CAAC;SACR,IAAI,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAqB;IAC7C,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC;IACzC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;QAAE,OAAO,EAAE,CAAC;IACvC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IACpC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC;YAAE,SAAS;QAC3F,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC;QAChC,MAAM,KAAK,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;QAClD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC;YACvC,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,YAAY,CAAC;gBAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7D,CAAC;QACD,KAAK,MAAM,KAAK,IAAI,KAAK,EAAE,CAAC;YAC1B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzD,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,CAAC,GAAG,SAAS,CAAC,CAAC;AACxB,CAAC;AAED,SAAS,UAAU,CAAC,SAAiB,EAAE,WAAoB;IACzD,MAAM,KAAK,GAAG,qBAAqB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;IAC5D,MAAM,KAAK,GAAmB,EAAE,CAAC;IACjC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAuB,CAAC;IACjD,IAAI,WAAW,GAAwB,IAAI,CAAC;IAC5C,IAAI,gBAAgB,GAAG,CAAC,CAAC,CAAC;IAC1B,KAAK,IAAI,SAAS,GAAG,CAAC,EAAE,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC;QAC9D,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,CAAiB,CAAC;YAC7D,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,eAAe,IAAI,CAAC,OAAO,CAAC,OAAO;gBAAE,SAAS;YAC5E,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC/D,MAAM,MAAM,GAAG,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBACvD,IAAI,CAAC,MAAM,IAAI,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC;oBAAE,SAAS;gBACzD,gBAAgB,IAAI,CAAC,CAAC;gBACtB,WAAW,GAAG;oBACZ,MAAM,EAAE,OAAO,CAAC,IAAI,IAAI,QAAQ,gBAAgB,EAAE;oBAClD,MAAM;oBACN,YAAY,EAAE,gBAAgB;oBAC9B,SAAS,EAAE,SAAS;oBACpB,SAAS,EAAE,IAAI,GAAG,EAAU;iBAC7B,CAAC;gBACF,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;gBACxB,SAAS;YACX,CAAC;YACD,IAAI,CAAC,WAAW;gBAAE,SAAS;YAC3B,KAAK,MAAM,QAAQ,IAAI,gBAAgB,CAAC,OAAO,CAAC,EAAE,CAAC;gBACjD,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBACpC,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;gBAC7C,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,WAAW,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,CAAC;gBAChE,SAAS,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;IACZ,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;AACrC,CAAC;AAED,SAAS,8BAA8B,CAAC,SAAiB,EAAE,QAAgB,EAAE,SAAiB,EAAE,SAAmC;IACjI,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC7C,MAAM,gBAAgB,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC;IAC/E,MAAM,OAAO,GAAG,sBAAsB,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC5D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IACtC,IAAI,CAAC,gBAAgB;QAAE,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,OAAO,IAAI,IAAI,CAAC;IAC3E,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,IAAI,gBAAgB,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC;IACpG,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,GAAG,YAAY,GAAG,CAAC,CAAC;IACtD,IAAI,WAAW,GAAG,CAAC;QAAE,OAAO,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,IAAI,IAAI,CAAC;IACxD,OAAO,OAAO,CAAC,WAAW,CAAC,EAAE,OAAO,IAAI,IAAI,CAAC;AAC/C,CAAC;AAED,SAAS,sBAAsB,CAAC,QAAgB;IAC9C,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;YAAE,OAAO,IAAI,CAAC;QAC1C,OAAO,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC5C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,WAAW,CAAC,KAAqB,EAAE,MAAc,EAAE,YAAqB;IAC/E,OAAO,KAAK,CAAC,IAAI,CACf,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,MAAM,IAAI,CAAC,YAAY,IAAI,IAAI,IAAI,IAAI,CAAC,YAAY,KAAK,YAAY,CAAC,CACjG,IAAI,IAAI,CAAC;AACZ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,SAAiB,EAAE,MAAc,EAAE,WAAoB,EAAE,YAAqB;IACjH,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,UAAU,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;IAChE,MAAM,IAAI,GAAG,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;IACtD,IAAI,IAAI,IAAI,IAAI;QAAE,OAAO,IAAI,CAAC;IAC9B,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACtC,IAAI,SAAS,GAAG,CAAC;QAAE,OAAO,IAAI,CAAC;IAC/B,MAAM,WAAW,GAA0B,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;QAC9E,MAAM,eAAe,GAAG,8BAA8B,CAAC,SAAS,EAAE,QAAQ,EAAE,IAAI,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;QAC1G,MAAM,eAAe,GAAG,eAAe,KAAK,IAAI,CAAC,CAAC,CAAC,sBAAsB,CAAC,SAAS,EAAE,QAAQ,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACvH,MAAM,cAAc,GAAG,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QACxD,OAAO;YACL,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,eAAe,KAAK,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,cAAc,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU;YACnG,cAAc;YACd,eAAe;YACf,eAAe;SACc,CAAC;IAClC,CAAC,CAAC,CAAC;IACH,OAAO;QACL,SAAS;QACT,MAAM;QACN,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC;QAClD,sBAAsB,EAAE,IAAI,CAAC,YAAY;QACzC,gBAAgB,EAAE,SAAS,KAAK,CAAC;QACjC,WAAW;KACZ,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,SAAiB,EAAE,MAAc,EAAE,WAAoB,EAAE,YAAqB;IAC1G,MAAM,OAAO,GAAG,oBAAoB,CAAC,SAAS,EAAE,MAAM,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;IACnF,IAAI,CAAC,OAAO;QAAE,OAAO,IAAI,CAAC;IAC1B,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,UAAU,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;IAC5D,MAAM,IAAI,GAAG,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;IACtD,IAAI,CAAC,IAAI;QAAE,OAAO,IAAI,CAAC;IACvB,MAAM,aAAa,GAAG,iBAAiB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;IAChE,MAAM,UAAU,GAAG,IAAI,GAAG,EAAyB,CAAC;IACpD,IAAI,CAAC;QACH,KAAK,MAAM,UAAU,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YAC7C,IAAI,UAAU,CAAC,eAAe,KAAK,IAAI;gBAAE,SAAS;YAClD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YAC/C,UAAU,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAC9F,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1D,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,UAAU,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;QAClE,CAAC;QACD,sBAAsB,CAAC,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,CAAC,CAAC;QAC/E,OAAO,OAAO,CAAC;IACjB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,KAAK,MAAM,CAAC,QAAQ,EAAE,OAAO,CAAC,IAAI,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC;YACvD,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;gBACrB,IAAI,CAAC;oBACH,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;wBAAE,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;gBACvD,CAAC;gBAAC,MAAM,CAAC,CAAA,CAAC;gBACV,SAAS;YACX,CAAC;YACD,IAAI,CAAC;gBACH,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC1D,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/C,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QACZ,CAAC;QACD,IAAI,aAAa,EAAE,CAAC;YAClB,MAAM,EAAE,WAAW,EAAE,GAAG,kBAAkB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;YACnE,IAAI,CAAC;gBACH,EAAE,CAAC,YAAY,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC;YAC9C,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QACZ,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,SAAiB,EAAE,WAAoB;IAC3E,MAAM,EAAE,GAAG,EAAE,GAAG,kBAAkB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;IAC3D,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAClC,IAAI,CAAC;QACH,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACnB,OAAO,kBAAkB,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;YAAS,CAAC;QACT,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;IAC7B,CAAC;AACH,CAAC"}
@@ -0,0 +1,142 @@
1
+ import assert from 'node:assert/strict';
2
+ import crypto from 'node:crypto';
3
+ import fs from 'node:fs';
4
+ import os from 'node:os';
5
+ import path from 'node:path';
6
+ import test from 'node:test';
7
+ import { applyRollback, buildRollbackPreview } from './sessionRollbackService.js';
8
+ function historyFileName(filePath) {
9
+ const hash = crypto.createHash('sha256').update(path.resolve(filePath)).digest('hex').slice(0, 16);
10
+ return `${hash}@v1`;
11
+ }
12
+ function withTempClaudeHome(run) {
13
+ const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'kumo-rollback-'));
14
+ const previousHome = process.env.USERPROFILE;
15
+ const previousCwd = process.cwd();
16
+ const projectDir = path.join(tempDir, 'project').replace(/\//g, '\\');
17
+ const sessionId = 'session-1';
18
+ const projectFolder = projectDir.replace(/:/g, '-').replace(/[\\/]/g, '-').replace(/_/g, '-');
19
+ const transcriptDir = path.join(tempDir, '.claude', 'projects', projectFolder);
20
+ const historyDir = path.join(tempDir, '.claude', 'file-history', sessionId);
21
+ const filePath = path.join(projectDir, 'note.txt').replace(/\//g, '\\');
22
+ fs.mkdirSync(projectDir, { recursive: true });
23
+ fs.mkdirSync(transcriptDir, { recursive: true });
24
+ fs.mkdirSync(historyDir, { recursive: true });
25
+ process.env.USERPROFILE = tempDir;
26
+ process.chdir(projectDir);
27
+ try {
28
+ run(tempDir, projectDir, sessionId, filePath);
29
+ }
30
+ finally {
31
+ process.chdir(previousCwd);
32
+ if (previousHome)
33
+ process.env.USERPROFILE = previousHome;
34
+ else
35
+ delete process.env.USERPROFILE;
36
+ fs.rmSync(tempDir, { recursive: true, force: true });
37
+ }
38
+ }
39
+ test('buildRollbackPreview returns affected files and prompt', () => {
40
+ withTempClaudeHome((rootDir, projectDir, sessionId, filePath) => {
41
+ fs.writeFileSync(filePath, 'after', 'utf-8');
42
+ const sessionFile = path.join(rootDir, '.claude', 'projects', projectDir.replace(/:/g, '-').replace(/[\\/]/g, '-').replace(/_/g, '-'), `${sessionId}.jsonl`);
43
+ fs.mkdirSync(path.dirname(sessionFile), { recursive: true });
44
+ fs.writeFileSync(sessionFile, `${JSON.stringify({
45
+ uuid: 'turn-1',
46
+ type: 'user',
47
+ message: { role: 'user', content: [{ type: 'text', text: 'fix file' }] },
48
+ })}\n${JSON.stringify({
49
+ uuid: 'assistant-1',
50
+ type: 'assistant',
51
+ message: {
52
+ role: 'assistant',
53
+ content: [{ type: 'tool_use', name: 'Write', input: { file_path: filePath } }],
54
+ },
55
+ })}\n`, 'utf-8');
56
+ const historyFile = path.join(rootDir, '.claude', 'file-history', sessionId, historyFileName(filePath));
57
+ fs.writeFileSync(historyFile, 'before', 'utf-8');
58
+ const preview = buildRollbackPreview(sessionId, 'turn-1', projectDir);
59
+ assert.ok(preview);
60
+ assert.equal(preview?.prompt, 'fix file');
61
+ assert.equal(preview?.startsNewSession, true);
62
+ assert.equal(preview?.fileChanges.length, 1);
63
+ assert.equal(preview?.fileChanges[0].rollbackContent, 'before');
64
+ });
65
+ });
66
+ test('buildRollbackPreview treats first user turn after meta lines as new session', () => {
67
+ withTempClaudeHome((rootDir, projectDir, sessionId, filePath) => {
68
+ const sessionFile = path.join(rootDir, '.claude', 'projects', projectDir.replace(/:/g, '-').replace(/[\\/]/g, '-').replace(/_/g, '-'), `${sessionId}.jsonl`);
69
+ fs.mkdirSync(path.dirname(sessionFile), { recursive: true });
70
+ fs.writeFileSync(sessionFile, `${JSON.stringify({ uuid: 'meta-1', type: 'system', message: { role: 'system', content: 'meta' } })}\n${JSON.stringify({
71
+ uuid: 'turn-1',
72
+ type: 'user',
73
+ message: { role: 'user', content: [{ type: 'text', text: 'first prompt' }] },
74
+ })}\n${JSON.stringify({
75
+ uuid: 'assistant-1',
76
+ type: 'assistant',
77
+ message: { role: 'assistant', content: [{ type: 'tool_use', name: 'Write', input: { file_path: filePath } }] },
78
+ })}\n`, 'utf-8');
79
+ const preview = buildRollbackPreview(sessionId, 'turn-1', projectDir);
80
+ assert.ok(preview);
81
+ assert.equal(preview?.startsNewSession, true);
82
+ });
83
+ });
84
+ test('applyRollback restores file content and truncates transcript', () => {
85
+ withTempClaudeHome((rootDir, projectDir, sessionId, filePath) => {
86
+ fs.writeFileSync(filePath, 'after', 'utf-8');
87
+ const sessionFile = path.join(rootDir, '.claude', 'projects', projectDir.replace(/:/g, '-').replace(/[\\/]/g, '-').replace(/_/g, '-'), `${sessionId}.jsonl`);
88
+ fs.mkdirSync(path.dirname(sessionFile), { recursive: true });
89
+ fs.writeFileSync(sessionFile, `${JSON.stringify({
90
+ uuid: 'turn-1',
91
+ type: 'user',
92
+ message: { role: 'user', content: [{ type: 'text', text: 'fix file' }] },
93
+ })}\n${JSON.stringify({
94
+ uuid: 'assistant-1',
95
+ type: 'assistant',
96
+ message: {
97
+ role: 'assistant',
98
+ content: [{ type: 'tool_use', name: 'Write', input: { file_path: filePath } }],
99
+ },
100
+ })}\n${JSON.stringify({
101
+ uuid: 'turn-2',
102
+ type: 'user',
103
+ message: { role: 'user', content: [{ type: 'text', text: 'later turn' }] },
104
+ })}\n`, 'utf-8');
105
+ const historyFile = path.join(rootDir, '.claude', 'file-history', sessionId, historyFileName(filePath));
106
+ fs.writeFileSync(historyFile, 'before', 'utf-8');
107
+ const preview = applyRollback(sessionId, 'turn-1', projectDir);
108
+ assert.ok(preview);
109
+ assert.equal(preview?.startsNewSession, true);
110
+ assert.equal(fs.readFileSync(filePath, 'utf-8'), 'before');
111
+ assert.equal(fs.readFileSync(sessionFile, 'utf-8'), '');
112
+ });
113
+ });
114
+ test('buildRollbackPreview keeps resume mode for later turns', () => {
115
+ withTempClaudeHome((rootDir, projectDir, sessionId, filePath) => {
116
+ fs.writeFileSync(filePath, 'after', 'utf-8');
117
+ const sessionFile = path.join(rootDir, '.claude', 'projects', projectDir.replace(/:/g, '-').replace(/[\\/]/g, '-').replace(/_/g, '-'), `${sessionId}.jsonl`);
118
+ fs.mkdirSync(path.dirname(sessionFile), { recursive: true });
119
+ fs.writeFileSync(sessionFile, `${JSON.stringify({
120
+ uuid: 'turn-1',
121
+ type: 'user',
122
+ message: { role: 'user', content: [{ type: 'text', text: 'first turn' }] },
123
+ })}\n${JSON.stringify({
124
+ uuid: 'turn-2',
125
+ type: 'user',
126
+ message: { role: 'user', content: [{ type: 'text', text: 'second turn' }] },
127
+ })}\n${JSON.stringify({
128
+ uuid: 'assistant-2',
129
+ type: 'assistant',
130
+ message: {
131
+ role: 'assistant',
132
+ content: [{ type: 'tool_use', name: 'Write', input: { file_path: filePath } }],
133
+ },
134
+ })}\n`, 'utf-8');
135
+ const historyFile = path.join(rootDir, '.claude', 'file-history', sessionId, historyFileName(filePath));
136
+ fs.writeFileSync(historyFile, 'before', 'utf-8');
137
+ const preview = buildRollbackPreview(sessionId, 'turn-2', projectDir);
138
+ assert.ok(preview);
139
+ assert.equal(preview?.startsNewSession, false);
140
+ });
141
+ });
142
+ //# sourceMappingURL=sessionRollbackService.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"sessionRollbackService.test.js","sourceRoot":"","sources":["../../src/services/sessionRollbackService.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AAElF,SAAS,eAAe,CAAC,QAAgB;IACvC,MAAM,IAAI,GAAG,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACnG,OAAO,GAAG,IAAI,KAAK,CAAC;AACtB,CAAC;AAED,SAAS,kBAAkB,CAAC,GAAuF;IACjH,MAAM,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,gBAAgB,CAAC,CAAC,CAAC;IACzE,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;IAC7C,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAClC,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACtE,MAAM,SAAS,GAAG,WAAW,CAAC;IAC9B,MAAM,aAAa,GAAG,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IAC9F,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,aAAa,CAAC,CAAC;IAC/E,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,SAAS,CAAC,CAAC;IAC5E,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACxE,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC9C,EAAE,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACjD,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC9C,OAAO,CAAC,GAAG,CAAC,WAAW,GAAG,OAAO,CAAC;IAClC,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IAC1B,IAAI,CAAC;QACH,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;IAChD,CAAC;YAAS,CAAC;QACT,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAC3B,IAAI,YAAY;YAAE,OAAO,CAAC,GAAG,CAAC,WAAW,GAAG,YAAY,CAAC;;YACpD,OAAO,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;QACpC,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACvD,CAAC;AACH,CAAC;AAED,IAAI,CAAC,wDAAwD,EAAE,GAAG,EAAE;IAClE,kBAAkB,CAAC,CAAC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE;QAC9D,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QAC7C,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,GAAG,SAAS,QAAQ,CAAC,CAAC;QAC7J,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7D,EAAE,CAAC,aAAa,CACd,WAAW,EACX,GAAG,IAAI,CAAC,SAAS,CAAC;YAChB,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE;SACzE,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC;YACpB,IAAI,EAAE,aAAa;YACnB,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE;gBACP,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,CAAC;aAC/E;SACF,CAAC,IAAI,EACN,OAAO,CACR,CAAC;QACF,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAC3B,OAAO,EACP,SAAS,EACT,cAAc,EACd,SAAS,EACT,eAAe,CAAC,QAAQ,CAAC,CAC1B,CAAC;QACF,EAAE,CAAC,aAAa,CAAC,WAAW,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;QAEjD,MAAM,OAAO,GAAG,oBAAoB,CAAC,SAAS,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;QAEtE,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;QACnB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QAC1C,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,gBAAgB,EAAE,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,6EAA6E,EAAE,GAAG,EAAE;IACvF,kBAAkB,CAAC,CAAC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE;QAC9D,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,GAAG,SAAS,QAAQ,CAAC,CAAC;QAC7J,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7D,EAAE,CAAC,aAAa,CACd,WAAW,EACX,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC;YACrH,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,EAAE;SAC7E,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC;YACpB,IAAI,EAAE,aAAa;YACnB,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE;SAC/G,CAAC,IAAI,EACN,OAAO,CACR,CAAC;QAEF,MAAM,OAAO,GAAG,oBAAoB,CAAC,SAAS,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;QAEtE,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;QACnB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,gBAAgB,EAAE,IAAI,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8DAA8D,EAAE,GAAG,EAAE;IACxE,kBAAkB,CAAC,CAAC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE;QAC9D,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QAC7C,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,GAAG,SAAS,QAAQ,CAAC,CAAC;QAC7J,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7D,EAAE,CAAC,aAAa,CACd,WAAW,EACX,GAAG,IAAI,CAAC,SAAS,CAAC;YAChB,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE;SACzE,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC;YACpB,IAAI,EAAE,aAAa;YACnB,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE;gBACP,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,CAAC;aAC/E;SACF,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC;YACpB,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,EAAE;SAC3E,CAAC,IAAI,EACN,OAAO,CACR,CAAC;QACF,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAC3B,OAAO,EACP,SAAS,EACT,cAAc,EACd,SAAS,EACT,eAAe,CAAC,QAAQ,CAAC,CAC1B,CAAC;QACF,EAAE,CAAC,aAAa,CAAC,WAAW,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;QAEjD,MAAM,OAAO,GAAG,aAAa,CAAC,SAAS,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;QAE/D,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;QACnB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,gBAAgB,EAAE,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,QAAQ,CAAC,CAAC;QAC3D,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,wDAAwD,EAAE,GAAG,EAAE;IAClE,kBAAkB,CAAC,CAAC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE;QAC9D,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QAC7C,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,GAAG,SAAS,QAAQ,CAAC,CAAC;QAC7J,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7D,EAAE,CAAC,aAAa,CACd,WAAW,EACX,GAAG,IAAI,CAAC,SAAS,CAAC;YAChB,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,EAAE;SAC3E,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC;YACpB,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE;SAC5E,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC;YACpB,IAAI,EAAE,aAAa;YACnB,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE;gBACP,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,CAAC;aAC/E;SACF,CAAC,IAAI,EACN,OAAO,CACR,CAAC;QACF,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAC3B,OAAO,EACP,SAAS,EACT,cAAc,EACd,SAAS,EACT,eAAe,CAAC,QAAQ,CAAC,CAC1B,CAAC;QACF,EAAE,CAAC,aAAa,CAAC,WAAW,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;QAEjD,MAAM,OAAO,GAAG,oBAAoB,CAAC,SAAS,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;QAEtE,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;QACnB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,gBAAgB,EAAE,KAAK,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -4,6 +4,7 @@ import path from 'node:path';
4
4
  import { log } from '../core/logger.js';
5
5
  import { cwdToProjectFolder } from '../utils/safePath.js';
6
6
  import { parseTranscriptMessage } from '../claude/transcriptParser.js';
7
+ import { hydratePromptImages } from '../handlers/promptImages.js';
7
8
  // remove any `--resume <id>` or `-r <id>` flag pairs
8
9
  export function removeResumeArgs(args) {
9
10
  const result = [];
@@ -91,10 +92,7 @@ export async function getSessionListForCwd(cwdOverride) {
91
92
  }
92
93
  // delete a session jsonl file for the given cwd
93
94
  export async function deleteSessionFile(sessionId, cwdOverride) {
94
- const cwd = cwdOverride && cwdOverride.length > 0 ? cwdOverride : process.cwd();
95
- const folderName = cwdToProjectFolder(cwd);
96
- const projectDir = path.join(os.homedir(), '.claude', 'projects', folderName);
97
- const sessionFile = path.join(projectDir, `${sessionId}.jsonl`);
95
+ const { sessionFile } = getSessionFileInfo(sessionId, cwdOverride);
98
96
  if (fs.existsSync(sessionFile)) {
99
97
  fs.unlinkSync(sessionFile);
100
98
  log(`[delete_session] deleted ${sessionFile}`);
@@ -103,10 +101,38 @@ export async function deleteSessionFile(sessionId, cwdOverride) {
103
101
  log(`[delete_session] file not found: ${sessionFile}`);
104
102
  }
105
103
  }
104
+ export function getSessionFileInfo(sessionId, cwdOverride) {
105
+ const cwd = cwdOverride && cwdOverride.length > 0 ? cwdOverride : process.cwd();
106
+ const folderName = cwdToProjectFolder(cwd);
107
+ const projectDir = path.join(os.homedir(), '.claude', 'projects', folderName);
108
+ const sessionFile = path.join(projectDir, `${sessionId}.jsonl`);
109
+ return { cwd, folderName, projectDir, sessionFile };
110
+ }
111
+ export function readSessionJsonlLines(sessionId, cwdOverride) {
112
+ const { sessionFile } = getSessionFileInfo(sessionId, cwdOverride);
113
+ if (!fs.existsSync(sessionFile))
114
+ return [];
115
+ return fs.readFileSync(sessionFile, 'utf-8').split('\n').filter((line) => line.trim().length > 0);
116
+ }
117
+ export function writeSessionJsonlLines(sessionId, lines, cwdOverride) {
118
+ const { projectDir, sessionFile } = getSessionFileInfo(sessionId, cwdOverride);
119
+ fs.mkdirSync(projectDir, { recursive: true });
120
+ const tempFile = `${sessionFile}.tmp`;
121
+ const nextContent = lines.length > 0 ? `${lines.join('\n')}\n` : '';
122
+ fs.writeFileSync(tempFile, nextContent, 'utf-8');
123
+ fs.renameSync(tempFile, sessionFile);
124
+ }
125
+ export function backupSessionFile(sessionId, cwdOverride) {
126
+ const { sessionFile } = getSessionFileInfo(sessionId, cwdOverride);
127
+ if (!fs.existsSync(sessionFile))
128
+ return null;
129
+ const backupFile = `${sessionFile}.rollback-${Date.now()}.bak`;
130
+ fs.copyFileSync(sessionFile, backupFile);
131
+ return backupFile;
132
+ }
106
133
  // read a session jsonl file and materialize messages for the app
107
134
  export function readSessionHistory(sessionId) {
108
- const folderName = cwdToProjectFolder(process.cwd());
109
- const sessionFile = path.join(os.homedir(), '.claude', 'projects', folderName, `${sessionId}.jsonl`);
135
+ const { sessionFile } = getSessionFileInfo(sessionId);
110
136
  if (!fs.existsSync(sessionFile))
111
137
  return [];
112
138
  const raw = fs.readFileSync(sessionFile, 'utf-8');
@@ -126,6 +152,6 @@ export function readSessionHistory(sessionId) {
126
152
  }
127
153
  catch { }
128
154
  }
129
- return messages;
155
+ return hydratePromptImages(messages);
130
156
  }
131
157
  //# sourceMappingURL=sessionService.js.map