38 lines
1.6 KiB
JavaScript
38 lines
1.6 KiB
JavaScript
#!/usr/bin/env node
|
||
'use strict';
|
||
|
||
// 默认预览；--apply 才会让服务端在 idle 状态下备份并原子迁移。
|
||
// CC_WEB_PASSWORD=... node scripts/migrate-session-history.js <会话 ID> [--apply]
|
||
const WebSocket = require('ws');
|
||
|
||
const sessionId = process.argv.slice(2).find((arg) => !arg.startsWith('--'));
|
||
if (!sessionId || !/^[0-9a-f-]{36}$/i.test(sessionId)) {
|
||
console.error('用法：node scripts/migrate-session-history.js <会话 ID> [--apply]');
|
||
process.exit(1);
|
||
}
|
||
const serverUrl = process.env.CC_WEB_SERVER_URL || `ws://127.0.0.1:${process.env.PORT || '8002'}/ws`;
|
||
const ws = new WebSocket(serverUrl);
|
||
const timer = setTimeout(() => finish({ ok: false, code: 'timeout' }), 15000);
|
||
let finished = false;
|
||
function finish(result) {
|
||
if (finished) return;
|
||
finished = true;
|
||
clearTimeout(timer);
|
||
console.log(JSON.stringify(result, null, 2));
|
||
process.exitCode = result.ok ? 0 : 1;
|
||
ws.terminate();
|
||
}
|
||
ws.on('open', () => ws.send(JSON.stringify({ type: 'auth', password: process.env.CC_WEB_PASSWORD || '' })));
|
||
ws.on('error', (error) => finish({ ok: false, code: 'connection_failed', message: error.message }));
|
||
ws.on('close', () => { if (!finished) finish({ ok: false, code: 'connection_closed' }); });
|
||
ws.on('message', (raw) => {
|
||
let msg;
|
||
try { msg = JSON.parse(String(raw)); } catch { return; }
|
||
if (msg.type === 'auth_result') {
|
||
if (!msg.success) return finish({ ok: false, code: 'authentication_failed' });
|
||
ws.send(JSON.stringify({ type: 'migrate_session_history', sessionId, apply: process.argv.includes('--apply') }));
|
||
} else if (msg.type === 'migrate_session_history_result' || msg.type === 'error') {
|
||
finish(msg);
|
||
}
|
||
});
|