@connorbritain/mssql-mcp-core 0.1.0

package/dist/tools/SearchSchemaTool.js

@@ -0,0 +1,237 @@
+import sql from "mssql";
+const clampEnvLimit = (value, fallback, max) => {
+    if (!value) {
+        return fallback;
+    }
+    const parsed = parseInt(value, 10);
+    if (Number.isNaN(parsed) || parsed <= 0) {
+        return fallback;
+    }
+    return Math.min(parsed, max);
+};
+const SEARCH_DEFAULT_LIMIT = clampEnvLimit(process.env.SEARCH_SCHEMA_DEFAULT_LIMIT, 50, 200);
+export class SearchSchemaTool {
+    constructor() {
+        this.name = "search_schema";
+        this.description = "Searches tables and columns using wildcard patterns to discover schema names.";
+        this.inputSchema = {
+            type: "object",
+            properties: {
+                tablePattern: {
+                    type: "string",
+                    description: "Wildcard pattern for table names (e.g. 'doc%')."
+                },
+                columnPattern: {
+                    type: "string",
+                    description: "Wildcard pattern for column names (e.g. '%id')."
+                },
+                limit: {
+                    type: "number",
+                    description: "Maximum rows to return per section (default 50, max 200)."
+                },
+                tableLimit: {
+                    type: "number",
+                    description: "Override for table section limit (defaults to limit)."
+                },
+                columnLimit: {
+                    type: "number",
+                    description: "Override for column section limit (defaults to limit)."
+                },
+                tableOffset: {
+                    type: "number",
+                    description: "Number of table rows to skip (for pagination)."
+                },
+                columnOffset: {
+                    type: "number",
+                    description: "Number of column rows to skip (for pagination)."
+                }
+            }
+        };
+    }
+    normalizePattern(value) {
+        if (!value) {
+            return null;
+        }
+        const trimmed = value.trim();
+        if (!trimmed) {
+            return null;
+        }
+        return `%${trimmed.replace(/%/g, "%").replace(/_/g, "_")}%`;
+    }
+    stripWildcards(value) {
+        if (!value) {
+            return null;
+        }
+        return value.replace(/[\%_]/g, "").trim() || null;
+    }
+    normalizeLimit(value, fallback = 50) {
+        if (typeof value !== "number" || Number.isNaN(value) || value <= 0) {
+            return fallback;
+        }
+        return Math.min(Math.floor(value), 200);
+    }
+    normalizeOffset(value) {
+        if (typeof value !== "number" || Number.isNaN(value) || value <= 0) {
+            return 0;
+        }
+        return Math.floor(value);
+    }
+    levenshteinDistance(a, b) {
+        const rows = a.length + 1;
+        const cols = b.length + 1;
+        const matrix = Array.from({ length: rows }, () => Array(cols).fill(0));
+        for (let i = 0; i < rows; i++) {
+            matrix[i][0] = i;
+        }
+        for (let j = 0; j < cols; j++) {
+            matrix[0][j] = j;
+        }
+        for (let i = 1; i < rows; i++) {
+            for (let j = 1; j < cols; j++) {
+                const cost = a[i - 1] === b[j - 1] ? 0 : 1;
+                matrix[i][j] = Math.min(matrix[i - 1][j] + 1, matrix[i][j - 1] + 1, matrix[i - 1][j - 1] + cost);
+            }
+        }
+        return matrix[rows - 1][cols - 1];
+    }
+    calculateSimilarity(a, b) {
+        if (!a || !b) {
+            return 0;
+        }
+        const lowerA = a.toLowerCase();
+        const lowerB = b.toLowerCase();
+        const distance = this.levenshteinDistance(lowerA, lowerB);
+        const maxLength = Math.max(lowerA.length, lowerB.length, 1);
+        return 1 - distance / maxLength;
+    }
+    async findSimilarTables(searchTerm, maxSuggestions, pool) {
+        const request = new sql.Request(pool);
+        const result = await request.query(`
+      SELECT TABLE_SCHEMA AS schemaName, TABLE_NAME AS tableName
+      FROM INFORMATION_SCHEMA.TABLES
+    `);
+        const suggestions = result.recordset
+            .map((row) => ({
+            ...row,
+            similarity: this.calculateSimilarity(row.tableName, searchTerm)
+        }))
+            .filter((row) => row.similarity >= 0.5)
+            .sort((a, b) => b.similarity - a.similarity)
+            .slice(0, maxSuggestions)
+            .map((row) => ({
+            schemaName: row.schemaName,
+            tableName: row.tableName,
+            similarity: Number(row.similarity.toFixed(3))
+        }));
+        return suggestions;
+    }
+    async run(params) {
+        try {
+            const rawTableSearch = this.stripWildcards(params.tablePattern);
+            const tablePattern = this.normalizePattern(params.tablePattern);
+            const columnPattern = this.normalizePattern(params.columnPattern);
+            const limit = typeof params.limit === "number" && params.limit > 0 ? Math.min(params.limit, 200) : SEARCH_DEFAULT_LIMIT;
+            const tableLimit = this.normalizeLimit(params.tableLimit, limit);
+            const columnLimit = this.normalizeLimit(params.columnLimit, limit);
+            const tableOffset = this.normalizeOffset(params.tableOffset);
+            const columnOffset = this.normalizeOffset(params.columnOffset);
+            if (!tablePattern && !columnPattern) {
+                return {
+                    success: false,
+                    message: "Provide at least one of tablePattern or columnPattern."
+                };
+            }
+            const pool = params.pool;
+            const tablesRequest = new sql.Request(pool);
+            tablesRequest.input("tablePattern", sql.NVarChar, tablePattern);
+            tablesRequest.input("tableOffset", sql.Int, tableOffset);
+            tablesRequest.input("tablesLimit", sql.Int, tableLimit);
+            const columnsRequest = new sql.Request(pool);
+            columnsRequest.input("tablePattern", sql.NVarChar, tablePattern);
+            columnsRequest.input("columnPattern", sql.NVarChar, columnPattern);
+            columnsRequest.input("columnOffset", sql.Int, columnOffset);
+            columnsRequest.input("columnLimit", sql.Int, columnLimit);
+            const tablesQuery = `
+        SELECT TABLE_SCHEMA AS schemaName, TABLE_NAME AS tableName
+        FROM INFORMATION_SCHEMA.TABLES
+        WHERE (@tablePattern IS NULL OR TABLE_NAME LIKE @tablePattern)
+        ORDER BY TABLE_SCHEMA, TABLE_NAME
+        OFFSET @tableOffset ROWS
+        FETCH NEXT @tablesLimit ROWS ONLY;
+
+        SELECT COUNT(*) AS total
+        FROM INFORMATION_SCHEMA.TABLES
+        WHERE (@tablePattern IS NULL OR TABLE_NAME LIKE @tablePattern)`;
+            const columnsQuery = `
+        SELECT TABLE_SCHEMA AS schemaName, TABLE_NAME AS tableName, COLUMN_NAME AS columnName, DATA_TYPE AS dataType
+        FROM INFORMATION_SCHEMA.COLUMNS
+        WHERE (@tablePattern IS NULL OR TABLE_NAME LIKE @tablePattern)
+          AND (@columnPattern IS NULL OR COLUMN_NAME LIKE @columnPattern)
+        ORDER BY TABLE_SCHEMA, TABLE_NAME, ORDINAL_POSITION
+        OFFSET @columnOffset ROWS
+        FETCH NEXT @columnLimit ROWS ONLY;
+
+        SELECT COUNT(*) AS total
+        FROM INFORMATION_SCHEMA.COLUMNS
+        WHERE (@tablePattern IS NULL OR TABLE_NAME LIKE @tablePattern)
+          AND (@columnPattern IS NULL OR COLUMN_NAME LIKE @columnPattern)`;
+            const [tablesResult, columnsResult] = await Promise.all([
+                tablesRequest.query(tablesQuery),
+                columnsRequest.query(columnsQuery)
+            ]);
+            const tableRecordsets = Array.isArray(tablesResult.recordsets) && tablesResult.recordsets.length
+                ? tablesResult.recordsets
+                : tablesResult.recordset
+                    ? [tablesResult.recordset]
+                    : [];
+            const columnRecordsets = Array.isArray(columnsResult.recordsets) && columnsResult.recordsets.length
+                ? columnsResult.recordsets
+                : columnsResult.recordset
+                    ? [columnsResult.recordset]
+                    : [];
+            const tableRecords = tableRecordsets[0] ?? [];
+            const tableTotalRecord = tableRecordsets[1]?.[0];
+            const totalTables = typeof tableTotalRecord?.total === "number" ? tableTotalRecord.total : tableRecords.length;
+            const columnRecords = columnRecordsets[0] ?? [];
+            const columnTotalRecord = columnRecordsets[1]?.[0];
+            const totalColumns = typeof columnTotalRecord?.total === "number" ? columnTotalRecord.total : columnRecords.length;
+            const response = {
+                success: true,
+                totalTables,
+                totalColumns,
+                tables: tableRecords,
+                columns: columnRecords,
+                tablesPage: {
+                    offset: tableOffset,
+                    limit: tableLimit,
+                    total: totalTables,
+                    hasMore: tableOffset + tableRecords.length < totalTables
+                },
+                columnsPage: {
+                    offset: columnOffset,
+                    limit: columnLimit,
+                    total: totalColumns,
+                    hasMore: columnOffset + columnRecords.length < totalColumns
+                }
+            };
+            const needsFuzzyTables = Boolean(rawTableSearch &&
+                !totalTables &&
+                !totalColumns);
+            if (needsFuzzyTables && rawTableSearch) {
+                const fuzzySuggestions = await this.findSimilarTables(rawTableSearch, 10, pool);
+                if (fuzzySuggestions.length) {
+                    response.fuzzySuggestions = fuzzySuggestions;
+                    response.fuzzySearchTerm = rawTableSearch;
+                }
+            }
+            return response;
+        }
+        catch (error) {
+            return {
+                success: false,
+                message: `Failed to search schema: ${error}`
+            };
+        }
+    }
+}
+//# sourceMappingURL=SearchSchemaTool.js.map

package/dist/tools/SearchSchemaTool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SearchSchemaTool.js","sourceRoot":"","sources":["../../src/tools/SearchSchemaTool.ts"],"names":[],"mappings":"AAAA,OAAO,GAAG,MAAM,OAAO,CAAC;AAGxB,MAAM,aAAa,GAAG,CAAC,KAAyB,EAAE,QAAgB,EAAE,GAAW,EAAE,EAAE;IACjF,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACnC,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,MAAM,IAAI,CAAC,EAAE,CAAC;QACxC,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,OAAO,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;AAC/B,CAAC,CAAC;AAEF,MAAM,oBAAoB,GAAG,aAAa,CAAC,OAAO,CAAC,GAAG,CAAC,2BAA2B,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;AAgC7F,MAAM,OAAO,gBAAgB;IAA7B;QAEE,SAAI,GAAG,eAAe,CAAC;QACvB,gBAAW,GAAG,+EAA+E,CAAC;QAC9F,gBAAW,GAAG;YACZ,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,YAAY,EAAE;oBACZ,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iDAAiD;iBAC/D;gBACD,aAAa,EAAE;oBACb,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iDAAiD;iBAC/D;gBACD,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,2DAA2D;iBACzE;gBACD,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,uDAAuD;iBACrE;gBACD,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,wDAAwD;iBACtE;gBACD,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,gDAAgD;iBAC9D;gBACD,YAAY,EAAE;oBACZ,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iDAAiD;iBAC/D;aACF;SACO,CAAC;IAyNb,CAAC;IAvNS,gBAAgB,CAAC,KAAqB;QAC5C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;QAC7B,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,GAAG,CAAC;IAC9D,CAAC;IAEO,cAAc,CAAC,KAAqB;QAC1C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,IAAI,CAAC;IACpD,CAAC;IAEO,cAAc,CAAC,KAAqB,EAAE,QAAQ,GAAG,EAAE;QACzD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;YACnE,OAAO,QAAQ,CAAC;QAClB,CAAC;QACD,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,GAAG,CAAC,CAAC;IAC1C,CAAC;IAEO,eAAe,CAAC,KAAqB;QAC3C,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;YACnE,OAAO,CAAC,CAAC;QACX,CAAC;QACD,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAC3B,CAAC;IAEO,mBAAmB,CAAC,CAAS,EAAE,CAAS;QAC9C,MAAM,IAAI,GAAG,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;QAC1B,MAAM,IAAI,GAAG,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;QAC1B,MAAM,MAAM,GAAe,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAEnF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9B,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;QACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9B,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;QAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9B,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC3C,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CACrB,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EACpB,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,EACpB,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAC5B,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;IACpC,CAAC;IAEO,mBAAmB,CAAC,CAAS,EAAE,CAAS;QAC9C,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;YACb,OAAO,CAAC,CAAC;QACX,CAAC;QACD,MAAM,MAAM,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC1D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC5D,OAAO,CAAC,GAAG,QAAQ,GAAG,SAAS,CAAC;IAClC,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAAC,UAAkB,EAAE,cAAsB,EAAE,IAAU;QACpF,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC;;;KAGlC,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,MAAM,CAAC,SAAS;aACjC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACb,GAAG,GAAG;YACN,UAAU,EAAE,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC;SAChE,CAAC,CAAC;aACF,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC;aACtC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC;aAC3C,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC;aACxB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACb,UAAU,EAAE,GAAG,CAAC,UAAU;YAC1B,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;SAC9C,CAAC,CAAC,CAAC;QAEN,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,MAAoB;QAC5B,IAAI,CAAC;YACH,MAAM,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;YAChE,MAAM,YAAY,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;YAChE,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAClE,MAAM,KAAK,GAAG,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,oBAAoB,CAAC;YACxH,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YACjE,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;YACnE,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAC7D,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;YAE/D,IAAI,CAAC,YAAY,IAAI,CAAC,aAAa,EAAE,CAAC;gBACpC,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,OAAO,EAAE,wDAAwD;iBAClE,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,GAAI,MAAc,CAAC,IAAI,CAAC;YAClC,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC5C,aAAa,CAAC,KAAK,CAAC,cAAc,EAAE,GAAG,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;YAChE,aAAa,CAAC,KAAK,CAAC,aAAa,EAAE,GAAG,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;YACzD,aAAa,CAAC,KAAK,CAAC,aAAa,EAAE,GAAG,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;YAExD,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC7C,cAAc,CAAC,KAAK,CAAC,cAAc,EAAE,GAAG,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;YACjE,cAAc,CAAC,KAAK,CAAC,eAAe,EAAE,GAAG,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;YACnE,cAAc,CAAC,KAAK,CAAC,cAAc,EAAE,GAAG,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;YAC5D,cAAc,CAAC,KAAK,CAAC,aAAa,EAAE,GAAG,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;YAE1D,MAAM,WAAW,GAAG;;;;;;;;;;uEAU6C,CAAC;YAElE,MAAM,YAAY,GAAG;;;;;;;;;;;;0EAY+C,CAAC;YAErE,MAAM,CAAC,YAAY,EAAE,aAAa,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBACtD,aAAa,CAAC,KAAK,CAAC,WAAW,CAAC;gBAChC,cAAc,CAAC,KAAK,CAAC,YAAY,CAAC;aACnC,CAAC,CAAC;YAEH,MAAM,eAAe,GAAG,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,YAAY,CAAC,UAAU,CAAC,MAAM;gBAC9F,CAAC,CAAC,YAAY,CAAC,UAAU;gBACzB,CAAC,CAAC,YAAY,CAAC,SAAS;oBACtB,CAAC,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC;oBAC1B,CAAC,CAAC,EAAE,CAAC;YACT,MAAM,gBAAgB,GAAG,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,aAAa,CAAC,UAAU,CAAC,MAAM;gBACjG,CAAC,CAAC,aAAa,CAAC,UAAU;gBAC1B,CAAC,CAAC,aAAa,CAAC,SAAS;oBACvB,CAAC,CAAC,CAAC,aAAa,CAAC,SAAS,CAAC;oBAC3B,CAAC,CAAC,EAAE,CAAC;YAET,MAAM,YAAY,GAAG,eAAe,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAC9C,MAAM,gBAAgB,GAAG,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAmC,CAAC;YACnF,MAAM,WAAW,GAAG,OAAO,gBAAgB,EAAE,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC;YAE/G,MAAM,aAAa,GAAG,gBAAgB,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAChD,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAmC,CAAC;YACrF,MAAM,YAAY,GAAG,OAAO,iBAAiB,EAAE,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC;YAEnH,MAAM,QAAQ,GAAiB;gBAC7B,OAAO,EAAE,IAAI;gBACb,WAAW;gBACX,YAAY;gBACZ,MAAM,EAAE,YAAY;gBACpB,OAAO,EAAE,aAAa;gBACtB,UAAU,EAAE;oBACV,MAAM,EAAE,WAAW;oBACnB,KAAK,EAAE,UAAU;oBACjB,KAAK,EAAE,WAAW;oBAClB,OAAO,EAAE,WAAW,GAAG,YAAY,CAAC,MAAM,GAAG,WAAW;iBACzD;gBACD,WAAW,EAAE;oBACX,MAAM,EAAE,YAAY;oBACpB,KAAK,EAAE,WAAW;oBAClB,KAAK,EAAE,YAAY;oBACnB,OAAO,EAAE,YAAY,GAAG,aAAa,CAAC,MAAM,GAAG,YAAY;iBAC5D;aACF,CAAC;YAEF,MAAM,gBAAgB,GAAG,OAAO,CAC9B,cAAc;gBACd,CAAC,WAAW;gBACZ,CAAC,YAAY,CACd,CAAC;YAEF,IAAI,gBAAgB,IAAI,cAAc,EAAE,CAAC;gBACvC,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,cAAc,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC;gBAChF,IAAI,gBAAgB,CAAC,MAAM,EAAE,CAAC;oBAC5B,QAAQ,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;oBAC7C,QAAQ,CAAC,eAAe,GAAG,cAAc,CAAC;gBAC5C,CAAC;YACH,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,4BAA4B,KAAK,EAAE;aAC7C,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}

package/dist/tools/TestConnectionTool.d.ts

@@ -0,0 +1,38 @@
+import { Tool } from "@modelcontextprotocol/sdk/types.js";
+export declare class TestConnectionTool implements Tool {
+    [key: string]: any;
+    name: string;
+    description: string;
+    inputSchema: any;
+    run(params: any): Promise<{
+        success: boolean;
+        message: string;
+        connected: boolean;
+        mcpServerVersion: any;
+        latency: {
+            connectionMs: number;
+            queryMs: number;
+            totalMs: number;
+        };
+        serverInfo: any;
+        error?: undefined;
+    } | {
+        success: boolean;
+        message: string;
+        connected: boolean;
+        latency: {
+            totalMs: number;
+            connectionMs?: undefined;
+            queryMs?: undefined;
+        };
+        error: {
+            code: string;
+            suggestion: string;
+        };
+        mcpServerVersion?: undefined;
+        serverInfo?: undefined;
+    }>;
+    private getEngineEditionName;
+    private categorizeError;
+}
+//# sourceMappingURL=TestConnectionTool.d.ts.map

package/dist/tools/TestConnectionTool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"TestConnectionTool.d.ts","sourceRoot":"","sources":["../../src/tools/TestConnectionTool.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,MAAM,oCAAoC,CAAC;AAG1D,qBAAa,kBAAmB,YAAW,IAAI;IAC7C,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;IACnB,IAAI,SAAqB;IACzB,WAAW,SAAsG;IAEjH,WAAW,EAaN,GAAG,CAAC;IAEH,GAAG,CAAC,MAAM,EAAE,GAAG;;;;;;;;;;;;;;;;;;;;;;kBAyG6B,MAAM;wBAAc,MAAM;;;;;IAf5E,OAAO,CAAC,oBAAoB;IAe5B,OAAO,CAAC,eAAe;CA2CxB"}

package/dist/tools/TestConnectionTool.js

@@ -0,0 +1,156 @@
+import sql from "mssql";
+import { getEnvironmentManager } from "../config/EnvironmentManager.js";
+export class TestConnectionTool {
+    constructor() {
+        this.name = "test_connection";
+        this.description = "Tests connectivity to a database environment and returns status, latency, and basic server info.";
+        this.inputSchema = {
+            type: "object",
+            properties: {
+                environment: {
+                    type: "string",
+                    description: "Optional environment name to test. If not provided, tests the default environment.",
+                },
+                verbose: {
+                    type: "boolean",
+                    description: "If true, returns additional server info (version, edition, etc.)",
+                },
+            },
+            required: [],
+        };
+    }
+    async run(params) {
+        const startTime = Date.now();
+        const environmentName = params?.environment;
+        const verbose = params?.verbose ?? false;
+        try {
+            const envManager = getEnvironmentManager();
+            const env = envManager.getEnvironment(environmentName);
+            // Get connection
+            const pool = await envManager.getConnection(environmentName);
+            const connectionTime = Date.now() - startTime;
+            // Run a simple query to verify connectivity
+            const queryStart = Date.now();
+            const request = new sql.Request(pool);
+            const result = await request.query("SELECT 1 AS connected");
+            const queryTime = Date.now() - queryStart;
+            let serverInfo = {
+                environment: env.name,
+                server: env.server,
+                database: env.database,
+                authMode: env.authMode,
+                readonly: env.readonly ?? false,
+            };
+            // Get detailed server info if verbose
+            if (verbose) {
+                try {
+                    const infoRequest = new sql.Request(pool);
+                    const infoResult = await infoRequest.query(`
+            SELECT 
+              SERVERPROPERTY('ProductVersion') AS version,
+              SERVERPROPERTY('ProductLevel') AS productLevel,
+              SERVERPROPERTY('Edition') AS edition,
+              SERVERPROPERTY('EngineEdition') AS engineEdition,
+              SERVERPROPERTY('MachineName') AS machineName,
+              SERVERPROPERTY('ServerName') AS serverName,
+              @@VERSION AS fullVersion
+          `);
+                    if (infoResult.recordset.length > 0) {
+                        const info = infoResult.recordset[0];
+                        serverInfo = {
+                            ...serverInfo,
+                            version: info.version,
+                            productLevel: info.productLevel,
+                            edition: info.edition,
+                            engineEdition: this.getEngineEditionName(info.engineEdition),
+                            machineName: info.machineName,
+                            serverName: info.serverName,
+                        };
+                    }
+                }
+                catch (infoError) {
+                    // Non-fatal: some queries may not work on all SQL Server editions
+                    serverInfo.versionInfo = "Unable to retrieve (may require elevated permissions)";
+                }
+            }
+            const totalTime = Date.now() - startTime;
+            return {
+                success: true,
+                message: `Successfully connected to '${env.name}' (${env.server}/${env.database})`,
+                connected: true,
+                mcpServerVersion: params?.mcpServerVersion,
+                latency: {
+                    connectionMs: connectionTime,
+                    queryMs: queryTime,
+                    totalMs: totalTime,
+                },
+                serverInfo,
+            };
+        }
+        catch (error) {
+            const totalTime = Date.now() - startTime;
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            return {
+                success: false,
+                message: `Failed to connect: ${errorMessage}`,
+                connected: false,
+                latency: {
+                    totalMs: totalTime,
+                },
+                error: this.categorizeError(errorMessage),
+            };
+        }
+    }
+    getEngineEditionName(edition) {
+        const editions = {
+            1: "Personal/Desktop Engine",
+            2: "Standard",
+            3: "Enterprise",
+            4: "Express",
+            5: "Azure SQL Database",
+            6: "Azure Synapse Analytics",
+            8: "Azure SQL Managed Instance",
+            9: "Azure SQL Edge",
+            11: "Azure Synapse Serverless",
+        };
+        return editions[edition] || `Unknown (${edition})`;
+    }
+    categorizeError(message) {
+        const lowerMessage = message.toLowerCase();
+        if (lowerMessage.includes("login failed") || lowerMessage.includes("authentication")) {
+            return {
+                code: "AUTH_FAILED",
+                suggestion: "Check username, password, and authentication mode. For AAD, ensure the browser auth completed.",
+            };
+        }
+        if (lowerMessage.includes("network") || lowerMessage.includes("enotfound") || lowerMessage.includes("econnrefused")) {
+            return {
+                code: "NETWORK_ERROR",
+                suggestion: "Check SERVER_NAME, SQL_PORT, and ensure the server is reachable. Verify firewall rules.",
+            };
+        }
+        if (lowerMessage.includes("timeout")) {
+            return {
+                code: "TIMEOUT",
+                suggestion: "Connection timed out. Check network latency and CONNECTION_TIMEOUT setting.",
+            };
+        }
+        if (lowerMessage.includes("certificate") || lowerMessage.includes("ssl")) {
+            return {
+                code: "CERTIFICATE_ERROR",
+                suggestion: "Set TRUST_SERVER_CERTIFICATE=true for self-signed certs, or verify the certificate chain.",
+            };
+        }
+        if (lowerMessage.includes("database") && lowerMessage.includes("not exist")) {
+            return {
+                code: "DATABASE_NOT_FOUND",
+                suggestion: "Check DATABASE_NAME. The specified database may not exist or the user lacks access.",
+            };
+        }
+        return {
+            code: "UNKNOWN",
+            suggestion: "Check all connection parameters and server logs for more details.",
+        };
+    }
+}
+//# sourceMappingURL=TestConnectionTool.js.map

package/dist/tools/TestConnectionTool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TestConnectionTool.js","sourceRoot":"","sources":["../../src/tools/TestConnectionTool.ts"],"names":[],"mappings":"AAAA,OAAO,GAAG,MAAM,OAAO,CAAC;AAExB,OAAO,EAAE,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AAExE,MAAM,OAAO,kBAAkB;IAA/B;QAEE,SAAI,GAAG,iBAAiB,CAAC;QACzB,gBAAW,GAAG,kGAAkG,CAAC;QAEjH,gBAAW,GAAG;YACZ,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,oFAAoF;iBAClG;gBACD,OAAO,EAAE;oBACP,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,kEAAkE;iBAChF;aACF;YACD,QAAQ,EAAE,EAAE;SACN,CAAC;IAsJX,CAAC;IApJC,KAAK,CAAC,GAAG,CAAC,MAAW;QACnB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,eAAe,GAAG,MAAM,EAAE,WAAW,CAAC;QAC5C,MAAM,OAAO,GAAG,MAAM,EAAE,OAAO,IAAI,KAAK,CAAC;QAEzC,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,qBAAqB,EAAE,CAAC;YAC3C,MAAM,GAAG,GAAG,UAAU,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;YAEvD,iBAAiB;YACjB,MAAM,IAAI,GAAG,MAAM,UAAU,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC;YAC7D,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAE9C,4CAA4C;YAC5C,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC9B,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAC5D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,UAAU,CAAC;YAE1C,IAAI,UAAU,GAAQ;gBACpB,WAAW,EAAE,GAAG,CAAC,IAAI;gBACrB,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,KAAK;aAChC,CAAC;YAEF,sCAAsC;YACtC,IAAI,OAAO,EAAE,CAAC;gBACZ,IAAI,CAAC;oBACH,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;oBAC1C,MAAM,UAAU,GAAG,MAAM,WAAW,CAAC,KAAK,CAAC;;;;;;;;;WAS1C,CAAC,CAAC;oBAEH,IAAI,UAAU,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACpC,MAAM,IAAI,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;wBACrC,UAAU,GAAG;4BACX,GAAG,UAAU;4BACb,OAAO,EAAE,IAAI,CAAC,OAAO;4BACrB,YAAY,EAAE,IAAI,CAAC,YAAY;4BAC/B,OAAO,EAAE,IAAI,CAAC,OAAO;4BACrB,aAAa,EAAE,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,aAAa,CAAC;4BAC5D,WAAW,EAAE,IAAI,CAAC,WAAW;4BAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;yBAC5B,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAAC,OAAO,SAAS,EAAE,CAAC;oBACnB,kEAAkE;oBAClE,UAAU,CAAC,WAAW,GAAG,uDAAuD,CAAC;gBACnF,CAAC;YACH,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAEzC,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,8BAA8B,GAAG,CAAC,IAAI,MAAM,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,QAAQ,GAAG;gBAClF,SAAS,EAAE,IAAI;gBACf,gBAAgB,EAAG,MAAc,EAAE,gBAAgB;gBACnD,OAAO,EAAE;oBACP,YAAY,EAAE,cAAc;oBAC5B,OAAO,EAAE,SAAS;oBAClB,OAAO,EAAE,SAAS;iBACnB;gBACD,UAAU;aACX,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACzC,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAE5E,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,sBAAsB,YAAY,EAAE;gBAC7C,SAAS,EAAE,KAAK;gBAChB,OAAO,EAAE;oBACP,OAAO,EAAE,SAAS;iBACnB;gBACD,KAAK,EAAE,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC;aAC1C,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,oBAAoB,CAAC,OAAe;QAC1C,MAAM,QAAQ,GAA2B;YACvC,CAAC,EAAE,yBAAyB;YAC5B,CAAC,EAAE,UAAU;YACb,CAAC,EAAE,YAAY;YACf,CAAC,EAAE,SAAS;YACZ,CAAC,EAAE,oBAAoB;YACvB,CAAC,EAAE,yBAAyB;YAC5B,CAAC,EAAE,4BAA4B;YAC/B,CAAC,EAAE,gBAAgB;YACnB,EAAE,EAAE,0BAA0B;SAC/B,CAAC;QACF,OAAO,QAAQ,CAAC,OAAO,CAAC,IAAI,YAAY,OAAO,GAAG,CAAC;IACrD,CAAC;IAEO,eAAe,CAAC,OAAe;QACrC,MAAM,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QAE3C,IAAI,YAAY,CAAC,QAAQ,CAAC,cAAc,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;YACrF,OAAO;gBACL,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,gGAAgG;aAC7G,CAAC;QACJ,CAAC;QAED,IAAI,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;YACpH,OAAO;gBACL,IAAI,EAAE,eAAe;gBACrB,UAAU,EAAE,yFAAyF;aACtG,CAAC;QACJ,CAAC;QAED,IAAI,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YACrC,OAAO;gBACL,IAAI,EAAE,SAAS;gBACf,UAAU,EAAE,6EAA6E;aAC1F,CAAC;QACJ,CAAC;QAED,IAAI,YAAY,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACzE,OAAO;gBACL,IAAI,EAAE,mBAAmB;gBACzB,UAAU,EAAE,2FAA2F;aACxG,CAAC;QACJ,CAAC;QAED,IAAI,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YAC5E,OAAO;gBACL,IAAI,EAAE,oBAAoB;gBAC1B,UAAU,EAAE,qFAAqF;aAClG,CAAC;QACJ,CAAC;QAED,OAAO;YACL,IAAI,EAAE,SAAS;YACf,UAAU,EAAE,mEAAmE;SAChF,CAAC;IACJ,CAAC;CACF"}

package/dist/tools/UpdateDataTool.d.ts

@@ -0,0 +1,61 @@
+import sql from "mssql";
+import { Tool } from "@modelcontextprotocol/sdk/types.js";
+export declare class UpdateDataTool implements Tool {
+    [key: string]: any;
+    name: string;
+    description: string;
+    inputSchema: any;
+    private static readonly MAX_ROWS_DEFAULT;
+    run(params: any): Promise<{
+        success: boolean;
+        message: string;
+        error: string;
+        affectedRows?: undefined;
+        maxAllowed?: undefined;
+        needsConfirmation?: undefined;
+        preview?: undefined;
+        updates?: undefined;
+        rowsAffected?: undefined;
+    } | {
+        success: boolean;
+        message: string;
+        error: string;
+        affectedRows: number;
+        maxAllowed?: undefined;
+        needsConfirmation?: undefined;
+        preview?: undefined;
+        updates?: undefined;
+        rowsAffected?: undefined;
+    } | {
+        success: boolean;
+        message: string;
+        error: string;
+        affectedRows: any;
+        maxAllowed: any;
+        needsConfirmation?: undefined;
+        preview?: undefined;
+        updates?: undefined;
+        rowsAffected?: undefined;
+    } | {
+        success: boolean;
+        needsConfirmation: boolean;
+        message: string;
+        affectedRows: any;
+        preview: sql.IRecordSet<any>;
+        updates: any;
+        error: string;
+        maxAllowed?: undefined;
+        rowsAffected?: undefined;
+    } | {
+        success: boolean;
+        message: string;
+        rowsAffected: number;
+        updates: any;
+        error?: undefined;
+        affectedRows?: undefined;
+        maxAllowed?: undefined;
+        needsConfirmation?: undefined;
+        preview?: undefined;
+    }>;
+}
+//# sourceMappingURL=UpdateDataTool.d.ts.map

package/dist/tools/UpdateDataTool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"UpdateDataTool.d.ts","sourceRoot":"","sources":["../../src/tools/UpdateDataTool.ts"],"names":[],"mappings":"AAAA,OAAO,GAAG,MAAM,OAAO,CAAC;AACxB,OAAO,EAAE,IAAI,EAAE,MAAM,oCAAoC,CAAC;AAE1D,qBAAa,cAAe,YAAW,IAAI;IACzC,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;IACnB,IAAI,SAAiB;IACrB,WAAW,SAAqG;IAChH,WAAW,EA6BN,GAAG,CAAC;IAET,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,gBAAgB,CAAQ;IAE1C,GAAG,CAAC,MAAM,EAAE,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAwFtB"}

package/dist/tools/UpdateDataTool.js

@@ -0,0 +1,117 @@
+import sql from "mssql";
+export class UpdateDataTool {
+    constructor() {
+        this.name = "update_data";
+        this.description = "Updates rows in an MSSQL table with preview and confirmation. Requires WHERE clause for safety.";
+        this.inputSchema = {
+            type: "object",
+            properties: {
+                tableName: {
+                    type: "string",
+                    description: "Name of the table to update"
+                },
+                updates: {
+                    type: "object",
+                    description: "Key-value pairs of columns to update. Example: { 'status': 'active', 'last_updated': '2025-01-01' }",
+                },
+                whereClause: {
+                    type: "string",
+                    description: "WHERE clause to identify which records to update. Example: \"genre = 'comedy' AND created_date <= '2025-07-05'\""
+                },
+                confirmUpdate: {
+                    type: "boolean",
+                    description: "Set to true to confirm and execute the update after preview. First call without this to see preview.",
+                },
+                maxRows: {
+                    type: "number",
+                    description: "Maximum number of rows allowed to update. Defaults to 1000 for safety.",
+                },
+                environment: {
+                    type: "string",
+                    description: "Optional environment name to target",
+                },
+            },
+            required: ["tableName", "updates", "whereClause"],
+        };
+    }
+    async run(params) {
+        let query;
+        try {
+            const { tableName, updates, whereClause, confirmUpdate, maxRows, environment } = params;
+            // Basic validation: ensure whereClause is not empty
+            if (!whereClause || whereClause.trim() === '') {
+                return {
+                    success: false,
+                    message: "WHERE clause is required for security reasons. Use 'WHERE 1=1' to update all rows (not recommended).",
+                    error: "MISSING_WHERE_CLAUSE",
+                };
+            }
+            const maxAllowed = maxRows || UpdateDataTool.MAX_ROWS_DEFAULT;
+            // Step 1: Get count of affected rows
+            const countQuery = `SELECT COUNT(*) as affectedRows FROM ${tableName} WHERE ${whereClause}`;
+            const countRequest = new sql.Request(params.pool);
+            const countResult = await countRequest.query(countQuery);
+            const affectedRows = countResult.recordset[0].affectedRows;
+            if (affectedRows === 0) {
+                return {
+                    success: false,
+                    message: "No rows match the WHERE clause. No update will be performed.",
+                    error: "NO_ROWS_MATCHED",
+                    affectedRows: 0,
+                };
+            }
+            if (affectedRows > maxAllowed) {
+                return {
+                    success: false,
+                    message: `Update would affect ${affectedRows} rows, which exceeds the maximum of ${maxAllowed}. Refine your WHERE clause or increase maxRows parameter.`,
+                    error: "TOO_MANY_ROWS",
+                    affectedRows,
+                    maxAllowed,
+                };
+            }
+            // Step 2: Show preview if not confirmed
+            if (!confirmUpdate) {
+                const previewQuery = `SELECT TOP 10 * FROM ${tableName} WHERE ${whereClause}`;
+                const previewRequest = new sql.Request(params.pool);
+                const previewResult = await previewRequest.query(previewQuery);
+                return {
+                    success: false,
+                    needsConfirmation: true,
+                    message: `Preview: ${affectedRows} row(s) will be updated. Review the preview below and re-run with confirmUpdate: true to proceed.`,
+                    affectedRows,
+                    preview: previewResult.recordset,
+                    updates,
+                    error: "CONFIRMATION_REQUIRED",
+                };
+            }
+            // Step 3: Execute the update
+            const request = new sql.Request(params.pool);
+            // Build SET clause with parameterized queries for security
+            const setClause = Object.keys(updates)
+                .map((key, index) => {
+                const paramName = `update_${index}`;
+                request.input(paramName, updates[key]);
+                return `[${key}] = @${paramName}`;
+            })
+                .join(", ");
+            query = `UPDATE ${tableName} SET ${setClause} WHERE ${whereClause}`;
+            const result = await request.query(query);
+            return {
+                success: true,
+                message: `Successfully updated ${result.rowsAffected[0]} row(s) in table '${tableName}'`,
+                rowsAffected: result.rowsAffected[0],
+                updates,
+            };
+        }
+        catch (error) {
+            console.error("Error updating data:", error);
+            return {
+                success: false,
+                message: `Failed to update data${query ? ` with '${query}'` : ''}: ${error}`,
+                error: "UPDATE_FAILED",
+            };
+        }
+    }
+}
+UpdateDataTool.MAX_ROWS_DEFAULT = 1000;
+//# sourceMappingURL=UpdateDataTool.js.map

package/dist/tools/UpdateDataTool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"UpdateDataTool.js","sourceRoot":"","sources":["../../src/tools/UpdateDataTool.ts"],"names":[],"mappings":"AAAA,OAAO,GAAG,MAAM,OAAO,CAAC;AAGxB,MAAM,OAAO,cAAc;IAA3B;QAEE,SAAI,GAAG,aAAa,CAAC;QACrB,gBAAW,GAAG,iGAAiG,CAAC;QAChH,gBAAW,GAAG;YACZ,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,SAAS,EAAE;oBACT,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,6BAA6B;iBAC3C;gBACD,OAAO,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,qGAAqG;iBACnH;gBACD,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,kHAAkH;iBAChI;gBACD,aAAa,EAAE;oBACb,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,sGAAsG;iBACpH;gBACD,OAAO,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,wEAAwE;iBACtF;gBACD,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,qCAAqC;iBACnD;aACF;YACD,QAAQ,EAAE,CAAC,WAAW,EAAE,SAAS,EAAE,aAAa,CAAC;SAC3C,CAAC;IA4FX,CAAC;IAxFC,KAAK,CAAC,GAAG,CAAC,MAAW;QACnB,IAAI,KAAyB,CAAC;QAC9B,IAAI,CAAC;YACH,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;YAExF,oDAAoD;YACpD,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;gBAC9C,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,OAAO,EAAE,sGAAsG;oBAC/G,KAAK,EAAE,sBAAsB;iBAC9B,CAAC;YACJ,CAAC;YAED,MAAM,UAAU,GAAG,OAAO,IAAI,cAAc,CAAC,gBAAgB,CAAC;YAE9D,qCAAqC;YACrC,MAAM,UAAU,GAAG,wCAAwC,SAAS,UAAU,WAAW,EAAE,CAAC;YAC5F,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,WAAW,GAAG,MAAM,YAAY,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YACzD,MAAM,YAAY,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC;YAE3D,IAAI,YAAY,KAAK,CAAC,EAAE,CAAC;gBACvB,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,OAAO,EAAE,8DAA8D;oBACvE,KAAK,EAAE,iBAAiB;oBACxB,YAAY,EAAE,CAAC;iBAChB,CAAC;YACJ,CAAC;YAED,IAAI,YAAY,GAAG,UAAU,EAAE,CAAC;gBAC9B,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,OAAO,EAAE,uBAAuB,YAAY,uCAAuC,UAAU,2DAA2D;oBACxJ,KAAK,EAAE,eAAe;oBACtB,YAAY;oBACZ,UAAU;iBACX,CAAC;YACJ,CAAC;YAED,wCAAwC;YACxC,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,MAAM,YAAY,GAAG,wBAAwB,SAAS,UAAU,WAAW,EAAE,CAAC;gBAC9E,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBACpD,MAAM,aAAa,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;gBAE/D,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,iBAAiB,EAAE,IAAI;oBACvB,OAAO,EAAE,YAAY,YAAY,mGAAmG;oBACpI,YAAY;oBACZ,OAAO,EAAE,aAAa,CAAC,SAAS;oBAChC,OAAO;oBACP,KAAK,EAAE,uBAAuB;iBAC/B,CAAC;YACJ,CAAC;YAED,6BAA6B;YAC7B,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAE7C,2DAA2D;YAC3D,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC;iBACnC,GAAG,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;gBAClB,MAAM,SAAS,GAAG,UAAU,KAAK,EAAE,CAAC;gBACpC,OAAO,CAAC,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;gBACvC,OAAO,IAAI,GAAG,QAAQ,SAAS,EAAE,CAAC;YACpC,CAAC,CAAC;iBACD,IAAI,CAAC,IAAI,CAAC,CAAC;YAEd,KAAK,GAAG,UAAU,SAAS,QAAQ,SAAS,UAAU,WAAW,EAAE,CAAC;YACpE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAE1C,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,wBAAwB,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,qBAAqB,SAAS,GAAG;gBACxF,YAAY,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC;gBACpC,OAAO;aACR,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;YAC7C,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,wBAAwB,KAAK,CAAC,CAAC,CAAC,UAAU,KAAK,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,KAAK,EAAE;gBAC5E,KAAK,EAAE,eAAe;aACvB,CAAC;QACJ,CAAC;IACH,CAAC;;AAzFuB,+BAAgB,GAAG,IAAI,AAAP,CAAQ"}