@wener/mcps 1.0.1 → 1.0.4

package/lib/server/server.js

@@ -5,38 +5,32 @@ import { LRUCache } from "lru-cache";
 import { isDevelopment } from "std-env";
 import { findMcpServerDef } from "../providers/findMcpServerDef.js";
 import { registerApiRoutes } from "./api-routes.js";
-import { auditMiddleware, configureAudit } from "./audit.js";
 import { registerChatRoutes } from "./chat-routes.js";
 import { loadConfig, loadEnvFiles, substituteEnvVars } from "./config.js";
+import { createMcpsEmitter, McpsEventType } from "./events.js";
 import { registerMcpRoutes } from "./mcp-routes.js";
 const log = consola.withTag("mcps");
 export function createServer(options = {}) {
-    const { cwd = process.cwd(), discoveryConfig: discoveryConfigOption } = options;
+    const { cwd = process.cwd(), discoveryConfig: discoveryConfigOption, setup } = options;
     const app = new Hono();
+    const emitter = createMcpsEmitter();
+    const apiRouters = {};
     // Request logging
     app.use(logger((v) => {
         log.debug(v);
     }));
-    // Audit middleware
-    app.use(auditMiddleware());
+    // Request event middleware - emits events for subscribers (audit, monitoring, etc.)
+    app.use(requestEventMiddleware(emitter));
     // Load .env files first
     loadEnvFiles(cwd);
     // Load config (with env var substitution)
     const config = substituteEnvVars(loadConfig(cwd));
-    // discoveryConfig: CLI option overrides config file, defaults to false
     const discoveryConfig = discoveryConfigOption ?? config.discoveryConfig ?? false;
     // Log available server types from registry
     const serverDefs = findMcpServerDef();
     log.info(`Available server types: ${serverDefs.map((d) => d.name).join(", ")}`);
     log.info(`Loaded ${Object.keys(config.servers).length} servers from config (discoveryConfig: ${discoveryConfig})`);
-    // Configure audit with lazy DB initialization
-    // DB will only be initialized when the first audit event needs persistence
-    configureAudit(config.audit, config.db);
-    const auditEnabled = config.audit?.enabled !== false;
-    const auditDbPath = config.audit?.db?.path ?? config.db?.path ?? ".mcps.db";
-    log.info(`Audit configured: enabled=${auditEnabled}, db=${auditDbPath} (lazy init)`);
-    // Unified cache for all MCP servers (both pre-configured and dynamic)
-    // Keyed by cache key from def.getCacheKey() or `config::${name}` for pre-configured
+    // Unified cache for all MCP servers
     const serverCache = new LRUCache({
         max: 100,
         dispose: (value, key) => {
@@ -44,76 +38,81 @@ export function createServer(options = {}) {
             value.close?.().catch((e) => log.error("Failed to close server", e));
         }
     });
-    // =========================================================================
-    // Register MCP routes (pre-configured and dynamic endpoints)
-    // =========================================================================
-    registerMcpRoutes({
+    const ctx = {
         app,
         config,
-        serverCache
-    });
-    // =========================================================================
-    // Register Chat/LLM Gateway routes
-    // =========================================================================
-    registerChatRoutes({
-        app,
-        config
-    });
-    // =========================================================================
-    // Health and info endpoints
-    // =========================================================================
-    app.get("/health", (c) => c.json({
-        status: "ok"
-    }));
-    app.get("/", (c) => {
-        return c.json({
-            message: "hello"
+        emitter,
+        serverCache,
+        apiRouters
+    };
+    const finalize = async () => {
+        // Allow plugins to set up before routes are registered
+        await setup?.(ctx);
+        // =========================================================================
+        // Register MCP routes (pre-configured and dynamic endpoints)
+        // =========================================================================
+        registerMcpRoutes({
+            app,
+            config,
+            serverCache
         });
-    });
-    // Server info - only available in dev mode or when discoveryConfig is enabled
-    if (isDevelopment || discoveryConfig) {
-        app.get("/info", (c) => {
-            // Dynamically get routes from Hono app
-            const routes = app.routes.map((r) => ({
-                method: r.method,
-                path: r.path
-            })).filter((r) => r.method !== "ALL" || r.path.startsWith("/mcp/")).sort((a, b) => a.path.localeCompare(b.path) || a.method.localeCompare(b.method));
-            // Group routes by category
-            const mcpRoutes = routes.filter((r) => r.path.startsWith("/mcp/")).map((r) => r.path);
-            const apiRoutes = routes.filter((r) => r.path.startsWith("/api/")).map((r) => `${r.method} ${r.path}`);
-            const chatRoutes = routes.filter((r) => r.path.startsWith("/v1/")).map((r) => `${r.method} ${r.path}`);
-            // Get unique MCP paths
-            const uniqueMcpPaths = [
-                ...new Set(mcpRoutes)
-            ];
-            return c.json({
-                name: "@wener/mcps",
-                version: "0.1.0",
-                servers: Object.keys(config.servers),
-                serverTypes: findMcpServerDef().map((d) => d.name),
-                models: config.models ? config.models.map((m) => m.name) : [],
-                endpoints: {
-                    mcp: uniqueMcpPaths,
-                    api: [
-                        ...new Set(apiRoutes)
-                    ],
-                    chat: [
-                        ...new Set(chatRoutes)
-                    ]
-                }
+        // =========================================================================
+        // Register Chat/LLM Gateway routes
+        // =========================================================================
+        registerChatRoutes({
+            app,
+            config
+        });
+        // =========================================================================
+        // Health and info endpoints
+        // =========================================================================
+        app.get("/health", (c) => c.json({
+            status: "ok"
+        }));
+        app.get("/", (c) => c.json({
+            message: "hello"
+        }));
+        if (isDevelopment || discoveryConfig) {
+            app.get("/info", (c) => {
+                const routes = app.routes.map((r) => ({
+                    method: r.method,
+                    path: r.path
+                })).filter((r) => r.method !== "ALL" || r.path.startsWith("/mcp/")).sort((a, b) => a.path.localeCompare(b.path) || a.method.localeCompare(b.method));
+                const mcpRoutes = routes.filter((r) => r.path.startsWith("/mcp/")).map((r) => r.path);
+                const apiRoutes = routes.filter((r) => r.path.startsWith("/api/")).map((r) => `${r.method} ${r.path}`);
+                const chatRoutes = routes.filter((r) => r.path.startsWith("/v1/")).map((r) => `${r.method} ${r.path}`);
+                const uniqueMcpPaths = [
+                    ...new Set(mcpRoutes)
+                ];
+                return c.json({
+                    name: "@wener/mcps",
+                    version: "0.1.0",
+                    servers: Object.keys(config.servers),
+                    serverTypes: findMcpServerDef().map((d) => d.name),
+                    models: config.models ? config.models.map((m) => m.name) : [],
+                    endpoints: {
+                        mcp: uniqueMcpPaths,
+                        api: [
+                            ...new Set(apiRoutes)
+                        ],
+                        chat: [
+                            ...new Set(chatRoutes)
+                        ]
+                    }
+                });
             });
+        }
+        // =========================================================================
+        // Register oRPC API routes (with any plugin-provided routers)
+        // =========================================================================
+        registerApiRoutes({
+            app,
+            config,
+            apiRouters,
+            statsProvider: ctx.statsProvider
         });
-    }
-    // =========================================================================
-    // Register oRPC API routes
-    // =========================================================================
-    registerApiRoutes({
-        app,
-        config
-    });
-    /**
-     * Print available endpoints grouped by category
-     */ const printEndpoints = () => {
+    };
+    const printEndpoints = () => {
         const routes = app.routes;
         const mcpPaths = new Set();
         const apiPaths = new Set();
@@ -135,32 +134,88 @@ export function createServer(options = {}) {
             }
         }
         log.info("Available endpoints:");
-        if (mcpPaths.size > 0) {
+        if (mcpPaths.size > 0)
             log.info(`  MCP: ${[
                 ...mcpPaths
             ].sort().join(", ")}`);
-        }
-        if (chatPaths.size > 0) {
+        if (chatPaths.size > 0)
             log.info(`  Chat: ${[
                 ...chatPaths
             ].sort().join(", ")}`);
-        }
-        if (apiPaths.size > 0) {
+        if (apiPaths.size > 0)
             log.info(`  API: ${[
                 ...apiPaths
             ].sort().join(", ")}`);
-        }
-        if (otherPaths.size > 0) {
+        if (otherPaths.size > 0)
             log.info(`  Other: ${[
                 ...otherPaths
             ].sort().join(", ")}`);
-        }
     };
     return {
         app,
         config,
+        emitter,
         serverCache,
-        printEndpoints
+        printEndpoints,
+        finalize
+    };
+}
+function headersToRecord(headers) {
+    const record = {};
+    headers.forEach((value, key) => {
+        record[key] = value;
+    });
+    return record;
+}
+/**
+ * Middleware that emits request events via the emitter.
+ * Subscribers (like audit plugin) can listen and handle these events.
+ */ function requestEventMiddleware(emitter) {
+    return async (c, next) => {
+        const startTime = Date.now();
+        const path = c.req.path;
+        let serverName;
+        let serverType;
+        const mcpMatch = path.match(/^\/mcp\/([^/]+)/);
+        if (mcpMatch) {
+            serverName = mcpMatch[1];
+            if ([
+                "tencent-cls",
+                "sql",
+                "prometheus",
+                "relay"
+            ].includes(serverName)) {
+                serverType = serverName;
+            }
+            else {
+                serverType = "custom";
+            }
+        }
+        if (path.startsWith("/v1/")) {
+            serverType = "chat";
+        }
+        let error;
+        try {
+            await next();
+        }
+        catch (e) {
+            error = e instanceof Error ? e.message : String(e);
+            throw e;
+        }
+        finally {
+            const durationMs = Date.now() - startTime;
+            emitter.emit(McpsEventType.Request, {
+                timestamp: new Date().toISOString(),
+                method: c.req.method,
+                path,
+                serverName,
+                serverType,
+                status: c.res.status,
+                durationMs,
+                error,
+                requestHeaders: headersToRecord(c.req.raw.headers)
+            });
+        }
     };
 }
 //# sourceMappingURL=server.js.map

package/lib/server/server.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/server/server.ts"],"sourcesContent":["import consola from 'consola';\nimport { Hono } from 'hono';\nimport { logger } from 'hono/logger';\nimport { LRUCache } from 'lru-cache';\nimport { isDevelopment } from 'std-env';\nimport type { McpServerInstance } from '@wener/ai/mcp';\nimport { findMcpServerDef } from '../providers/findMcpServerDef';\nimport { registerApiRoutes } from './api-routes';\nimport { auditMiddleware, configureAudit } from './audit';\nimport { registerChatRoutes } from './chat-routes';\nimport { loadConfig, loadEnvFiles, substituteEnvVars } from './config';\nimport { registerMcpRoutes } from './mcp-routes';\n\nconst log = consola.withTag('mcps');\n\nexport interface CreateServerOptions {\n\tcwd?: string;\n\tport?: number;\n\t/** Enable server config discovery endpoints (default: false) */\n\tdiscoveryConfig?: boolean;\n}\n\nexport function createServer(options: CreateServerOptions = {}) {\n\tconst { cwd = process.cwd(), discoveryConfig: discoveryConfigOption } = options;\n\n\tconst app = new Hono();\n\n\t// Request logging\n\tapp.use(\n\t\tlogger((v) => {\n\t\t\tlog.debug(v);\n\t\t}),\n\t);\n\n\t// Audit middleware\n\tapp.use(auditMiddleware());\n\n\t// Load .env files first\n\tloadEnvFiles(cwd);\n\n\t// Load config (with env var substitution)\n\tconst config = substituteEnvVars(loadConfig(cwd));\n\t// discoveryConfig: CLI option overrides config file, defaults to false\n\tconst discoveryConfig = discoveryConfigOption ?? config.discoveryConfig ?? false;\n\n\t// Log available server types from registry\n\tconst serverDefs = findMcpServerDef();\n\tlog.info(`Available server types: ${serverDefs.map((d) => d.name).join(', ')}`);\n\tlog.info(`Loaded ${Object.keys(config.servers).length} servers from config (discoveryConfig: ${discoveryConfig})`);\n\n\t// Configure audit with lazy DB initialization\n\t// DB will only be initialized when the first audit event needs persistence\n\tconfigureAudit(config.audit, config.db);\n\tconst auditEnabled = config.audit?.enabled !== false;\n\tconst auditDbPath = config.audit?.db?.path ?? config.db?.path ?? '.mcps.db';\n\tlog.info(`Audit configured: enabled=${auditEnabled}, db=${auditDbPath} (lazy init)`);\n\n\t// Unified cache for all MCP servers (both pre-configured and dynamic)\n\t// Keyed by cache key from def.getCacheKey() or `config::${name}` for pre-configured\n\tconst serverCache = new LRUCache<string, McpServerInstance>({\n\t\tmax: 100,\n\t\tdispose: (value, key) => {\n\t\t\tlog.info(`Closing expired MCP server: ${key}`);\n\t\t\tvalue.close?.().catch((e: unknown) => log.error('Failed to close server', e));\n\t\t},\n\t});\n\n\t// =========================================================================\n\t// Register MCP routes (pre-configured and dynamic endpoints)\n\t// =========================================================================\n\tregisterMcpRoutes({ app, config, serverCache });\n\n\t// =========================================================================\n\t// Register Chat/LLM Gateway routes\n\t// =========================================================================\n\tregisterChatRoutes({ app, config });\n\n\t// =========================================================================\n\t// Health and info endpoints\n\t// =========================================================================\n\tapp.get('/health', (c) => c.json({ status: 'ok' }));\n\n\tapp.get('/', (c) => {\n\t\treturn c.json({ message: 'hello' });\n\t});\n\n\t// Server info - only available in dev mode or when discoveryConfig is enabled\n\tif (isDevelopment || discoveryConfig) {\n\t\tapp.get('/info', (c) => {\n\t\t\t// Dynamically get routes from Hono app\n\t\t\tconst routes = app.routes\n\t\t\t\t.map((r) => ({ method: r.method, path: r.path }))\n\t\t\t\t.filter((r) => r.method !== 'ALL' || r.path.startsWith('/mcp/'))\n\t\t\t\t.sort((a, b) => a.path.localeCompare(b.path) || a.method.localeCompare(b.method));\n\n\t\t\t// Group routes by category\n\t\t\tconst mcpRoutes = routes.filter((r) => r.path.startsWith('/mcp/')).map((r) => r.path);\n\t\t\tconst apiRoutes = routes.filter((r) => r.path.startsWith('/api/')).map((r) => `${r.method} ${r.path}`);\n\t\t\tconst chatRoutes = routes.filter((r) => r.path.startsWith('/v1/')).map((r) => `${r.method} ${r.path}`);\n\n\t\t\t// Get unique MCP paths\n\t\t\tconst uniqueMcpPaths = [...new Set(mcpRoutes)];\n\n\t\t\treturn c.json({\n\t\t\t\tname: '@wener/mcps',\n\t\t\t\tversion: '0.1.0',\n\t\t\t\tservers: Object.keys(config.servers),\n\t\t\t\tserverTypes: findMcpServerDef().map((d) => d.name),\n\t\t\t\tmodels: config.models ? config.models.map((m) => m.name) : [],\n\t\t\t\tendpoints: {\n\t\t\t\t\tmcp: uniqueMcpPaths,\n\t\t\t\t\tapi: [...new Set(apiRoutes)],\n\t\t\t\t\tchat: [...new Set(chatRoutes)],\n\t\t\t\t},\n\t\t\t});\n\t\t});\n\t}\n\n\t// =========================================================================\n\t// Register oRPC API routes\n\t// =========================================================================\n\tregisterApiRoutes({ app, config });\n\n\t/**\n\t * Print available endpoints grouped by category\n\t */\n\tconst printEndpoints = () => {\n\t\tconst routes = app.routes;\n\t\tconst mcpPaths = new Set<string>();\n\t\tconst apiPaths = new Set<string>();\n\t\tconst chatPaths = new Set<string>();\n\t\tconst otherPaths = new Set<string>();\n\n\t\tfor (const route of routes) {\n\t\t\tconst path = route.path;\n\t\t\tif (path.startsWith('/mcp/')) {\n\t\t\t\tmcpPaths.add(path);\n\t\t\t} else if (path.startsWith('/api/')) {\n\t\t\t\tapiPaths.add(`${route.method} ${path}`);\n\t\t\t} else if (path.startsWith('/v1/')) {\n\t\t\t\tchatPaths.add(`${route.method} ${path}`);\n\t\t\t} else if (route.method !== 'ALL') {\n\t\t\t\totherPaths.add(`${route.method} ${path}`);\n\t\t\t}\n\t\t}\n\n\t\tlog.info('Available endpoints:');\n\t\tif (mcpPaths.size > 0) {\n\t\t\tlog.info(`  MCP: ${[...mcpPaths].sort().join(', ')}`);\n\t\t}\n\t\tif (chatPaths.size > 0) {\n\t\t\tlog.info(`  Chat: ${[...chatPaths].sort().join(', ')}`);\n\t\t}\n\t\tif (apiPaths.size > 0) {\n\t\t\tlog.info(`  API: ${[...apiPaths].sort().join(', ')}`);\n\t\t}\n\t\tif (otherPaths.size > 0) {\n\t\t\tlog.info(`  Other: ${[...otherPaths].sort().join(', ')}`);\n\t\t}\n\t};\n\n\treturn { app, config, serverCache, printEndpoints };\n}\n"],"names":["consola","Hono","logger","LRUCache","isDevelopment","findMcpServerDef","registerApiRoutes","auditMiddleware","configureAudit","registerChatRoutes","loadConfig","loadEnvFiles","substituteEnvVars","registerMcpRoutes","log","withTag","createServer","options","cwd","process","discoveryConfig","discoveryConfigOption","app","use","v","debug","config","serverDefs","info","map","d","name","join","Object","keys","servers","length","audit","db","auditEnabled","enabled","auditDbPath","path","serverCache","max","dispose","value","key","close","catch","e","error","get","c","json","status","message","routes","r","method","filter","startsWith","sort","a","b","localeCompare","mcpRoutes","apiRoutes","chatRoutes","uniqueMcpPaths","Set","version","serverTypes","models","m","endpoints","mcp","api","chat","printEndpoints","mcpPaths","apiPaths","chatPaths","otherPaths","route","add","size"],"mappings":"AAAA,OAAOA,aAAa,UAAU;AAC9B,SAASC,IAAI,QAAQ,OAAO;AAC5B,SAASC,MAAM,QAAQ,cAAc;AACrC,SAASC,QAAQ,QAAQ,YAAY;AACrC,SAASC,aAAa,QAAQ,UAAU;AAExC,SAASC,gBAAgB,QAAQ,gCAAgC;AACjE,SAASC,iBAAiB,QAAQ,eAAe;AACjD,SAASC,eAAe,EAAEC,cAAc,QAAQ,UAAU;AAC1D,SAASC,kBAAkB,QAAQ,gBAAgB;AACnD,SAASC,UAAU,EAAEC,YAAY,EAAEC,iBAAiB,QAAQ,WAAW;AACvE,SAASC,iBAAiB,QAAQ,eAAe;AAEjD,MAAMC,MAAMd,QAAQe,OAAO,CAAC;AAS5B,OAAO,SAASC,aAAaC,UAA+B,CAAC,CAAC;IAC7D,MAAM,EAAEC,MAAMC,QAAQD,GAAG,EAAE,EAAEE,iBAAiBC,qBAAqB,EAAE,GAAGJ;IAExE,MAAMK,MAAM,IAAIrB;IAEhB,kBAAkB;IAClBqB,IAAIC,GAAG,CACNrB,OAAO,CAACsB;QACPV,IAAIW,KAAK,CAACD;IACX;IAGD,mBAAmB;IACnBF,IAAIC,GAAG,CAAChB;IAER,wBAAwB;IACxBI,aAAaO;IAEb,0CAA0C;IAC1C,MAAMQ,SAASd,kBAAkBF,WAAWQ;IAC5C,uEAAuE;IACvE,MAAME,kBAAkBC,yBAAyBK,OAAON,eAAe,IAAI;IAE3E,2CAA2C;IAC3C,MAAMO,aAAatB;IACnBS,IAAIc,IAAI,CAAC,CAAC,wBAAwB,EAAED,WAAWE,GAAG,CAAC,CAACC,IAAMA,EAAEC,IAAI,EAAEC,IAAI,CAAC,OAAO;IAC9ElB,IAAIc,IAAI,CAAC,CAAC,OAAO,EAAEK,OAAOC,IAAI,CAACR,OAAOS,OAAO,EAAEC,MAAM,CAAC,uCAAuC,EAAEhB,gBAAgB,CAAC,CAAC;IAEjH,8CAA8C;IAC9C,2EAA2E;IAC3EZ,eAAekB,OAAOW,KAAK,EAAEX,OAAOY,EAAE;IACtC,MAAMC,eAAeb,OAAOW,KAAK,EAAEG,YAAY;IAC/C,MAAMC,cAAcf,OAAOW,KAAK,EAAEC,IAAII,QAAQhB,OAAOY,EAAE,EAAEI,QAAQ;IACjE5B,IAAIc,IAAI,CAAC,CAAC,0BAA0B,EAAEW,aAAa,KAAK,EAAEE,YAAY,YAAY,CAAC;IAEnF,sEAAsE;IACtE,oFAAoF;IACpF,MAAME,cAAc,IAAIxC,SAAoC;QAC3DyC,KAAK;QACLC,SAAS,CAACC,OAAOC;YAChBjC,IAAIc,IAAI,CAAC,CAAC,4BAA4B,EAAEmB,KAAK;YAC7CD,MAAME,KAAK,KAAKC,MAAM,CAACC,IAAepC,IAAIqC,KAAK,CAAC,0BAA0BD;QAC3E;IACD;IAEA,4EAA4E;IAC5E,6DAA6D;IAC7D,4EAA4E;IAC5ErC,kBAAkB;QAAES;QAAKI;QAAQiB;IAAY;IAE7C,4EAA4E;IAC5E,mCAAmC;IACnC,4EAA4E;IAC5ElC,mBAAmB;QAAEa;QAAKI;IAAO;IAEjC,4EAA4E;IAC5E,4BAA4B;IAC5B,4EAA4E;IAC5EJ,IAAI8B,GAAG,CAAC,WAAW,CAACC,IAAMA,EAAEC,IAAI,CAAC;YAAEC,QAAQ;QAAK;IAEhDjC,IAAI8B,GAAG,CAAC,KAAK,CAACC;QACb,OAAOA,EAAEC,IAAI,CAAC;YAAEE,SAAS;QAAQ;IAClC;IAEA,8EAA8E;IAC9E,IAAIpD,iBAAiBgB,iBAAiB;QACrCE,IAAI8B,GAAG,CAAC,SAAS,CAACC;YACjB,uCAAuC;YACvC,MAAMI,SAASnC,IAAImC,MAAM,CACvB5B,GAAG,CAAC,CAAC6B,IAAO,CAAA;oBAAEC,QAAQD,EAAEC,MAAM;oBAAEjB,MAAMgB,EAAEhB,IAAI;gBAAC,CAAA,GAC7CkB,MAAM,CAAC,CAACF,IAAMA,EAAEC,MAAM,KAAK,SAASD,EAAEhB,IAAI,CAACmB,UAAU,CAAC,UACtDC,IAAI,CAAC,CAACC,GAAGC,IAAMD,EAAErB,IAAI,CAACuB,aAAa,CAACD,EAAEtB,IAAI,KAAKqB,EAAEJ,MAAM,CAACM,aAAa,CAACD,EAAEL,MAAM;YAEhF,2BAA2B;YAC3B,MAAMO,YAAYT,OAAOG,MAAM,CAAC,CAACF,IAAMA,EAAEhB,IAAI,CAACmB,UAAU,CAAC,UAAUhC,GAAG,CAAC,CAAC6B,IAAMA,EAAEhB,IAAI;YACpF,MAAMyB,YAAYV,OAAOG,MAAM,CAAC,CAACF,IAAMA,EAAEhB,IAAI,CAACmB,UAAU,CAAC,UAAUhC,GAAG,CAAC,CAAC6B,IAAM,GAAGA,EAAEC,MAAM,CAAC,CAAC,EAAED,EAAEhB,IAAI,EAAE;YACrG,MAAM0B,aAAaX,OAAOG,MAAM,CAAC,CAACF,IAAMA,EAAEhB,IAAI,CAACmB,UAAU,CAAC,SAAShC,GAAG,CAAC,CAAC6B,IAAM,GAAGA,EAAEC,MAAM,CAAC,CAAC,EAAED,EAAEhB,IAAI,EAAE;YAErG,uBAAuB;YACvB,MAAM2B,iBAAiB;mBAAI,IAAIC,IAAIJ;aAAW;YAE9C,OAAOb,EAAEC,IAAI,CAAC;gBACbvB,MAAM;gBACNwC,SAAS;gBACTpC,SAASF,OAAOC,IAAI,CAACR,OAAOS,OAAO;gBACnCqC,aAAanE,mBAAmBwB,GAAG,CAAC,CAACC,IAAMA,EAAEC,IAAI;gBACjD0C,QAAQ/C,OAAO+C,MAAM,GAAG/C,OAAO+C,MAAM,CAAC5C,GAAG,CAAC,CAAC6C,IAAMA,EAAE3C,IAAI,IAAI,EAAE;gBAC7D4C,WAAW;oBACVC,KAAKP;oBACLQ,KAAK;2BAAI,IAAIP,IAAIH;qBAAW;oBAC5BW,MAAM;2BAAI,IAAIR,IAAIF;qBAAY;gBAC/B;YACD;QACD;IACD;IAEA,4EAA4E;IAC5E,2BAA2B;IAC3B,4EAA4E;IAC5E9D,kBAAkB;QAAEgB;QAAKI;IAAO;IAEhC;;EAEC,GACD,MAAMqD,iBAAiB;QACtB,MAAMtB,SAASnC,IAAImC,MAAM;QACzB,MAAMuB,WAAW,IAAIV;QACrB,MAAMW,WAAW,IAAIX;QACrB,MAAMY,YAAY,IAAIZ;QACtB,MAAMa,aAAa,IAAIb;QAEvB,KAAK,MAAMc,SAAS3B,OAAQ;YAC3B,MAAMf,OAAO0C,MAAM1C,IAAI;YACvB,IAAIA,KAAKmB,UAAU,CAAC,UAAU;gBAC7BmB,SAASK,GAAG,CAAC3C;YACd,OAAO,IAAIA,KAAKmB,UAAU,CAAC,UAAU;gBACpCoB,SAASI,GAAG,CAAC,GAAGD,MAAMzB,MAAM,CAAC,CAAC,EAAEjB,MAAM;YACvC,OAAO,IAAIA,KAAKmB,UAAU,CAAC,SAAS;gBACnCqB,UAAUG,GAAG,CAAC,GAAGD,MAAMzB,MAAM,CAAC,CAAC,EAAEjB,MAAM;YACxC,OAAO,IAAI0C,MAAMzB,MAAM,KAAK,OAAO;gBAClCwB,WAAWE,GAAG,CAAC,GAAGD,MAAMzB,MAAM,CAAC,CAAC,EAAEjB,MAAM;YACzC;QACD;QAEA5B,IAAIc,IAAI,CAAC;QACT,IAAIoD,SAASM,IAAI,GAAG,GAAG;YACtBxE,IAAIc,IAAI,CAAC,CAAC,OAAO,EAAE;mBAAIoD;aAAS,CAAClB,IAAI,GAAG9B,IAAI,CAAC,OAAO;QACrD;QACA,IAAIkD,UAAUI,IAAI,GAAG,GAAG;YACvBxE,IAAIc,IAAI,CAAC,CAAC,QAAQ,EAAE;mBAAIsD;aAAU,CAACpB,IAAI,GAAG9B,IAAI,CAAC,OAAO;QACvD;QACA,IAAIiD,SAASK,IAAI,GAAG,GAAG;YACtBxE,IAAIc,IAAI,CAAC,CAAC,OAAO,EAAE;mBAAIqD;aAAS,CAACnB,IAAI,GAAG9B,IAAI,CAAC,OAAO;QACrD;QACA,IAAImD,WAAWG,IAAI,GAAG,GAAG;YACxBxE,IAAIc,IAAI,CAAC,CAAC,SAAS,EAAE;mBAAIuD;aAAW,CAACrB,IAAI,GAAG9B,IAAI,CAAC,OAAO;QACzD;IACD;IAEA,OAAO;QAAEV;QAAKI;QAAQiB;QAAaoC;IAAe;AACnD"}
+{"version":3,"sources":["../../src/server/server.ts"],"sourcesContent":["import consola from 'consola';\nimport type { Context, Next } from 'hono';\nimport { Hono } from 'hono';\nimport { logger } from 'hono/logger';\nimport { LRUCache } from 'lru-cache';\nimport { isDevelopment } from 'std-env';\nimport type { McpServerInstance } from '@wener/ai/mcp';\nimport { findMcpServerDef } from '../providers/findMcpServerDef';\nimport { registerApiRoutes } from './api-routes';\nimport { registerChatRoutes } from './chat-routes';\nimport { loadConfig, loadEnvFiles, substituteEnvVars } from './config';\nimport { createMcpsEmitter, McpsEventType, type McpsEmitter } from './events';\nimport { registerMcpRoutes } from './mcp-routes';\n\nconst log = consola.withTag('mcps');\n\nexport interface McpsServerContext {\n\tapp: Hono;\n\tconfig: import('./schema').McpsConfig;\n\temitter: McpsEmitter;\n\tserverCache: LRUCache<string, McpServerInstance>;\n\t/** Plugins can register additional oRPC routers here */\n\tapiRouters: Record<string, any>;\n\t/** Plugins can register stats providers here */\n\tstatsProvider?: StatsProvider;\n}\n\nexport interface StatsProvider {\n\tgetStats(options: { from?: string | null; to?: string | null }): {\n\t\ttotalRequests: number;\n\t\ttotalErrors: number;\n\t\tavgDurationMs: number;\n\t\tbyServer: Array<{ name: string; count: number }>;\n\t\tbyMethod: Array<{ method: string; count: number }>;\n\t};\n\tqueryEvents(options: { limit?: number }): { events: Array<{ path: string }>; total: number };\n}\n\nexport interface CreateServerOptions {\n\tcwd?: string;\n\tport?: number;\n\t/** Enable server config discovery endpoints (default: false) */\n\tdiscoveryConfig?: boolean;\n\t/**\n\t * Called after core server is created but before routes are registered.\n\t * Use this to set up optional plugins like audit via the emitter.\n\t */\n\tsetup?: (ctx: McpsServerContext) => void | Promise<void>;\n}\n\nexport function createServer(options: CreateServerOptions = {}) {\n\tconst { cwd = process.cwd(), discoveryConfig: discoveryConfigOption, setup } = options;\n\n\tconst app = new Hono();\n\tconst emitter = createMcpsEmitter();\n\tconst apiRouters: Record<string, any> = {};\n\n\t// Request logging\n\tapp.use(\n\t\tlogger((v) => {\n\t\t\tlog.debug(v);\n\t\t}),\n\t);\n\n\t// Request event middleware - emits events for subscribers (audit, monitoring, etc.)\n\tapp.use(requestEventMiddleware(emitter));\n\n\t// Load .env files first\n\tloadEnvFiles(cwd);\n\n\t// Load config (with env var substitution)\n\tconst config = substituteEnvVars(loadConfig(cwd));\n\tconst discoveryConfig = discoveryConfigOption ?? config.discoveryConfig ?? false;\n\n\t// Log available server types from registry\n\tconst serverDefs = findMcpServerDef();\n\tlog.info(`Available server types: ${serverDefs.map((d) => d.name).join(', ')}`);\n\tlog.info(`Loaded ${Object.keys(config.servers).length} servers from config (discoveryConfig: ${discoveryConfig})`);\n\n\t// Unified cache for all MCP servers\n\tconst serverCache = new LRUCache<string, McpServerInstance>({\n\t\tmax: 100,\n\t\tdispose: (value, key) => {\n\t\t\tlog.info(`Closing expired MCP server: ${key}`);\n\t\t\tvalue.close?.().catch((e: unknown) => log.error('Failed to close server', e));\n\t\t},\n\t});\n\n\tconst ctx: McpsServerContext = { app, config, emitter, serverCache, apiRouters };\n\n\tconst finalize = async () => {\n\t\t// Allow plugins to set up before routes are registered\n\t\tawait setup?.(ctx);\n\n\t\t// =========================================================================\n\t\t// Register MCP routes (pre-configured and dynamic endpoints)\n\t\t// =========================================================================\n\t\tregisterMcpRoutes({ app, config, serverCache });\n\n\t\t// =========================================================================\n\t\t// Register Chat/LLM Gateway routes\n\t\t// =========================================================================\n\t\tregisterChatRoutes({ app, config });\n\n\t\t// =========================================================================\n\t\t// Health and info endpoints\n\t\t// =========================================================================\n\t\tapp.get('/health', (c) => c.json({ status: 'ok' }));\n\t\tapp.get('/', (c) => c.json({ message: 'hello' }));\n\n\t\tif (isDevelopment || discoveryConfig) {\n\t\t\tapp.get('/info', (c) => {\n\t\t\t\tconst routes = app.routes\n\t\t\t\t\t.map((r) => ({ method: r.method, path: r.path }))\n\t\t\t\t\t.filter((r) => r.method !== 'ALL' || r.path.startsWith('/mcp/'))\n\t\t\t\t\t.sort((a, b) => a.path.localeCompare(b.path) || a.method.localeCompare(b.method));\n\n\t\t\t\tconst mcpRoutes = routes.filter((r) => r.path.startsWith('/mcp/')).map((r) => r.path);\n\t\t\t\tconst apiRoutes = routes.filter((r) => r.path.startsWith('/api/')).map((r) => `${r.method} ${r.path}`);\n\t\t\t\tconst chatRoutes = routes.filter((r) => r.path.startsWith('/v1/')).map((r) => `${r.method} ${r.path}`);\n\t\t\t\tconst uniqueMcpPaths = [...new Set(mcpRoutes)];\n\n\t\t\t\treturn c.json({\n\t\t\t\t\tname: '@wener/mcps',\n\t\t\t\t\tversion: '0.1.0',\n\t\t\t\t\tservers: Object.keys(config.servers),\n\t\t\t\t\tserverTypes: findMcpServerDef().map((d) => d.name),\n\t\t\t\t\tmodels: config.models ? config.models.map((m) => m.name) : [],\n\t\t\t\t\tendpoints: {\n\t\t\t\t\t\tmcp: uniqueMcpPaths,\n\t\t\t\t\t\tapi: [...new Set(apiRoutes)],\n\t\t\t\t\t\tchat: [...new Set(chatRoutes)],\n\t\t\t\t\t},\n\t\t\t\t});\n\t\t\t});\n\t\t}\n\n\t\t// =========================================================================\n\t\t// Register oRPC API routes (with any plugin-provided routers)\n\t\t// =========================================================================\n\t\tregisterApiRoutes({ app, config, apiRouters, statsProvider: ctx.statsProvider });\n\t};\n\n\tconst printEndpoints = () => {\n\t\tconst routes = app.routes;\n\t\tconst mcpPaths = new Set<string>();\n\t\tconst apiPaths = new Set<string>();\n\t\tconst chatPaths = new Set<string>();\n\t\tconst otherPaths = new Set<string>();\n\n\t\tfor (const route of routes) {\n\t\t\tconst path = route.path;\n\t\t\tif (path.startsWith('/mcp/')) {\n\t\t\t\tmcpPaths.add(path);\n\t\t\t} else if (path.startsWith('/api/')) {\n\t\t\t\tapiPaths.add(`${route.method} ${path}`);\n\t\t\t} else if (path.startsWith('/v1/')) {\n\t\t\t\tchatPaths.add(`${route.method} ${path}`);\n\t\t\t} else if (route.method !== 'ALL') {\n\t\t\t\totherPaths.add(`${route.method} ${path}`);\n\t\t\t}\n\t\t}\n\n\t\tlog.info('Available endpoints:');\n\t\tif (mcpPaths.size > 0) log.info(`  MCP: ${[...mcpPaths].sort().join(', ')}`);\n\t\tif (chatPaths.size > 0) log.info(`  Chat: ${[...chatPaths].sort().join(', ')}`);\n\t\tif (apiPaths.size > 0) log.info(`  API: ${[...apiPaths].sort().join(', ')}`);\n\t\tif (otherPaths.size > 0) log.info(`  Other: ${[...otherPaths].sort().join(', ')}`);\n\t};\n\n\treturn { app, config, emitter, serverCache, printEndpoints, finalize };\n}\n\nfunction headersToRecord(headers: Headers): Record<string, string> {\n\tconst record: Record<string, string> = {};\n\theaders.forEach((value, key) => {\n\t\trecord[key] = value;\n\t});\n\treturn record;\n}\n\n/**\n * Middleware that emits request events via the emitter.\n * Subscribers (like audit plugin) can listen and handle these events.\n */\nfunction requestEventMiddleware(emitter: McpsEmitter) {\n\treturn async (c: Context, next: Next) => {\n\t\tconst startTime = Date.now();\n\t\tconst path = c.req.path;\n\n\t\tlet serverName: string | undefined;\n\t\tlet serverType: string | undefined;\n\n\t\tconst mcpMatch = path.match(/^\\/mcp\\/([^/]+)/);\n\t\tif (mcpMatch) {\n\t\t\tserverName = mcpMatch[1];\n\t\t\tif (['tencent-cls', 'sql', 'prometheus', 'relay'].includes(serverName)) {\n\t\t\t\tserverType = serverName;\n\t\t\t} else {\n\t\t\t\tserverType = 'custom';\n\t\t\t}\n\t\t}\n\n\t\tif (path.startsWith('/v1/')) {\n\t\t\tserverType = 'chat';\n\t\t}\n\n\t\tlet error: string | undefined;\n\n\t\ttry {\n\t\t\tawait next();\n\t\t} catch (e) {\n\t\t\terror = e instanceof Error ? e.message : String(e);\n\t\t\tthrow e;\n\t\t} finally {\n\t\t\tconst durationMs = Date.now() - startTime;\n\n\t\t\temitter.emit(McpsEventType.Request, {\n\t\t\t\ttimestamp: new Date().toISOString(),\n\t\t\t\tmethod: c.req.method,\n\t\t\t\tpath,\n\t\t\t\tserverName,\n\t\t\t\tserverType,\n\t\t\t\tstatus: c.res.status,\n\t\t\t\tdurationMs,\n\t\t\t\terror,\n\t\t\t\trequestHeaders: headersToRecord(c.req.raw.headers),\n\t\t\t});\n\t\t}\n\t};\n}\n"],"names":["consola","Hono","logger","LRUCache","isDevelopment","findMcpServerDef","registerApiRoutes","registerChatRoutes","loadConfig","loadEnvFiles","substituteEnvVars","createMcpsEmitter","McpsEventType","registerMcpRoutes","log","withTag","createServer","options","cwd","process","discoveryConfig","discoveryConfigOption","setup","app","emitter","apiRouters","use","v","debug","requestEventMiddleware","config","serverDefs","info","map","d","name","join","Object","keys","servers","length","serverCache","max","dispose","value","key","close","catch","e","error","ctx","finalize","get","c","json","status","message","routes","r","method","path","filter","startsWith","sort","a","b","localeCompare","mcpRoutes","apiRoutes","chatRoutes","uniqueMcpPaths","Set","version","serverTypes","models","m","endpoints","mcp","api","chat","statsProvider","printEndpoints","mcpPaths","apiPaths","chatPaths","otherPaths","route","add","size","headersToRecord","headers","record","forEach","next","startTime","Date","now","req","serverName","serverType","mcpMatch","match","includes","Error","String","durationMs","emit","Request","timestamp","toISOString","res","requestHeaders","raw"],"mappings":"AAAA,OAAOA,aAAa,UAAU;AAE9B,SAASC,IAAI,QAAQ,OAAO;AAC5B,SAASC,MAAM,QAAQ,cAAc;AACrC,SAASC,QAAQ,QAAQ,YAAY;AACrC,SAASC,aAAa,QAAQ,UAAU;AAExC,SAASC,gBAAgB,QAAQ,gCAAgC;AACjE,SAASC,iBAAiB,QAAQ,eAAe;AACjD,SAASC,kBAAkB,QAAQ,gBAAgB;AACnD,SAASC,UAAU,EAAEC,YAAY,EAAEC,iBAAiB,QAAQ,WAAW;AACvE,SAASC,iBAAiB,EAAEC,aAAa,QAA0B,WAAW;AAC9E,SAASC,iBAAiB,QAAQ,eAAe;AAEjD,MAAMC,MAAMd,QAAQe,OAAO,CAAC;AAoC5B,OAAO,SAASC,aAAaC,UAA+B,CAAC,CAAC;IAC7D,MAAM,EAAEC,MAAMC,QAAQD,GAAG,EAAE,EAAEE,iBAAiBC,qBAAqB,EAAEC,KAAK,EAAE,GAAGL;IAE/E,MAAMM,MAAM,IAAItB;IAChB,MAAMuB,UAAUb;IAChB,MAAMc,aAAkC,CAAC;IAEzC,kBAAkB;IAClBF,IAAIG,GAAG,CACNxB,OAAO,CAACyB;QACPb,IAAIc,KAAK,CAACD;IACX;IAGD,oFAAoF;IACpFJ,IAAIG,GAAG,CAACG,uBAAuBL;IAE/B,wBAAwB;IACxBf,aAAaS;IAEb,0CAA0C;IAC1C,MAAMY,SAASpB,kBAAkBF,WAAWU;IAC5C,MAAME,kBAAkBC,yBAAyBS,OAAOV,eAAe,IAAI;IAE3E,2CAA2C;IAC3C,MAAMW,aAAa1B;IACnBS,IAAIkB,IAAI,CAAC,CAAC,wBAAwB,EAAED,WAAWE,GAAG,CAAC,CAACC,IAAMA,EAAEC,IAAI,EAAEC,IAAI,CAAC,OAAO;IAC9EtB,IAAIkB,IAAI,CAAC,CAAC,OAAO,EAAEK,OAAOC,IAAI,CAACR,OAAOS,OAAO,EAAEC,MAAM,CAAC,uCAAuC,EAAEpB,gBAAgB,CAAC,CAAC;IAEjH,oCAAoC;IACpC,MAAMqB,cAAc,IAAItC,SAAoC;QAC3DuC,KAAK;QACLC,SAAS,CAACC,OAAOC;YAChB/B,IAAIkB,IAAI,CAAC,CAAC,4BAA4B,EAAEa,KAAK;YAC7CD,MAAME,KAAK,KAAKC,MAAM,CAACC,IAAelC,IAAImC,KAAK,CAAC,0BAA0BD;QAC3E;IACD;IAEA,MAAME,MAAyB;QAAE3B;QAAKO;QAAQN;QAASiB;QAAahB;IAAW;IAE/E,MAAM0B,WAAW;QAChB,uDAAuD;QACvD,MAAM7B,QAAQ4B;QAEd,4EAA4E;QAC5E,6DAA6D;QAC7D,4EAA4E;QAC5ErC,kBAAkB;YAAEU;YAAKO;YAAQW;QAAY;QAE7C,4EAA4E;QAC5E,mCAAmC;QACnC,4EAA4E;QAC5ElC,mBAAmB;YAAEgB;YAAKO;QAAO;QAEjC,4EAA4E;QAC5E,4BAA4B;QAC5B,4EAA4E;QAC5EP,IAAI6B,GAAG,CAAC,WAAW,CAACC,IAAMA,EAAEC,IAAI,CAAC;gBAAEC,QAAQ;YAAK;QAChDhC,IAAI6B,GAAG,CAAC,KAAK,CAACC,IAAMA,EAAEC,IAAI,CAAC;gBAAEE,SAAS;YAAQ;QAE9C,IAAIpD,iBAAiBgB,iBAAiB;YACrCG,IAAI6B,GAAG,CAAC,SAAS,CAACC;gBACjB,MAAMI,SAASlC,IAAIkC,MAAM,CACvBxB,GAAG,CAAC,CAACyB,IAAO,CAAA;wBAAEC,QAAQD,EAAEC,MAAM;wBAAEC,MAAMF,EAAEE,IAAI;oBAAC,CAAA,GAC7CC,MAAM,CAAC,CAACH,IAAMA,EAAEC,MAAM,KAAK,SAASD,EAAEE,IAAI,CAACE,UAAU,CAAC,UACtDC,IAAI,CAAC,CAACC,GAAGC,IAAMD,EAAEJ,IAAI,CAACM,aAAa,CAACD,EAAEL,IAAI,KAAKI,EAAEL,MAAM,CAACO,aAAa,CAACD,EAAEN,MAAM;gBAEhF,MAAMQ,YAAYV,OAAOI,MAAM,CAAC,CAACH,IAAMA,EAAEE,IAAI,CAACE,UAAU,CAAC,UAAU7B,GAAG,CAAC,CAACyB,IAAMA,EAAEE,IAAI;gBACpF,MAAMQ,YAAYX,OAAOI,MAAM,CAAC,CAACH,IAAMA,EAAEE,IAAI,CAACE,UAAU,CAAC,UAAU7B,GAAG,CAAC,CAACyB,IAAM,GAAGA,EAAEC,MAAM,CAAC,CAAC,EAAED,EAAEE,IAAI,EAAE;gBACrG,MAAMS,aAAaZ,OAAOI,MAAM,CAAC,CAACH,IAAMA,EAAEE,IAAI,CAACE,UAAU,CAAC,SAAS7B,GAAG,CAAC,CAACyB,IAAM,GAAGA,EAAEC,MAAM,CAAC,CAAC,EAAED,EAAEE,IAAI,EAAE;gBACrG,MAAMU,iBAAiB;uBAAI,IAAIC,IAAIJ;iBAAW;gBAE9C,OAAOd,EAAEC,IAAI,CAAC;oBACbnB,MAAM;oBACNqC,SAAS;oBACTjC,SAASF,OAAOC,IAAI,CAACR,OAAOS,OAAO;oBACnCkC,aAAapE,mBAAmB4B,GAAG,CAAC,CAACC,IAAMA,EAAEC,IAAI;oBACjDuC,QAAQ5C,OAAO4C,MAAM,GAAG5C,OAAO4C,MAAM,CAACzC,GAAG,CAAC,CAAC0C,IAAMA,EAAExC,IAAI,IAAI,EAAE;oBAC7DyC,WAAW;wBACVC,KAAKP;wBACLQ,KAAK;+BAAI,IAAIP,IAAIH;yBAAW;wBAC5BW,MAAM;+BAAI,IAAIR,IAAIF;yBAAY;oBAC/B;gBACD;YACD;QACD;QAEA,4EAA4E;QAC5E,8DAA8D;QAC9D,4EAA4E;QAC5E/D,kBAAkB;YAAEiB;YAAKO;YAAQL;YAAYuD,eAAe9B,IAAI8B,aAAa;QAAC;IAC/E;IAEA,MAAMC,iBAAiB;QACtB,MAAMxB,SAASlC,IAAIkC,MAAM;QACzB,MAAMyB,WAAW,IAAIX;QACrB,MAAMY,WAAW,IAAIZ;QACrB,MAAMa,YAAY,IAAIb;QACtB,MAAMc,aAAa,IAAId;QAEvB,KAAK,MAAMe,SAAS7B,OAAQ;YAC3B,MAAMG,OAAO0B,MAAM1B,IAAI;YACvB,IAAIA,KAAKE,UAAU,CAAC,UAAU;gBAC7BoB,SAASK,GAAG,CAAC3B;YACd,OAAO,IAAIA,KAAKE,UAAU,CAAC,UAAU;gBACpCqB,SAASI,GAAG,CAAC,GAAGD,MAAM3B,MAAM,CAAC,CAAC,EAAEC,MAAM;YACvC,OAAO,IAAIA,KAAKE,UAAU,CAAC,SAAS;gBACnCsB,UAAUG,GAAG,CAAC,GAAGD,MAAM3B,MAAM,CAAC,CAAC,EAAEC,MAAM;YACxC,OAAO,IAAI0B,MAAM3B,MAAM,KAAK,OAAO;gBAClC0B,WAAWE,GAAG,CAAC,GAAGD,MAAM3B,MAAM,CAAC,CAAC,EAAEC,MAAM;YACzC;QACD;QAEA9C,IAAIkB,IAAI,CAAC;QACT,IAAIkD,SAASM,IAAI,GAAG,GAAG1E,IAAIkB,IAAI,CAAC,CAAC,OAAO,EAAE;eAAIkD;SAAS,CAACnB,IAAI,GAAG3B,IAAI,CAAC,OAAO;QAC3E,IAAIgD,UAAUI,IAAI,GAAG,GAAG1E,IAAIkB,IAAI,CAAC,CAAC,QAAQ,EAAE;eAAIoD;SAAU,CAACrB,IAAI,GAAG3B,IAAI,CAAC,OAAO;QAC9E,IAAI+C,SAASK,IAAI,GAAG,GAAG1E,IAAIkB,IAAI,CAAC,CAAC,OAAO,EAAE;eAAImD;SAAS,CAACpB,IAAI,GAAG3B,IAAI,CAAC,OAAO;QAC3E,IAAIiD,WAAWG,IAAI,GAAG,GAAG1E,IAAIkB,IAAI,CAAC,CAAC,SAAS,EAAE;eAAIqD;SAAW,CAACtB,IAAI,GAAG3B,IAAI,CAAC,OAAO;IAClF;IAEA,OAAO;QAAEb;QAAKO;QAAQN;QAASiB;QAAawC;QAAgB9B;IAAS;AACtE;AAEA,SAASsC,gBAAgBC,OAAgB;IACxC,MAAMC,SAAiC,CAAC;IACxCD,QAAQE,OAAO,CAAC,CAAChD,OAAOC;QACvB8C,MAAM,CAAC9C,IAAI,GAAGD;IACf;IACA,OAAO+C;AACR;AAEA;;;CAGC,GACD,SAAS9D,uBAAuBL,OAAoB;IACnD,OAAO,OAAO6B,GAAYwC;QACzB,MAAMC,YAAYC,KAAKC,GAAG;QAC1B,MAAMpC,OAAOP,EAAE4C,GAAG,CAACrC,IAAI;QAEvB,IAAIsC;QACJ,IAAIC;QAEJ,MAAMC,WAAWxC,KAAKyC,KAAK,CAAC;QAC5B,IAAID,UAAU;YACbF,aAAaE,QAAQ,CAAC,EAAE;YACxB,IAAI;gBAAC;gBAAe;gBAAO;gBAAc;aAAQ,CAACE,QAAQ,CAACJ,aAAa;gBACvEC,aAAaD;YACd,OAAO;gBACNC,aAAa;YACd;QACD;QAEA,IAAIvC,KAAKE,UAAU,CAAC,SAAS;YAC5BqC,aAAa;QACd;QAEA,IAAIlD;QAEJ,IAAI;YACH,MAAM4C;QACP,EAAE,OAAO7C,GAAG;YACXC,QAAQD,aAAauD,QAAQvD,EAAEQ,OAAO,GAAGgD,OAAOxD;YAChD,MAAMA;QACP,SAAU;YACT,MAAMyD,aAAaV,KAAKC,GAAG,KAAKF;YAEhCtE,QAAQkF,IAAI,CAAC9F,cAAc+F,OAAO,EAAE;gBACnCC,WAAW,IAAIb,OAAOc,WAAW;gBACjClD,QAAQN,EAAE4C,GAAG,CAACtC,MAAM;gBACpBC;gBACAsC;gBACAC;gBACA5C,QAAQF,EAAEyD,GAAG,CAACvD,MAAM;gBACpBkD;gBACAxD;gBACA8D,gBAAgBtB,gBAAgBpC,EAAE4C,GAAG,CAACe,GAAG,CAACtB,OAAO;YAClD;QACD;IACD;AACD"}

package/package.json

@@ -1,6 +1,7 @@
 {
   "name": "@wener/mcps",
-  "version": "1.0.1",
+  "version": "1.0.4",
+  "type": "module",
   "description": "",
   "author": "",
   "bin": {
@@ -18,6 +19,30 @@
     "./server": {
       "types": "./src/server/index.ts",
       "default": "./lib/server/index.js"
+    },
+    "./entities": {
+      "types": "./src/entities/index.ts",
+      "default": "./lib/entities/index.js"
+    },
+    "./contracts": {
+      "types": "./src/contracts/index.ts",
+      "default": "./lib/contracts/index.js"
+    },
+    "./chat": {
+      "types": "./src/chat/index.ts",
+      "default": "./lib/chat/index.js"
+    },
+    "./providers": {
+      "types": "./src/providers/McpServerHandlerDef.ts",
+      "default": "./lib/providers/McpServerHandlerDef.js"
+    },
+    "./cli": {
+      "types": "./src/cli.ts",
+      "default": "./lib/cli.js"
+    },
+    "./audit": {
+      "types": "./src/server/audit-plugin.ts",
+      "default": "./lib/server/audit-plugin.js"
     }
   },
   "files": [
@@ -27,17 +52,21 @@
     "README.md"
   ],
   "keywords": [],
+  "dependencies": {
+    "emittery": "^1.2.0"
+  },
   "devDependencies": {
     "@ai-sdk/mcp": "^1.0.18",
     "@ai-sdk/openai-compatible": "^2.0.26",
     "@ai-sdk/react": "^3.0.71",
     "@base-ui/react": "^1.1.0",
     "@hono/mcp": "^0.2.3",
+    "@larksuiteoapi/node-sdk": "^1.59.0",
     "@hono/node-server": "^1.19.9",
     "@hono/vite-dev-server": "^0.24.1",
-    "@mikro-orm/core": "^7.0.0-rc.0",
-    "@mikro-orm/decorators": "^7.0.0-rc.0",
-    "@mikro-orm/sqlite": "^7.0.0-rc.0",
+    "@mikro-orm/core": "^7.0.0-rc.2",
+    "@mikro-orm/decorators": "^7.0.0-rc.2",
+    "@mikro-orm/sql": "^7.0.0-rc.2",
     "@modelcontextprotocol/sdk": "^1.25.3",
     "@orpc/client": "^1.13.4",
     "@orpc/contract": "^1.13.4",
@@ -71,7 +100,8 @@
     "tedious": "^19.2.0",
     "yaml": "^2.8.2",
     "zod": "^4.3.6",
-    "@wener/ai": "0.1.0"
+    "@wener/ai": "0.1.2",
+    "@wener/server": "2.0.5"
   },
   "scripts": {
     "build": "bun run ./src/scripts/bundle.ts",

package/src/chat/handler.ts

@@ -747,7 +747,7 @@ export function createChatHandler(options: ChatHandlerOptions = {}) {
 			let previousContext: { input: unknown; output: unknown[] } | null = null;
 			if (request.previous_response_id) {
 				try {
-					const { isDbInitialized, getEntityManager } = await import('../server/db');
+					const { isDbInitialized, getEntityManager } = await import('../server/audit-db');
 					const { ResponseEntity } = await import('../entities');
 					if (isDbInitialized()) {
 						const em = getEntityManager().fork();
@@ -840,7 +840,7 @@ export function createChatHandler(options: ChatHandlerOptions = {}) {
 
 			// Store response for future previous_response_id lookups
 			try {
-				const { isDbInitialized, getEntityManager } = await import('../server/db');
+				const { isDbInitialized, getEntityManager } = await import('../server/audit-db');
 				const { ResponseEntity } = await import('../entities');
 				if (isDbInitialized()) {
 					const em = getEntityManager().fork();

package/src/cli-start.ts

@@ -0,0 +1,43 @@
+import { serve } from '@hono/node-server';
+import consola from 'consola';
+import { createServer, type CreateServerOptions } from './server/server';
+
+const log = consola.withTag('mcps');
+
+let serverHandle: ReturnType<typeof serve> | undefined;
+
+function shutdown() {
+	log.info('Shutting down...');
+	if (serverHandle) {
+		serverHandle.close(() => process.exit(0));
+		setTimeout(() => process.exit(1), 3000);
+	} else {
+		process.exit(0);
+	}
+}
+
+process.on('SIGINT', shutdown);
+process.on('SIGTERM', shutdown);
+
+export async function startServer(options: { port: string; cwd: string; discoveryConfig: boolean; setup?: CreateServerOptions['setup'] }) {
+	const port = Number.parseInt(options.port, 10);
+	const { app, printEndpoints, finalize } = createServer({
+		cwd: options.cwd,
+		port,
+		discoveryConfig: options.discoveryConfig,
+		setup: options.setup,
+	});
+
+	await finalize();
+
+	log.info(`Starting MCPS server on port ${port}`);
+
+	serverHandle = serve({
+		fetch: app.fetch,
+		port,
+		hostname: '0.0.0.0',
+	});
+
+	log.success(`MCPS server running at http://localhost:${port}`);
+	printEndpoints();
+}

package/src/cli.ts

@@ -0,0 +1,45 @@
+import { Command } from 'commander';
+import type { CreateServerOptions } from './server/server';
+
+export interface CreateProgramOptions {
+	name?: string;
+	description?: string;
+	version?: string;
+	defaultPort?: string;
+	/** Called before the server starts, allows registering additional providers via side-effect imports */
+	beforeStart?: () => void | Promise<void>;
+	/** Server setup callback, passed to createServer. Use to register plugins like audit. */
+	setup?: CreateServerOptions['setup'];
+}
+
+/**
+ * Create a reusable Commander program for MCPS server.
+ * Consumers can register providers before calling `program.parseAsync()`.
+ */
+export function createProgram(options: CreateProgramOptions = {}): Command {
+	const {
+		name = 'mcps',
+		description = 'MCP Proxy Server - Unified MCP service with relay, SQL, CLS, and Prometheus support',
+		version = '0.1.0',
+		defaultPort = '8036',
+		beforeStart,
+		setup,
+	} = options;
+
+	const program = new Command();
+
+	program
+		.name(name)
+		.description(description)
+		.version(version)
+		.option('-p, --port <port>', 'Port to listen on', defaultPort)
+		.option('-c, --cwd <path>', 'Working directory for config files', process.cwd())
+		.option('--discovery-config', 'Enable server config discovery endpoints', false)
+		.action(async (opts) => {
+			await beforeStart?.();
+			const { startServer } = await import('./cli-start.js');
+			await startServer({ ...opts, setup });
+		});
+
+	return program;
+}

package/src/dev.server.ts

@@ -1,6 +1,13 @@
+import { setupAudit } from '#/server/audit-plugin';
 import { createServer } from '#/server/server';
 
-const { app } = createServer({});
+const { app, finalize } = createServer({
+	setup: (ctx) => {
+		setupAudit(ctx);
+	},
+});
+
+await finalize();
 
 export default {
 	fetch: app.fetch,

package/src/index.ts

@@ -1 +1,47 @@
-export {};
+// Core server
+export { createServer, type CreateServerOptions, type McpsServerContext, type StatsProvider } from './server/server';
+
+// Events (emittery-based decoupling)
+export { McpsEventType, createMcpsEmitter, type McpsEmitter, type McpsEventData, type McpsRequestEvent } from './server/events';
+
+// Configuration
+export { loadConfig, substituteEnvVars, loadEnvFiles } from './server/config';
+export {
+	type McpsConfig,
+	type ServerConfig,
+	type ModelConfig,
+	type AuditConfig,
+	type DbConfig,
+	McpsConfigSchema,
+	ServerConfigSchema,
+	ModelConfigSchema,
+	AuditConfigSchema,
+	DbConfigSchema,
+	HeaderNames,
+} from './server/schema';
+
+// MCP routes and handler
+export { registerMcpRoutes } from './server/mcp-routes';
+export { registerChatRoutes } from './server/chat-routes';
+export { registerApiRoutes } from './server/api-routes';
+export { createMcpLoggingHandler } from './server/mcp-handler';
+export { createMcpsRouter } from './server/mcps-router';
+
+// Providers
+export {
+	type McpServerHandlerDef,
+	type McpServerHandlerDef as McpServerDef,
+	type HeaderMapping,
+	type DefineMcpServerHandlerOptions,
+	defineMcpServerHandler,
+	registerMcpServerHandler,
+	getAllMcpServerHandlerDefs,
+	getMcpServerHandlerDef,
+} from './providers/McpServerHandlerDef';
+export { findMcpServerDef, resolveMcpServerDef, getMcpServerDefCount } from './providers/findMcpServerDef';
+
+// Contracts
+export * from './contracts';
+
+// Chat
+export { createChatHandler, type ChatHandlerOptions } from './chat';