tokenlean 0.1.0 → 0.2.0

package/bin/tl-diff.mjs

@@ -21,6 +21,30 @@ if (process.argv.includes('--prompt')) {
 }
 
 import { execSync } from 'child_process';
+import {
+  createOutput,
+  parseCommonArgs,
+  formatTokens,
+  COMMON_OPTIONS_HELP
+} from '../src/output.mjs';
+
+const HELP = `
+tl-diff - Token-efficient git diff summary
+
+Usage: tl-diff [ref] [options]
+
+Options:
+  --staged             Show staged changes only
+  --stat-only          Show just the summary (no file list)
+${COMMON_OPTIONS_HELP}
+
+Examples:
+  tl-diff                     # Working directory changes
+  tl-diff --staged            # Staged changes
+  tl-diff HEAD~3              # Last 3 commits
+  tl-diff main                # Changes vs main branch
+  tl-diff -j                  # JSON output
+`;
 
 function run(cmd) {
   try {
@@ -30,15 +54,6 @@ function run(cmd) {
   }
 }
 
-function estimateTokens(content) {
-  return Math.ceil(content.length / 4);
-}
-
-function formatTokens(tokens) {
-  if (tokens >= 1000) return `${(tokens / 1000).toFixed(1)}k`;
-  return String(tokens);
-}
-
 function parseDiffStat(stat) {
   const lines = stat.trim().split('\n');
   const files = [];
@@ -96,66 +111,30 @@ function categorizeChanges(files) {
   return categories;
 }
 
-function printSummary(files, categories, options) {
-  const totalChanges = files.reduce((sum, f) => sum + f.changes, 0);
-  const totalAdditions = files.reduce((sum, f) => sum + f.additions, 0);
-  const totalDeletions = files.reduce((sum, f) => sum + f.deletions, 0);
-
-  console.log(`\n📊 Diff Summary`);
-  console.log(`   ${files.length} files changed, ~${formatTokens(totalChanges * 4)} tokens of changes`);
-  console.log(`   +${totalAdditions} additions, -${totalDeletions} deletions\n`);
-
-  const order = ['components', 'hooks', 'store', 'types', 'manuscripts', 'tests', 'config', 'other'];
-  const labels = {
-    components: '🧩 Components',
-    hooks: '🪝 Hooks',
-    store: '📦 Store',
-    types: '📝 Types',
-    manuscripts: '📖 Manuscripts',
-    tests: '🧪 Tests',
-    config: '⚙️  Config',
-    other: '📄 Other'
-  };
-
-  for (const cat of order) {
-    const catFiles = categories[cat];
-    if (catFiles.length === 0) continue;
-
-    console.log(`${labels[cat]} (${catFiles.length})`);
-
-    // Sort by changes descending
-    catFiles.sort((a, b) => b.changes - a.changes);
-
-    for (const f of catFiles.slice(0, 10)) {
-      const bar = '+'.repeat(Math.min(f.additions, 20)) + '-'.repeat(Math.min(f.deletions, 20));
-      console.log(`  ${f.path}`);
-      console.log(`    ${f.changes} changes ${bar}`);
-    }
-
-    if (catFiles.length > 10) {
-      console.log(`  ... and ${catFiles.length - 10} more`);
-    }
-
-    console.log();
-  }
-}
-
 // Main
 const args = process.argv.slice(2);
+const options = parseCommonArgs(args);
+
+// Parse tool-specific options
 let ref = '';
 let staged = false;
 let statOnly = false;
 
-for (let i = 0; i < args.length; i++) {
-  if (args[i] === '--staged') {
+for (const arg of options.remaining) {
+  if (arg === '--staged') {
     staged = true;
-  } else if (args[i] === '--stat-only') {
+  } else if (arg === '--stat-only') {
     statOnly = true;
-  } else if (!args[i].startsWith('-')) {
-    ref = args[i];
+  } else if (!arg.startsWith('-')) {
+    ref = arg;
   }
 }
 
+if (options.help) {
+  console.log(HELP);
+  process.exit(0);
+}
+
 // Build git diff command
 let diffCmd = 'git diff';
 if (staged) {
@@ -167,17 +146,71 @@ diffCmd += ' --stat=200';
 
 const stat = run(diffCmd);
 
+const out = createOutput(options);
+
 if (!stat.trim()) {
-  console.log('\n✨ No changes detected\n');
+  out.header('No changes detected');
+  out.print();
   process.exit(0);
 }
 
 const files = parseDiffStat(stat);
 const categories = categorizeChanges(files);
 
-printSummary(files, categories, { statOnly });
+const totalChanges = files.reduce((sum, f) => sum + f.changes, 0);
+const totalAdditions = files.reduce((sum, f) => sum + f.additions, 0);
+const totalDeletions = files.reduce((sum, f) => sum + f.deletions, 0);
+
+// Set JSON data
+out.setData('files', files);
+out.setData('categories', categories);
+out.setData('totalFiles', files.length);
+out.setData('totalChanges', totalChanges);
+out.setData('estimatedTokens', totalChanges * 4);
+
+// Summary header
+out.header('Diff Summary');
+out.header(`${files.length} files changed, ~${formatTokens(totalChanges * 4)} tokens of changes`);
+out.header(`+${totalAdditions} additions, -${totalDeletions} deletions`);
+out.blank();
 
 if (!statOnly) {
-  console.log('💡 Tip: Use --stat-only for just the summary, or check specific files with:');
-  console.log('   git diff [ref] -- path/to/file.ts\n');
+  const order = ['components', 'hooks', 'store', 'types', 'manuscripts', 'tests', 'config', 'other'];
+  const labels = {
+    components: 'Components',
+    hooks: 'Hooks',
+    store: 'Store',
+    types: 'Types',
+    manuscripts: 'Manuscripts',
+    tests: 'Tests',
+    config: 'Config',
+    other: 'Other'
+  };
+
+  for (const cat of order) {
+    const catFiles = categories[cat];
+    if (catFiles.length === 0) continue;
+
+    out.add(`${labels[cat]} (${catFiles.length})`);
+
+    // Sort by changes descending
+    catFiles.sort((a, b) => b.changes - a.changes);
+
+    for (const f of catFiles.slice(0, 10)) {
+      const bar = '+'.repeat(Math.min(f.additions, 20)) + '-'.repeat(Math.min(f.deletions, 20));
+      out.add(`  ${f.path}`);
+      out.add(`    ${f.changes} changes ${bar}`);
+    }
+
+    if (catFiles.length > 10) {
+      out.add(`  ... and ${catFiles.length - 10} more`);
+    }
+
+    out.blank();
+  }
+
+  out.header('Tip: Use --stat-only for just the summary, or check specific files with:');
+  out.header('   git diff [ref] -- path/to/file.ts');
 }
+
+out.print();

package/bin/tl-related.mjs

@@ -23,41 +23,37 @@ if (process.argv.includes('--prompt')) {
 import { readFileSync, existsSync, readdirSync, statSync } from 'fs';
 import { join, dirname, basename, relative, extname } from 'path';
 import { execSync } from 'child_process';
-import { shellEscape } from '../src/output.mjs';
-
-try { execSync('which rg', { stdio: 'ignore' }); } catch {
-  console.error('⚠️  ripgrep (rg) not found. Install: brew install ripgrep');
+import {
+  createOutput,
+  parseCommonArgs,
+  estimateTokens,
+  formatTokens,
+  shellEscape,
+  COMMON_OPTIONS_HELP
+} from '../src/output.mjs';
+import { findProjectRoot } from '../src/project.mjs';
+
+const HELP = `
+tl-related - Find related files (tests, types, usages)
+
+Usage: tl-related <file> [options]
+${COMMON_OPTIONS_HELP}
+
+Examples:
+  tl-related src/Button.tsx        # Find tests, types, importers
+  tl-related src/api.ts -j         # JSON output
+  tl-related src/utils.ts -q       # Quiet (file paths only)
+`;
+
+// Check for ripgrep
+try {
+  execSync('which rg', { stdio: 'ignore' });
+} catch {
+  console.error('ripgrep (rg) not found. Install: brew install ripgrep');
   process.exit(1);
 }
 
-const SKIP_DIRS = new Set([
-  'node_modules', '.git', 'android', 'ios', 'dist', 'build', '.expo', '.next'
-]);
-
-function findProjectRoot() {
-  let dir = process.cwd();
-  while (dir !== '/') {
-    if (existsSync(join(dir, 'package.json'))) return dir;
-    dir = dirname(dir);
-  }
-  return process.cwd();
-}
-
-function estimateTokens(filePath) {
-  try {
-    const content = readFileSync(filePath, 'utf-8');
-    return Math.ceil(content.length / 4);
-  } catch {
-    return 0;
-  }
-}
-
-function formatTokens(tokens) {
-  if (tokens >= 1000) return `${(tokens / 1000).toFixed(1)}k`;
-  return String(tokens);
-}
-
-function findTestFiles(filePath, projectRoot) {
+function findTestFiles(filePath) {
   const dir = dirname(filePath);
   const name = basename(filePath, extname(filePath));
   const tests = [];
@@ -66,10 +62,15 @@ function findTestFiles(filePath, projectRoot) {
   const patterns = [
     join(dir, `${name}.test.ts`),
     join(dir, `${name}.test.tsx`),
+    join(dir, `${name}.test.js`),
+    join(dir, `${name}.test.jsx`),
     join(dir, `${name}.spec.ts`),
     join(dir, `${name}.spec.tsx`),
+    join(dir, `${name}.spec.js`),
+    join(dir, `${name}.spec.jsx`),
     join(dir, '__tests__', `${name}.test.ts`),
     join(dir, '__tests__', `${name}.test.tsx`),
+    join(dir, '__tests__', `${name}.test.js`),
     join(dir, '__tests__', `${name}.spec.ts`),
     join(dir, '__tests__', `${name}.spec.tsx`),
   ];
@@ -104,10 +105,12 @@ function findTypeFiles(filePath, projectRoot) {
   // Check project-wide types directory
   const globalTypes = join(projectRoot, 'src', 'types');
   if (existsSync(globalTypes)) {
-    const typeFiles = readdirSync(globalTypes).filter(f => f.endsWith('.ts'));
-    for (const tf of typeFiles.slice(0, 5)) {
-      types.push(join(globalTypes, tf));
-    }
+    try {
+      const typeFiles = readdirSync(globalTypes).filter(f => f.endsWith('.ts'));
+      for (const tf of typeFiles.slice(0, 5)) {
+        types.push(join(globalTypes, tf));
+      }
+    } catch { /* permission error */ }
   }
 
   return types;
@@ -115,7 +118,6 @@ function findTypeFiles(filePath, projectRoot) {
 
 function findImporters(filePath, projectRoot) {
   const name = basename(filePath, extname(filePath));
-
   const importers = new Set();
 
   // Search for files that might import this module
@@ -141,7 +143,7 @@ function findImporters(filePath, projectRoot) {
         }
       } catch { /* skip unreadable files */ }
     }
-  } catch (e) { /* rg not found or no matches */ }
+  } catch { /* rg not found or no matches */ }
 
   return Array.from(importers);
 }
@@ -153,43 +155,48 @@ function findSiblings(filePath) {
   try {
     const files = readdirSync(dir).filter(f => {
       const fullPath = join(dir, f);
-      return statSync(fullPath).isFile() &&
-        f !== basename(filePath) &&
-        !f.includes('.test.') &&
-        !f.includes('.spec.') &&
-        (f.endsWith('.ts') || f.endsWith('.tsx'));
+      try {
+        return statSync(fullPath).isFile() &&
+          f !== basename(filePath) &&
+          !f.includes('.test.') &&
+          !f.includes('.spec.') &&
+          (f.endsWith('.ts') || f.endsWith('.tsx') || f.endsWith('.js') || f.endsWith('.jsx'));
+      } catch {
+        return false;
+      }
     });
 
     for (const f of files.slice(0, 5)) {
       siblings.push(join(dir, f));
     }
-  } catch (e) { /* permission error */ }
+  } catch { /* permission error */ }
 
   return siblings;
 }
 
-function printSection(title, files, projectRoot) {
-  if (files.length === 0) return;
-
-  console.log(`\n${title}`);
-  for (const f of files) {
-    const rel = relative(projectRoot, f);
-    const tokens = estimateTokens(f);
-    console.log(`  ${rel} (~${formatTokens(tokens)})`);
+function getFileInfo(filePath) {
+  try {
+    const content = readFileSync(filePath, 'utf-8');
+    return {
+      tokens: estimateTokens(content),
+      lines: content.split('\n').length
+    };
+  } catch {
+    return { tokens: 0, lines: 0 };
   }
 }
 
 // Main
 const args = process.argv.slice(2);
-const targetFile = args[0];
+const options = parseCommonArgs(args);
+const targetFile = options.remaining.find(a => !a.startsWith('-'));
 
-if (!targetFile) {
-  console.log('\nUsage: claude-related <file>\n');
-  console.log('Finds tests, types, and importers for a given file.');
-  process.exit(1);
+if (options.help || !targetFile) {
+  console.log(HELP);
+  process.exit(options.help ? 0 : 1);
 }
 
-const fullPath = join(process.cwd(), targetFile);
+const fullPath = targetFile.startsWith('/') ? targetFile : join(process.cwd(), targetFile);
 if (!existsSync(fullPath)) {
   console.error(`File not found: ${targetFile}`);
   process.exit(1);
@@ -197,31 +204,65 @@ if (!existsSync(fullPath)) {
 
 const projectRoot = findProjectRoot();
 const relPath = relative(projectRoot, fullPath);
+const out = createOutput(options);
 
-console.log(`\n📎 Related files for: ${relPath}`);
-
-const tests = findTestFiles(fullPath, projectRoot);
+const tests = findTestFiles(fullPath);
 const types = findTypeFiles(fullPath, projectRoot);
 const importers = findImporters(fullPath, projectRoot);
 const siblings = findSiblings(fullPath);
 
-printSection('🧪 Tests', tests, projectRoot);
-printSection('📝 Types', types, projectRoot);
-printSection('📥 Imported by', importers.slice(0, 10), projectRoot);
-printSection('👥 Siblings', siblings, projectRoot);
+// Collect file info for JSON
+const testsInfo = tests.map(f => ({ path: relative(projectRoot, f), ...getFileInfo(f) }));
+const typesInfo = types.map(f => ({ path: relative(projectRoot, f), ...getFileInfo(f) }));
+const importersInfo = importers.slice(0, 10).map(f => ({ path: relative(projectRoot, f), ...getFileInfo(f) }));
+const siblingsInfo = siblings.map(f => ({ path: relative(projectRoot, f), ...getFileInfo(f) }));
+
+// Set JSON data
+out.setData('file', relPath);
+out.setData('tests', testsInfo);
+out.setData('types', typesInfo);
+out.setData('importers', importersInfo);
+out.setData('siblings', siblingsInfo);
+out.setData('totalImporters', importers.length);
+
+// Header
+out.header(`Related files for: ${relPath}`);
+out.blank();
+
+// Sections
+function addSection(title, files) {
+  if (files.length === 0) return;
+  out.add(title);
+  for (const f of files) {
+    const rel = relative(projectRoot, f);
+    const info = getFileInfo(f);
+    out.add(`  ${rel} (~${formatTokens(info.tokens)})`);
+  }
+  out.blank();
+}
+
+addSection('Tests:', tests);
+addSection('Types:', types);
+addSection('Imported by:', importers.slice(0, 10));
+addSection('Siblings:', siblings);
 
 const totalFiles = tests.length + types.length + Math.min(importers.length, 10) + siblings.length;
 if (totalFiles === 0) {
-  console.log('\n  No related files found.');
+  out.add('  No related files found.');
+  out.blank();
 }
 
 // Summary
-const totalTokens = [...tests, ...types, ...importers.slice(0, 10), ...siblings]
-  .reduce((sum, f) => sum + estimateTokens(f), 0);
+const allFiles = [...tests, ...types, ...importers.slice(0, 10), ...siblings];
+const totalTokens = allFiles.reduce((sum, f) => sum + getFileInfo(f).tokens, 0);
 
-console.log(`\n📊 Total: ${totalFiles} related files, ~${formatTokens(totalTokens)} tokens`);
+out.setData('totalFiles', totalFiles);
+out.setData('totalTokens', totalTokens);
+
+out.header(`Total: ${totalFiles} related files, ~${formatTokens(totalTokens)} tokens`);
 
 if (importers.length > 10) {
-  console.log(`   (${importers.length - 10} more importers not shown)`);
+  out.header(`(${importers.length - 10} more importers not shown)`);
 }
-console.log();
+
+out.print();