token-pilot 0.12.0 → 0.14.1

package/dist/index.js

@@ -3,6 +3,7 @@ import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'
 import { readFileSync } from 'node:fs';
 import { execFile } from 'node:child_process';
 import { promisify } from 'node:util';
+import { fileURLToPath } from 'node:url';
 import { createServer } from './server.js';
 import { installHook, uninstallHook } from './hooks/installer.js';
 import { findBinary, installBinary, checkBinaryUpdate, isNewerVersion } from './ast-index/binary-manager.js';
@@ -10,14 +11,14 @@ import { loadConfig } from './config/loader.js';
 import { isDangerousRoot } from './core/validation.js';
 const execFileAsync = promisify(execFile);
 const HOOK_DENY_THRESHOLD = 500;
-const CODE_EXTENSIONS = new Set([
+export const CODE_EXTENSIONS = new Set([
     'ts', 'tsx', 'js', 'jsx', 'mjs', 'py', 'go', 'rs', 'java', 'kt', 'kts',
     'swift', 'cs', 'cpp', 'cc', 'cxx', 'hpp', 'c', 'h', 'php', 'rb', 'scala',
     'dart', 'lua', 'sh', 'bash', 'sql', 'r', 'vue', 'svelte', 'pl', 'pm',
     'ex', 'exs', 'groovy', 'm', 'proto', 'bsl',
     'lisp', 'lsp', 'cl', 'asd',
 ]);
-function getVersion() {
+export function getVersion() {
     try {
         const pkgPath = new URL('../package.json', import.meta.url).pathname;
         const pkg = JSON.parse(readFileSync(pkgPath, 'utf-8'));
@@ -27,50 +28,48 @@ function getVersion() {
         return '0.0.0';
     }
 }
-const args = process.argv.slice(2);
-switch (args[0]) {
-    case 'hook-read':
-        handleHookRead(args[1]);
-        break;
-    case 'hook-edit':
-        handleHookEdit();
-        break;
-    case 'install-hook':
-        handleInstallHook(args[1] || process.cwd());
-        break;
-    case 'uninstall-hook':
-        handleUninstallHook(args[1] || process.cwd());
-        break;
-    case 'install-ast-index':
-        handleInstallAstIndex();
-        break;
-    case 'doctor':
-        handleDoctor();
-        break;
-    case 'init':
-        handleInit(args[1] || process.cwd());
-        break;
-    case '--version':
-    case '-v':
-        console.log(getVersion());
-        process.exit(0);
-        break;
-    case '--help':
-    case '-h':
-        printHelp();
-        break;
-    default:
-        startServer().catch(err => {
-            console.error(`[token-pilot] Fatal: ${err instanceof Error ? err.message : err}`);
-            process.exit(1);
-        });
-        break;
+export async function main(cliArgs = process.argv.slice(2)) {
+    switch (cliArgs[0]) {
+        case 'hook-read':
+            handleHookRead(cliArgs[1]);
+            return;
+        case 'hook-edit':
+            handleHookEdit();
+            return;
+        case 'install-hook':
+            await handleInstallHook(cliArgs[1] || process.cwd());
+            return;
+        case 'uninstall-hook':
+            await handleUninstallHook(cliArgs[1] || process.cwd());
+            return;
+        case 'install-ast-index':
+            await handleInstallAstIndex();
+            return;
+        case 'doctor':
+            await handleDoctor();
+            return;
+        case 'init':
+            await handleInit(cliArgs[1] || process.cwd());
+            return;
+        case '--version':
+        case '-v':
+            console.log(getVersion());
+            process.exit(0);
+            return;
+        case '--help':
+        case '-h':
+            printHelp();
+            return;
+        default:
+            await startServer(cliArgs);
+            return;
+    }
 }
-async function startServer() {
-    let projectRoot = args[0] || process.cwd();
+export async function startServer(cliArgs = process.argv.slice(2)) {
+    let projectRoot = cliArgs[0] || process.cwd();
     // Detect git root for reliable project root
     // Try multiple sources: args[0] → INIT_CWD (npm/npx invoking dir) → PWD → cwd
-    if (!args[0]) {
+    if (!cliArgs[0]) {
         const candidates = [
             process.env.INIT_CWD, // npm/npx sets this to invoking directory
             process.env.PWD, // shell working directory (may differ from cwd)
@@ -140,7 +139,7 @@ async function startServer() {
         process.exit(0);
     });
 }
-function handleHookRead(filePathArg) {
+export function handleHookRead(filePathArg) {
     // Parse stdin (Claude Code hook format) to get tool_input
     let filePath = filePathArg;
     let hasOffset = false;
@@ -193,7 +192,7 @@ function handleHookRead(filePathArg) {
     process.stdout.write(deny);
     process.exit(0);
 }
-function handleHookEdit() {
+export function handleHookEdit() {
     // Parse stdin for Edit tool_input
     let filePath;
     try {
@@ -223,17 +222,17 @@ function handleHookEdit() {
     process.stdout.write(context);
     process.exit(0);
 }
-async function handleInstallHook(projectRoot) {
+export async function handleInstallHook(projectRoot) {
     const result = await installHook(projectRoot);
     console.log(result.message);
-    process.exit(result.installed ? 0 : 1);
+    process.exit(result.fatal ? 1 : 0);
 }
-async function handleUninstallHook(projectRoot) {
+export async function handleUninstallHook(projectRoot) {
     const result = await uninstallHook(projectRoot);
     console.log(result.message);
-    process.exit(result.removed ? 0 : 1);
+    process.exit(result.fatal ? 1 : 0);
 }
-async function handleInstallAstIndex() {
+export async function handleInstallAstIndex() {
     const status = await findBinary();
     if (status.available) {
         // Check if update is available
@@ -256,7 +255,7 @@ async function handleInstallAstIndex() {
         process.exit(1);
     }
 }
-async function handleDoctor() {
+export async function handleDoctor() {
     const version = getVersion();
     const { existsSync } = await import('node:fs');
     const { join } = await import('node:path');
@@ -324,7 +323,7 @@ async function handleDoctor() {
     console.log('');
     process.exit(0);
 }
-async function handleInit(targetDir) {
+export async function handleInit(targetDir) {
     const { existsSync, readFileSync: readFs, writeFileSync } = await import('node:fs');
     const { join } = await import('node:path');
     const mcpPath = join(targetDir, '.mcp.json');
@@ -379,7 +378,7 @@ async function handleInit(targetDir) {
 // ──────────────────────────────────────────────
 // Update checking
 // ──────────────────────────────────────────────
-async function checkNpmLatest(packageName) {
+export async function checkNpmLatest(packageName) {
     try {
         const controller = new AbortController();
         const timeout = setTimeout(() => controller.abort(), 3000);
@@ -396,7 +395,7 @@ async function checkNpmLatest(packageName) {
         return null;
     }
 }
-async function checkAllUpdates(config, binaryStatus) {
+export async function checkAllUpdates(config, binaryStatus) {
     if (!config.updates.checkOnStartup)
         return;
     const [tpLatest, astUpdate, cmLatest] = await Promise.allSettled([
@@ -427,7 +426,7 @@ async function checkAllUpdates(config, binaryStatus) {
         // On startup, we only notify if explicitly useful.
     }
 }
-function printHelp() {
+export function printHelp() {
     console.log(`token-pilot v${getVersion()} — MCP server for token-efficient code reading
 
 Usage:
@@ -450,4 +449,11 @@ MCP Tools (18):
 `);
     process.exit(0);
 }
+const isDirectRun = process.argv[1] !== undefined && fileURLToPath(import.meta.url) === process.argv[1];
+if (isDirectRun) {
+    main().catch(err => {
+        console.error(`[token-pilot] Fatal: ${err instanceof Error ? err.message : err}`);
+        process.exit(1);
+    });
+}
 //# sourceMappingURL=index.js.map