Move registry into its own dir

Author: tsck

src/server.ts

@@ -22,7 +22,7 @@ import { packageInfo } from "./common/packageInfo.js";
 import { type ConnectionErrorHandler } from "./common/connectionErrorHandler.js";
 import type { Elicitation } from "./elicitation.js";
 import { AllTools } from "./tools/index.js";
-import { UIRegistry } from "./ui/registry.js";
+import { UIRegistry } from "./ui/registry/index.js";
 
 export interface ServerOptions {
     session: Session;

src/tools/tool.ts

@@ -9,7 +9,7 @@ import type { UserConfig } from "../common/config/userConfig.js";
 import type { Server } from "../server.js";
 import type { Elicitation } from "../elicitation.js";
 import type { PreviewFeature } from "../common/schemas.js";
-import type { UIRegistry } from "../ui/registry.js";
+import type { UIRegistry } from "../ui/registry/index.js";
 
 export type ToolArgs<T extends ZodRawShape> = {
     [K in keyof T]: z.infer<T[K]>;

src/ui/index.ts

@@ -1,3 +1,3 @@
-export { UIRegistry, getDefaultUIRegistry, setDefaultUIRegistry } from "./registry.js";
+export { UIRegistry, getDefaultUIRegistry, setDefaultUIRegistry } from "./registry/index.js";
 export { useRenderData } from "./hooks/index.js";
 export { createUIBuildConfig } from "./build/index.js";

src/ui/registry/index.ts

@@ -0,0 +1 @@
+export { UIRegistry, getDefaultUIRegistry, setDefaultUIRegistry } from "./registry.js";

src/ui/registry.ts src/ui/registry/registry.tssrc/ui/registry.ts renamed to src/ui/registry/registry.ts

@@ -1,5 +1,6 @@
 import { readFileSync, existsSync } from "fs";
 import { join, dirname } from "path";
+import { uiMap } from "./uiMap.js";
 
 /**
  * Get the default UI dist path by finding the package root.
@@ -19,15 +20,6 @@ function getDefaultUIDistPath(): string {
     }
 }
 
-/**
- * Mapping from tool names to their bundled UI HTML file names
- * Tool names use kebab-case (e.g., 'list-databases')
- * HTML files use PascalCase (e.g., 'ListDatabases.html')
- */
-const TOOL_TO_UI_FILE: Record<string, string> = {
-    "list-databases": "ListDatabases.html",
-};
-
 /**
  * UI Registry that manages bundled UI HTML strings for tools.
  *
@@ -75,7 +67,7 @@ export class UIRegistry {
      * @param toolName The name of the tool
      */
     has(toolName: string): boolean {
-        return this.overrides.has(toolName) || toolName in TOOL_TO_UI_FILE;
+        return this.overrides.has(toolName) || toolName in uiMap;
     }
 
     /**
@@ -91,7 +83,7 @@ export class UIRegistry {
         }
 
         // Check if we have a mapping for this tool
-        const htmlFileName = TOOL_TO_UI_FILE[toolName];
+        const htmlFileName = uiMap[toolName];
         if (!htmlFileName) {
             return undefined;
         }
@@ -128,7 +120,7 @@ export class UIRegistry {
      * Get all registered tool names (both overrides and defaults)
      */
     getRegisteredTools(): string[] {
-        const tools = new Set<string>([...this.overrides.keys(), ...Object.keys(TOOL_TO_UI_FILE)]);
+        const tools = new Set<string>([...this.overrides.keys(), ...Object.keys(uiMap)]);
         return Array.from(tools);
     }
 }

src/ui/registry/uiMap.ts

@@ -0,0 +1,8 @@
+/**
+ * Mapping from tool names to their bundled UI HTML file names
+ * Tool names use kebab-case (e.g., 'list-databases')
+ * HTML files use PascalCase (e.g., 'ListDatabases.html')
+ */
+export const uiMap: Record<string, string> = {
+    "list-databases": "ListDatabases.html",
+};