token-pilot 0.12.0 → 0.14.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "token-pilot",
-  "version": "0.12.0",
+  "version": "0.14.1",
   "description": "Save 60-80% tokens when AI reads code — MCP server for token-efficient code navigation, AST-aware structural reading instead of dumping full files into context window",
   "type": "module",
   "main": "dist/index.js",
@@ -18,14 +18,15 @@
     "CHANGELOG.md"
   ],
   "scripts": {
-    "prebuild": "rm -rf dist",
+    "prebuild": "node --input-type=module -e \"import { rm } from 'node:fs/promises'; await rm('dist', { recursive: true, force: true });\"",
     "build": "tsc",
     "dev": "tsc --watch",
     "start": "node dist/index.js",
     "test": "vitest run",
+    "test:coverage": "vitest run --coverage",
     "test:watch": "vitest",
     "lint": "tsc --noEmit",
-    "prepublishOnly": "npm run build && chmod +x dist/index.js"
+    "prepublishOnly": "npm run build && node --input-type=module -e \"import { chmod } from 'node:fs/promises'; await chmod('dist/index.js', 0o755);\""
   },
   "keywords": [
     "token-savings",
@@ -61,6 +62,7 @@
     "chokidar": "^4.0.3"
   },
   "devDependencies": {
+    "@vitest/coverage-v8": "^3.2.4",
     "@types/node": "^22.0.0",
     "typescript": "^5.7.0",
     "vitest": "^3.0.0"

package/start.sh

@@ -1,27 +1,28 @@
-#!/usr/bin/env bash
-# Token Pilot bootstrap script for Claude Code plugin system.
-# Handles auto-install of dependencies and build on first run.
-
-set -e
-
-SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
-cd "$SCRIPT_DIR"
-
-# 1. Install node_modules if missing
-if [ ! -d "node_modules" ]; then
-  echo "[token-pilot] Installing dependencies..." >&2
-  npm install --production --no-audit --no-fund 2>&1 >&2
-fi
-
-# 2. Build if dist/ is missing
-if [ ! -f "dist/index.js" ]; then
-  echo "[token-pilot] Building..." >&2
-  # Need devDependencies for build
-  npm install --no-audit --no-fund 2>&1 >&2
-  npm run build 2>&1 >&2
-fi
-
-# 3. Start the MCP server
-# Pass CLAUDE_PROJECT_DIR as project root if available, otherwise cwd
-PROJECT_ROOT="${CLAUDE_PROJECT_DIR:-$(pwd)}"
-exec node "$SCRIPT_DIR/dist/index.js" "$PROJECT_ROOT"
+#!/usr/bin/env bash
+# Token Pilot bootstrap script for Claude Code plugin system.
+# Handles auto-install of dependencies and build on first run.
+
+set -e
+
+SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
+cd "$SCRIPT_DIR"
+
+# 1. Install runtime dependencies when the package is incomplete.
+if [ ! -f "node_modules/@modelcontextprotocol/sdk/package.json" ]; then
+  echo "[token-pilot] Installing runtime dependencies..." >&2
+  npm install --production --no-audit --no-fund 2>&1 >&2
+fi
+
+# 2. Build if dist/ is missing.
+if [ ! -f "dist/index.js" ]; then
+  echo "[token-pilot] Building..." >&2
+  if [ ! -f "node_modules/typescript/bin/tsc" ]; then
+    npm install --no-audit --no-fund 2>&1 >&2
+  fi
+  npm run build 2>&1 >&2
+fi
+
+# 3. Start the MCP server.
+# Pass CLAUDE_PROJECT_DIR as project root if available, otherwise cwd.
+PROJECT_ROOT="${CLAUDE_PROJECT_DIR:-$(pwd)}"
+exec node "$SCRIPT_DIR/dist/index.js" "$PROJECT_ROOT"

package/dist/handlers/class-hierarchy.d.ts

@@ -1,11 +0,0 @@
-import type { AstIndexClient } from '../ast-index/client.js';
-export interface ClassHierarchyArgs {
-    name: string;
-}
-export declare function handleClassHierarchy(args: ClassHierarchyArgs, astIndex: AstIndexClient): Promise<{
-    content: Array<{
-        type: 'text';
-        text: string;
-    }>;
-}>;
-//# sourceMappingURL=class-hierarchy.d.ts.map

package/dist/handlers/class-hierarchy.js

@@ -1,28 +0,0 @@
-export async function handleClassHierarchy(args, astIndex) {
-    const tree = await astIndex.hierarchy(args.name);
-    if (!tree) {
-        return {
-            content: [{
-                    type: 'text',
-                    text: `No hierarchy found for "${args.name}".`,
-                }],
-        };
-    }
-    const lines = [
-        `HIERARCHY: "${args.name}"`,
-        '',
-    ];
-    formatNode(tree, lines, 0);
-    return { content: [{ type: 'text', text: lines.join('\n') }] };
-}
-function formatNode(node, lines, depth) {
-    const indent = '  '.repeat(depth);
-    const loc = node.file ? ` (${node.file}:${node.line})` : '';
-    lines.push(`${indent}${node.kind} ${node.name}${loc}`);
-    if (node.children) {
-        for (const child of node.children) {
-            formatNode(child, lines, depth + 1);
-        }
-    }
-}
-//# sourceMappingURL=class-hierarchy.js.map

package/dist/handlers/export-ast-index.d.ts

@@ -1,22 +0,0 @@
-import type { AstIndexClient } from '../ast-index/client.js';
-import type { FileCache } from '../core/file-cache.js';
-export interface ExportAstIndexArgs {
-    paths?: string[];
-    format?: 'markdown' | 'json';
-    all_indexed?: boolean;
-}
-/**
- * Export AST structural data in a format suitable for context-mode's BM25 index.
- *
- * Generates a markdown document with headings per file and symbols as sections,
- * which context-mode can index via its `index` tool for cross-tool search.
- *
- * When all_indexed=true, exports all files known to ast-index (not just cached ones).
- */
-export declare function handleExportAstIndex(args: ExportAstIndexArgs, astIndex: AstIndexClient, fileCache: FileCache): Promise<{
-    content: Array<{
-        type: 'text';
-        text: string;
-    }>;
-}>;
-//# sourceMappingURL=export-ast-index.d.ts.map

package/dist/handlers/export-ast-index.js

@@ -1,175 +0,0 @@
-/**
- * Export AST structural data in a format suitable for context-mode's BM25 index.
- *
- * Generates a markdown document with headings per file and symbols as sections,
- * which context-mode can index via its `index` tool for cross-tool search.
- *
- * When all_indexed=true, exports all files known to ast-index (not just cached ones).
- */
-export async function handleExportAstIndex(args, astIndex, fileCache) {
-    const format = args.format ?? 'markdown';
-    // When all_indexed is requested, use ast-index to get all file outlines directly
-    if (args.all_indexed) {
-        return exportAllIndexed(astIndex, format, args.paths);
-    }
-    // Gather all cached files or specified subset
-    const cachedPaths = fileCache.cachedPaths();
-    const targetPaths = args.paths && args.paths.length > 0
-        ? args.paths.filter(p => cachedPaths.includes(p))
-        : cachedPaths;
-    if (targetPaths.length === 0) {
-        return {
-            content: [{
-                    type: 'text',
-                    text: 'No cached files available.\nHINT: Use all_indexed=true to export all files from the ast-index, or use smart_read on files first to populate the cache.',
-                }],
-        };
-    }
-    if (format === 'json') {
-        return exportJson(targetPaths, fileCache);
-    }
-    return exportMarkdown(targetPaths, fileCache);
-}
-function exportMarkdown(paths, fileCache) {
-    const sections = [
-        '# Token Pilot AST Index Export',
-        '',
-        `Exported ${paths.length} files. Index this content via context-mode for cross-tool search.`,
-        '',
-    ];
-    for (const filePath of paths) {
-        const cached = fileCache.get(filePath);
-        if (!cached)
-            continue;
-        const { structure } = cached;
-        sections.push(`## ${structure.path}`);
-        sections.push('');
-        sections.push(`Language: ${structure.language} | Lines: ${structure.meta.lines}`);
-        sections.push('');
-        // Imports as a section
-        if (structure.imports.length > 0) {
-            sections.push('### Imports');
-            for (const imp of structure.imports) {
-                sections.push(`- \`${imp.specifiers.join(', ')}\` from \`${imp.source}\``);
-            }
-            sections.push('');
-        }
-        // Symbols as searchable sections
-        for (const sym of structure.symbols) {
-            formatSymbolMarkdown(sym, sections, 3);
-        }
-        sections.push('---');
-        sections.push('');
-    }
-    sections.push('');
-    sections.push('To index in context-mode, pass this content to the `index` tool with source: "token-pilot-ast".');
-    return { content: [{ type: 'text', text: sections.join('\n') }] };
-}
-function exportJson(paths, fileCache) {
-    const data = [];
-    for (const filePath of paths) {
-        const cached = fileCache.get(filePath);
-        if (!cached)
-            continue;
-        const symbols = flattenSymbols(cached.structure.symbols);
-        data.push({
-            path: cached.structure.path,
-            language: cached.structure.language,
-            lines: cached.structure.meta.lines,
-            symbols,
-        });
-    }
-    return { content: [{ type: 'text', text: JSON.stringify(data, null, 2) }] };
-}
-function formatSymbolMarkdown(sym, sections, headingLevel) {
-    const heading = '#'.repeat(Math.min(headingLevel, 6));
-    sections.push(`${heading} ${sym.kind} \`${sym.name}\``);
-    sections.push('');
-    sections.push(`Signature: \`${sym.signature}\` (L${sym.location.startLine}-${sym.location.endLine})`);
-    if (sym.doc) {
-        sections.push('');
-        sections.push(sym.doc);
-    }
-    sections.push('');
-    for (const child of sym.children) {
-        formatSymbolMarkdown(child, sections, headingLevel + 1);
-    }
-}
-/**
- * Export all files from ast-index directly (bypasses cache).
- * Uses ast-index outline for each file to get structure.
- */
-async function exportAllIndexed(astIndex, format, filterPaths) {
-    let allFiles = await astIndex.listFiles();
-    if (filterPaths && filterPaths.length > 0) {
-        allFiles = allFiles.filter(f => filterPaths.some(p => f.includes(p)));
-    }
-    if (allFiles.length === 0) {
-        return {
-            content: [{
-                    type: 'text',
-                    text: 'No files in ast-index. The index may not be built yet.',
-                }],
-        };
-    }
-    // For large projects, just export file list with symbol counts
-    // Getting full outlines for 1000+ files would be too slow
-    if (allFiles.length > 50) {
-        const sections = [
-            '# Token Pilot AST Index Export',
-            '',
-            `Total indexed files: ${allFiles.length}`,
-            '',
-            '## Indexed Files',
-            '',
-        ];
-        for (const f of allFiles) {
-            sections.push(`- ${f}`);
-        }
-        sections.push('');
-        sections.push('HINT: Use export_ast_index(paths=["src/specific-dir/"]) to export outlines for a subset.');
-        return { content: [{ type: 'text', text: sections.join('\n') }] };
-    }
-    // For smaller sets, get full outlines
-    const sections = [
-        '# Token Pilot AST Index Export',
-        '',
-        `Exported ${allFiles.length} files from ast-index.`,
-        '',
-    ];
-    for (const filePath of allFiles) {
-        const structure = await astIndex.outline(filePath);
-        if (!structure) {
-            sections.push(`## ${filePath}`);
-            sections.push('(no AST structure available)');
-            sections.push('');
-            continue;
-        }
-        sections.push(`## ${structure.path}`);
-        sections.push('');
-        sections.push(`Language: ${structure.language} | Lines: ${structure.meta.lines}`);
-        sections.push('');
-        for (const sym of structure.symbols) {
-            formatSymbolMarkdown(sym, sections, 3);
-        }
-        sections.push('---');
-        sections.push('');
-    }
-    return { content: [{ type: 'text', text: sections.join('\n') }] };
-}
-function flattenSymbols(symbols) {
-    const result = [];
-    for (const sym of symbols) {
-        result.push({
-            name: sym.name,
-            kind: sym.kind,
-            signature: sym.signature,
-            location: `L${sym.location.startLine}-${sym.location.endLine}`,
-        });
-        if (sym.children.length > 0) {
-            result.push(...flattenSymbols(sym.children));
-        }
-    }
-    return result;
-}
-//# sourceMappingURL=export-ast-index.js.map

package/dist/handlers/find-implementations.d.ts

@@ -1,11 +0,0 @@
-import type { AstIndexClient } from '../ast-index/client.js';
-export interface FindImplementationsArgs {
-    name: string;
-}
-export declare function handleFindImplementations(args: FindImplementationsArgs, astIndex: AstIndexClient): Promise<{
-    content: Array<{
-        type: 'text';
-        text: string;
-    }>;
-}>;
-//# sourceMappingURL=find-implementations.d.ts.map

package/dist/handlers/find-implementations.js

@@ -1,27 +0,0 @@
-export async function handleFindImplementations(args, astIndex) {
-    const results = await astIndex.implementations(args.name);
-    if (results.length === 0) {
-        const hints = [`No implementations found for "${args.name}".`];
-        if (!astIndex.isAvailable()) {
-            hints.push('WARNING: ast-index is not available. Install it: cargo install ast-index');
-        }
-        else {
-            hints.push('TIP: Index may not cover this language/project. Run `ast-index build` in the project root.');
-        }
-        return { content: [{ type: 'text', text: hints.join('\n') }] };
-    }
-    const lines = [
-        `IMPLEMENTATIONS: "${args.name}" (${results.length} found)`,
-        '',
-    ];
-    for (const impl of results) {
-        const methods = impl.methods?.length
-            ? ` — methods: ${impl.methods.join(', ')}`
-            : '';
-        lines.push(`  ${impl.kind} ${impl.name} (${impl.file}:${impl.line})${methods}`);
-    }
-    lines.push('');
-    lines.push('HINT: Use read_symbol() to inspect a specific implementation.');
-    return { content: [{ type: 'text', text: lines.join('\n') }] };
-}
-//# sourceMappingURL=find-implementations.js.map

package/dist/handlers/search-code.d.ts

@@ -1,14 +0,0 @@
-import type { AstIndexClient } from '../ast-index/client.js';
-export interface SearchCodeArgs {
-    query: string;
-    in_file?: string;
-    max_results?: number;
-    fuzzy?: boolean;
-}
-export declare function handleSearchCode(args: SearchCodeArgs, astIndex: AstIndexClient): Promise<{
-    content: Array<{
-        type: 'text';
-        text: string;
-    }>;
-}>;
-//# sourceMappingURL=search-code.d.ts.map

package/dist/handlers/search-code.js

@@ -1,32 +0,0 @@
-export async function handleSearchCode(args, astIndex) {
-    const results = await astIndex.search(args.query, {
-        inFile: args.in_file,
-        maxResults: args.max_results ?? 20,
-        fuzzy: args.fuzzy,
-    });
-    if (results.length === 0) {
-        const hints = [`No results found for "${args.query}".`];
-        if (!args.fuzzy)
-            hints.push('TIP: Try fuzzy=true for broader matching.');
-        if (!astIndex.isAvailable()) {
-            hints.push('WARNING: ast-index is not available. Install it: cargo install ast-index');
-        }
-        else {
-            hints.push('TIP: Index may not cover this language/project. Run `ast-index build` in the project root.');
-            hints.push('TIP: Check that the symbol exists and the index is up to date (project_overview shows index status).');
-        }
-        return { content: [{ type: 'text', text: hints.join('\n') }] };
-    }
-    const lines = [
-        `SEARCH: "${args.query}" (${results.length} results)`,
-        '',
-    ];
-    for (const r of results) {
-        lines.push(`  ${r.file}:${r.line}`);
-        lines.push(`    ${r.text.trim()}`);
-    }
-    lines.push('');
-    lines.push('HINT: Use read_symbol() or read_range() to load specific results.');
-    return { content: [{ type: 'text', text: lines.join('\n') }] };
-}
-//# sourceMappingURL=search-code.js.map