worklog / src / core / session-reader.ts

import { createReadStream } from 'fs';
import * as readline from 'readline';

import type { MessageContent, ParsedMessage, ParsedSession, RawSessionEntry, SessionStats, ToolUse } from '../types';

/**
 * Get the "effective date" for a timestamp using a 3am boundary.
 * Work done before 3am counts as the previous day (aligns with sleep cycle).
 */
function getEffectiveDate(timestamp: string): string {
  const d = new Date(timestamp);
  d.setHours(d.getHours() - 3); // Shift 3am boundary to midnight
  return d.toISOString().split('T')[0];
}

/**
 * Stream-parse a JSONL session file
 */
export async function* parseJSONLStream(filePath: string): AsyncGenerator<RawSessionEntry> {
  const rl = readline.createInterface({
    input: createReadStream(filePath),
    crlfDelay: Infinity,
  });

  for await (const line of rl) {
    if (line.trim() === '') continue;
    try {
      yield JSON.parse(line) as RawSessionEntry;
    } catch {
      // Skip invalid JSON lines
    }
  }
}

/**
 * Parse a session file into a structured format
 */
export async function parseSessionFile(
  filePath: string,
  projectPath: string,
  projectName: string,
): Promise<ParsedSession> {
  const messages: ParsedMessage[] = [];
  const toolCalls: Record<string, number> = {};
  let sessionId = '';
  let gitBranch = '';
  let startTime = '';
  let endTime = '';
  let totalInputTokens = 0;
  let totalOutputTokens = 0;
  let userMessages = 0;
  let assistantMessages = 0;

  const seen = new Set<string>();

  for await (const entry of parseJSONLStream(filePath)) {
    // Deduplication - use uuid (unique per chunk) not message.id (same across streaming chunks)
    if (seen.has(entry.uuid)) continue;
    seen.add(entry.uuid);

    // Extract metadata from first entry
    if (sessionId === '' && typeof entry.sessionId === 'string' && entry.sessionId !== '') {
      sessionId = entry.sessionId;
    }
    if (gitBranch === '' && entry.gitBranch !== undefined) {
      gitBranch = entry.gitBranch;
    }

    // Track timestamps (only if valid)
    if (entry.timestamp && typeof entry.timestamp === 'string') {
      if (startTime === '' || entry.timestamp < startTime) {
        startTime = entry.timestamp;
      }
      if (endTime === '' || entry.timestamp > endTime) {
        endTime = entry.timestamp;
      }
    }

    // Skip entries without a valid message (malformed JSONL entries)
    if (!entry.message) {
      continue;
    }

    // Extract token usage from assistant messages
    if (entry.type === 'assistant' && entry.message.usage !== undefined) {
      const usage = entry.message.usage;
      totalInputTokens += usage.input_tokens;
      totalOutputTokens += usage.output_tokens;
      totalInputTokens += usage.cache_creation_input_tokens ?? 0;
      totalInputTokens += usage.cache_read_input_tokens ?? 0;
    }

    // Parse message content
    const text = extractText(entry.message.content);
    const toolUses = extractToolUses(entry.message.content);

    // Count tool calls
    for (const tool of toolUses) {
      toolCalls[tool.name] = (toolCalls[tool.name] ?? 0) + 1;
    }

    if (entry.type === 'user') userMessages++;
    if (entry.type === 'assistant') assistantMessages++;

    messages.push({
      type: entry.type,
      timestamp: entry.timestamp,
      text,
      toolUses,
    });
  }

  // Use filename as sessionId fallback
  if (sessionId === '') {
    sessionId = filePath.split('/').pop()?.replace('.jsonl', '') ?? 'unknown';
  }

  // Provide default timestamps if none found
  const now = new Date().toISOString();
  if (startTime === '') {
    startTime = now;
  }
  if (endTime === '') {
    endTime = startTime;
  }

  // Derive date from endTime with 3am boundary (aligns with sleep cycle)
  // Work done before 3am counts as the previous day
  const date = getEffectiveDate(endTime);

  const stats: SessionStats = {
    userMessages,
    assistantMessages,
    toolCalls,
    totalInputTokens,
    totalOutputTokens,
  };

  return {
    sessionId,
    filePath,
    projectPath,
    projectName,
    gitBranch,
    startTime,
    endTime,
    date,
    messages,
    stats,
  };
}

/**
 * Extract text from message content array
 */
function extractText(content: MessageContent[]): string {
  if (!Array.isArray(content)) return '';

  const texts: string[] = [];
  for (const item of content) {
    if (item.type === 'text') {
      // Handle both formats: { text: "..." } and { content: "..." }
      const text = 'text' in item ? item.text : 'content' in item ? item.content : '';
      if (text !== '') texts.push(text);
    }
  }
  return texts.join('\n');
}

/**
 * Extract tool uses from message content
 */
function extractToolUses(content: MessageContent[]): ToolUse[] {
  if (!Array.isArray(content)) return [];

  const tools: ToolUse[] = [];
  for (const item of content) {
    if (item.type === 'tool_use') {
      tools.push({
        name: item.name,
        input: summarizeToolInput(item.name, item.input),
        rawInput: item.input,
      });
    }
  }
  return tools;
}

/**
 * Summarize tool input for display (truncate long content)
 */
function summarizeToolInput(toolName: string, input: Record<string, unknown>): string {
  const MAX_LENGTH = 200;

  switch (toolName) {
    case 'Bash':
      return truncate(typeof input.command === 'string' ? input.command : '', MAX_LENGTH);
    case 'Read':
      return truncate(typeof input.file_path === 'string' ? input.file_path : '', MAX_LENGTH);
    case 'Write':
    case 'Edit':
      return truncate(typeof input.file_path === 'string' ? input.file_path : '', MAX_LENGTH);
    case 'Glob':
      return truncate(typeof input.pattern === 'string' ? input.pattern : '', MAX_LENGTH);
    case 'Grep':
      return truncate(typeof input.pattern === 'string' ? input.pattern : '', MAX_LENGTH);
    case 'Task':
      return truncate(typeof input.description === 'string' ? input.description : '', MAX_LENGTH);
    default:
      return truncate(JSON.stringify(input), MAX_LENGTH);
  }
}

function truncate(str: string, maxLength: number): string {
  if (str.length <= maxLength) return str;
  return str.slice(0, maxLength - 3) + '...';
}

/**
 * Work type classification based on files changed
 */
export type WorkType = 'feature' | 'infrastructure' | 'tests' | 'docs' | 'mixed';

export interface WorkScope {
  frontend: number;
  backend: number;
  tests: number;
  types: number;
  config: number;
  docs: number;
}

export interface WorkClassification {
  type: WorkType;
  signals: string[]; // Human-readable explanation of why
  scope: WorkScope;
  scopeSummary: string; // e.g., "frontend, backend" or "tests"
}

/**
 * Check if a file path looks like frontend code
 */
function isFrontend(file: string): boolean {
  const lower = file.toLowerCase();
  return (
    lower.includes('/components/') ||
    lower.includes('/pages/') ||
    lower.includes('/screens/') ||
    lower.includes('/views/') ||
    lower.includes('/ui/') ||
    lower.includes('/app/') ||
    lower.includes('/apps/web/') ||
    lower.includes('/web/') ||
    lower.includes('/frontend/') ||
    lower.includes('/client/') ||
    lower.endsWith('.tsx') ||
    lower.endsWith('.jsx') ||
    lower.endsWith('.css') ||
    lower.endsWith('.scss')
  );
}

/**
 * Check if a file path looks like backend code
 */
function isBackend(file: string): boolean {
  const lower = file.toLowerCase();
  return (
    lower.includes('/api/') ||
    lower.includes('/server/') ||
    lower.includes('/services/') ||
    lower.includes('/lib/') ||
    lower.includes('/core/') ||
    lower.includes('/packages/') ||
    lower.includes('/backend/') ||
    lower.includes('/handlers/') ||
    lower.includes('/routes/') ||
    lower.includes('/controllers/') ||
    lower.includes('/models/') ||
    lower.includes('/utils/') ||
    (lower.endsWith('.ts') &&
      !lower.endsWith('.test.ts') &&
      !lower.endsWith('.spec.ts') &&
      !lower.endsWith('.d.ts') &&
      !isFrontend(file))
  );
}

/**
 * Classify the type of work based on file paths
 */
export function classifyWork(files: string[]): WorkClassification {
  const signals: string[] = [];
  const scope: WorkScope = {
    frontend: 0,
    backend: 0,
    tests: 0,
    types: 0,
    config: 0,
    docs: 0,
  };

  let featureFiles = 0;

  for (const file of files) {
    const lower = file.toLowerCase();
    const filename = file.split('/').pop() ?? '';

    // Tests
    if (
      lower.includes('.test.') ||
      lower.includes('.spec.') ||
      lower.includes('__tests__') ||
      lower.includes('/test/') ||
      lower.includes('/tests/')
    ) {
      scope.tests++;
      continue;
    }

    // Types/interfaces
    if (
      filename === 'types.ts' ||
      filename === 'interfaces.ts' ||
      lower.endsWith('.d.ts') ||
      lower.includes('/types/') ||
      lower.includes('/interfaces/')
    ) {
      scope.types++;
      continue;
    }

    // Config/devops
    if (
      lower.includes('.config.') ||
      lower.includes('/config/') ||
      lower.includes('.github/') ||
      lower.includes('dockerfile') ||
      lower.includes('.yml') ||
      lower.includes('.yaml') ||
      filename.startsWith('.') ||
      filename === 'package.json' ||
      filename === 'tsconfig.json'
    ) {
      scope.config++;
      continue;
    }

    // Docs
    if (lower.endsWith('.md') || lower.includes('/docs/') || lower.includes('/documentation/')) {
      scope.docs++;
      continue;
    }

    // Feature work - classify as frontend or backend
    featureFiles++;
    if (isFrontend(file)) {
      scope.frontend++;
    } else if (isBackend(file)) {
      scope.backend++;
    } else {
      // Default to backend for unclassified .ts files
      scope.backend++;
    }
  }

  const total = files.length;
  if (total === 0) {
    return {
      type: 'mixed',
      signals: ['no files changed'],
      scope,
      scopeSummary: '',
    };
  }

  // Build scope summary - simplified to frontend/backend/both
  let scopeSummary = '';
  const hasFrontend = scope.frontend > 0;
  const hasBackend = scope.backend > 0;
  if (hasFrontend && hasBackend) {
    scopeSummary = 'frontend, backend';
  } else if (hasFrontend) {
    scopeSummary = 'frontend';
  } else if (hasBackend) {
    scopeSummary = 'backend';
  } else if (scope.tests > 0) {
    scopeSummary = 'tests';
  } else if (scope.docs > 0) {
    scopeSummary = 'docs';
  } else if (scope.config > 0) {
    scopeSummary = 'config';
  }

  // Determine primary type (>50% of files)
  const threshold = total * 0.5;

  if (scope.tests > threshold) {
    signals.push(`${scope.tests.toString()}/${total.toString()} files are tests`);
    return { type: 'tests', signals, scope, scopeSummary };
  }

  if (scope.docs > threshold) {
    signals.push(`${scope.docs.toString()}/${total.toString()} files are documentation`);
    return { type: 'docs', signals, scope, scopeSummary };
  }

  if (scope.types + scope.config > threshold) {
    if (scope.types > scope.config) {
      signals.push(`${scope.types.toString()}/${total.toString()} files are types`);
    } else {
      signals.push(`${scope.config.toString()}/${total.toString()} files are config`);
    }
    return { type: 'infrastructure', signals, scope, scopeSummary };
  }

  if (featureFiles > threshold) {
    signals.push(`${featureFiles.toString()}/${total.toString()} files are feature code`);
    return { type: 'feature', signals, scope, scopeSummary };
  }

  // Mixed - build a description
  if (featureFiles > 0) signals.push(`${featureFiles.toString()} feature`);
  if (scope.tests > 0) signals.push(`${scope.tests.toString()} test`);
  if (scope.types > 0) signals.push(`${scope.types.toString()} type`);
  if (scope.config > 0) signals.push(`${scope.config.toString()} config`);
  if (scope.docs > 0) signals.push(`${scope.docs.toString()} doc`);

  return { type: 'mixed', signals, scope, scopeSummary };
}

/**
 * Create a condensed transcript for LLM summarization
 * Leads with action summary (files changed) to ensure implementation work is captured
 */
export function createCondensedTranscript(session: ParsedSession): string {
  const parts: string[] = [];

  parts.push(`Project: ${session.projectName}`);
  if (session.gitBranch !== '') {
    parts.push(`Branch: ${session.gitBranch}`);
  }
  parts.push(`Duration: ${formatDuration(session.startTime, session.endTime)}`);
  parts.push('');

  // LEAD with files changed - this is the most important signal of actual work
  const filesWritten: string[] = [];
  const filesEdited: string[] = [];
  const commandsRun: string[] = [];

  for (const msg of session.messages) {
    if (msg.type === 'assistant') {
      for (const tool of msg.toolUses) {
        if (tool.name === 'Write') {
          const rawInput = tool.rawInput;
          const filePath = rawInput?.file_path;
          const path = typeof filePath === 'string' ? filePath : '';
          if (path !== '' && !filesWritten.includes(path)) {
            filesWritten.push(path);
          }
        } else if (tool.name === 'Edit') {
          const rawInput = tool.rawInput;
          const filePath = rawInput?.file_path;
          const path = typeof filePath === 'string' ? filePath : '';
          if (path !== '' && !filesEdited.includes(path)) {
            filesEdited.push(path);
          }
        } else if (tool.name === 'Bash') {
          const rawInput = tool.rawInput;
          const command = rawInput?.command;
          const cmd = typeof command === 'string' ? command.slice(0, 100) : '';
          if (cmd !== '' && commandsRun.length < 10) {
            commandsRun.push(cmd);
          }
        }
      }
    }
  }

  // Classify the work based on file paths
  const allFiles = [...filesWritten, ...filesEdited];
  const classification = classifyWork(allFiles);
  parts.push(`WORK TYPE: ${classification.type}`);
  if (classification.scopeSummary !== '') {
    parts.push(`SCOPE: ${classification.scopeSummary}`);
  }
  parts.push('');

  // Show action summary at the TOP
  if (filesWritten.length > 0) {
    parts.push(`FILES CREATED (${filesWritten.length.toString()}):`);
    filesWritten.slice(0, 15).forEach((f) => parts.push(`  - ${f}`));
    if (filesWritten.length > 15) parts.push(`  ... and ${(filesWritten.length - 15).toString()} more`);
    parts.push('');
  }

  if (filesEdited.length > 0) {
    parts.push(`FILES EDITED (${filesEdited.length.toString()}):`);
    filesEdited.slice(0, 15).forEach((f) => parts.push(`  - ${f}`));
    if (filesEdited.length > 15) parts.push(`  ... and ${(filesEdited.length - 15).toString()} more`);
    parts.push('');
  }

  if (commandsRun.length > 0) {
    parts.push(`COMMANDS RUN (${commandsRun.length.toString()}):`);
    commandsRun.slice(0, 5).forEach((c) => parts.push(`  $ ${c}`));
    parts.push('');
  }

  // Then show conversation context (but less of it)
  parts.push('CONVERSATION:');
  let messageCount = 0;
  for (const msg of session.messages) {
    if (messageCount > 20) break; // Limit to avoid overwhelming

    if (msg.type === 'user' && msg.text !== '') {
      const text = msg.text.slice(0, 300);
      parts.push(`User: ${text}`);
      messageCount++;
    } else if (msg.type === 'assistant' && msg.text !== '') {
      const text = msg.text.slice(0, 200);
      parts.push(`Assistant: ${text}`);
      messageCount++;
    }
  }

  // Add stats at end
  parts.push('');
  const toolSummary = Object.entries(session.stats.toolCalls)
    .sort((a, b) => b[1] - a[1])
    .slice(0, 10)
    .map(([name, count]) => `${name}(${count.toString()})`)
    .join(', ');
  if (toolSummary !== '') {
    parts.push(`Tool usage: ${toolSummary}`);
  }

  return parts.join('\n');
}

function formatDuration(start: string, end: string): string {
  if (start === '' || end === '') return 'unknown';

  const startDate = new Date(start);
  const endDate = new Date(end);
  const diffMs = endDate.getTime() - startDate.getTime();

  const minutes = Math.floor(diffMs / 60000);
  if (minutes < 60) return `${minutes.toString()} min`;

  const hours = Math.floor(minutes / 60);
  const remainingMinutes = minutes % 60;
  return `${hours.toString()}h ${remainingMinutes.toString()}m`;
}