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.
- package/dist/claude/fileHistory.js +55 -11
- package/dist/claude/fileHistory.js.map +1 -1
- package/dist/handlers/promptImages.js +113 -0
- package/dist/handlers/promptImages.js.map +1 -0
- package/dist/handlers/promptImages.test.js +52 -0
- package/dist/handlers/promptImages.test.js.map +1 -0
- package/dist/handlers/runtimeHandlers.js +15 -3
- package/dist/handlers/runtimeHandlers.js.map +1 -1
- package/dist/handlers/sessionHandlers.js +51 -0
- package/dist/handlers/sessionHandlers.js.map +1 -1
- package/dist/index.js +1 -0
- package/dist/index.js.map +1 -1
- package/dist/pty/spawn.js +31 -5
- package/dist/pty/spawn.js.map +1 -1
- package/dist/pty/spawn.test.js +16 -0
- package/dist/pty/spawn.test.js.map +1 -0
- package/dist/services/checkpointService.js +113 -0
- package/dist/services/checkpointService.js.map +1 -0
- package/dist/services/checkpointService.test.js +61 -0
- package/dist/services/checkpointService.test.js.map +1 -0
- package/dist/services/cliRuntime.js.map +1 -1
- package/dist/services/sessionRollbackService.js +195 -0
- package/dist/services/sessionRollbackService.js.map +1 -0
- package/dist/services/sessionRollbackService.test.js +142 -0
- package/dist/services/sessionRollbackService.test.js.map +1 -0
- package/dist/services/sessionService.js +33 -7
- package/dist/services/sessionService.js.map +1 -1
- package/package.json +2 -2
- package/scripts/run-tests.mjs +1 -0
|
@@ -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;
|
|
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
|
|
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
|
|
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
|