kumo-cli 1.0.6 → 1.1.0
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 +131 -127
- package/dist/auth/credentialsStore.js +1 -1
- package/dist/auth/credentialsStore.js.map +1 -1
- package/dist/claude/transcriptParser.js +110 -11
- package/dist/claude/transcriptParser.js.map +1 -1
- package/dist/core/config.js +1 -4
- package/dist/core/config.js.map +1 -1
- package/dist/core/logger.js +6 -18
- package/dist/core/logger.js.map +1 -1
- package/dist/engine/claudeArgs.js +102 -0
- package/dist/engine/claudeArgs.js.map +1 -0
- package/dist/engine/claudeBinary.js +85 -0
- package/dist/engine/claudeBinary.js.map +1 -0
- package/dist/engine/claudeConfig.js +113 -0
- package/dist/engine/claudeConfig.js.map +1 -0
- package/dist/engine/controlChannel.js +63 -0
- package/dist/engine/controlChannel.js.map +1 -0
- package/dist/engine/liveTranscript.js +101 -0
- package/dist/engine/liveTranscript.js.map +1 -0
- package/dist/engine/ndjson.js +43 -0
- package/dist/engine/ndjson.js.map +1 -0
- package/dist/engine/permissions.js +98 -0
- package/dist/engine/permissions.js.map +1 -0
- package/dist/engine/sessionManager.js +429 -0
- package/dist/engine/sessionManager.js.map +1 -0
- package/dist/engine/settingsCommands.js +204 -0
- package/dist/engine/settingsCommands.js.map +1 -0
- package/dist/engine/streamSession.js +509 -0
- package/dist/engine/streamSession.js.map +1 -0
- package/dist/engine/taskTracker.js +110 -0
- package/dist/engine/taskTracker.js.map +1 -0
- package/dist/handlers/promptImages.js +8 -39
- package/dist/handlers/promptImages.js.map +1 -1
- package/dist/handlers/runtimeHandlers.js +195 -27
- package/dist/handlers/runtimeHandlers.js.map +1 -1
- package/dist/handlers/sessionHandlers.js +41 -70
- package/dist/handlers/sessionHandlers.js.map +1 -1
- package/dist/index.js +35 -40
- package/dist/index.js.map +1 -1
- package/dist/services/cliRuntime.js +37 -30
- package/dist/services/cliRuntime.js.map +1 -1
- package/dist/services/effortService.js +13 -60
- package/dist/services/effortService.js.map +1 -1
- package/dist/services/forkService.js +120 -0
- package/dist/services/forkService.js.map +1 -0
- package/dist/services/modelService.js +37 -84
- package/dist/services/modelService.js.map +1 -1
- package/dist/services/sessionListService.js +7 -9
- package/dist/services/sessionListService.js.map +1 -1
- package/dist/services/sessionService.js +77 -57
- package/dist/services/sessionService.js.map +1 -1
- package/dist/services/tunnelService.js +4 -5
- package/dist/services/tunnelService.js.map +1 -1
- package/dist/transport/wsClient.js +0 -4
- package/dist/transport/wsClient.js.map +1 -1
- package/dist/utils/ignorePaths.js +3 -3
- package/dist/utils/ignorePaths.js.map +1 -1
- package/package.json +62 -69
- package/dist/ansiCodes.js +0 -107
- package/dist/ansiCodes.js.map +0 -1
- package/dist/auth/credentialsStore.test.js +0 -60
- package/dist/auth/credentialsStore.test.js.map +0 -1
- package/dist/claude/fileHistory.js +0 -79
- package/dist/claude/fileHistory.js.map +0 -1
- package/dist/claude/generateHookSettings.js +0 -23
- package/dist/claude/generateHookSettings.js.map +0 -1
- package/dist/claude/hookServer.js +0 -33
- package/dist/claude/hookServer.js.map +0 -1
- package/dist/claude/sessionScanner.js +0 -162
- package/dist/claude/sessionScanner.js.map +0 -1
- package/dist/claude/sessionScanner.test.js +0 -24
- package/dist/claude/sessionScanner.test.js.map +0 -1
- package/dist/claude/transcriptParser.test.js +0 -78
- package/dist/claude/transcriptParser.test.js.map +0 -1
- package/dist/commands/cliArgs.test.js +0 -25
- package/dist/commands/cliArgs.test.js.map +0 -1
- package/dist/config.js +0 -40
- package/dist/config.js.map +0 -1
- package/dist/credentialsStore.js +0 -57
- package/dist/credentialsStore.js.map +0 -1
- package/dist/generateHookSettings.js +0 -30
- package/dist/generateHookSettings.js.map +0 -1
- package/dist/handlers/fileHandlerPayloads.test.js +0 -55
- package/dist/handlers/fileHandlerPayloads.test.js.map +0 -1
- package/dist/handlers/promptImages.test.js +0 -52
- package/dist/handlers/promptImages.test.js.map +0 -1
- package/dist/hookServer.js +0 -32
- package/dist/hookServer.js.map +0 -1
- package/dist/ignorePaths.js +0 -68
- package/dist/ignorePaths.js.map +0 -1
- package/dist/loadDotEnv.js +0 -79
- package/dist/loadDotEnv.js.map +0 -1
- package/dist/logger.js +0 -29
- package/dist/logger.js.map +0 -1
- package/dist/portForwardManager.js +0 -97
- package/dist/portForwardManager.js.map +0 -1
- package/dist/projectHistoryReader.js +0 -85
- package/dist/projectHistoryReader.js.map +0 -1
- package/dist/pty/ansiCodes.js +0 -14
- package/dist/pty/ansiCodes.js.map +0 -1
- package/dist/pty/notificationDetector.js +0 -29
- package/dist/pty/notificationDetector.js.map +0 -1
- package/dist/pty/notificationDetector.test.js +0 -24
- package/dist/pty/notificationDetector.test.js.map +0 -1
- package/dist/pty/ptyMenuParser.js +0 -91
- package/dist/pty/ptyMenuParser.js.map +0 -1
- package/dist/pty/ptyMenuParser.test.js +0 -26
- package/dist/pty/ptyMenuParser.test.js.map +0 -1
- package/dist/pty/ptySnapshotRenderer.js +0 -71
- package/dist/pty/ptySnapshotRenderer.js.map +0 -1
- package/dist/pty/selectionMenuDetector.js +0 -49
- package/dist/pty/selectionMenuDetector.js.map +0 -1
- package/dist/pty/selectionMenuDetector.test.js +0 -62
- package/dist/pty/selectionMenuDetector.test.js.map +0 -1
- package/dist/pty/selectionMenuMeta.js +0 -54
- package/dist/pty/selectionMenuMeta.js.map +0 -1
- package/dist/pty/selectionMenuMeta.test.js +0 -26
- package/dist/pty/selectionMenuMeta.test.js.map +0 -1
- package/dist/pty/selectionMenuParser.js +0 -83
- package/dist/pty/selectionMenuParser.js.map +0 -1
- package/dist/pty/selectionMenuParser.test.js +0 -30
- package/dist/pty/selectionMenuParser.test.js.map +0 -1
- package/dist/pty/spawn.js +0 -104
- package/dist/pty/spawn.js.map +0 -1
- package/dist/pty/spawn.test.js +0 -17
- package/dist/pty/spawn.test.js.map +0 -1
- package/dist/pty/statusSessionDetector.js +0 -28
- package/dist/pty/statusSessionDetector.js.map +0 -1
- package/dist/pty/statusSessionDetector.test.js +0 -23
- package/dist/pty/statusSessionDetector.test.js.map +0 -1
- package/dist/pty/textInputDetector.js +0 -60
- package/dist/pty/textInputDetector.js.map +0 -1
- package/dist/pty/textInputDetector.test.js +0 -32
- package/dist/pty/textInputDetector.test.js.map +0 -1
- package/dist/ptyMenuParser.js +0 -470
- package/dist/ptyMenuParser.js.map +0 -1
- package/dist/ptySnapshotRenderer.js +0 -78
- package/dist/ptySnapshotRenderer.js.map +0 -1
- package/dist/services/checkpointService.js +0 -113
- package/dist/services/checkpointService.js.map +0 -1
- package/dist/services/checkpointService.test.js +0 -61
- package/dist/services/checkpointService.test.js.map +0 -1
- package/dist/services/claudeMenuBridge.js +0 -116
- package/dist/services/claudeMenuBridge.js.map +0 -1
- package/dist/services/claudeMenuBridge.test.js +0 -90
- package/dist/services/claudeMenuBridge.test.js.map +0 -1
- package/dist/services/claudeService.js +0 -146
- package/dist/services/claudeService.js.map +0 -1
- package/dist/services/claudeService.test.js +0 -39
- package/dist/services/claudeService.test.js.map +0 -1
- package/dist/services/cliRuntime.test.js +0 -55
- package/dist/services/cliRuntime.test.js.map +0 -1
- package/dist/services/modelAdapter.test.js +0 -47
- package/dist/services/modelAdapter.test.js.map +0 -1
- package/dist/services/modelArgs.test.js +0 -64
- package/dist/services/modelArgs.test.js.map +0 -1
- package/dist/services/modelBaseUrl.test.js +0 -32
- package/dist/services/modelBaseUrl.test.js.map +0 -1
- package/dist/services/modelService.test.js +0 -136
- package/dist/services/modelService.test.js.map +0 -1
- package/dist/services/notificationDedupe.js +0 -23
- package/dist/services/notificationDedupe.js.map +0 -1
- package/dist/services/notificationDedupe.test.js +0 -20
- package/dist/services/notificationDedupe.test.js.map +0 -1
- package/dist/services/sessionListService.test.js +0 -22
- package/dist/services/sessionListService.test.js.map +0 -1
- package/dist/services/sessionRollbackService.js +0 -195
- package/dist/services/sessionRollbackService.js.map +0 -1
- package/dist/services/sessionRollbackService.test.js +0 -142
- package/dist/services/sessionRollbackService.test.js.map +0 -1
- package/dist/services/tunnelHelpers.test.js +0 -42
- package/dist/services/tunnelHelpers.test.js.map +0 -1
- package/dist/services/tunnelService.test.js +0 -91
- package/dist/services/tunnelService.test.js.map +0 -1
- package/dist/sessionScanner.js +0 -271
- package/dist/sessionScanner.js.map +0 -1
- package/dist/snapshotScanner.js +0 -154
- package/dist/snapshotScanner.js.map +0 -1
- package/dist/spawn.js +0 -79
- package/dist/spawn.js.map +0 -1
- package/dist/terminalManager.js +0 -73
- package/dist/terminalManager.js.map +0 -1
- package/dist/transport/directWsServer.js +0 -135
- package/dist/transport/directWsServer.js.map +0 -1
- package/dist/transport/webrtcTransport.js +0 -145
- package/dist/transport/webrtcTransport.js.map +0 -1
- package/dist/transport/webrtcTransport.test.js +0 -104
- package/dist/transport/webrtcTransport.test.js.map +0 -1
- package/dist/transport/webrtcTypes.js +0 -29
- package/dist/transport/webrtcTypes.js.map +0 -1
- package/dist/transport/webrtcTypes.test.js +0 -20
- package/dist/transport/webrtcTypes.test.js.map +0 -1
- package/dist/utils/ptyBuffer.js +0 -46
- package/dist/utils/ptyBuffer.js.map +0 -1
- package/scripts/claude_launcher.cjs +0 -71
- package/scripts/run-tests.mjs +0 -46
- package/scripts/session_hook_forwarder.cjs +0 -44
|
@@ -1,195 +0,0 @@
|
|
|
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
|
|
@@ -1 +0,0 @@
|
|
|
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"}
|
|
@@ -1,142 +0,0 @@
|
|
|
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
|
|
@@ -1 +0,0 @@
|
|
|
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"}
|
|
@@ -1,42 +0,0 @@
|
|
|
1
|
-
import test from 'node:test';
|
|
2
|
-
import assert from 'node:assert/strict';
|
|
3
|
-
import { decodeTunnelBody, filterTunnelRequestHeaders, normalizeTunnelPath, resolveTunnelHostHeader, rewriteTunnelResponseHeaders, } from './tunnelHelpers.js';
|
|
4
|
-
test('normalizeTunnelPath falls back to root for malformed paths', () => {
|
|
5
|
-
assert.equal(normalizeTunnelPath('/api/health'), '/api/health');
|
|
6
|
-
assert.equal(normalizeTunnelPath('//evil'), '/');
|
|
7
|
-
});
|
|
8
|
-
test('resolveTunnelHostHeader appends port when missing', () => {
|
|
9
|
-
assert.equal(resolveTunnelHostHeader(3000, 'localhost'), 'localhost:3000');
|
|
10
|
-
assert.equal(resolveTunnelHostHeader(3000, 'localhost:4000'), 'localhost:4000');
|
|
11
|
-
});
|
|
12
|
-
test('decodeTunnelBody skips get requests and decodes post payloads', () => {
|
|
13
|
-
assert.equal(decodeTunnelBody({ requestId: '1', method: 'GET', path: '/', headers: {}, port: 3000, body: 'aGk=' }), undefined);
|
|
14
|
-
assert.equal(decodeTunnelBody({ requestId: '1', method: 'POST', path: '/', headers: {}, port: 3000, body: Buffer.from('hi').toString('base64') })?.toString('utf-8'), 'hi');
|
|
15
|
-
});
|
|
16
|
-
test('filterTunnelRequestHeaders strips hop-by-hop headers and rewrites local origins', () => {
|
|
17
|
-
const headers = filterTunnelRequestHeaders({
|
|
18
|
-
Host: 'example.com',
|
|
19
|
-
Connection: 'keep-alive',
|
|
20
|
-
Origin: 'http://localhost:8080',
|
|
21
|
-
Referer: 'http://127.0.0.1:8080/app',
|
|
22
|
-
Accept: 'application/json',
|
|
23
|
-
}, '127.0.0.1', 3000, 'localhost:3000');
|
|
24
|
-
assert.deepEqual(headers, {
|
|
25
|
-
Origin: 'http://127.0.0.1:3000',
|
|
26
|
-
Referer: 'http://127.0.0.1:3000/app',
|
|
27
|
-
Accept: 'application/json',
|
|
28
|
-
host: 'localhost:3000',
|
|
29
|
-
});
|
|
30
|
-
});
|
|
31
|
-
test('rewriteTunnelResponseHeaders strips local redirect origins and cookie domain', () => {
|
|
32
|
-
const headers = rewriteTunnelResponseHeaders({
|
|
33
|
-
location: 'http://127.0.0.1:3000/login?next=%2F',
|
|
34
|
-
'set-cookie': 'sid=abc; Domain=localhost; Path=/; Secure',
|
|
35
|
-
'content-length': '10',
|
|
36
|
-
}, '127.0.0.1', 3000, 'localhost:3000');
|
|
37
|
-
assert.deepEqual(headers, {
|
|
38
|
-
location: '/login?next=%2F',
|
|
39
|
-
'set-cookie': 'sid=abc; Path=/',
|
|
40
|
-
});
|
|
41
|
-
});
|
|
42
|
-
//# sourceMappingURL=tunnelHelpers.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"tunnelHelpers.test.js","sourceRoot":"","sources":["../../src/services/tunnelHelpers.test.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EACL,gBAAgB,EAChB,0BAA0B,EAC1B,mBAAmB,EACnB,uBAAuB,EACvB,4BAA4B,GAC7B,MAAM,oBAAoB,CAAC;AAE5B,IAAI,CAAC,4DAA4D,EAAE,GAAG,EAAE;IACtE,MAAM,CAAC,KAAK,CAAC,mBAAmB,CAAC,aAAa,CAAC,EAAE,aAAa,CAAC,CAAC;IAChE,MAAM,CAAC,KAAK,CAAC,mBAAmB,CAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,CAAC;AACnD,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,mDAAmD,EAAE,GAAG,EAAE;IAC7D,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,IAAI,EAAE,WAAW,CAAC,EAAE,gBAAgB,CAAC,CAAC;IAC3E,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,IAAI,EAAE,gBAAgB,CAAC,EAAE,gBAAgB,CAAC,CAAC;AAClF,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+DAA+D,EAAE,GAAG,EAAE;IACzE,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;IAC/H,MAAM,CAAC,KAAK,CACV,gBAAgB,CAAC,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,QAAQ,CAAC,OAAO,CAAC,EACvJ,IAAI,CACL,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,iFAAiF,EAAE,GAAG,EAAE;IAC3F,MAAM,OAAO,GAAG,0BAA0B,CACxC;QACE,IAAI,EAAE,aAAa;QACnB,UAAU,EAAE,YAAY;QACxB,MAAM,EAAE,uBAAuB;QAC/B,OAAO,EAAE,2BAA2B;QACpC,MAAM,EAAE,kBAAkB;KAC3B,EACD,WAAW,EACX,IAAI,EACJ,gBAAgB,CACjB,CAAC;IAEF,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE;QACxB,MAAM,EAAE,uBAAuB;QAC/B,OAAO,EAAE,2BAA2B;QACpC,MAAM,EAAE,kBAAkB;QAC1B,IAAI,EAAE,gBAAgB;KACvB,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8EAA8E,EAAE,GAAG,EAAE;IACxF,MAAM,OAAO,GAAG,4BAA4B,CAC1C;QACE,QAAQ,EAAE,sCAAsC;QAChD,YAAY,EAAE,2CAA2C;QACzD,gBAAgB,EAAE,IAAI;KACvB,EACD,WAAW,EACX,IAAI,EACJ,gBAAgB,CACjB,CAAC;IAEF,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE;QACxB,QAAQ,EAAE,iBAAiB;QAC3B,YAAY,EAAE,iBAAiB;KAChC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1,91 +0,0 @@
|
|
|
1
|
-
import test from 'node:test';
|
|
2
|
-
import assert from 'node:assert/strict';
|
|
3
|
-
import http from 'node:http';
|
|
4
|
-
import { handleTunnelRequest } from './tunnelService.js';
|
|
5
|
-
test('preserves browser client hint headers through tunnel proxy', async () => {
|
|
6
|
-
const seenHeaders = await new Promise((resolve, reject) => {
|
|
7
|
-
const server = http.createServer((req, res) => {
|
|
8
|
-
resolve(req.headers);
|
|
9
|
-
res.end('ok');
|
|
10
|
-
server.close();
|
|
11
|
-
});
|
|
12
|
-
server.on('error', reject);
|
|
13
|
-
server.listen(0, '127.0.0.1', async () => {
|
|
14
|
-
const address = server.address();
|
|
15
|
-
if (!address || typeof address === 'string')
|
|
16
|
-
return reject(new Error('server address unavailable'));
|
|
17
|
-
const response = await handleTunnelRequest({
|
|
18
|
-
requestId: 'req-1',
|
|
19
|
-
method: 'GET',
|
|
20
|
-
path: '/',
|
|
21
|
-
port: address.port,
|
|
22
|
-
headers: {
|
|
23
|
-
'user-agent': 'Mozilla/5.0 Chrome/120.0.0.0',
|
|
24
|
-
'sec-ch-ua': '"Chromium";v="120"',
|
|
25
|
-
'sec-ch-ua-mobile': '?0',
|
|
26
|
-
'sec-ch-ua-platform': '"Windows"',
|
|
27
|
-
'upgrade-insecure-requests': '1',
|
|
28
|
-
connection: 'keep-alive',
|
|
29
|
-
host: 'provider.ecologynet.local:3000',
|
|
30
|
-
},
|
|
31
|
-
});
|
|
32
|
-
assert.equal(response.status, 200);
|
|
33
|
-
});
|
|
34
|
-
});
|
|
35
|
-
assert.equal(seenHeaders['user-agent'], 'Mozilla/5.0 Chrome/120.0.0.0');
|
|
36
|
-
assert.equal(seenHeaders['sec-ch-ua'], '"Chromium";v="120"');
|
|
37
|
-
assert.equal(seenHeaders['sec-ch-ua-mobile'], '?0');
|
|
38
|
-
assert.equal(seenHeaders['sec-ch-ua-platform'], '"Windows"');
|
|
39
|
-
assert.equal(seenHeaders['upgrade-insecure-requests'], '1');
|
|
40
|
-
assert.equal(seenHeaders.connection, 'keep-alive');
|
|
41
|
-
assert.notEqual(seenHeaders.host, 'provider.ecologynet.local:3000');
|
|
42
|
-
});
|
|
43
|
-
test('replaces non-browser user agents for browser-gated local apps', async () => {
|
|
44
|
-
const seenHeaders = await new Promise((resolve, reject) => {
|
|
45
|
-
const server = http.createServer((req, res) => {
|
|
46
|
-
resolve(req.headers);
|
|
47
|
-
res.end('ok');
|
|
48
|
-
server.close();
|
|
49
|
-
});
|
|
50
|
-
server.on('error', reject);
|
|
51
|
-
server.listen(0, '127.0.0.1', async () => {
|
|
52
|
-
const address = server.address();
|
|
53
|
-
if (!address || typeof address === 'string')
|
|
54
|
-
return reject(new Error('server address unavailable'));
|
|
55
|
-
const response = await handleTunnelRequest({
|
|
56
|
-
requestId: 'req-2',
|
|
57
|
-
method: 'GET',
|
|
58
|
-
path: '/',
|
|
59
|
-
port: address.port,
|
|
60
|
-
headers: { 'user-agent': 'Dart/3.0 (dart:io)' },
|
|
61
|
-
});
|
|
62
|
-
assert.equal(response.status, 200);
|
|
63
|
-
});
|
|
64
|
-
});
|
|
65
|
-
assert.match(seenHeaders['user-agent'] ?? '', /Chrome\/120/);
|
|
66
|
-
});
|
|
67
|
-
test('modernizes old browser user agents for Rails allow_browser modern gate', async () => {
|
|
68
|
-
const seenHeaders = await new Promise((resolve, reject) => {
|
|
69
|
-
const server = http.createServer((req, res) => {
|
|
70
|
-
resolve(req.headers);
|
|
71
|
-
res.end('ok');
|
|
72
|
-
server.close();
|
|
73
|
-
});
|
|
74
|
-
server.on('error', reject);
|
|
75
|
-
server.listen(0, '127.0.0.1', async () => {
|
|
76
|
-
const address = server.address();
|
|
77
|
-
if (!address || typeof address === 'string')
|
|
78
|
-
return reject(new Error('server address unavailable'));
|
|
79
|
-
const response = await handleTunnelRequest({
|
|
80
|
-
requestId: 'req-3',
|
|
81
|
-
method: 'GET',
|
|
82
|
-
path: '/',
|
|
83
|
-
port: address.port,
|
|
84
|
-
headers: { 'user-agent': 'Mozilla/5.0 AppleWebKit/537.36 Chrome/119.0.0.0 Safari/537.36' },
|
|
85
|
-
});
|
|
86
|
-
assert.equal(response.status, 200);
|
|
87
|
-
});
|
|
88
|
-
});
|
|
89
|
-
assert.equal(seenHeaders['user-agent'], 'Mozilla/5.0 AppleWebKit/537.36 Chrome/120.0.0.0 Safari/537.36');
|
|
90
|
-
});
|
|
91
|
-
//# sourceMappingURL=tunnelService.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"tunnelService.test.js","sourceRoot":"","sources":["../../src/services/tunnelService.test.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAEzD,IAAI,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;IAC5E,MAAM,WAAW,GAAG,MAAM,IAAI,OAAO,CAA2B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAClF,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YAC5C,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACrB,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACd,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ;gBAAE,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC,CAAC;YACpG,MAAM,QAAQ,GAAG,MAAM,mBAAmB,CAAC;gBACzC,SAAS,EAAE,OAAO;gBAClB,MAAM,EAAE,KAAK;gBACb,IAAI,EAAE,GAAG;gBACT,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,OAAO,EAAE;oBACP,YAAY,EAAE,8BAA8B;oBAC5C,WAAW,EAAE,oBAAoB;oBACjC,kBAAkB,EAAE,IAAI;oBACxB,oBAAoB,EAAE,WAAW;oBACjC,2BAA2B,EAAE,GAAG;oBAChC,UAAU,EAAE,YAAY;oBACxB,IAAI,EAAE,gCAAgC;iBACvC;aACF,CAAC,CAAC;YACH,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,YAAY,CAAC,EAAE,8BAA8B,CAAC,CAAC;IACxE,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE,oBAAoB,CAAC,CAAC;IAC7D,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,kBAAkB,CAAC,EAAE,IAAI,CAAC,CAAC;IACpD,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,oBAAoB,CAAC,EAAE,WAAW,CAAC,CAAC;IAC7D,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,2BAA2B,CAAC,EAAE,GAAG,CAAC,CAAC;IAC5D,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;IACnD,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,EAAE,gCAAgC,CAAC,CAAC;AACtE,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;IAC/E,MAAM,WAAW,GAAG,MAAM,IAAI,OAAO,CAA2B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAClF,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YAC5C,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACrB,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACd,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ;gBAAE,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC,CAAC;YACpG,MAAM,QAAQ,GAAG,MAAM,mBAAmB,CAAC;gBACzC,SAAS,EAAE,OAAO;gBAClB,MAAM,EAAE,KAAK;gBACb,IAAI,EAAE,GAAG;gBACT,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,OAAO,EAAE,EAAE,YAAY,EAAE,oBAAoB,EAAE;aAChD,CAAC,CAAC;YACH,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,YAAY,CAAC,IAAI,EAAE,EAAE,aAAa,CAAC,CAAC;AAC/D,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;IACxF,MAAM,WAAW,GAAG,MAAM,IAAI,OAAO,CAA2B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAClF,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YAC5C,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACrB,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACd,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ;gBAAE,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC,CAAC;YACpG,MAAM,QAAQ,GAAG,MAAM,mBAAmB,CAAC;gBACzC,SAAS,EAAE,OAAO;gBAClB,MAAM,EAAE,KAAK;gBACb,IAAI,EAAE,GAAG;gBACT,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,OAAO,EAAE,EAAE,YAAY,EAAE,+DAA+D,EAAE;aAC3F,CAAC,CAAC;YACH,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,YAAY,CAAC,EAAE,+DAA+D,CAAC,CAAC;AAC3G,CAAC,CAAC,CAAC"}
|