@connorbritain/mssql-mcp-core 0.1.0

package/dist/routing/IntentRouter.js

@@ -0,0 +1,243 @@
+import { getEnvironmentManager } from "../config/EnvironmentManager.js";
+export class IntentRouter {
+    constructor(options) {
+        this.tools = options.tools;
+        this.allowMutations = options.allowMutations;
+        this.requireConfirmationForMutations = options.requireConfirmationForMutations;
+    }
+    async route(params) {
+        const prompt = typeof params.prompt === "string" ? params.prompt.trim() : "";
+        if (!prompt) {
+            return {
+                success: false,
+                message: "Prompt is required to route intent.",
+                error: "MISSING_PROMPT",
+            };
+        }
+        const toolArguments = this.normalizeArguments(params.toolArguments ?? {});
+        const confirmIntent = Boolean(params.confirmIntent);
+        const preferredToolName = params.preferredToolName;
+        const normalizedPrompt = prompt.toLowerCase();
+        // Infer environment from prompt if not explicitly provided
+        const environment = params.environment || this.inferEnvironment(normalizedPrompt);
+        if (environment) {
+            toolArguments.environment = environment;
+        }
+        const inferredIntent = this.inferIntent(normalizedPrompt, toolArguments);
+        const eligibleTools = this.tools.filter((tool) => this.isToolEligible(tool));
+        const candidates = eligibleTools
+            .map((tool) => this.scoreTool(tool, normalizedPrompt, toolArguments, inferredIntent, preferredToolName))
+            .filter((candidate) => candidate.score > Number.NEGATIVE_INFINITY)
+            .sort((a, b) => b.score - a.score);
+        const bestCandidate = candidates[0];
+        if (!bestCandidate || bestCandidate.score <= 0) {
+            return {
+                success: false,
+                message: "Unable to determine an appropriate SQL tool for the provided prompt. Try specifying the desired action more concretely (e.g., 'list tables', 'describe table X', 'run SELECT ...').",
+                error: "NO_TOOL_MATCH",
+            };
+        }
+        const missingArgs = this.getMissingArguments(bestCandidate.config, toolArguments);
+        if (missingArgs.length) {
+            return {
+                success: false,
+                routedTool: bestCandidate.config.name,
+                message: `Selected tool '${bestCandidate.config.name}' requires argument(s): ${missingArgs.join(", ")}. Provide them in the request arguments.`,
+                error: "MISSING_ARGUMENTS",
+            };
+        }
+        const requiresConfirmation = (bestCandidate.config.requiresConfirmation ||
+            bestCandidate.config.mutatesData ||
+            bestCandidate.config.schemaChange) &&
+            this.requireConfirmationForMutations;
+        if (requiresConfirmation && !confirmIntent) {
+            return {
+                success: false,
+                routedTool: bestCandidate.config.name,
+                message: "This operation modifies data or schema. Re-run with confirmIntent: true to proceed.",
+                error: "CONFIRMATION_REQUIRED",
+            };
+        }
+        try {
+            const result = await bestCandidate.config.tool.run(toolArguments);
+            return {
+                success: true,
+                routedTool: bestCandidate.config.name,
+                intent: inferredIntent,
+                reasoning: bestCandidate.reasons,
+                toolResult: result,
+                selectedEnvironment: environment,
+            };
+        }
+        catch (error) {
+            return {
+                success: false,
+                routedTool: bestCandidate.config.name,
+                message: `Routed tool '${bestCandidate.config.name}' failed: ${error}`,
+                error: "ROUTED_TOOL_FAILED",
+                selectedEnvironment: environment,
+            };
+        }
+    }
+    inferEnvironment(prompt) {
+        const envManager = getEnvironmentManager();
+        const environments = envManager.listEnvironments();
+        for (const env of environments) {
+            const patterns = [
+                new RegExp(`\\b${env.name}\\b`, "i"),
+                new RegExp(`\\b${env.name.replace(/-/g, "\\s")}\\b`, "i"),
+            ];
+            for (const pattern of patterns) {
+                if (pattern.test(prompt)) {
+                    return env.name;
+                }
+            }
+        }
+        const envKeywords = {
+            prod: ["production", "prod", "live"],
+            staging: ["staging", "stage", "uat"],
+            dev: ["development", "dev", "local"],
+        };
+        for (const [envSuffix, keywords] of Object.entries(envKeywords)) {
+            for (const keyword of keywords) {
+                if (prompt.includes(keyword)) {
+                    const matchingEnv = environments.find((e) => e.name.toLowerCase().includes(envSuffix));
+                    if (matchingEnv) {
+                        return matchingEnv.name;
+                    }
+                }
+            }
+        }
+        return undefined;
+    }
+    normalizeArguments(args) {
+        if (!args || typeof args !== "object") {
+            return {};
+        }
+        const cloned = {};
+        for (const [key, value] of Object.entries(args)) {
+            if (value !== undefined) {
+                cloned[key] = value;
+            }
+        }
+        return cloned;
+    }
+    isToolEligible(tool) {
+        if (this.allowMutations) {
+            return true;
+        }
+        return !tool.mutatesData && !tool.schemaChange;
+    }
+    inferIntent(prompt, toolArguments) {
+        const detectors = [
+            {
+                intent: "schema_change",
+                keywords: ["create table", "drop table", "create index", "drop index", "alter", "ddl"],
+            },
+            {
+                intent: "data_write",
+                keywords: ["update", "insert", "delete", "fix", "modify", "change", "correct"],
+            },
+            {
+                intent: "metadata",
+                keywords: ["profile", "sample", "statistics", "distribution", "quality"],
+            },
+            {
+                intent: "schema_discovery",
+                keywords: ["describe", "columns", "list tables", "show tables", "schema", "search"],
+            },
+            {
+                intent: "data_read",
+                keywords: ["select", "query", "fetch", "count", "report", "view"],
+            },
+        ];
+        for (const detector of detectors) {
+            if (detector.keywords.some((keyword) => prompt.includes(keyword))) {
+                return detector.intent;
+            }
+        }
+        const sqlSnippet = this.extractSqlSnippet(toolArguments);
+        if (sqlSnippet) {
+            if (sqlSnippet.startsWith("select")) {
+                return "data_read";
+            }
+            if (sqlSnippet.startsWith("update") || sqlSnippet.startsWith("delete")) {
+                return "data_write";
+            }
+            if (sqlSnippet.startsWith("insert")) {
+                return "data_write";
+            }
+            if (sqlSnippet.startsWith("create") || sqlSnippet.startsWith("drop")) {
+                return "schema_change";
+            }
+        }
+        if (toolArguments?.tablePattern || toolArguments?.columnPattern) {
+            return "schema_discovery";
+        }
+        return "data_read";
+    }
+    extractSqlSnippet(args) {
+        const sqlCandidate = typeof args?.query === "string" ? args.query : typeof args?.sql === "string" ? args.sql : null;
+        return sqlCandidate?.trim().toLowerCase() ?? null;
+    }
+    scoreTool(config, prompt, toolArguments, inferredIntent, preferredToolName) {
+        let score = config.baseScore ?? 0.5;
+        const reasons = [];
+        if (config.intents.includes(inferredIntent)) {
+            score += 5;
+            reasons.push(`intent match (${inferredIntent})`);
+        }
+        if (preferredToolName && config.name === preferredToolName) {
+            score += 3;
+            reasons.push("preferred tool match");
+        }
+        if (config.keywords?.length) {
+            for (const keyword of config.keywords) {
+                if (prompt.includes(keyword)) {
+                    score += 2;
+                    reasons.push(`keyword '${keyword}'`);
+                }
+            }
+        }
+        if (config.requiredArgs?.length) {
+            for (const arg of config.requiredArgs) {
+                if (this.hasArgument(toolArguments, arg)) {
+                    score += 1;
+                }
+                else {
+                    score -= 1;
+                }
+            }
+        }
+        if ((config.mutatesData || config.schemaChange) && !this.allowMutations) {
+            score = Number.NEGATIVE_INFINITY;
+        }
+        return { config, score, reasons };
+    }
+    hasArgument(args, key) {
+        if (!args) {
+            return false;
+        }
+        const value = args[key];
+        if (value === null || value === undefined) {
+            return false;
+        }
+        if (typeof value === "string") {
+            return value.trim().length > 0;
+        }
+        if (Array.isArray(value)) {
+            return value.length > 0;
+        }
+        if (typeof value === "object") {
+            return Object.keys(value).length > 0;
+        }
+        return true;
+    }
+    getMissingArguments(config, args) {
+        if (!config.requiredArgs || config.requiredArgs.length === 0) {
+            return [];
+        }
+        return config.requiredArgs.filter((arg) => !this.hasArgument(args, arg));
+    }
+}
+//# sourceMappingURL=IntentRouter.js.map

package/dist/routing/IntentRouter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"IntentRouter.js","sourceRoot":"","sources":["../../src/routing/IntentRouter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AAUxE,MAAM,OAAO,YAAY;IAKvB,YAAY,OAA4B;QACtC,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC3B,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;QAC7C,IAAI,CAAC,+BAA+B,GAAG,OAAO,CAAC,+BAA+B,CAAC;IACjF,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,MAAM,MAAM,GAAG,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC7E,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,qCAAqC;gBAC9C,KAAK,EAAE,gBAAgB;aACxB,CAAC;QACJ,CAAC;QAED,MAAM,aAAa,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC;QAC1E,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QACpD,MAAM,iBAAiB,GAAG,MAAM,CAAC,iBAAiB,CAAC;QACnD,MAAM,gBAAgB,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC;QAE9C,2DAA2D;QAC3D,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,IAAI,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,CAAC;QAClF,IAAI,WAAW,EAAE,CAAC;YAChB,aAAa,CAAC,WAAW,GAAG,WAAW,CAAC;QAC1C,CAAC;QAED,MAAM,cAAc,GAAG,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE,aAAa,CAAC,CAAC;QACzE,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC;QAE7E,MAAM,UAAU,GAAG,aAAa;aAC7B,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACZ,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,gBAAgB,EAAE,aAAa,EAAE,cAAc,EAAE,iBAAiB,CAAC,CACzF;aACA,MAAM,CAAC,CAAC,SAAS,EAAiC,EAAE,CAAC,SAAS,CAAC,KAAK,GAAG,MAAM,CAAC,iBAAiB,CAAC;aAChG,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QAErC,MAAM,aAAa,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;QACpC,IAAI,CAAC,aAAa,IAAI,aAAa,CAAC,KAAK,IAAI,CAAC,EAAE,CAAC;YAC/C,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,OAAO,EACL,qLAAqL;gBACvL,KAAK,EAAE,eAAe;aACvB,CAAC;QACJ,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,mBAAmB,CAAC,aAAa,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QAClF,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;YACvB,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,UAAU,EAAE,aAAa,CAAC,MAAM,CAAC,IAAI;gBACrC,OAAO,EAAE,kBAAkB,aAAa,CAAC,MAAM,CAAC,IAAI,2BAA2B,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,0CAA0C;gBAC/I,KAAK,EAAE,mBAAmB;aAC3B,CAAC;QACJ,CAAC;QAED,MAAM,oBAAoB,GACxB,CAAC,aAAa,CAAC,MAAM,CAAC,oBAAoB;YACxC,aAAa,CAAC,MAAM,CAAC,WAAW;YAChC,aAAa,CAAC,MAAM,CAAC,YAAY,CAAC;YACpC,IAAI,CAAC,+BAA+B,CAAC;QAEvC,IAAI,oBAAoB,IAAI,CAAC,aAAa,EAAE,CAAC;YAC3C,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,UAAU,EAAE,aAAa,CAAC,MAAM,CAAC,IAAI;gBACrC,OAAO,EACL,qFAAqF;gBACvF,KAAK,EAAE,uBAAuB;aAC/B,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;YAClE,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,UAAU,EAAE,aAAa,CAAC,MAAM,CAAC,IAAI;gBACrC,MAAM,EAAE,cAAc;gBACtB,SAAS,EAAE,aAAa,CAAC,OAAO;gBAChC,UAAU,EAAE,MAAM;gBAClB,mBAAmB,EAAE,WAAW;aACjC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,UAAU,EAAE,aAAa,CAAC,MAAM,CAAC,IAAI;gBACrC,OAAO,EAAE,gBAAgB,aAAa,CAAC,MAAM,CAAC,IAAI,aAAa,KAAK,EAAE;gBACtE,KAAK,EAAE,oBAAoB;gBAC3B,mBAAmB,EAAE,WAAW;aACjC,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,MAAc;QACrC,MAAM,UAAU,GAAG,qBAAqB,EAAE,CAAC;QAC3C,MAAM,YAAY,GAAG,UAAU,CAAC,gBAAgB,EAAE,CAAC;QAEnD,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAC/B,MAAM,QAAQ,GAAG;gBACf,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,KAAK,EAAE,GAAG,CAAC;gBACpC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC;aAC1D,CAAC;YAEF,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;oBACzB,OAAO,GAAG,CAAC,IAAI,CAAC;gBAClB,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,WAAW,GAA6B;YAC5C,IAAI,EAAE,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,CAAC;YACpC,OAAO,EAAE,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC;YACpC,GAAG,EAAE,CAAC,aAAa,EAAE,KAAK,EAAE,OAAO,CAAC;SACrC,CAAC;QAEF,KAAK,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;YAChE,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC7B,MAAM,WAAW,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAC1C,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,CACzC,CAAC;oBACF,IAAI,WAAW,EAAE,CAAC;wBAChB,OAAO,WAAW,CAAC,IAAI,CAAC;oBAC1B,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,kBAAkB,CAAC,IAAS;QAClC,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YACtC,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,MAAM,GAAwB,EAAE,CAAC;QACvC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YAChD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YACtB,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,cAAc,CAAC,IAAuB;QAC5C,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC;IACjD,CAAC;IAEO,WAAW,CAAC,MAAc,EAAE,aAAkC;QACpE,MAAM,SAAS,GAA0D;YACvE;gBACE,MAAM,EAAE,eAAe;gBACvB,QAAQ,EAAE,CAAC,cAAc,EAAE,YAAY,EAAE,cAAc,EAAE,YAAY,EAAE,OAAO,EAAE,KAAK,CAAC;aACvF;YACD;gBACE,MAAM,EAAE,YAAY;gBACpB,QAAQ,EAAE,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,CAAC;aAC/E;YACD;gBACE,MAAM,EAAE,UAAU;gBAClB,QAAQ,EAAE,CAAC,SAAS,EAAE,QAAQ,EAAE,YAAY,EAAE,cAAc,EAAE,SAAS,CAAC;aACzE;YACD;gBACE,MAAM,EAAE,kBAAkB;gBAC1B,QAAQ,EAAE,CAAC,UAAU,EAAE,SAAS,EAAE,aAAa,EAAE,aAAa,EAAE,QAAQ,EAAE,QAAQ,CAAC;aACpF;YACD;gBACE,MAAM,EAAE,WAAW;gBACnB,QAAQ,EAAE,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC;aAClE;SACF,CAAC;QAEF,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,IAAI,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;gBAClE,OAAO,QAAQ,CAAC,MAAM,CAAC;YACzB,CAAC;QACH,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;QACzD,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,UAAU,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACpC,OAAO,WAAW,CAAC;YACrB,CAAC;YACD,IAAI,UAAU,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,UAAU,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACvE,OAAO,YAAY,CAAC;YACtB,CAAC;YACD,IAAI,UAAU,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACpC,OAAO,YAAY,CAAC;YACtB,CAAC;YACD,IAAI,UAAU,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;gBACrE,OAAO,eAAe,CAAC;YACzB,CAAC;QACH,CAAC;QAED,IAAI,aAAa,EAAE,YAAY,IAAI,aAAa,EAAE,aAAa,EAAE,CAAC;YAChE,OAAO,kBAAkB,CAAC;QAC5B,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAEO,iBAAiB,CAAC,IAAyB;QACjD,MAAM,YAAY,GAChB,OAAO,IAAI,EAAE,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,IAAI,EAAE,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;QACjG,OAAO,YAAY,EAAE,IAAI,EAAE,CAAC,WAAW,EAAE,IAAI,IAAI,CAAC;IACpD,CAAC;IAEO,SAAS,CACf,MAAyB,EACzB,MAAc,EACd,aAAkC,EAClC,cAA8B,EAC9B,iBAA0B;QAE1B,IAAI,KAAK,GAAG,MAAM,CAAC,SAAS,IAAI,GAAG,CAAC;QACpC,MAAM,OAAO,GAAa,EAAE,CAAC;QAE7B,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;YAC5C,KAAK,IAAI,CAAC,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,iBAAiB,cAAc,GAAG,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,iBAAiB,IAAI,MAAM,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;YAC3D,KAAK,IAAI,CAAC,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,MAAM,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC;YAC5B,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACtC,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC7B,KAAK,IAAI,CAAC,CAAC;oBACX,OAAO,CAAC,IAAI,CAAC,YAAY,OAAO,GAAG,CAAC,CAAC;gBACvC,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,YAAY,EAAE,MAAM,EAAE,CAAC;YAChC,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;gBACtC,IAAI,IAAI,CAAC,WAAW,CAAC,aAAa,EAAE,GAAG,CAAC,EAAE,CAAC;oBACzC,KAAK,IAAI,CAAC,CAAC;gBACb,CAAC;qBAAM,CAAC;oBACN,KAAK,IAAI,CAAC,CAAC;gBACb,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACxE,KAAK,GAAG,MAAM,CAAC,iBAAiB,CAAC;QACnC,CAAC;QAED,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;IACpC,CAAC;IAEO,WAAW,CAAC,IAAyB,EAAE,GAAW;QACxD,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;QACxB,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YAC1C,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,OAAO,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;QACjC,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;QAC1B,CAAC;QACD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;QACvC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,mBAAmB,CAAC,MAAyB,EAAE,IAAyB;QAC9E,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,MAAM,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7D,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,OAAO,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC;IAC3E,CAAC;CACF"}

package/dist/server/createMcpServer.d.ts

@@ -0,0 +1,7 @@
+import type { McpServerConfig } from "../types.js";
+/**
+ * Start an MCP server with the given tier configuration.
+ * This is the single entry point that reader/writer/server all call.
+ */
+export declare function startMcpServer(config: McpServerConfig): Promise<void>;
+//# sourceMappingURL=createMcpServer.d.ts.map

package/dist/server/createMcpServer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"createMcpServer.d.ts","sourceRoot":"","sources":["../../src/server/createMcpServer.ts"],"names":[],"mappings":"AAUA,OAAO,KAAK,EAAE,eAAe,EAAgB,MAAM,aAAa,CAAC;AAiBjE;;;GAGG;AACH,wBAAsB,cAAc,CAAC,MAAM,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC,CAqG3E"}

package/dist/server/createMcpServer.js

@@ -0,0 +1,100 @@
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { CallToolRequestSchema, ListToolsRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
+import * as crypto from "crypto";
+import { initShims } from "../shims.js";
+import { getEnvironmentManager } from "../config/EnvironmentManager.js";
+import { IntentRouter } from "../routing/IntentRouter.js";
+import { wrapToolRun } from "./wrapToolRun.js";
+import { createAllToolInstances, getReaderTools, getWriterTools, getAdminTools, buildToolRegistry, READER_MUTATING_TOOLS, WRITER_MUTATING_TOOLS, ADMIN_MUTATING_TOOLS, READER_APPROVAL_EXEMPT, WRITER_APPROVAL_EXEMPT, ADMIN_APPROVAL_EXEMPT, } from "./toolsets.js";
+/**
+ * Start an MCP server with the given tier configuration.
+ * This is the single entry point that reader/writer/server all call.
+ */
+export async function startMcpServer(config) {
+    // 1. Runtime shims (SlowBuffer, dotenv)
+    initShims();
+    // 2. Session ID for audit correlation
+    const sessionId = crypto.randomUUID();
+    // 3. Environment manager (reads ENVIRONMENTS_CONFIG_PATH, etc.)
+    const environmentManager = getEnvironmentManager();
+    // 4. Create all tool instances
+    const tools = createAllToolInstances();
+    // 5. Select tools and policies based on tier
+    let exposedTools;
+    let mutatingToolNames;
+    let approvalExemptTools;
+    switch (config.tier) {
+        case "reader":
+            exposedTools = getReaderTools(tools);
+            mutatingToolNames = READER_MUTATING_TOOLS;
+            approvalExemptTools = READER_APPROVAL_EXEMPT;
+            break;
+        case "writer":
+            exposedTools = getWriterTools(tools);
+            mutatingToolNames = WRITER_MUTATING_TOOLS;
+            approvalExemptTools = WRITER_APPROVAL_EXEMPT;
+            break;
+        case "admin":
+            exposedTools = getAdminTools(tools);
+            mutatingToolNames = ADMIN_MUTATING_TOOLS;
+            approvalExemptTools = ADMIN_APPROVAL_EXEMPT;
+            break;
+    }
+    // 6. Wrap all exposed tools with policy enforcement + pool injection
+    for (const tool of exposedTools) {
+        wrapToolRun(tool, {
+            environmentManager,
+            sessionId,
+            serverVersion: config.version,
+            mutatingToolNames,
+            approvalExemptTools,
+        });
+    }
+    // 7. Build intent router (optional, used internally — not exposed as a tool)
+    const isReadOnly = config.tier === "reader" || process.env.READONLY === "true";
+    const requireMutationConfirmation = process.env.REQUIRE_MUTATION_CONFIRMATION !== "false";
+    const toolRegistry = buildToolRegistry(tools);
+    const _intentRouter = new IntentRouter({
+        tools: toolRegistry,
+        allowMutations: !isReadOnly,
+        requireConfirmationForMutations: requireMutationConfirmation,
+    });
+    // 8. Build a Map for fast tool dispatch
+    const toolMap = new Map();
+    for (const tool of exposedTools) {
+        toolMap.set(tool.name, tool);
+    }
+    // 9. Create MCP server
+    const server = new Server({ name: config.name, version: config.version }, { capabilities: { tools: {} } });
+    // 10. Register request handlers
+    server.setRequestHandler(ListToolsRequestSchema, async () => ({
+        tools: exposedTools,
+    }));
+    server.setRequestHandler(CallToolRequestSchema, async (request) => {
+        const { name, arguments: args } = request.params;
+        try {
+            const tool = toolMap.get(name);
+            if (!tool) {
+                return {
+                    content: [{ type: "text", text: `Unknown tool: ${name}` }],
+                    isError: true,
+                };
+            }
+            const result = await tool.run(args);
+            return {
+                content: [{ type: "text", text: JSON.stringify(result, null, 2) }],
+            };
+        }
+        catch (error) {
+            return {
+                content: [{ type: "text", text: `Error occurred: ${error}` }],
+                isError: true,
+            };
+        }
+    });
+    // 11. Connect transport and start
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+}
+//# sourceMappingURL=createMcpServer.js.map

package/dist/server/createMcpServer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"createMcpServer.js","sourceRoot":"","sources":["../../src/server/createMcpServer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EACL,qBAAqB,EACrB,sBAAsB,GACvB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AAEjC,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AAExE,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EACL,sBAAsB,EACtB,cAAc,EACd,cAAc,EACd,aAAa,EACb,iBAAiB,EACjB,qBAAqB,EACrB,qBAAqB,EACrB,oBAAoB,EACpB,sBAAsB,EACtB,sBAAsB,EACtB,qBAAqB,GACtB,MAAM,eAAe,CAAC;AAEvB;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,MAAuB;IAC1D,wCAAwC;IACxC,SAAS,EAAE,CAAC;IAEZ,sCAAsC;IACtC,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;IAEtC,gEAAgE;IAChE,MAAM,kBAAkB,GAAG,qBAAqB,EAAE,CAAC;IAEnD,+BAA+B;IAC/B,MAAM,KAAK,GAAG,sBAAsB,EAAE,CAAC;IAEvC,6CAA6C;IAC7C,IAAI,YAA4B,CAAC;IACjC,IAAI,iBAA8B,CAAC;IACnC,IAAI,mBAAgC,CAAC;IAErC,QAAQ,MAAM,CAAC,IAAI,EAAE,CAAC;QACpB,KAAK,QAAQ;YACX,YAAY,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;YACrC,iBAAiB,GAAG,qBAAqB,CAAC;YAC1C,mBAAmB,GAAG,sBAAsB,CAAC;YAC7C,MAAM;QACR,KAAK,QAAQ;YACX,YAAY,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;YACrC,iBAAiB,GAAG,qBAAqB,CAAC;YAC1C,mBAAmB,GAAG,sBAAsB,CAAC;YAC7C,MAAM;QACR,KAAK,OAAO;YACV,YAAY,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;YACpC,iBAAiB,GAAG,oBAAoB,CAAC;YACzC,mBAAmB,GAAG,qBAAqB,CAAC;YAC5C,MAAM;IACV,CAAC;IAED,qEAAqE;IACrE,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;QAChC,WAAW,CAAC,IAAI,EAAE;YAChB,kBAAkB;YAClB,SAAS;YACT,aAAa,EAAE,MAAM,CAAC,OAAO;YAC7B,iBAAiB;YACjB,mBAAmB;SACpB,CAAC,CAAC;IACL,CAAC;IAED,6EAA6E;IAC7E,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,KAAK,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,MAAM,CAAC;IAC/E,MAAM,2BAA2B,GAAG,OAAO,CAAC,GAAG,CAAC,6BAA6B,KAAK,OAAO,CAAC;IAC1F,MAAM,YAAY,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;IAE9C,MAAM,aAAa,GAAG,IAAI,YAAY,CAAC;QACrC,KAAK,EAAE,YAAY;QACnB,cAAc,EAAE,CAAC,UAAU;QAC3B,+BAA+B,EAAE,2BAA2B;KAC7D,CAAC,CAAC;IAEH,wCAAwC;IACxC,MAAM,OAAO,GAAG,IAAI,GAAG,EAAwB,CAAC;IAChD,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;QAChC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAC/B,CAAC;IAED,uBAAuB;IACvB,MAAM,MAAM,GAAG,IAAI,MAAM,CACvB,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,EAC9C,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAChC,CAAC;IAEF,gCAAgC;IAChC,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;QAC5D,KAAK,EAAE,YAAY;KACpB,CAAC,CAAC,CAAC;IAEJ,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAChE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QACjD,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC/B,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,IAAI,EAAE,EAAE,CAAC;oBAC1D,OAAO,EAAE,IAAI;iBACd,CAAC;YACJ,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACpC,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;aACnE,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,KAAK,EAAE,EAAE,CAAC;gBAC7D,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,kCAAkC;IAClC,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;AAClC,CAAC"}

package/dist/server/toolsets.d.ts

@@ -0,0 +1,42 @@
+import type { RunnableTool, ToolRoutingConfig } from "../types.js";
+export declare function createAllToolInstances(): {
+    createIndexTool: RunnableTool;
+    createTableTool: RunnableTool;
+    deleteDataTool: RunnableTool;
+    describeTableTool: RunnableTool;
+    dropTableTool: RunnableTool;
+    explainQueryTool: RunnableTool;
+    insertDataTool: RunnableTool;
+    inspectDependenciesTool: RunnableTool;
+    listDatabasesTool: RunnableTool;
+    listEnvironmentsTool: RunnableTool;
+    listScriptsTool: RunnableTool;
+    listTableTool: RunnableTool;
+    profileTableTool: RunnableTool;
+    readDataTool: RunnableTool;
+    relationshipInspectorTool: RunnableTool;
+    runScriptTool: RunnableTool;
+    searchSchemaTool: RunnableTool;
+    testConnectionTool: RunnableTool;
+    updateDataTool: RunnableTool;
+    validateEnvironmentConfigTool: RunnableTool;
+};
+type AllTools = ReturnType<typeof createAllToolInstances>;
+/** 14 read-only tools exposed by the reader tier */
+export declare function getReaderTools(t: AllTools): RunnableTool[];
+/** 17 tools: reader tools + insert/delete/update */
+export declare function getWriterTools(t: AllTools): RunnableTool[];
+/** 20 tools: writer tools + create table/index, drop table */
+export declare function getAdminTools(t: AllTools): RunnableTool[];
+/** Reader: no mutations */
+export declare const READER_MUTATING_TOOLS: Set<string>;
+/** Writer: data mutation only */
+export declare const WRITER_MUTATING_TOOLS: Set<string>;
+/** Admin/server: data mutation + DDL */
+export declare const ADMIN_MUTATING_TOOLS: Set<string>;
+export declare const READER_APPROVAL_EXEMPT: Set<string>;
+export declare const WRITER_APPROVAL_EXEMPT: Set<string>;
+export declare const ADMIN_APPROVAL_EXEMPT: Set<string>;
+export declare function buildToolRegistry(t: AllTools): ToolRoutingConfig[];
+export {};
+//# sourceMappingURL=toolsets.d.ts.map

package/dist/server/toolsets.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolsets.d.ts","sourceRoot":"","sources":["../../src/server/toolsets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAyBnE,wBAAgB,sBAAsB;qBAEQ,YAAY;qBACZ,YAAY;oBACd,YAAY;uBACN,YAAY;mBACpB,YAAY;sBACN,YAAY;oBAChB,YAAY;6BACM,YAAY;uBACxB,YAAY;0BACN,YAAY;qBACtB,YAAY;mBAChB,YAAY;sBACN,YAAY;kBACpB,YAAY;+BACc,YAAY;mBACpC,YAAY;sBACN,YAAY;wBACR,YAAY;oBACpB,YAAY;mCACkB,YAAY;EAErF;AAED,KAAK,QAAQ,GAAG,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAI1D,oDAAoD;AACpD,wBAAgB,cAAc,CAAC,CAAC,EAAE,QAAQ,GAAG,YAAY,EAAE,CAiB1D;AAED,oDAAoD;AACpD,wBAAgB,cAAc,CAAC,CAAC,EAAE,QAAQ,GAAG,YAAY,EAAE,CAO1D;AAED,8DAA8D;AAC9D,wBAAgB,aAAa,CAAC,CAAC,EAAE,QAAQ,GAAG,YAAY,EAAE,CAOzD;AAID,2BAA2B;AAC3B,eAAO,MAAM,qBAAqB,aAAoB,CAAC;AAEvD,iCAAiC;AACjC,eAAO,MAAM,qBAAqB,aAIhC,CAAC;AAEH,wCAAwC;AACxC,eAAO,MAAM,oBAAoB,aAO/B,CAAC;AAIH,eAAO,MAAM,sBAAsB,aAejC,CAAC;AAEH,eAAO,MAAM,sBAAsB,aAWjC,CAAC;AAEH,eAAO,MAAM,qBAAqB,aAchC,CAAC;AAIH,wBAAgB,iBAAiB,CAAC,CAAC,EAAE,QAAQ,GAAG,iBAAiB,EAAE,CA2JlE"}