106 lines
3.8 KiB
JavaScript
106 lines
3.8 KiB
JavaScript
|
#!/usr/bin/env node
|
|||
|
|
/**
|
|||
|
|
* UserPromptSubmit Hook - 会话启动恢复器
|
|||
|
|
* 1. 检测 handoff.json 存在时注入恢复上下文并归档
|
|||
|
|
* 2. 每次会话首次 prompt 时重置 heartbeat 计数器（解决 /clear 后计数不归零问题）
|
|||
|
|
*/
|
|||
|
|
const fs = require('fs');
|
|||
|
|
const path = require('path');
|
|||
|
|
|
|||
|
|
const CLAUDE_ROOT = require('./lib/root.js');
|
|||
|
|
const readStdin = require('./lib/read-stdin.js');
|
|||
|
|
|
|||
|
|
const SESSION_STATE_DIR = path.join(CLAUDE_ROOT, 'session-state');
|
|||
|
|
const HANDOFF_PATH = path.join(SESSION_STATE_DIR, 'handoff.json');
|
|||
|
|
const RESTORE_MARKER = path.join(SESSION_STATE_DIR, '.session-restored');
|
|||
|
|
const HEARTBEAT_FILE = path.join(CLAUDE_ROOT, 'debug', 'session-heartbeat.json');
|
|||
|
|
|
|||
|
|
(async () => {
|
|||
|
|
try {
|
|||
|
|
let hookData = {};
|
|||
|
|
try { hookData = await readStdin(); } catch (_) {}
|
|||
|
|
|
|||
|
|
const sessionId = hookData.session_id || '';
|
|||
|
|
let messages = [];
|
|||
|
|
|
|||
|
|
// === 1. 会话 ID 变化检测 → 重置 heartbeat ===
|
|||
|
|
let lastSessionId = '';
|
|||
|
|
try {
|
|||
|
|
if (fs.existsSync(RESTORE_MARKER)) {
|
|||
|
|
lastSessionId = fs.readFileSync(RESTORE_MARKER, 'utf8').trim();
|
|||
|
|
}
|
|||
|
|
} catch (_) {}
|
|||
|
|
|
|||
|
|
if (sessionId && sessionId !== lastSessionId) {
|
|||
|
|
// 新会话（/clear 或新窗口），重置 heartbeat 计数器
|
|||
|
|
if (fs.existsSync(HEARTBEAT_FILE)) {
|
|||
|
|
fs.writeFileSync(HEARTBEAT_FILE, JSON.stringify({
|
|||
|
|
count: 0, lastActivity: Date.now(), notified: []
|
|||
|
|
}), 'utf8');
|
|||
|
|
}
|
|||
|
|
// 记录当前 session_id
|
|||
|
|
if (!fs.existsSync(SESSION_STATE_DIR)) {
|
|||
|
|
fs.mkdirSync(SESSION_STATE_DIR, { recursive: true });
|
|||
|
|
}
|
|||
|
|
fs.writeFileSync(RESTORE_MARKER, sessionId, 'utf8');
|
|||
|
|
|
|||
|
|
// === 1b. Auto Git Pull（新会话拉取最新代码）===
|
|||
|
|
try {
|
|||
|
|
const sync = require('../scripts/auto-git-sync.js');
|
|||
|
|
const pullMsgs = sync.pullLatest();
|
|||
|
|
if (pullMsgs && pullMsgs.length > 0) {
|
|||
|
|
messages.push(pullMsgs.join('\n'));
|
|||
|
|
}
|
|||
|
|
} catch (_) {} // fail-open
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// === 2. Handoff 恢复 ===
|
|||
|
|
if (fs.existsSync(HANDOFF_PATH)) {
|
|||
|
|
let handoff = {};
|
|||
|
|
try {
|
|||
|
|
handoff = JSON.parse(fs.readFileSync(HANDOFF_PATH, 'utf8'));
|
|||
|
|
} catch (_) {
|
|||
|
|
fs.unlinkSync(HANDOFF_PATH);
|
|||
|
|
console.log(JSON.stringify({ continue: true, suppressOutput: true }));
|
|||
|
|
return process.exit(0);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查是否过期（24 小时）
|
|||
|
|
const age = Date.now() - new Date(handoff.timestamp || 0).getTime();
|
|||
|
|
if (age > 24 * 60 * 60 * 1000) {
|
|||
|
|
const archiveName = `handoff-expired-${Date.now()}.json`;
|
|||
|
|
fs.renameSync(HANDOFF_PATH, path.join(SESSION_STATE_DIR, archiveName));
|
|||
|
|
console.log(JSON.stringify({ continue: true, suppressOutput: true }));
|
|||
|
|
return process.exit(0);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 归档并注入恢复上下文
|
|||
|
|
const archiveName = `handoff-${Date.now()}.json`;
|
|||
|
|
fs.renameSync(HANDOFF_PATH, path.join(SESSION_STATE_DIR, archiveName));
|
|||
|
|
|
|||
|
|
messages.push([
|
|||
|
|
'[SESSION_RESTORE] 检测到上次会话的 handoff 记录:',
|
|||
|
|
`- 时间: ${handoff.timestamp}`,
|
|||
|
|
`- 工作目录: ${handoff.working_directory || 'unknown'}`,
|
|||
|
|
`- 工具调用数: ${handoff.tool_call_count || 'unknown'}`,
|
|||
|
|
`- 摘要: ${handoff.conversation_summary || '无'}`,
|
|||
|
|
'',
|
|||
|
|
'请检查 memory 和 task 列表以恢复上下文。'
|
|||
|
|
].join('\n'));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (messages.length > 0) {
|
|||
|
|
console.log(JSON.stringify({
|
|||
|
|
continue: true,
|
|||
|
|
suppressOutput: false,
|
|||
|
|
systemMessage: messages.join('\n')
|
|||
|
|
}));
|
|||
|
|
} else {
|
|||
|
|
console.log(JSON.stringify({ continue: true, suppressOutput: true }));
|
|||
|
|
}
|
|||
|
|
} catch {
|
|||
|
|
console.log(JSON.stringify({ continue: true, suppressOutput: true }));
|
|||
|
|
}
|
|||
|
|
process.exit(0);
|
|||
|
|
})();
|