agent-dir 0.5.0 → 1.1.0

package/dist/bin/agent-dir.js

@@ -5,6 +5,7 @@ import { createRequire } from "node:module";
 import { resolve } from "node:path";
 import { createInterface } from "node:readline/promises";
 import { parseArgs } from "node:util";
+import { CodeGraphIntegration } from "../src/codegraph.js";
 import { configPath, loadConfig, mergeProfile, saveConfig } from "../src/config.js";
 import { printBanner, printHelp, printRequestHeader } from "../src/logging.js";
 import { startServer } from "../src/server.js";
@@ -51,7 +52,6 @@ const { values, positionals } = parseArgs({
         command: { type: "string" },
         blacklist: { type: "string" },
         token: { type: "string" },
-        git: { type: "boolean" },
         rotate: { type: "boolean" },
         "no-tunnel": { type: "boolean" },
         random: { type: "boolean" },
@@ -344,8 +344,6 @@ if (command === "config") {
         const blacklist = parseList(values.blacklist);
         if (blacklist !== undefined)
             profile.blacklistedCommands = blacklist;
-        if (values.git !== undefined)
-            profile.git = values.git;
         if (profile.tunnel === "wormhole" && !profile.subdomain && !values.random) {
             fail("Wormhole profiles require --subdomain <name>, or use --random.");
         }
@@ -411,8 +409,6 @@ if (commands !== undefined)
     overrides.commands = commands;
 if (blacklist !== undefined)
     overrides.blacklistedCommands = blacklist;
-if (values.git !== undefined)
-    overrides.git = values.git;
 const commandConfig = mergeProfile(profile ?? { directory, port, tunnel }, overrides);
 const telemetryConfigId = profile
     ? command
@@ -427,7 +423,6 @@ const telemetryConfigId = profile
         npm: commandConfig.npm?.allowedScripts ?? [],
         commands: commandConfig.commands ?? [],
         blacklistedCommands: commandConfig.blacklistedCommands ?? [],
-        git: commandConfig.git ?? commandConfig.commands?.includes("git") ?? false,
     }))
         .digest("hex")
         .slice(0, 12);
@@ -451,7 +446,10 @@ if (allowedCommands)
 if (commandConfig.blacklistedCommands?.length) {
     bannerOptions.blacklistedCommands = commandConfig.blacklistedCommands;
 }
-bannerOptions.gitEnabled = commandConfig.git ?? commandConfig.commands?.includes("git") ?? false;
+const codeGraph = new CodeGraphIntegration(directory);
+const codeGraphCapability = await codeGraph.prepare();
+if (codeGraphCapability.tool)
+    bannerOptions.codeGraph = true;
 printBanner(bannerOptions);
 const telemetrySessionId = randomUUID();
 const server = await startServer({
@@ -459,12 +457,16 @@ const server = await startServer({
     port,
     token,
     commandConfig,
+    codeGraph,
     telemetry: {
         ...(config.telemetry ?? { level: "none" }),
         ...(telemetryConfigId ? { configId: telemetryConfigId } : {}),
         sessionId: telemetrySessionId,
         persist: true,
     },
+}).catch(async (error) => {
+    await codeGraph.close();
+    throw error;
 });
 let tunnelResult;
 console.log("  ✓ SERVER ONLINE");

package/dist/src/auth.js

@@ -14,7 +14,9 @@ function secureCompare(left, right) {
     return timingSafeEqual(leftHash, rightHash);
 }
 export function unauthorized() {
-    return new Response(JSON.stringify({ error: "Unauthorized" }), {
+    return new Response(JSON.stringify({
+        error: "Unauthorized: provide a valid Authorization: Bearer <token> header.",
+    }), {
         status: 401,
         headers: {
             "content-type": "application/json; charset=utf-8",

package/dist/src/codegraph.d.ts

@@ -18,6 +18,7 @@ export declare class CodeGraphIntegration {
     private readonly pending;
     constructor(root: string);
     capability(): Promise<CodeGraphCapability>;
+    prepare(): Promise<CodeGraphCapability>;
     explore(query: string, maxFiles?: number): Promise<Record<string, unknown>>;
     close(): Promise<void>;
     private detect;

package/dist/src/codegraph.js

@@ -27,6 +27,18 @@ export class CodeGraphIntegration {
             this.capabilityPromise = this.detect();
         return this.capabilityPromise;
     }
+    async prepare() {
+        const capability = await this.capability();
+        if (capability.status === "available") {
+            try {
+                await this.ensureStarted();
+            }
+            catch {
+                // Keep Agent Dir startup available even if CodeGraph fails to start.
+            }
+        }
+        return this.capability();
+    }
     async explore(query, maxFiles) {
         const capability = await this.capability();
         if (capability.status === "not_installed")

package/dist/src/config.js

@@ -117,7 +117,5 @@ function isProfile(value) {
             profile.blacklistedCommands.some((item) => typeof item !== "string"))
             return false;
     }
-    if (profile.git !== undefined && typeof profile.git !== "boolean")
-        return false;
     return true;
 }

package/dist/src/logging.d.ts

@@ -11,7 +11,7 @@ export declare function printBanner(options: {
     npmScripts?: string[];
     commands?: string[];
     blacklistedCommands?: string[];
-    gitEnabled?: boolean;
+    codeGraph?: boolean;
 }): void;
 export declare function printRequestLog(entry: RequestLog): void;
 export declare function logRequest(entry: RequestLog): void;

package/dist/src/logging.js

@@ -9,53 +9,24 @@ const BRIGHT_BLUE = "\x1b[94m";
 const BRIGHT_CYAN = "\x1b[96m";
 const BRIGHT_MAGENTA = "\x1b[95m";
 const BRIGHT_YELLOW = "\x1b[93m";
-const BRIGHT_RED = "\x1b[91m";
-const BRIGHT_GREEN = "\x1b[92m";
 const writeLine = (line = "") => {
     process.stdout.write(`${line}\r\n`);
 };
 const color = (value, code) => `${code}${value}${RESET}`;
 const TOOL_ICONS = {
-    list_files: "📂",
-    list_dirs: "🗂️",
     read_range: "📐",
-    read_files: "📚",
-    write_files: "📝",
-    patch_files: "🧩",
-    delete_files: "🗑️",
     run_npm_batch: "🔧",
     run_command_batch: "🛠️",
-    search_files: "🔎",
     find_files: "🧭",
     search_code: "🧠",
     find_symbol: "🔹",
-    find_definition: "🎯",
     find_references: "🔗",
-    find_imports: "↪",
-    find_exports: "↗",
-    git_status: "🌿",
-    git_stage: "➕",
-    git_unstage: "➖",
-    git_commit: "●",
-    git_restore: "↩",
-    git_push: "⬆",
-    git_diff: "📝",
-    git_log: "📜",
-    project_overview: "🗺️",
-    package_info: "📦",
-    allowed_commands: "🔐",
     file_info: "ℹ️",
-    diagnostics: "🩺",
     project_context: "🧭",
-    inspect: "🔎",
-    locate: "🎯",
-    read_relevant: "🎯",
-    git_changes: "📋",
     apply_changes: "✏️",
     validate: "✓",
 };
 const MCP_METHOD_ICONS = {
-    initialize: "🚀",
     "notifications/initialized": "✓",
     "notifications/cancelled": "✖",
     "notifications/progress": "⏳",
@@ -80,19 +51,8 @@ const MCP_METHOD_ICONS = {
 const TOOL_NAME_WIDTH = Math.max(...Object.keys(TOOL_ICONS).map((name) => name.length));
 const MCP_METHOD_NAME_WIDTH = Math.max(...Object.keys(MCP_METHOD_ICONS).map((name) => name.length));
 const TOOL_COLORS = {
-    list_files: CYAN,
-    list_dirs: BRIGHT_CYAN,
-    read_files: BRIGHT_GREEN,
-    write_files: BRIGHT_YELLOW,
-    patch_files: BRIGHT_MAGENTA,
-    delete_files: BRIGHT_RED,
     run_npm_batch: BRIGHT_BLUE,
     run_command_batch: BRIGHT_CYAN,
-    git_stage: YELLOW,
-    git_unstage: MAGENTA,
-    git_commit: BRIGHT_GREEN,
-    git_restore: RED,
-    git_push: BRIGHT_BLUE,
 };
 const NARROW_TOOL_ICONS = new Set(["✎", "✖", "⚙", "⌘", "↪", "↗"]);
 const TOOL_ICON_SLOT_WIDTH = 3;
@@ -126,8 +86,6 @@ export function printHelp() {
     writeLine("      --npm <scripts>        Comma-separated allowed npm scripts");
     writeLine("      --command <commands>   Comma-separated allowed commands");
     writeLine("      --blacklist <commands> Comma-separated blocked command prefixes");
-    writeLine("      --git                  Enable dedicated Git MCP tools");
-    writeLine("      --no-git               Disable dedicated Git MCP tools");
     writeLine("      --token <token>        Explicit Bearer token");
     writeLine("      --no-tunnel            Disable tunneling");
     writeLine("      --random               Use a random Wormhole URL (no subdomain)");
@@ -180,38 +138,20 @@ export function printBanner(options) {
     writeLine(`    ${color("⚡ /mcp", CYAN)}`);
     writeLine("");
     writeLine("  TOOLS");
-    writeLine(`    ${color(toolLabel("list_files"), CYAN)}`);
-    writeLine(`    ${color(toolLabel("list_dirs"), BRIGHT_CYAN)}`);
-    writeLine(`    ${color(toolLabel("read_range"), GREEN)}`);
-    writeLine(`    ${color(toolLabel("read_files"), BRIGHT_GREEN)}`);
-    writeLine(`    ${color(toolLabel("write_files"), BRIGHT_YELLOW)}`);
-    writeLine(`    ${color(toolLabel("patch_files"), BRIGHT_MAGENTA)}`);
-    writeLine(`    ${color(toolLabel("delete_files"), BRIGHT_RED)}`);
-    writeLine(`    ${color(toolLabel("run_npm_batch"), BRIGHT_BLUE)}`);
-    writeLine(`    ${color(toolLabel("run_command_batch"), BRIGHT_CYAN)}`);
-    writeLine(`    ${color(toolLabel("search_files"), CYAN)}`);
+    writeLine(`    ${color(toolLabel("project_context"), BRIGHT_CYAN)}`);
     writeLine(`    ${color(toolLabel("find_files"), CYAN)}`);
     writeLine(`    ${color(toolLabel("search_code"), BRIGHT_CYAN)}`);
     writeLine(`    ${color(toolLabel("find_symbol"), GREEN)}`);
-    writeLine(`    ${color(toolLabel("find_definition"), GREEN)}`);
     writeLine(`    ${color(toolLabel("find_references"), GREEN)}`);
-    writeLine(`    ${color(toolLabel("find_imports"), BRIGHT_GREEN)}`);
-    writeLine(`    ${color(toolLabel("find_exports"), BRIGHT_GREEN)}`);
-    if (options.gitEnabled) {
-        writeLine(`    ${color(toolLabel("git_status"), CYAN)}`);
-        writeLine(`    ${color(toolLabel("git_stage"), YELLOW)}`);
-        writeLine(`    ${color(toolLabel("git_unstage"), MAGENTA)}`);
-        writeLine(`    ${color(toolLabel("git_commit"), BRIGHT_GREEN)}`);
-        writeLine(`    ${color(toolLabel("git_restore"), RED)}`);
-        writeLine(`    ${color(toolLabel("git_push"), BRIGHT_BLUE)}`);
-        writeLine(`    ${color(toolLabel("git_diff"), CYAN)}`);
-        writeLine(`    ${color(toolLabel("git_log"), CYAN)}`);
-    }
-    writeLine(`    ${color(toolLabel("project_overview"), BRIGHT_CYAN)}`);
-    writeLine(`    ${color(toolLabel("package_info"), YELLOW)}`);
-    writeLine(`    ${color(toolLabel("allowed_commands"), YELLOW)}`);
+    writeLine(`    ${color(toolLabel("read_range"), GREEN)}`);
+    writeLine(`    ${color(toolLabel("apply_changes"), BRIGHT_MAGENTA)}`);
+    writeLine(`    ${color(toolLabel("validate"), BRIGHT_YELLOW)}`);
+    writeLine(`    ${color(toolLabel("run_npm_batch"), BRIGHT_BLUE)}`);
+    writeLine(`    ${color(toolLabel("run_command_batch"), BRIGHT_CYAN)}`);
     writeLine(`    ${color(toolLabel("file_info"), YELLOW)}`);
-    writeLine(`    ${color(toolLabel("diagnostics"), BRIGHT_YELLOW)}`);
+    if (options.codeGraph) {
+        writeLine(`    ${color(toolLabel("codegraph_explore"), BRIGHT_MAGENTA)}`);
+    }
     writeLine("");
     if (options.npmScripts?.length) {
         writeLine("  NPM COMMANDS");

package/dist/src/mcp.d.ts

@@ -1,3 +1,4 @@
+import { CodeGraphIntegration } from "./codegraph.js";
 import { TelemetryRecorder } from "./telemetry/recorder.js";
 import type { TelemetryConfig } from "./telemetry/types.js";
 import type { CommandConfig } from "./types.js";
@@ -6,6 +7,6 @@ interface McpHandler {
     closeSubscriptions: () => void;
     telemetrySnapshot: () => ReturnType<TelemetryRecorder["snapshot"]>;
 }
-export declare function createMcpHandler(root: string, commandConfig?: CommandConfig, telemetry?: TelemetryConfig): McpHandler;
+export declare function createMcpHandler(root: string, commandConfig?: CommandConfig, telemetry?: TelemetryConfig, codeGraphIntegration?: CodeGraphIntegration): McpHandler;
 export declare function validateMcpParamHeaders(inputSchema: Record<string, unknown>, args: Record<string, unknown>, headers: Headers): string | undefined;
 export {};