@townco/debugger 0.1.66 → 0.1.68

package/src/lib/toolRegistry.ts

@@ -0,0 +1,424 @@
+import type { LucideIcon } from "lucide-react";
+import {
+  CheckSquare,
+  Code,
+  Eye,
+  FileText,
+  FolderOpen,
+  FoldVertical,
+  Globe,
+  Image,
+  ListTree,
+  MessageSquare,
+  Pencil,
+  Search,
+  Terminal,
+  Trash,
+  Wrench,
+} from "lucide-react";
+
+// ============================================================================
+// Types
+// ============================================================================
+
+/**
+ * Category of tool operation
+ */
+export type ToolCategory =
+  | "file" // File system operations
+  | "code" // Code analysis and search
+  | "web" // Web/browser operations
+  | "ai" // AI/LLM operations
+  | "task" // Task management and todos
+  | "subagent" // Subagent/delegation
+  | "compaction" // Context compression
+  | "sandbox" // Sandbox/execution
+  | "other"; // Unknown/other
+
+/**
+ * Metadata for a tool
+ */
+export interface ToolMetadata {
+  /** Display name for the tool */
+  name: string;
+  /** Icon to use in UI */
+  icon: LucideIcon;
+  /** Category for grouping/filtering */
+  category: ToolCategory;
+  /** Optional color override (defaults to category color) */
+  color?: string;
+  /** Optional description */
+  description?: string;
+}
+
+/**
+ * Pattern matcher for dynamic tool names
+ */
+interface ToolPattern {
+  /** Test function to match tool names */
+  test: (toolName: string) => boolean;
+  /** Metadata to return if matched */
+  metadata: Omit<ToolMetadata, "name">;
+}
+
+// ============================================================================
+// Tool Registry
+// ============================================================================
+
+/**
+ * Registry of known tools with their metadata
+ * Add new tools here as exact matches
+ */
+const KNOWN_TOOLS: Record<string, Omit<ToolMetadata, "name">> = {
+  // Task management
+  todo_write: {
+    icon: CheckSquare,
+    category: "task",
+    description: "Create or update todo items",
+  },
+
+  // Subagents
+  Task: {
+    icon: ListTree,
+    category: "subagent",
+    description: "Delegate to a subagent",
+  },
+
+  // Web operations
+  WebSearch: {
+    icon: Search,
+    category: "web",
+    description: "Search the web",
+  },
+  WebFetch: {
+    icon: Globe,
+    category: "web",
+    description: "Fetch web page content",
+  },
+
+  // Document operations
+  document_extract: {
+    icon: FileText,
+    category: "file",
+    description: "Extract text from documents",
+  },
+
+  // File operations
+  Read: {
+    icon: FileText,
+    category: "file",
+    description: "Read file contents",
+  },
+  read_file: {
+    icon: FileText,
+    category: "file",
+    description: "Read file contents",
+  },
+  Write: {
+    icon: Pencil,
+    category: "file",
+    description: "Write to a file",
+  },
+  write: {
+    icon: Pencil,
+    category: "file",
+    description: "Write to a file",
+  },
+  search_replace: {
+    icon: Pencil,
+    category: "file",
+    description: "Search and replace in files",
+  },
+  delete_file: {
+    icon: Trash,
+    category: "file",
+    description: "Delete a file",
+  },
+  list_dir: {
+    icon: FolderOpen,
+    category: "file",
+    description: "List directory contents",
+  },
+
+  // Code operations
+  Grep: {
+    icon: Search,
+    category: "code",
+    description: "Search code with grep",
+  },
+  grep: {
+    icon: Search,
+    category: "code",
+    description: "Search code with grep",
+  },
+  codebase_search: {
+    icon: Search,
+    category: "code",
+    description: "Semantic code search",
+  },
+  read_lints: {
+    icon: Code,
+    category: "code",
+    description: "Read linter errors",
+  },
+
+  // Terminal operations
+  run_terminal_cmd: {
+    icon: Terminal,
+    category: "sandbox",
+    description: "Run terminal command",
+  },
+
+  // Image generation
+  Sandbox_GenerateImage: {
+    icon: Image,
+    category: "sandbox",
+    description: "Generate an image",
+  },
+};
+
+/**
+ * Pattern matchers for dynamic tool names
+ * These are checked in order if no exact match is found
+ */
+const TOOL_PATTERNS: ToolPattern[] = [
+  // Sandbox operations (prefix-based)
+  {
+    test: (name) => name.startsWith("Sandbox_"),
+    metadata: {
+      icon: Terminal,
+      category: "sandbox",
+      description: "Sandbox operation",
+    },
+  },
+
+  // Browser operations (prefix-based)
+  {
+    test: (name) =>
+      name.startsWith("Browser") ||
+      name.toLowerCase().startsWith("mcp_cursor-browser"),
+    metadata: {
+      icon: Globe,
+      category: "web",
+      description: "Browser operation",
+    },
+  },
+
+  // Subagent tools (pattern-based)
+  {
+    test: (name) => name.toLowerCase().includes("subagent"),
+    metadata: {
+      icon: ListTree,
+      category: "subagent",
+      description: "Subagent operation",
+    },
+  },
+
+  // File operations (keyword-based)
+  {
+    test: (name) => {
+      const lower = name.toLowerCase();
+      return lower.includes("read") || lower.includes("file");
+    },
+    metadata: {
+      icon: FileText,
+      category: "file",
+    },
+  },
+  {
+    test: (name) => {
+      const lower = name.toLowerCase();
+      return lower.includes("write") || lower.includes("edit");
+    },
+    metadata: {
+      icon: Pencil,
+      category: "file",
+    },
+  },
+  {
+    test: (name) => name.toLowerCase().includes("delete"),
+    metadata: {
+      icon: Trash,
+      category: "file",
+    },
+  },
+  {
+    test: (name) => {
+      const lower = name.toLowerCase();
+      return lower.includes("list") || lower.includes("dir");
+    },
+    metadata: {
+      icon: FolderOpen,
+      category: "file",
+    },
+  },
+
+  // Code operations (keyword-based)
+  {
+    test: (name) => {
+      const lower = name.toLowerCase();
+      return (
+        lower.includes("grep") ||
+        lower.includes("search") ||
+        lower.includes("codebase")
+      );
+    },
+    metadata: {
+      icon: Search,
+      category: "code",
+    },
+  },
+  {
+    test: (name) =>
+      name.toLowerCase().includes("terminal") ||
+      name.toLowerCase().includes("run"),
+    metadata: {
+      icon: Terminal,
+      category: "sandbox",
+    },
+  },
+  {
+    test: (name) => {
+      const lower = name.toLowerCase();
+      return lower.includes("lint") || lower.includes("analyze");
+    },
+    metadata: {
+      icon: Code,
+      category: "code",
+    },
+  },
+
+  // Web operations (keyword-based)
+  {
+    test: (name) => name.toLowerCase().includes("web"),
+    metadata: {
+      icon: Globe,
+      category: "web",
+    },
+  },
+
+  // AI operations (keyword-based)
+  {
+    test: (name) => {
+      const lower = name.toLowerCase();
+      return lower.includes("chat") || lower.includes("message");
+    },
+    metadata: {
+      icon: MessageSquare,
+      category: "ai",
+    },
+  },
+
+  // View operations (keyword-based)
+  {
+    test: (name) => {
+      const lower = name.toLowerCase();
+      return lower.includes("view") || lower.includes("show");
+    },
+    metadata: {
+      icon: Eye,
+      category: "file",
+    },
+  },
+
+  // Compaction operations (keyword-based)
+  {
+    test: (name) => {
+      const lower = name.toLowerCase();
+      return (
+        lower.includes("compact") ||
+        lower.includes("compression") ||
+        lower.includes("summarize") ||
+        lower.includes("summarization")
+      );
+    },
+    metadata: {
+      icon: FoldVertical,
+      category: "compaction",
+    },
+  },
+];
+
+/**
+ * Default icons by category
+ */
+const CATEGORY_DEFAULTS: Record<ToolCategory, LucideIcon> = {
+  file: FileText,
+  code: Code,
+  web: Globe,
+  ai: MessageSquare,
+  task: CheckSquare,
+  subagent: ListTree,
+  compaction: FoldVertical,
+  sandbox: Terminal,
+  other: Wrench,
+};
+
+// ============================================================================
+// Public API
+// ============================================================================
+
+/**
+ * Get metadata for a tool by name
+ *
+ * Lookup order:
+ * 1. Exact match in KNOWN_TOOLS
+ * 2. Pattern match in TOOL_PATTERNS
+ * 3. Fallback to "other" category
+ */
+export function getToolMetadata(toolName: string): ToolMetadata {
+  // Check exact matches first
+  const exactMatch = KNOWN_TOOLS[toolName];
+  if (exactMatch) {
+    return {
+      name: toolName,
+      ...exactMatch,
+    };
+  }
+
+  // Check patterns
+  for (const pattern of TOOL_PATTERNS) {
+    if (pattern.test(toolName)) {
+      return {
+        name: toolName,
+        ...pattern.metadata,
+      };
+    }
+  }
+
+  // Fallback
+  return {
+    name: toolName,
+    icon: CATEGORY_DEFAULTS.other,
+    category: "other",
+    description: "Unknown tool",
+  };
+}
+
+/**
+ * Get just the icon for a tool (convenience method)
+ */
+export function getToolIcon(toolName: string): LucideIcon {
+  return getToolMetadata(toolName).icon;
+}
+
+/**
+ * Get the default icon for a category
+ */
+export function getCategoryIcon(category: ToolCategory): LucideIcon {
+  return CATEGORY_DEFAULTS[category];
+}
+
+/**
+ * Get all registered tool names (for autocomplete, documentation, etc.)
+ */
+export function getRegisteredTools(): string[] {
+  return Object.keys(KNOWN_TOOLS);
+}
+
+/**
+ * Check if a tool is registered
+ */
+export function isRegisteredTool(toolName: string): boolean {
+  return toolName in KNOWN_TOOLS;
+}