airkjw/claude-mem
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Enhance context hook output formatting and add file aggregation for sessions

Browse Source 
- Updated contextHook to support colorized output for terminal and JSON format for hooks.
- Introduced ANSI color codes for improved readability in terminal output.
- Modified the output structure to include session details with color formatting.
- Added a new method in SessionStore to aggregate files read and modified from observations for a session.
- Improved error handling for JSON parsing of file data in the new method.
This commit is contained in:
Alex Newman
2025-10-21 16:37:16 -04:00
parent fe861a85bd
commit df0f3fd96c
10 changed files with 406 additions and 219 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/hooks/context.ts
+165 -59
View File
@@ -11,13 +11,26 @@ export interface SessionStartInput {
[key: string]: any;
}
// ANSI color codes for terminal output
const colors = {
reset: '\x1b[0m',
bright: '\x1b[1m',
dim: '\x1b[2m',
cyan: '\x1b[36m',
green: '\x1b[32m',
yellow: '\x1b[33m',
blue: '\x1b[34m',
magenta: '\x1b[35m',
gray: '\x1b[90m',
};
/**
* Context Hook - SessionStart
* Shows user what happened in recent sessions
*
* Output: Returns formatted context string to be wrapped in hookSpecificOutput
*/
export function contextHook(input?: SessionStartInput): string {
export function contextHook(input?: SessionStartInput, useColors: boolean = false): string {
// v4.0.0: Ensure worker is running before loading context
ensureWorkerRunning();
const cwd = input?.cwd ?? process.cwd();
@@ -29,20 +42,37 @@ export function contextHook(input?: SessionStartInput): string {
const sessions = db.getRecentSessionsWithStatus(project, 3);
if (sessions.length === 0) {
if (useColors) {
return `\n${colors.bright}${colors.cyan}📝 Recent Session Context${colors.reset}\n\n${colors.dim}No previous sessions found for this project yet.${colors.reset}\n`;
}
return '# Recent Session Context\n\nNo previous sessions found for this project yet.';
}
const output: string[] = [];
output.push('# Recent Session Context');
output.push('');
output.push(`Showing last ${sessions.length} session(s) for **${project}**:`);
output.push('');
if (useColors) {
output.push('');
output.push(`${colors.bright}${colors.cyan}📝 Recent Session Context${colors.reset}`);
output.push(`${colors.gray}${'─'.repeat(60)}${colors.reset}`);
output.push(`${colors.dim}Showing last ${sessions.length} session(s) for ${colors.reset}${colors.bright}${project}${colors.reset}`);
output.push('');
} else {
output.push('# Recent Session Context');
output.push('');
output.push(`Showing last ${sessions.length} session(s) for **${project}**:`);
output.push('');
}
for (const session of sessions) {
if (!session.sdk_session_id) continue;
output.push('---');
output.push('');
if (useColors) {
output.push(`${colors.gray}${'─'.repeat(60)}${colors.reset}`);
output.push('');
} else {
output.push('---');
output.push('');
}
// Check if session has a summary
if (session.has_summary) {
@@ -50,99 +80,175 @@ export function contextHook(input?: SessionStartInput): string {
if (summary) {
const promptLabel = summary.prompt_number ? ` (Prompt #${summary.prompt_number})` : '';
output.push(`**Summary${promptLabel}**`);
output.push('');
if (useColors) {
output.push(`${colors.bright}${colors.green}✓ Summary${promptLabel}${colors.reset}`);
output.push('');
} else {
output.push(`**Summary${promptLabel}**`);
output.push('');
}
if (summary.request) {
output.push(`**Request:** ${summary.request}`);
}
if (summary.completed) {
output.push(`**Completed:** ${summary.completed}`);
}
if (summary.learned) {
output.push(`**Learned:** ${summary.learned}`);
}
if (summary.next_steps) {
output.push(`**Next Steps:** ${summary.next_steps}`);
}
if (summary.files_read) {
try {
const files = JSON.parse(summary.files_read);
if (Array.isArray(files) && files.length > 0) {
output.push(`**Files Read:** ${files.join(', ')}`);
}
} catch {
if (summary.files_read.trim()) {
output.push(`**Files Read:** ${summary.files_read}`);
}
if (useColors) {
output.push(`${colors.bright}${colors.yellow}Request:${colors.reset} ${summary.request}`);
output.push('');
} else {
output.push(`**Request:** ${summary.request}`);
}
}
if (summary.files_edited) {
try {
const files = JSON.parse(summary.files_edited);
if (Array.isArray(files) && files.length > 0) {
output.push(`**Files Edited:** ${files.join(', ')}`);
}
} catch {
if (summary.files_edited.trim()) {
output.push(`**Files Edited:** ${summary.files_edited}`);
}
if (summary.completed) {
if (useColors) {
output.push(`${colors.bright}${colors.green}Completed:${colors.reset} ${summary.completed}`);
output.push('');
} else {
output.push(`**Completed:** ${summary.completed}`);
}
}
if (summary.learned) {
if (useColors) {
output.push(`${colors.bright}${colors.blue}Learned:${colors.reset} ${summary.learned}`);
output.push('');
} else {
output.push(`**Learned:** ${summary.learned}`);
}
}
if (summary.next_steps) {
if (useColors) {
output.push(`${colors.bright}${colors.magenta}Next Steps:${colors.reset} ${summary.next_steps}`);
output.push('');
} else {
output.push(`**Next Steps:** ${summary.next_steps}`);
}
}
// Get files from observations (not from summary which is never populated)
const sessionFiles = db.getFilesForSession(session.sdk_session_id);
if (sessionFiles.filesRead.length > 0) {
if (useColors) {
output.push(`${colors.dim}Files Read: ${sessionFiles.filesRead.join(', ')}${colors.reset}`);
output.push('');
} else {
output.push(`**Files Read:** ${sessionFiles.filesRead.join(', ')}`);
}
}
if (sessionFiles.filesModified.length > 0) {
if (useColors) {
output.push(`${colors.dim}Files Modified: ${sessionFiles.filesModified.join(', ')}${colors.reset}`);
output.push('');
} else {
output.push(`**Files Modified:** ${sessionFiles.filesModified.join(', ')}`);
}
}
const dateTime = new Date(summary.created_at).toLocaleString();
output.push(`**Date:** ${dateTime}`);
if (useColors) {
output.push(`${colors.dim}Date: ${dateTime}${colors.reset}`);
} else {
output.push(`**Date:** ${dateTime}`);
}
}
} else if (session.status === 'active') {
// Active session without summary - show observation titles
output.push(`**In Progress**`);
output.push('');
if (useColors) {
output.push(`${colors.bright}${colors.yellow}⏳ In Progress${colors.reset}`);
output.push('');
} else {
output.push(`**In Progress**`);
output.push('');
}
if (session.user_prompt) {
output.push(`**Request:** ${session.user_prompt}`);
if (useColors) {
output.push(`${colors.bright}${colors.yellow}Request:${colors.reset} ${session.user_prompt}`);
output.push('');
} else {
output.push(`**Request:** ${session.user_prompt}`);
}
}
const observations = db.getObservationsForSession(session.sdk_session_id);
if (observations.length > 0) {
output.push('');
output.push(`**Observations (${observations.length}):**`);
for (const obs of observations) {
output.push(`- ${obs.title}`);
if (useColors) {
output.push(`${colors.bright}Observations (${observations.length}):${colors.reset}`);
for (const obs of observations) {
output.push(` ${colors.dim}•${colors.reset} ${obs.title}`);
}
output.push('');
} else {
output.push(`**Observations (${observations.length}):**`);
for (const obs of observations) {
output.push(`- ${obs.title}`);
}
}
} else {
output.push('');
output.push('*No observations yet*');
if (useColors) {
output.push(`${colors.dim}No observations yet${colors.reset}`);
output.push('');
} else {
output.push('*No observations yet*');
}
}
output.push('');
output.push(`**Status:** Active - summary pending`);
const activeDateTime = new Date(session.started_at).toLocaleString();
output.push(`**Date:** ${activeDateTime}`);
if (useColors) {
output.push(`${colors.dim}Status: Active - summary pending${colors.reset}`);
output.push(`${colors.dim}Date: ${activeDateTime}${colors.reset}`);
} else {
output.push(`**Status:** Active - summary pending`);
output.push(`**Date:** ${activeDateTime}`);
}
} else {
// Failed or completed session without summary
const displayStatus = session.status === 'failed' ? 'stopped' : session.status;
output.push(`**${displayStatus.charAt(0).toUpperCase() + displayStatus.slice(1)}**`);
output.push('');
const statusIcon = session.status === 'failed' ? '⚠️' : '○';
if (useColors) {
const statusColor = session.status === 'failed' ? colors.yellow : colors.gray;
output.push(`${colors.bright}${statusColor}${statusIcon} ${displayStatus.charAt(0).toUpperCase() + displayStatus.slice(1)}${colors.reset}`);
output.push('');
} else {
output.push(`**${displayStatus.charAt(0).toUpperCase() + displayStatus.slice(1)}**`);
output.push('');
}
if (session.user_prompt) {
output.push(`**Request:** ${session.user_prompt}`);
if (useColors) {
output.push(`${colors.bright}${colors.yellow}Request:${colors.reset} ${session.user_prompt}`);
output.push('');
} else {
output.push(`**Request:** ${session.user_prompt}`);
}
}
output.push('');
output.push(`**Status:** ${displayStatus} - no summary available`);
const failedDateTime = new Date(session.started_at).toLocaleString();
output.push(`**Date:** ${failedDateTime}`);
if (useColors) {
output.push(`${colors.dim}Status: ${displayStatus} - no summary available${colors.reset}`);
output.push(`${colors.dim}Date: ${failedDateTime}${colors.reset}`);
} else {
output.push(`**Status:** ${displayStatus} - no summary available`);
output.push(`**Date:** ${failedDateTime}`);
}
}
output.push('');
}
if (useColors) {
output.push(`${colors.gray}${'─'.repeat(60)}${colors.reset}`);
output.push('');
}
return output.join('\n');
} finally {
db.close();