@tejasanik/postgres-mcp-server 2.1.0 → 2.2.0

package/dist/tools/sql/utils/connection-utils.js

@@ -0,0 +1,129 @@
+/**
+ * Connection Utilities
+ *
+ * Helper functions for connection management and override handling.
+ * Centralizes connection override logic to reduce duplication.
+ */
+import { getDbManager } from '../../../db-manager.js';
+/**
+ * Build a ConnectionOverride object from optional parameters.
+ * Returns undefined if no override parameters are provided.
+ *
+ * @param params - Optional connection override parameters
+ * @returns ConnectionOverride object or undefined
+ */
+export function buildConnectionOverride(params) {
+    const hasOverride = params.server || params.database || params.schema;
+    if (!hasOverride) {
+        return undefined;
+    }
+    return {
+        server: params.server,
+        database: params.database,
+        schema: params.schema,
+    };
+}
+/**
+ * Check if connection override parameters conflict with transaction usage.
+ *
+ * @param params - Connection override parameters
+ * @param transactionId - Transaction ID if any
+ * @throws Error if override is used with transaction
+ */
+export function validateOverrideWithTransaction(params, transactionId) {
+    const hasOverride = params.server || params.database || params.schema;
+    if (hasOverride && transactionId) {
+        throw new Error('Connection override (server/database/schema) cannot be used with transactions. ' +
+            'Transactions are bound to the main connection.');
+    }
+}
+/**
+ * Acquire a client with optional connection override.
+ * Handles both regular and override client acquisition.
+ *
+ * @param override - Optional connection override
+ * @returns Acquired client with release function
+ */
+export async function acquireClient(override) {
+    const dbManager = getDbManager();
+    if (override) {
+        const result = await dbManager.getClientWithOverride(override);
+        return {
+            client: result.client,
+            release: result.release,
+            isOverride: true,
+            connectionInfo: {
+                server: result.server,
+                database: result.database,
+                schema: result.schema,
+            },
+        };
+    }
+    const client = await dbManager.getClient();
+    return {
+        client,
+        release: () => client.release(),
+        isOverride: false,
+    };
+}
+/**
+ * Execute a function with an acquired client, ensuring proper cleanup.
+ *
+ * @param override - Optional connection override
+ * @param fn - Function to execute with the client
+ * @returns Result of the function
+ */
+export async function withClient(override, fn) {
+    const { client, release } = await acquireClient(override);
+    try {
+        return await fn(client);
+    }
+    finally {
+        release();
+    }
+}
+/**
+ * Execute a function with a transaction, ensuring proper rollback on error.
+ *
+ * @param override - Optional connection override
+ * @param fn - Function to execute within transaction
+ * @param rollbackOnComplete - If true, rollback even on success (for dry-run)
+ * @returns Result of the function
+ */
+export async function withTransaction(override, fn, rollbackOnComplete = false) {
+    const { client, release } = await acquireClient(override);
+    try {
+        await client.query('BEGIN');
+        const result = await fn(client);
+        if (rollbackOnComplete) {
+            await client.query('ROLLBACK');
+        }
+        else {
+            await client.query('COMMIT');
+        }
+        return result;
+    }
+    catch (error) {
+        try {
+            await client.query('ROLLBACK');
+        }
+        catch {
+            // Ignore rollback errors
+        }
+        throw error;
+    }
+    finally {
+        release();
+    }
+}
+/**
+ * Execute a dry-run transaction (always rollback).
+ *
+ * @param override - Optional connection override
+ * @param fn - Function to execute within transaction
+ * @returns Result of the function
+ */
+export async function withDryRunTransaction(override, fn) {
+    return withTransaction(override, fn, true);
+}
+//# sourceMappingURL=connection-utils.js.map

package/dist/tools/sql/utils/connection-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"connection-utils.js","sourceRoot":"","sources":["../../../../src/tools/sql/utils/connection-utils.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AA2BtD;;;;;;GAMG;AACH,MAAM,UAAU,uBAAuB,CACrC,MAAgC;IAEhC,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,MAAM,CAAC;IACtE,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO;QACL,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,MAAM,EAAE,MAAM,CAAC,MAAM;KACtB,CAAC;AACJ,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,+BAA+B,CAC7C,MAAgC,EAChC,aAAsB;IAEtB,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,MAAM,CAAC;IACtE,IAAI,WAAW,IAAI,aAAa,EAAE,CAAC;QACjC,MAAM,IAAI,KAAK,CACb,iFAAiF;YACjF,gDAAgD,CACjD,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,QAA6B;IAE7B,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;IAEjC,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;QAC/D,OAAO;YACL,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,UAAU,EAAE,IAAI;YAChB,cAAc,EAAE;gBACd,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,MAAM,EAAE,MAAM,CAAC,MAAM;aACtB;SACF,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,SAAS,EAAE,CAAC;IAC3C,OAAO;QACL,MAAM;QACN,OAAO,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE;QAC/B,UAAU,EAAE,KAAK;KAClB,CAAC;AACJ,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,QAAwC,EACxC,EAAsC;IAEtC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,aAAa,CAAC,QAAQ,CAAC,CAAC;IAC1D,IAAI,CAAC;QACH,OAAO,MAAM,EAAE,CAAC,MAAM,CAAC,CAAC;IAC1B,CAAC;YAAS,CAAC;QACT,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,QAAwC,EACxC,EAAsC,EACtC,qBAA8B,KAAK;IAEnC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,aAAa,CAAC,QAAQ,CAAC,CAAC;IAC1D,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC5B,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,MAAM,CAAC,CAAC;QAChC,IAAI,kBAAkB,EAAE,CAAC;YACvB,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QACjC,CAAC;aAAM,CAAC;YACN,MAAM,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QACjC,CAAC;QAAC,MAAM,CAAC;YACP,yBAAyB;QAC3B,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;YAAS,CAAC;QACT,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,QAAwC,EACxC,EAAsC;IAEtC,OAAO,eAAe,CAAC,QAAQ,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC;AAC7C,CAAC"}

package/dist/tools/sql/utils/constants.d.ts

@@ -0,0 +1,55 @@
+/**
+ * SQL Tools Constants
+ *
+ * Centralized configuration values for SQL execution tools.
+ * All magic numbers are defined here for maintainability.
+ */
+/** Maximum characters in response before writing to temp file */
+export declare const MAX_OUTPUT_CHARS = 50000;
+/** Default maximum rows to return in direct response */
+export declare const MAX_ROWS_DEFAULT = 1000;
+/** Absolute maximum rows allowed */
+export declare const MAX_ROWS_LIMIT = 100000;
+/** Default SQL query length limit (100KB) */
+export declare const DEFAULT_SQL_LENGTH_LIMIT = 100000;
+/** Maximum number of query parameters */
+export declare const MAX_PARAMS = 100;
+/** Maximum SQL file size (50MB) */
+export declare const MAX_SQL_FILE_SIZE: number;
+/** Maximum sample rows to return in dry-run */
+export declare const MAX_DRY_RUN_SAMPLE_ROWS = 10;
+/** Maximum statements to preview in file preview */
+export declare const MAX_PREVIEW_STATEMENTS = 100;
+/** Default statements to show in preview */
+export declare const DEFAULT_PREVIEW_STATEMENTS = 20;
+/** Maximum batch queries in single call */
+export declare const MAX_BATCH_QUERIES = 20;
+/** Maximum sample size for mutation preview */
+export declare const MAX_MUTATION_SAMPLE_SIZE = 20;
+/** Default sample size for mutation preview */
+export declare const DEFAULT_MUTATION_SAMPLE_SIZE = 5;
+/** Transaction timeout in milliseconds (10 minutes for active operations) */
+export declare const TRANSACTION_TIMEOUT_MS: number;
+/** Transaction auto-cleanup timeout in milliseconds (45 minutes) */
+export declare const TRANSACTION_CLEANUP_TIMEOUT_MS: number;
+/** Transaction cleanup check interval in milliseconds (5 minutes) */
+export declare const TRANSACTION_CLEANUP_INTERVAL_MS: number;
+/** Maximum SQL length for ReDoS-safe regex operations (100KB) */
+export declare const MAX_SQL_LENGTH_FOR_REGEX = 100000;
+/** Maximum hypothetical indexes per query */
+export declare const MAX_HYPOTHETICAL_INDEXES = 10;
+/** Maximum columns per hypothetical index */
+export declare const MAX_HYPOTHETICAL_INDEX_COLUMNS = 10;
+/** Maximum statements in dry-run file results */
+export declare const MAX_DRY_RUN_STATEMENTS = 200;
+/** Default statements in dry-run file results */
+export declare const DEFAULT_DRY_RUN_STATEMENTS = 50;
+/** Maximum rows per statement in multi-statement execution */
+export declare const MAX_ROWS_PER_STATEMENT = 100;
+/** Short SQL truncation limit for display */
+export declare const SQL_TRUNCATION_SHORT = 200;
+/** Long SQL truncation limit for detailed display */
+export declare const SQL_TRUNCATION_LONG = 300;
+/** Maximum tables to analyze for schema hints */
+export declare const MAX_TABLES_TO_ANALYZE = 10;
+//# sourceMappingURL=constants.d.ts.map

package/dist/tools/sql/utils/constants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../../../../src/tools/sql/utils/constants.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,iEAAiE;AACjE,eAAO,MAAM,gBAAgB,QAAQ,CAAC;AAEtC,wDAAwD;AACxD,eAAO,MAAM,gBAAgB,OAAO,CAAC;AAErC,oCAAoC;AACpC,eAAO,MAAM,cAAc,SAAS,CAAC;AAErC,6CAA6C;AAC7C,eAAO,MAAM,wBAAwB,SAAS,CAAC;AAE/C,yCAAyC;AACzC,eAAO,MAAM,UAAU,MAAM,CAAC;AAE9B,mCAAmC;AACnC,eAAO,MAAM,iBAAiB,QAAmB,CAAC;AAElD,+CAA+C;AAC/C,eAAO,MAAM,uBAAuB,KAAK,CAAC;AAE1C,oDAAoD;AACpD,eAAO,MAAM,sBAAsB,MAAM,CAAC;AAE1C,4CAA4C;AAC5C,eAAO,MAAM,0BAA0B,KAAK,CAAC;AAE7C,2CAA2C;AAC3C,eAAO,MAAM,iBAAiB,KAAK,CAAC;AAEpC,+CAA+C;AAC/C,eAAO,MAAM,wBAAwB,KAAK,CAAC;AAE3C,+CAA+C;AAC/C,eAAO,MAAM,4BAA4B,IAAI,CAAC;AAE9C,6EAA6E;AAC7E,eAAO,MAAM,sBAAsB,QAAiB,CAAC;AAErD,oEAAoE;AACpE,eAAO,MAAM,8BAA8B,QAAiB,CAAC;AAE7D,qEAAqE;AACrE,eAAO,MAAM,+BAA+B,QAAgB,CAAC;AAE7D,iEAAiE;AACjE,eAAO,MAAM,wBAAwB,SAAS,CAAC;AAE/C,6CAA6C;AAC7C,eAAO,MAAM,wBAAwB,KAAK,CAAC;AAE3C,6CAA6C;AAC7C,eAAO,MAAM,8BAA8B,KAAK,CAAC;AAEjD,iDAAiD;AACjD,eAAO,MAAM,sBAAsB,MAAM,CAAC;AAE1C,iDAAiD;AACjD,eAAO,MAAM,0BAA0B,KAAK,CAAC;AAE7C,8DAA8D;AAC9D,eAAO,MAAM,sBAAsB,MAAM,CAAC;AAE1C,6CAA6C;AAC7C,eAAO,MAAM,oBAAoB,MAAM,CAAC;AAExC,qDAAqD;AACrD,eAAO,MAAM,mBAAmB,MAAM,CAAC;AAEvC,iDAAiD;AACjD,eAAO,MAAM,qBAAqB,KAAK,CAAC"}

package/dist/tools/sql/utils/constants.js

@@ -0,0 +1,55 @@
+/**
+ * SQL Tools Constants
+ *
+ * Centralized configuration values for SQL execution tools.
+ * All magic numbers are defined here for maintainability.
+ */
+/** Maximum characters in response before writing to temp file */
+export const MAX_OUTPUT_CHARS = 50000;
+/** Default maximum rows to return in direct response */
+export const MAX_ROWS_DEFAULT = 1000;
+/** Absolute maximum rows allowed */
+export const MAX_ROWS_LIMIT = 100000;
+/** Default SQL query length limit (100KB) */
+export const DEFAULT_SQL_LENGTH_LIMIT = 100000;
+/** Maximum number of query parameters */
+export const MAX_PARAMS = 100;
+/** Maximum SQL file size (50MB) */
+export const MAX_SQL_FILE_SIZE = 50 * 1024 * 1024;
+/** Maximum sample rows to return in dry-run */
+export const MAX_DRY_RUN_SAMPLE_ROWS = 10;
+/** Maximum statements to preview in file preview */
+export const MAX_PREVIEW_STATEMENTS = 100;
+/** Default statements to show in preview */
+export const DEFAULT_PREVIEW_STATEMENTS = 20;
+/** Maximum batch queries in single call */
+export const MAX_BATCH_QUERIES = 20;
+/** Maximum sample size for mutation preview */
+export const MAX_MUTATION_SAMPLE_SIZE = 20;
+/** Default sample size for mutation preview */
+export const DEFAULT_MUTATION_SAMPLE_SIZE = 5;
+/** Transaction timeout in milliseconds (10 minutes for active operations) */
+export const TRANSACTION_TIMEOUT_MS = 10 * 60 * 1000;
+/** Transaction auto-cleanup timeout in milliseconds (45 minutes) */
+export const TRANSACTION_CLEANUP_TIMEOUT_MS = 45 * 60 * 1000;
+/** Transaction cleanup check interval in milliseconds (5 minutes) */
+export const TRANSACTION_CLEANUP_INTERVAL_MS = 5 * 60 * 1000;
+/** Maximum SQL length for ReDoS-safe regex operations (100KB) */
+export const MAX_SQL_LENGTH_FOR_REGEX = 100000;
+/** Maximum hypothetical indexes per query */
+export const MAX_HYPOTHETICAL_INDEXES = 10;
+/** Maximum columns per hypothetical index */
+export const MAX_HYPOTHETICAL_INDEX_COLUMNS = 10;
+/** Maximum statements in dry-run file results */
+export const MAX_DRY_RUN_STATEMENTS = 200;
+/** Default statements in dry-run file results */
+export const DEFAULT_DRY_RUN_STATEMENTS = 50;
+/** Maximum rows per statement in multi-statement execution */
+export const MAX_ROWS_PER_STATEMENT = 100;
+/** Short SQL truncation limit for display */
+export const SQL_TRUNCATION_SHORT = 200;
+/** Long SQL truncation limit for detailed display */
+export const SQL_TRUNCATION_LONG = 300;
+/** Maximum tables to analyze for schema hints */
+export const MAX_TABLES_TO_ANALYZE = 10;
+//# sourceMappingURL=constants.js.map

package/dist/tools/sql/utils/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../../src/tools/sql/utils/constants.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,iEAAiE;AACjE,MAAM,CAAC,MAAM,gBAAgB,GAAG,KAAK,CAAC;AAEtC,wDAAwD;AACxD,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,CAAC;AAErC,oCAAoC;AACpC,MAAM,CAAC,MAAM,cAAc,GAAG,MAAM,CAAC;AAErC,6CAA6C;AAC7C,MAAM,CAAC,MAAM,wBAAwB,GAAG,MAAM,CAAC;AAE/C,yCAAyC;AACzC,MAAM,CAAC,MAAM,UAAU,GAAG,GAAG,CAAC;AAE9B,mCAAmC;AACnC,MAAM,CAAC,MAAM,iBAAiB,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC;AAElD,+CAA+C;AAC/C,MAAM,CAAC,MAAM,uBAAuB,GAAG,EAAE,CAAC;AAE1C,oDAAoD;AACpD,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAAG,CAAC;AAE1C,4CAA4C;AAC5C,MAAM,CAAC,MAAM,0BAA0B,GAAG,EAAE,CAAC;AAE7C,2CAA2C;AAC3C,MAAM,CAAC,MAAM,iBAAiB,GAAG,EAAE,CAAC;AAEpC,+CAA+C;AAC/C,MAAM,CAAC,MAAM,wBAAwB,GAAG,EAAE,CAAC;AAE3C,+CAA+C;AAC/C,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAAC,CAAC;AAE9C,6EAA6E;AAC7E,MAAM,CAAC,MAAM,sBAAsB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AAErD,oEAAoE;AACpE,MAAM,CAAC,MAAM,8BAA8B,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AAE7D,qEAAqE;AACrE,MAAM,CAAC,MAAM,+BAA+B,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;AAE7D,iEAAiE;AACjE,MAAM,CAAC,MAAM,wBAAwB,GAAG,MAAM,CAAC;AAE/C,6CAA6C;AAC7C,MAAM,CAAC,MAAM,wBAAwB,GAAG,EAAE,CAAC;AAE3C,6CAA6C;AAC7C,MAAM,CAAC,MAAM,8BAA8B,GAAG,EAAE,CAAC;AAEjD,iDAAiD;AACjD,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAAG,CAAC;AAE1C,iDAAiD;AACjD,MAAM,CAAC,MAAM,0BAA0B,GAAG,EAAE,CAAC;AAE7C,8DAA8D;AAC9D,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAAG,CAAC;AAE1C,6CAA6C;AAC7C,MAAM,CAAC,MAAM,oBAAoB,GAAG,GAAG,CAAC;AAExC,qDAAqD;AACrD,MAAM,CAAC,MAAM,mBAAmB,GAAG,GAAG,CAAC;AAEvC,iDAAiD;AACjD,MAAM,CAAC,MAAM,qBAAqB,GAAG,EAAE,CAAC"}

package/dist/tools/sql/utils/dry-run-utils.d.ts

@@ -0,0 +1,31 @@
+/**
+ * Dry-Run Utilities
+ *
+ * Helper functions for transaction-based dry-run operations.
+ * Handles error extraction and detection of non-rollbackable operations.
+ */
+import { DryRunError, NonRollbackableWarning } from '../../../types.js';
+/**
+ * Extract detailed error information from a PostgreSQL error.
+ * Captures all available fields to help AI quickly identify and fix issues.
+ */
+export declare function extractDryRunError(error: unknown): DryRunError;
+/**
+ * Check if a SQL statement contains operations that cannot be fully rolled back
+ * or have side effects even within a transaction.
+ *
+ * @param sql - The SQL statement to check
+ * @param statementIndex - Optional statement index for error reporting
+ * @param lineNumber - Optional line number for error reporting
+ * @returns Array of warnings about non-rollbackable operations
+ */
+export declare function detectNonRollbackableOperations(sql: string, statementIndex?: number, lineNumber?: number): NonRollbackableWarning[];
+/**
+ * Check if any warning requires skipping the statement.
+ */
+export declare function hasMustSkipWarning(warnings: NonRollbackableWarning[]): boolean;
+/**
+ * Get skip reason from must-skip warnings.
+ */
+export declare function getSkipReason(warnings: NonRollbackableWarning[]): string;
+//# sourceMappingURL=dry-run-utils.d.ts.map

package/dist/tools/sql/utils/dry-run-utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"dry-run-utils.d.ts","sourceRoot":"","sources":["../../../../src/tools/sql/utils/dry-run-utils.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,WAAW,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAgFxE;;;GAGG;AACH,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,OAAO,GAAG,WAAW,CAsC9D;AAED;;;;;;;;GAQG;AACH,wBAAgB,+BAA+B,CAC7C,GAAG,EAAE,MAAM,EACX,cAAc,CAAC,EAAE,MAAM,EACvB,UAAU,CAAC,EAAE,MAAM,GAClB,sBAAsB,EAAE,CAkB1B;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,sBAAsB,EAAE,GAAG,OAAO,CAE9E;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,QAAQ,EAAE,sBAAsB,EAAE,GAAG,MAAM,CAKxE"}

package/dist/tools/sql/utils/dry-run-utils.js

@@ -0,0 +1,173 @@
+/**
+ * Dry-Run Utilities
+ *
+ * Helper functions for transaction-based dry-run operations.
+ * Handles error extraction and detection of non-rollbackable operations.
+ */
+/**
+ * Non-rollbackable operation patterns.
+ * Operations that cannot run inside a transaction or have permanent side effects.
+ */
+const NON_ROLLBACKABLE_PATTERNS = [
+    // Operations that cannot run inside a transaction at all
+    {
+        pattern: /\bVACUUM\b/,
+        operation: 'VACUUM',
+        message: 'VACUUM cannot run inside a transaction block. Statement skipped.',
+        mustSkip: true,
+    },
+    {
+        pattern: /\bCLUSTER\b(?!.*CREATE)/,
+        operation: 'CLUSTER',
+        message: 'CLUSTER cannot run inside a transaction block. Statement skipped.',
+        mustSkip: true,
+    },
+    {
+        pattern: /\bREINDEX\b.*\bCONCURRENTLY\b/,
+        operation: 'REINDEX_CONCURRENTLY',
+        message: 'REINDEX CONCURRENTLY cannot run inside a transaction block. Statement skipped.',
+        mustSkip: true,
+    },
+    {
+        pattern: /\bCREATE\s+INDEX\b.*\bCONCURRENTLY\b/,
+        operation: 'CREATE_INDEX_CONCURRENTLY',
+        message: 'CREATE INDEX CONCURRENTLY cannot run inside a transaction block. Statement skipped.',
+        mustSkip: true,
+    },
+    {
+        pattern: /\bCREATE\s+DATABASE\b/,
+        operation: 'CREATE_DATABASE',
+        message: 'CREATE DATABASE cannot run inside a transaction block. Statement skipped.',
+        mustSkip: true,
+    },
+    {
+        pattern: /\bDROP\s+DATABASE\b/,
+        operation: 'DROP_DATABASE',
+        message: 'DROP DATABASE cannot run inside a transaction block. Statement skipped.',
+        mustSkip: true,
+    },
+    // Operations with permanent side effects
+    {
+        pattern: /\bNEXTVAL\s*\(/,
+        operation: 'SEQUENCE',
+        message: 'NEXTVAL increments sequence even when transaction is rolled back. Statement skipped to prevent sequence consumption.',
+        mustSkip: true,
+    },
+    {
+        pattern: /\bSETVAL\s*\(/,
+        operation: 'SEQUENCE',
+        message: 'SETVAL modifies sequence. Statement skipped to prevent side effects.',
+        mustSkip: true,
+    },
+    // Warning-only operations (still executed)
+    {
+        pattern: /\bINSERT\s+INTO\b/,
+        operation: 'SEQUENCE',
+        message: 'INSERT may consume sequence values (for SERIAL/BIGSERIAL columns) even when rolled back.',
+        mustSkip: false,
+    },
+    {
+        pattern: /\bNOTIFY\b/,
+        operation: 'NOTIFY',
+        message: 'NOTIFY sends notifications on commit. Since dry-run rolls back, notifications will NOT be sent.',
+        mustSkip: false,
+    },
+];
+/**
+ * Extract detailed error information from a PostgreSQL error.
+ * Captures all available fields to help AI quickly identify and fix issues.
+ */
+export function extractDryRunError(error) {
+    // Extract message - prioritize Error.message, then object.message, then String conversion
+    let message;
+    if (error instanceof Error) {
+        message = error.message;
+    }
+    else if (error && typeof error === 'object' && 'message' in error) {
+        message = String(error.message);
+    }
+    else {
+        message = String(error);
+    }
+    const result = { message };
+    if (error && typeof error === 'object') {
+        const pgError = error;
+        // Extract string fields
+        if (pgError.code)
+            result.code = String(pgError.code);
+        if (pgError.severity)
+            result.severity = String(pgError.severity);
+        if (pgError.detail)
+            result.detail = String(pgError.detail);
+        if (pgError.hint)
+            result.hint = String(pgError.hint);
+        if (pgError.internalQuery)
+            result.internalQuery = String(pgError.internalQuery);
+        if (pgError.where)
+            result.where = String(pgError.where);
+        if (pgError.schema)
+            result.schema = String(pgError.schema);
+        if (pgError.table)
+            result.table = String(pgError.table);
+        if (pgError.column)
+            result.column = String(pgError.column);
+        if (pgError.dataType)
+            result.dataType = String(pgError.dataType);
+        if (pgError.constraint)
+            result.constraint = String(pgError.constraint);
+        if (pgError.file)
+            result.file = String(pgError.file);
+        if (pgError.line)
+            result.line = String(pgError.line);
+        if (pgError.routine)
+            result.routine = String(pgError.routine);
+        // Extract number fields
+        if (pgError.position !== undefined)
+            result.position = Number(pgError.position);
+        if (pgError.internalPosition !== undefined)
+            result.internalPosition = Number(pgError.internalPosition);
+    }
+    return result;
+}
+/**
+ * Check if a SQL statement contains operations that cannot be fully rolled back
+ * or have side effects even within a transaction.
+ *
+ * @param sql - The SQL statement to check
+ * @param statementIndex - Optional statement index for error reporting
+ * @param lineNumber - Optional line number for error reporting
+ * @returns Array of warnings about non-rollbackable operations
+ */
+export function detectNonRollbackableOperations(sql, statementIndex, lineNumber) {
+    const warnings = [];
+    const upperSql = sql.toUpperCase().trim();
+    const clusterPattern = /\bCLUSTER\b/;
+    for (const { pattern, operation, message, mustSkip } of NON_ROLLBACKABLE_PATTERNS) {
+        // Special handling for CLUSTER (must not be part of CREATE)
+        if (operation === 'CLUSTER') {
+            if (clusterPattern.test(upperSql) && !upperSql.includes('CREATE')) {
+                warnings.push({ operation, message, statementIndex, lineNumber, mustSkip });
+            }
+        }
+        else if (pattern.test(upperSql)) {
+            warnings.push({ operation, message, statementIndex, lineNumber, mustSkip });
+        }
+    }
+    return warnings;
+}
+/**
+ * Check if any warning requires skipping the statement.
+ */
+export function hasMustSkipWarning(warnings) {
+    return warnings.some((w) => w.mustSkip);
+}
+/**
+ * Get skip reason from must-skip warnings.
+ */
+export function getSkipReason(warnings) {
+    return warnings
+        .filter((w) => w.mustSkip)
+        .map((w) => w.message)
+        .join('; ');
+}
+//# sourceMappingURL=dry-run-utils.js.map

package/dist/tools/sql/utils/dry-run-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"dry-run-utils.js","sourceRoot":"","sources":["../../../../src/tools/sql/utils/dry-run-utils.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAOH;;;GAGG;AACH,MAAM,yBAAyB,GAK1B;IACH,yDAAyD;IACzD;QACE,OAAO,EAAE,YAAY;QACrB,SAAS,EAAE,QAAQ;QACnB,OAAO,EAAE,kEAAkE;QAC3E,QAAQ,EAAE,IAAI;KACf;IACD;QACE,OAAO,EAAE,yBAAyB;QAClC,SAAS,EAAE,SAAS;QACpB,OAAO,EAAE,mEAAmE;QAC5E,QAAQ,EAAE,IAAI;KACf;IACD;QACE,OAAO,EAAE,+BAA+B;QACxC,SAAS,EAAE,sBAAsB;QACjC,OAAO,EAAE,gFAAgF;QACzF,QAAQ,EAAE,IAAI;KACf;IACD;QACE,OAAO,EAAE,sCAAsC;QAC/C,SAAS,EAAE,2BAA2B;QACtC,OAAO,EAAE,qFAAqF;QAC9F,QAAQ,EAAE,IAAI;KACf;IACD;QACE,OAAO,EAAE,uBAAuB;QAChC,SAAS,EAAE,iBAAiB;QAC5B,OAAO,EAAE,2EAA2E;QACpF,QAAQ,EAAE,IAAI;KACf;IACD;QACE,OAAO,EAAE,qBAAqB;QAC9B,SAAS,EAAE,eAAe;QAC1B,OAAO,EAAE,yEAAyE;QAClF,QAAQ,EAAE,IAAI;KACf;IACD,yCAAyC;IACzC;QACE,OAAO,EAAE,gBAAgB;QACzB,SAAS,EAAE,UAAU;QACrB,OAAO,EAAE,sHAAsH;QAC/H,QAAQ,EAAE,IAAI;KACf;IACD;QACE,OAAO,EAAE,eAAe;QACxB,SAAS,EAAE,UAAU;QACrB,OAAO,EAAE,sEAAsE;QAC/E,QAAQ,EAAE,IAAI;KACf;IACD,2CAA2C;IAC3C;QACE,OAAO,EAAE,mBAAmB;QAC5B,SAAS,EAAE,UAAU;QACrB,OAAO,EAAE,0FAA0F;QACnG,QAAQ,EAAE,KAAK;KAChB;IACD;QACE,OAAO,EAAE,YAAY;QACrB,SAAS,EAAE,QAAQ;QACnB,OAAO,EAAE,iGAAiG;QAC1G,QAAQ,EAAE,KAAK;KAChB;CACF,CAAC;AAEF;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAAc;IAC/C,0FAA0F;IAC1F,IAAI,OAAe,CAAC;IACpB,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;IAC1B,CAAC;SAAM,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,SAAS,IAAI,KAAK,EAAE,CAAC;QACpE,OAAO,GAAG,MAAM,CAAE,KAA8B,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;IAED,MAAM,MAAM,GAAgB,EAAE,OAAO,EAAE,CAAC;IAExC,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACvC,MAAM,OAAO,GAAG,KAAgC,CAAC;QAEjD,wBAAwB;QACxB,IAAI,OAAO,CAAC,IAAI;YAAE,MAAM,CAAC,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACrD,IAAI,OAAO,CAAC,QAAQ;YAAE,MAAM,CAAC,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACjE,IAAI,OAAO,CAAC,MAAM;YAAE,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC3D,IAAI,OAAO,CAAC,IAAI;YAAE,MAAM,CAAC,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACrD,IAAI,OAAO,CAAC,aAAa;YAAE,MAAM,CAAC,aAAa,GAAG,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAChF,IAAI,OAAO,CAAC,KAAK;YAAE,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACxD,IAAI,OAAO,CAAC,MAAM;YAAE,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC3D,IAAI,OAAO,CAAC,KAAK;YAAE,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACxD,IAAI,OAAO,CAAC,MAAM;YAAE,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC3D,IAAI,OAAO,CAAC,QAAQ;YAAE,MAAM,CAAC,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACjE,IAAI,OAAO,CAAC,UAAU;YAAE,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACvE,IAAI,OAAO,CAAC,IAAI;YAAE,MAAM,CAAC,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACrD,IAAI,OAAO,CAAC,IAAI;YAAE,MAAM,CAAC,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACrD,IAAI,OAAO,CAAC,OAAO;YAAE,MAAM,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAE9D,wBAAwB;QACxB,IAAI,OAAO,CAAC,QAAQ,KAAK,SAAS;YAAE,MAAM,CAAC,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC/E,IAAI,OAAO,CAAC,gBAAgB,KAAK,SAAS;YAAE,MAAM,CAAC,gBAAgB,GAAG,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IACzG,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,+BAA+B,CAC7C,GAAW,EACX,cAAuB,EACvB,UAAmB;IAEnB,MAAM,QAAQ,GAA6B,EAAE,CAAC;IAC9C,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;IAE1C,MAAM,cAAc,GAAG,aAAa,CAAC;IAErC,KAAK,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,yBAAyB,EAAE,CAAC;QAClF,4DAA4D;QAC5D,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC5B,IAAI,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAClE,QAAQ,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC9E,CAAC;QACH,CAAC;aAAM,IAAI,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YAClC,QAAQ,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC9E,CAAC;IACH,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,QAAkC;IACnE,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;AAC1C,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa,CAAC,QAAkC;IAC9D,OAAO,QAAQ;SACZ,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC;SACzB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;SACrB,IAAI,CAAC,IAAI,CAAC,CAAC;AAChB,CAAC"}

package/dist/tools/sql/utils/file-handler.d.ts

@@ -0,0 +1,57 @@
+/**
+ * File Handler Utilities
+ *
+ * Functions for validating and processing SQL files.
+ * Handles file validation, reading, and preprocessing.
+ */
+/**
+ * Result of file validation.
+ */
+export interface FileValidationResult {
+    isValid: boolean;
+    resolvedPath: string;
+    fileSize: number;
+    error?: string;
+}
+/**
+ * Validates a SQL file path and returns file information.
+ *
+ * @param filePath - Path to the SQL file
+ * @returns Validation result with resolved path and file size
+ */
+export declare function validateSqlFile(filePath: string): FileValidationResult;
+/**
+ * Reads a SQL file and optionally preprocesses it.
+ *
+ * @param resolvedPath - Resolved path to the SQL file
+ * @param stripPatterns - Optional patterns to remove from content
+ * @param stripAsRegex - If true, patterns are regex; if false, literal strings
+ * @returns Preprocessed SQL content
+ */
+export declare function readSqlFile(resolvedPath: string, stripPatterns?: string[], stripAsRegex?: boolean): string;
+/**
+ * Preprocess SQL content by removing patterns.
+ * Supports both literal string matching and regex patterns.
+ *
+ * @param sql - The SQL content to preprocess
+ * @param patterns - Array of patterns to remove from SQL content
+ * @param isRegex - If true, patterns are treated as regex; if false, as literal strings
+ * @returns Preprocessed SQL content
+ */
+export declare function preprocessSqlContent(sql: string, patterns: string[], isRegex?: boolean): string;
+/**
+ * Format file size in human-readable format.
+ *
+ * @param bytes - File size in bytes
+ * @returns Human-readable file size (e.g., "1.5 MB")
+ */
+export declare function formatFileSize(bytes: number): string;
+/**
+ * Ensure a file path is safe (no path traversal attacks).
+ *
+ * @param basePath - Base directory path
+ * @param filePath - File path to validate
+ * @returns True if the file path is within the base path
+ */
+export declare function isPathSafe(basePath: string, filePath: string): boolean;
+//# sourceMappingURL=file-handler.d.ts.map

package/dist/tools/sql/utils/file-handler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"file-handler.d.ts","sourceRoot":"","sources":["../../../../src/tools/sql/utils/file-handler.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAMH;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,OAAO,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;;;;GAKG;AACH,wBAAgB,eAAe,CAAC,QAAQ,EAAE,MAAM,GAAG,oBAAoB,CA+DtE;AAED;;;;;;;GAOG;AACH,wBAAgB,WAAW,CACzB,YAAY,EAAE,MAAM,EACpB,aAAa,CAAC,EAAE,MAAM,EAAE,EACxB,YAAY,GAAE,OAAe,GAC5B,MAAM,CAQR;AAED;;;;;;;;GAQG;AACH,wBAAgB,oBAAoB,CAClC,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE,MAAM,EAAE,EAClB,OAAO,GAAE,OAAe,GACvB,MAAM,CA2BR;AAED;;;;;GAKG;AACH,wBAAgB,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAYpD;AAED;;;;;;GAMG;AACH,wBAAgB,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAItE"}