cli.js

#!/usr/bin/env node
import fs from 'fs';
import path from 'path';
import { program } from 'commander';
import inquirer from 'inquirer';
import chalk from 'chalk';

const CLAUDE_DIR = path.join(process.env.HOME, '.claude/projects');

function getProjects() {
  return fs.readdirSync(CLAUDE_DIR).filter(f => !f.startsWith('.'));
}

function getSessions(project) {
  const projectPath = path.join(CLAUDE_DIR, project);
  return fs.readdirSync(projectPath)
    .filter(f => f.endsWith('.jsonl'))
    .map(f => {
      const filePath = path.join(projectPath, f);
      const stats = fs.statSync(filePath);
      return { file: f, path: filePath, size: stats.size, modified: stats.mtime };
    })
    .sort((a, b) => b.modified - a.modified);
}

function parseSession(filePath) {
  const lines = fs.readFileSync(filePath, 'utf8').split('\n').filter(line => line.trim());
  return lines.map(line => {
    try {
      return JSON.parse(line);
    } catch (e) {
      console.warn(`Skipping invalid JSON line: ${line.substring(0, 50)}...`);
      return null;
    }
  }).filter(Boolean);
}

function analyzeSession(entries) {
  const stats = {
    userMessages: 0,
    assistantMessages: 0,
    toolCalls: 0,
    toolResults: 0,
    subagents: 0,
    types: new Set()
  };

  entries.forEach(e => {
    stats.types.add(e.type);
    if (e.type === 'user') stats.userMessages++;
    if (e.type === 'assistant') stats.assistantMessages++;
    if (e.message?.content) {
      const content = Array.isArray(e.message.content) ? e.message.content : [e.message.content];
      content.forEach(c => {
        if (c.type === 'tool_use') stats.toolCalls++;
        if (c.type === 'tool_result') stats.toolResults++;
      });
    }
  });

  return stats;
}

function getCurrentProject() {
  const cwd = process.cwd();
  // Claude encodes paths: /Users/foo/My Project -> -Users-foo-My-Project
  // Replace both / and spaces with -, strip leading -
  const projectKey = cwd.replace(/[/ ]/g, '-').replace(/^-/, '');
  const projects = getProjects();
  // Try exact match first, then partial match on the last segments
  return projects.find(p => p === projectKey)
    || projects.find(p => p.endsWith(projectKey.split('-').slice(-3).join('-')))
    || projects.find(p => p.endsWith(projectKey.split('-').slice(-2).join('-')))
    || null;
}

async function selectSession() {
  const projects = getProjects();
  const currentProject = getCurrentProject();

  let project;
  if (currentProject) {
    console.log(chalk.cyan(`📁 Detected project: ${currentProject}`));
    const { useDetected } = await inquirer.prompt([{
      type: 'confirm',
      name: 'useDetected',
      message: `Use detected project?`,
      default: true
    }]);
    if (useDetected) {
      project = currentProject;
    }
  }

  if (!project) {
    const answer = await inquirer.prompt([{
      type: 'list',
      name: 'project',
      message: 'Select project:',
      choices: projects
    }]);
    project = answer.project;
  }

  const sessions = getSessions(project);
  const { session } = await inquirer.prompt([{
    type: 'list',
    name: 'session',
    message: 'Select session:',
    choices: sessions.map(s => ({
      name: `${s.file} (${(s.size/1024/1024).toFixed(2)}MB, ${s.modified.toLocaleDateString()})`,
      value: s
    }))
  }]);

  return session;
}

async function interactiveExport() {
  const session = await selectSession();
  const entries = parseSession(session.path);
  const stats = analyzeSession(entries);

  console.log(chalk.cyan(`\n📊 Session Stats:`));
  console.log(`  User messages: ${stats.userMessages}`);
  console.log(`  Assistant messages: ${stats.assistantMessages}`);
  console.log(`  Tool calls: ${stats.toolCalls}`);
  console.log(`  Tool results: ${stats.toolResults}\n`);

  const { filters } = await inquirer.prompt([{
    type: 'checkbox',
    name: 'filters',
    message: 'What to include:',
    choices: [
      { name: 'User messages', value: 'user', checked: true },
      { name: 'Assistant messages', value: 'assistant', checked: true },
      { name: 'Tool calls (inputs)', value: 'tool_use', checked: true },
      { name: 'Tool results (outputs)', value: 'tool_result', checked: true },
      { name: 'System messages', value: 'system', checked: false },
      { name: 'Progress events', value: 'progress', checked: false },
      { name: 'File snapshots', value: 'file-history-snapshot', checked: false }
    ]
  }]);

  const { format } = await inquirer.prompt([{
    type: 'list',
    name: 'format',
    message: 'Export format:',
    choices: ['markdown', 'json', 'jsonl', 'xml']
  }]);

  const filtered = entries.filter(e => {
    if (filters.includes(e.type)) return true;
    if (e.type === 'assistant' || e.type === 'user') {
      if (!e.message?.content) return filters.includes(e.type);
      const content = Array.isArray(e.message.content) ? e.message.content : [e.message.content];
      const hasMatchingContent = content.some(c =>
        (c?.type && filters.includes(c.type)) || filters.includes(e.type)
      );
      return hasMatchingContent;
    }
    return false;
  });

  const output = format === 'markdown' ? toMarkdown(filtered) :
                 format === 'json' ? JSON.stringify(filtered, null, 2) :
                 format === 'jsonl' ? filtered.map(e => JSON.stringify(e)).join('\n') :
                 toXML(filtered);

  const outFile = `export-${Date.now()}.${format === 'markdown' ? 'md' : format === 'xml' ? 'xml' : 'json'}`;
  fs.writeFileSync(outFile, output);
  console.log(chalk.green(`\n✅ Exported ${filtered.length} entries to ${outFile}`));
}

function toMarkdown(entries) {
  let md = '# Claude Code Session Export\n\n';
  entries.forEach((e, idx) => {
    const timestamp = e.timestamp ? new Date(e.timestamp).toLocaleString() : '';

    if (e.type === 'user') {
      md += `## 👤 User ${timestamp ? `(${timestamp})` : ''}\n`;
      if (!e.message) return;
      if (typeof e.message.content === 'string') {
        md += `${e.message.content}\n\n`;
      } else if (Array.isArray(e.message.content)) {
        e.message.content.forEach(c => {
          if (typeof c === 'string') md += `${c}\n\n`;
          else if (c?.type === 'text') md += `${c.text}\n\n`;
          else if (c?.type === 'tool_result') {
            md += `**📥 Tool Result** \`${c.tool_use_id}\`\n\`\`\`\n${typeof c.content === 'string' ? c.content : JSON.stringify(c.content, null, 2)}\n\`\`\`\n\n`;
          } else if (c?.type === 'image') {
            md += `**🖼️ Image**: ${c.source?.type || 'unknown'}\n\n`;
          }
        });
      }
    } else if (e.type === 'assistant') {
      md += `## 🤖 Assistant ${timestamp ? `(${timestamp})` : ''}\n`;
      if (!e.message?.content) return;
      const content = Array.isArray(e.message.content) ? e.message.content : [e.message.content];
      content.forEach(c => {
        if (typeof c === 'string') md += `${c}\n\n`;
        else if (c?.type === 'text') md += `${c.text}\n\n`;
        else if (c?.type === 'tool_use') {
          md += `**🔧 Tool: \`${c.name}\`** (ID: \`${c.id}\`)\n\`\`\`json\n${JSON.stringify(c.input, null, 2)}\n\`\`\`\n\n`;
        } else if (c?.type === 'thinking') {
          md += `**💭 Thinking**\n\`\`\`\n${c.thinking}\n\`\`\`\n\n`;
        }
      });
    } else if (e.type === 'progress') {
      md += `_⏳ Progress: ${e.data?.type || 'unknown'}_\n\n`;
    } else if (e.type === 'system') {
      md += `**⚙️ System**\n\`\`\`\n${typeof e.message === 'string' ? e.message : JSON.stringify(e.message, null, 2)}\n\`\`\`\n\n`;
    }
  });
  return md;
}

function toXML(entries) {
  const escape = s => String(s).replace(/[<>&'"]/g, c => ({'<':'&lt;','>':'&gt;','&':'&amp;',"'":'&apos;','"':'&quot;'}[c]));
  let xml = '<?xml version="1.0" encoding="UTF-8"?>\n<session>\n';
  entries.forEach(e => {
    xml += `  <entry type="${e.type}" timestamp="${e.timestamp || ''}" uuid="${e.uuid || ''}">\n`;
    if (e.message?.content) {
      const content = Array.isArray(e.message.content) ? e.message.content : [e.message.content];
      content.forEach(c => {
        if (typeof c === 'string') xml += `    <text>${escape(c)}</text>\n`;
        else if (c?.type === 'text') xml += `    <text>${escape(c.text)}</text>\n`;
        else if (c?.type === 'tool_use') xml += `    <tool_use name="${c.name}" id="${c.id}">${escape(JSON.stringify(c.input))}</tool_use>\n`;
        else if (c?.type === 'tool_result') xml += `    <tool_result id="${c.tool_use_id}">${escape(typeof c.content === 'string' ? c.content : JSON.stringify(c.content))}</tool_result>\n`;
      });
    }
    xml += `  </entry>\n`;
  });
  xml += '</session>';
  return xml;
}

program
  .name('claude-export')
  .description('Export Claude Code sessions with full control')
  .action(interactiveExport);

program.parse();