kumo-cli 1.0.3 → 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/README.md +13 -5
- package/dist/auth/credentialsStore.js +61 -14
- package/dist/auth/credentialsStore.js.map +1 -1
- package/dist/auth/credentialsStore.test.js +60 -0
- package/dist/auth/credentialsStore.test.js.map +1 -0
- package/dist/claude/fileHistory.js +79 -0
- package/dist/claude/fileHistory.js.map +1 -0
- package/dist/claude/sessionScanner.js +7 -95
- package/dist/claude/sessionScanner.js.map +1 -1
- package/dist/claude/sessionScanner.test.js +24 -0
- package/dist/claude/sessionScanner.test.js.map +1 -0
- package/dist/claude/transcriptParser.js +83 -0
- package/dist/claude/transcriptParser.js.map +1 -0
- package/dist/claude/transcriptParser.test.js +78 -0
- package/dist/claude/transcriptParser.test.js.map +1 -0
- package/dist/commands/agentPrompt.js +18 -0
- package/dist/commands/agentPrompt.js.map +1 -0
- package/dist/commands/cliArgs.js +56 -0
- package/dist/commands/cliArgs.js.map +1 -0
- package/dist/commands/cliArgs.test.js +25 -0
- package/dist/commands/cliArgs.test.js.map +1 -0
- package/dist/core/logger.js +7 -2
- package/dist/core/logger.js.map +1 -1
- package/dist/core/projectFileWatcher.js +56 -0
- package/dist/core/projectFileWatcher.js.map +1 -0
- package/dist/handlers/fileHandlerPayloads.js +72 -0
- package/dist/handlers/fileHandlerPayloads.js.map +1 -0
- package/dist/handlers/fileHandlerPayloads.test.js +55 -0
- package/dist/handlers/fileHandlerPayloads.test.js.map +1 -0
- package/dist/handlers/fileHandlers.js +74 -0
- package/dist/handlers/fileHandlers.js.map +1 -0
- package/dist/handlers/messageRouter.js +8 -205
- package/dist/handlers/messageRouter.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/router.types.js +2 -0
- package/dist/handlers/router.types.js.map +1 -0
- package/dist/handlers/runtimeHandlers.js +67 -0
- package/dist/handlers/runtimeHandlers.js.map +1 -0
- package/dist/handlers/sessionHandlers.js +119 -0
- package/dist/handlers/sessionHandlers.js.map +1 -0
- package/dist/handlers/terminalHandlers.js +40 -0
- package/dist/handlers/terminalHandlers.js.map +1 -0
- package/dist/index.js +60 -145
- package/dist/index.js.map +1 -1
- package/dist/pty/notificationDetector.js +29 -0
- package/dist/pty/notificationDetector.js.map +1 -0
- package/dist/pty/notificationDetector.test.js +24 -0
- package/dist/pty/notificationDetector.test.js.map +1 -0
- package/dist/pty/ptyMenuParser.js +11 -277
- package/dist/pty/ptyMenuParser.js.map +1 -1
- package/dist/pty/ptyMenuParser.test.js +26 -0
- package/dist/pty/ptyMenuParser.test.js.map +1 -0
- package/dist/pty/selectionMenuDetector.js +49 -0
- package/dist/pty/selectionMenuDetector.js.map +1 -0
- package/dist/pty/selectionMenuDetector.test.js +62 -0
- package/dist/pty/selectionMenuDetector.test.js.map +1 -0
- package/dist/pty/selectionMenuMeta.js +54 -0
- package/dist/pty/selectionMenuMeta.js.map +1 -0
- package/dist/pty/selectionMenuMeta.test.js +26 -0
- package/dist/pty/selectionMenuMeta.test.js.map +1 -0
- package/dist/pty/selectionMenuParser.js +83 -0
- package/dist/pty/selectionMenuParser.js.map +1 -0
- package/dist/pty/selectionMenuParser.test.js +30 -0
- package/dist/pty/selectionMenuParser.test.js.map +1 -0
- 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/pty/statusSessionDetector.js +28 -0
- package/dist/pty/statusSessionDetector.js.map +1 -0
- package/dist/pty/statusSessionDetector.test.js +23 -0
- package/dist/pty/statusSessionDetector.test.js.map +1 -0
- package/dist/pty/terminalManager.js +11 -6
- package/dist/pty/terminalManager.js.map +1 -1
- package/dist/pty/textInputDetector.js +60 -0
- package/dist/pty/textInputDetector.js.map +1 -0
- package/dist/pty/textInputDetector.test.js +32 -0
- package/dist/pty/textInputDetector.test.js.map +1 -0
- package/dist/services/bootstrapService.js +28 -0
- package/dist/services/bootstrapService.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/claudeMenuBridge.js +116 -0
- package/dist/services/claudeMenuBridge.js.map +1 -0
- package/dist/services/claudeMenuBridge.test.js +90 -0
- package/dist/services/claudeMenuBridge.test.js.map +1 -0
- package/dist/services/claudeService.js +5 -98
- package/dist/services/claudeService.js.map +1 -1
- package/dist/services/cliRuntime.js +49 -0
- package/dist/services/cliRuntime.js.map +1 -0
- package/dist/services/cliRuntime.test.js +55 -0
- package/dist/services/cliRuntime.test.js.map +1 -0
- package/dist/services/fileService.js +19 -23
- package/dist/services/fileService.js.map +1 -1
- package/dist/services/modelAdapter.test.js +47 -0
- package/dist/services/modelAdapter.test.js.map +1 -0
- package/dist/services/modelArgs.test.js +64 -0
- package/dist/services/modelArgs.test.js.map +1 -0
- package/dist/services/modelBaseUrl.test.js +32 -0
- package/dist/services/modelBaseUrl.test.js.map +1 -0
- package/dist/services/modelService.js +20 -9
- package/dist/services/modelService.js.map +1 -1
- package/dist/services/modelService.test.js +60 -0
- package/dist/services/modelService.test.js.map +1 -1
- package/dist/services/notificationDedupe.js +23 -0
- package/dist/services/notificationDedupe.js.map +1 -0
- package/dist/services/notificationDedupe.test.js +20 -0
- package/dist/services/notificationDedupe.test.js.map +1 -0
- package/dist/services/pairingService.js +17 -8
- package/dist/services/pairingService.js.map +1 -1
- package/dist/services/sessionListService.js +19 -0
- package/dist/services/sessionListService.js.map +1 -0
- package/dist/services/sessionListService.test.js +22 -0
- package/dist/services/sessionListService.test.js.map +1 -0
- 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 +41 -52
- package/dist/services/sessionService.js.map +1 -1
- package/dist/services/tunnelHelpers.js +69 -0
- package/dist/services/tunnelHelpers.js.map +1 -0
- package/dist/services/tunnelHelpers.test.js +42 -0
- package/dist/services/tunnelHelpers.test.js.map +1 -0
- package/dist/services/tunnelService.js +69 -25
- package/dist/services/tunnelService.js.map +1 -1
- package/dist/services/tunnelService.test.js +91 -0
- package/dist/services/tunnelService.test.js.map +1 -0
- package/dist/transport/webrtcTransport.js +145 -0
- package/dist/transport/webrtcTransport.js.map +1 -0
- package/dist/transport/webrtcTransport.test.js +104 -0
- package/dist/transport/webrtcTransport.test.js.map +1 -0
- package/dist/transport/webrtcTypes.js +29 -0
- package/dist/transport/webrtcTypes.js.map +1 -0
- package/dist/transport/webrtcTypes.test.js +20 -0
- package/dist/transport/webrtcTypes.test.js.map +1 -0
- package/dist/transport/wsClient.js +16 -3
- package/dist/transport/wsClient.js.map +1 -1
- package/dist/utils/safePath.js +18 -6
- package/dist/utils/safePath.js.map +1 -1
- package/package.json +5 -2
- package/scripts/run-tests.mjs +46 -0
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import test from 'node:test';
|
|
2
|
+
import assert from 'node:assert/strict';
|
|
3
|
+
import { mergeSessionActivity } from './sessionListService.js';
|
|
4
|
+
test('mergeSessionActivity marks persisted active sessions', () => {
|
|
5
|
+
const sessions = [
|
|
6
|
+
{ sessionId: 's1', firstMessage: 'hello', lastTimestamp: '2026-06-22T10:00:00.000Z' },
|
|
7
|
+
{ sessionId: 's2', firstMessage: 'world', lastTimestamp: '2026-06-22T09:00:00.000Z' },
|
|
8
|
+
];
|
|
9
|
+
const result = mergeSessionActivity(sessions, new Set(['s2']));
|
|
10
|
+
assert.deepEqual(result, [
|
|
11
|
+
{ sessionId: 's1', firstMessage: 'hello', lastTimestamp: '2026-06-22T10:00:00.000Z', isActive: false },
|
|
12
|
+
{ sessionId: 's2', firstMessage: 'world', lastTimestamp: '2026-06-22T09:00:00.000Z', isActive: true },
|
|
13
|
+
]);
|
|
14
|
+
});
|
|
15
|
+
test('mergeSessionActivity includes the current runtime session id', () => {
|
|
16
|
+
const sessions = [
|
|
17
|
+
{ sessionId: 's1', firstMessage: 'hello', lastTimestamp: '2026-06-22T10:00:00.000Z' },
|
|
18
|
+
];
|
|
19
|
+
const result = mergeSessionActivity(sessions, new Set(), 's1');
|
|
20
|
+
assert.equal(result[0]?.isActive, true);
|
|
21
|
+
});
|
|
22
|
+
//# sourceMappingURL=sessionListService.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sessionListService.test.js","sourceRoot":"","sources":["../../src/services/sessionListService.test.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAE/D,IAAI,CAAC,sDAAsD,EAAE,GAAG,EAAE;IAChE,MAAM,QAAQ,GAAG;QACf,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,OAAO,EAAE,aAAa,EAAE,0BAA0B,EAAE;QACrF,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,OAAO,EAAE,aAAa,EAAE,0BAA0B,EAAE;KACtF,CAAC;IAEF,MAAM,MAAM,GAAG,oBAAoB,CAAC,QAAQ,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAE/D,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE;QACvB,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,OAAO,EAAE,aAAa,EAAE,0BAA0B,EAAE,QAAQ,EAAE,KAAK,EAAE;QACtG,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,OAAO,EAAE,aAAa,EAAE,0BAA0B,EAAE,QAAQ,EAAE,IAAI,EAAE;KACtG,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8DAA8D,EAAE,GAAG,EAAE;IACxE,MAAM,QAAQ,GAAG;QACf,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,OAAO,EAAE,aAAa,EAAE,0BAA0B,EAAE;KACtF,CAAC;IAEF,MAAM,MAAM,GAAG,oBAAoB,CAAC,QAAQ,EAAE,IAAI,GAAG,EAAU,EAAE,IAAI,CAAC,CAAC;IAEvE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;AAC1C,CAAC,CAAC,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"}
|
|
@@ -3,7 +3,8 @@ import os from 'node:os';
|
|
|
3
3
|
import path from 'node:path';
|
|
4
4
|
import { log } from '../core/logger.js';
|
|
5
5
|
import { cwdToProjectFolder } from '../utils/safePath.js';
|
|
6
|
-
import {
|
|
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');
|
|
@@ -116,53 +142,16 @@ export function readSessionHistory(sessionId) {
|
|
|
116
142
|
if (!line.trim())
|
|
117
143
|
continue;
|
|
118
144
|
try {
|
|
119
|
-
const
|
|
120
|
-
|
|
121
|
-
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
|
|
125
|
-
|
|
126
|
-
if (!role || role === 'system')
|
|
127
|
-
continue;
|
|
128
|
-
if (entry['isMeta'] || entry['sourceToolUseID'])
|
|
129
|
-
continue;
|
|
130
|
-
const uuid = entry['uuid'];
|
|
131
|
-
if (uuid && seenUuids.has(uuid))
|
|
132
|
-
continue;
|
|
133
|
-
if (uuid)
|
|
134
|
-
seenUuids.add(uuid);
|
|
135
|
-
const timestamp = entry['timestamp'];
|
|
136
|
-
let content = '';
|
|
137
|
-
const toolUses = [];
|
|
138
|
-
if (typeof msgObj['content'] === 'string')
|
|
139
|
-
content = msgObj['content'];
|
|
140
|
-
else if (Array.isArray(msgObj['content'])) {
|
|
141
|
-
const arr = msgObj['content'];
|
|
142
|
-
content = arr
|
|
143
|
-
.filter((b) => b['type'] === 'text' && b['text'] && b['text'].trim() !== 'No response requested.')
|
|
144
|
-
.map((b) => b['text']).join('');
|
|
145
|
-
for (const b of arr) {
|
|
146
|
-
if (b['type'] === 'tool_use' && b['name']) {
|
|
147
|
-
const tu = { toolName: b['name'], input: b['input'] ?? {} };
|
|
148
|
-
if (tu.toolName === 'Write') {
|
|
149
|
-
const fp = tu.input['file_path'] ?? tu.input['path'] ?? '';
|
|
150
|
-
if (fp) {
|
|
151
|
-
const oldContent = readFileHistoryPrev(sessionId, fp);
|
|
152
|
-
if (oldContent !== null)
|
|
153
|
-
tu.input['old_content'] = oldContent;
|
|
154
|
-
}
|
|
155
|
-
}
|
|
156
|
-
toolUses.push(tu);
|
|
157
|
-
}
|
|
158
|
-
}
|
|
159
|
-
}
|
|
160
|
-
if (!content && toolUses.length === 0)
|
|
161
|
-
continue;
|
|
162
|
-
messages.push({ role, content, id: uuid, timestamp, toolUses: toolUses.length > 0 ? toolUses : undefined });
|
|
145
|
+
const parsed = parseTranscriptMessage(JSON.parse(line), {
|
|
146
|
+
sessionId,
|
|
147
|
+
seenUuids,
|
|
148
|
+
includeNoise: true,
|
|
149
|
+
});
|
|
150
|
+
if (parsed)
|
|
151
|
+
messages.push(parsed);
|
|
163
152
|
}
|
|
164
153
|
catch { }
|
|
165
154
|
}
|
|
166
|
-
return messages;
|
|
155
|
+
return hydratePromptImages(messages);
|
|
167
156
|
}
|
|
168
157
|
//# sourceMappingURL=sessionService.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"sessionService.js","sourceRoot":"","sources":["../../src/services/sessionService.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,GAAG,EAAE,MAAM,mBAAmB,CAAC;AACxC,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"sessionService.js","sourceRoot":"","sources":["../../src/services/sessionService.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,GAAG,EAAE,MAAM,mBAAmB,CAAC;AACxC,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,sBAAsB,EAAuC,MAAM,+BAA+B,CAAC;AAC5G,OAAO,EAAE,mBAAmB,EAAoB,MAAM,6BAA6B,CAAC;AAwBpF,qDAAqD;AACrD,MAAM,UAAU,gBAAgB,CAAC,IAAc;IAC7C,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACrC,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,UAAU,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI;YAAE,CAAC,EAAE,CAAC;;YAC/C,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5B,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,6DAA6D;AAC7D,MAAM,CAAC,KAAK,UAAU,mBAAmB;IACvC,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;IACnE,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAC;IACjC,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QAC7E,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;gBAChF,IAAI,IAAI,CAAC,SAAS;oBAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACjD,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QACZ,CAAC;IACH,CAAC;IAAC,MAAM,CAAC,CAAA,CAAC;IACV,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,qFAAqF;AACrF,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,WAAoB;IAC7D,MAAM,GAAG,GAAG,WAAW,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;IAChF,MAAM,UAAU,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAC;IAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;IAC9E,GAAG,CAAC,sBAAsB,GAAG,YAAY,UAAU,SAAS,UAAU,EAAE,CAAC,CAAC;IAC1E,MAAM,QAAQ,GAAuB,EAAE,CAAC;IACxC,IAAI,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC;YAAE,OAAO,QAAQ,CAAC;QAChD,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC7E,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;YAChD,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;gBAClE,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;gBACtD,IAAI,YAAY,GAAG,EAAE,CAAC;gBACtB,IAAI,aAAa,GAAG,EAAE,CAAC;gBACvB,IAAI,WAAW,GAAG,KAAK,CAAC;gBACxB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,IAAI,CAAC;wBACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAA4B,CAAC;wBAC1D,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,WAAW;4BAAE,SAAS;wBACxE,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,CAAwC,CAAC;wBACvE,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,KAAK,QAAQ;4BAAE,SAAS;wBACxE,WAAW,GAAG,IAAI,CAAC;wBACnB,MAAM,EAAE,GAAG,KAAK,CAAC,WAAW,CAAuB,CAAC;wBACpD,IAAI,EAAE;4BAAE,aAAa,GAAG,EAAE,CAAC;wBAC3B,IAAI,CAAC,YAAY,IAAI,MAAM,CAAC,MAAM,CAAC,KAAK,MAAM,EAAE,CAAC;4BAC/C,IAAI,IAAI,GAAG,EAAE,CAAC;4BACd,IAAI,OAAO,MAAM,CAAC,SAAS,CAAC,KAAK,QAAQ;gCAAE,IAAI,GAAI,MAAM,CAAC,SAAS,CAAY,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;iCACzF,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC;gCAC1C,IAAI,GAAI,MAAM,CAAC,SAAS,CAAoC;qCACzD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,MAAM,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC;qCAChD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAW,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;4BAC5D,CAAC;4BACD,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC;gCAAE,YAAY,GAAG,IAAI,CAAC;wBAC1D,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC,CAAA,CAAC;gBACZ,CAAC;gBACD,IAAI,WAAW;oBAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,CAAC,CAAC;YAC7E,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QACZ,CAAC;IACH,CAAC;IAAC,MAAM,CAAC,CAAA,CAAC;IACV,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,CAAC;IACtF,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,gDAAgD;AAChD,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,SAAiB,EAAE,WAAoB;IAC7E,MAAM,EAAE,WAAW,EAAE,GAAG,kBAAkB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;IACnE,IAAI,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;QAC/B,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QAC3B,GAAG,CAAC,4BAA4B,WAAW,EAAE,CAAC,CAAC;IACjD,CAAC;SAAM,CAAC;QACN,GAAG,CAAC,oCAAoC,WAAW,EAAE,CAAC,CAAC;IACzD,CAAC;AACH,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,SAAiB,EAAE,WAAoB;IACxE,MAAM,GAAG,GAAG,WAAW,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;IAChF,MAAM,UAAU,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAC;IAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;IAC9E,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,SAAS,QAAQ,CAAC,CAAC;IAChE,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,CAAC;AACtD,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,SAAiB,EAAE,WAAoB;IAC3E,MAAM,EAAE,WAAW,EAAE,GAAG,kBAAkB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;IACnE,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC;QAAE,OAAO,EAAE,CAAC;IAC3C,OAAO,EAAE,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACpG,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,SAAiB,EAAE,KAAe,EAAE,WAAoB;IAC7F,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,GAAG,kBAAkB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;IAC/E,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC9C,MAAM,QAAQ,GAAG,GAAG,WAAW,MAAM,CAAC;IACtC,MAAM,WAAW,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;IACpE,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;IACjD,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;AACvC,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,SAAiB,EAAE,WAAoB;IACvE,MAAM,EAAE,WAAW,EAAE,GAAG,kBAAkB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;IACnE,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC;QAAE,OAAO,IAAI,CAAC;IAC7C,MAAM,UAAU,GAAG,GAAG,WAAW,aAAa,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC;IAC/D,EAAE,CAAC,YAAY,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;IACzC,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,iEAAiE;AACjE,MAAM,UAAU,kBAAkB,CAAC,SAAiB;IAClD,MAAM,EAAE,WAAW,EAAE,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;IACtD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC;QAAE,OAAO,EAAE,CAAC;IAC3C,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;IAClD,MAAM,QAAQ,GAAqB,EAAE,CAAC;IACtC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IACpC,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;QACnC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;YAAE,SAAS;QAC3B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,sBAAsB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAiB,EAAE;gBACtE,SAAS;gBACT,SAAS;gBACT,YAAY,EAAE,IAAI;aACnB,CAAC,CAAC;YACH,IAAI,MAAM;gBAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpC,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;IACZ,CAAC;IACD,OAAO,mBAAmB,CAAC,QAAQ,CAAC,CAAC;AACvC,CAAC"}
|
|
@@ -0,0 +1,69 @@
|
|
|
1
|
+
export const DEFAULT_TUNNEL_HOST = '127.0.0.1';
|
|
2
|
+
// build a stable host header for the proxied local target
|
|
3
|
+
export function resolveTunnelHostHeader(port, requestedHost) {
|
|
4
|
+
const trimmedHost = typeof requestedHost === 'string' ? requestedHost.trim() : '';
|
|
5
|
+
if (!trimmedHost)
|
|
6
|
+
return undefined;
|
|
7
|
+
return trimmedHost.includes(':') ? trimmedHost : `${trimmedHost}:${port}`;
|
|
8
|
+
}
|
|
9
|
+
// validate and normalize the tunnel path to avoid malformed local requests
|
|
10
|
+
export function normalizeTunnelPath(pathValue) {
|
|
11
|
+
return typeof pathValue === 'string' && pathValue.startsWith('/') && !pathValue.startsWith('//') ? pathValue : '/';
|
|
12
|
+
}
|
|
13
|
+
// decode a tunnel request body only for methods that may legally send one
|
|
14
|
+
export function decodeTunnelBody(request) {
|
|
15
|
+
if (!request.body || request.method === 'GET' || request.method === 'HEAD') {
|
|
16
|
+
return undefined;
|
|
17
|
+
}
|
|
18
|
+
return Buffer.from(request.body, 'base64');
|
|
19
|
+
}
|
|
20
|
+
// drop hop-by-hop headers and rewrite local origins before proxying the request
|
|
21
|
+
export function filterTunnelRequestHeaders(headers, targetHost, port, hostHeader) {
|
|
22
|
+
const filteredHeaders = Object.fromEntries(Object.entries(headers)
|
|
23
|
+
.filter(([key]) => {
|
|
24
|
+
const lowerKey = key.toLowerCase();
|
|
25
|
+
return !lowerKey.startsWith('sec-') && !['host', 'connection', 'accept-encoding', 'upgrade-insecure-requests'].includes(lowerKey);
|
|
26
|
+
})
|
|
27
|
+
.map(([key, value]) => {
|
|
28
|
+
const lowerKey = key.toLowerCase();
|
|
29
|
+
if (lowerKey === 'origin' || lowerKey === 'referer') {
|
|
30
|
+
return [key, value.replace(/http:\/\/(?:localhost|127\.0\.0\.1):\d+/g, `http://${targetHost}:${port}`)];
|
|
31
|
+
}
|
|
32
|
+
return [key, value];
|
|
33
|
+
}));
|
|
34
|
+
if (hostHeader)
|
|
35
|
+
filteredHeaders.host = hostHeader;
|
|
36
|
+
return filteredHeaders;
|
|
37
|
+
}
|
|
38
|
+
// remove transport-specific headers and rewrite redirects/cookies for the app side
|
|
39
|
+
export function rewriteTunnelResponseHeaders(headers, targetHost, port, hostHeader) {
|
|
40
|
+
const responseHeaders = Object.fromEntries(Object.entries(headers).filter(([key]) => !['content-encoding', 'transfer-encoding', 'content-length'].includes(key.toLowerCase())));
|
|
41
|
+
const remoteOrigin = `http://${targetHost}:${port}`;
|
|
42
|
+
if (responseHeaders.location) {
|
|
43
|
+
const location = String(responseHeaders.location);
|
|
44
|
+
const absoluteLike = location.startsWith('http://') || location.startsWith('https://') || location.startsWith('//');
|
|
45
|
+
if (absoluteLike) {
|
|
46
|
+
const parsed = new URL(location.startsWith('//') ? `http:${location}` : location);
|
|
47
|
+
const origin = `${parsed.protocol}//${parsed.host}`;
|
|
48
|
+
const originsToStrip = new Set([
|
|
49
|
+
`http://${targetHost}:${port}`,
|
|
50
|
+
`http://localhost:${port}`,
|
|
51
|
+
hostHeader ? `http://${hostHeader}` : '',
|
|
52
|
+
hostHeader ? `https://${hostHeader}` : '',
|
|
53
|
+
].filter(Boolean));
|
|
54
|
+
if (originsToStrip.has(origin)) {
|
|
55
|
+
responseHeaders.location = `${parsed.pathname}${parsed.search}${parsed.hash}`;
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
else {
|
|
59
|
+
responseHeaders.location = location.replace(remoteOrigin, '');
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
if (responseHeaders['set-cookie']) {
|
|
63
|
+
responseHeaders['set-cookie'] = responseHeaders['set-cookie']
|
|
64
|
+
.replace(/;\s*domain=[^;]*/gi, '')
|
|
65
|
+
.replace(/;\s*secure/gi, '');
|
|
66
|
+
}
|
|
67
|
+
return responseHeaders;
|
|
68
|
+
}
|
|
69
|
+
//# sourceMappingURL=tunnelHelpers.js.map
|