rawsql-ts 0.12.0 → 0.12.1

package/dist/transformers/DDLDiffGenerator.js

@@ -0,0 +1,313 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DDLDiffGenerator = void 0;
+const SqlParser_1 = require("../parsers/SqlParser");
+const SqlFormatter_1 = require("./SqlFormatter");
+const DDLGeneralizer_1 = require("./DDLGeneralizer");
+const MultiQuerySplitter_1 = require("../utils/MultiQuerySplitter");
+const CreateTableQuery_1 = require("../models/CreateTableQuery");
+const DDLStatements_1 = require("../models/DDLStatements");
+const ValueComponent_1 = require("../models/ValueComponent");
+class DDLDiffGenerator {
+    static generateDiff(currentSql, expectedSql, options = {}) {
+        const currentAst = this.parseAndGeneralize(currentSql);
+        const expectedAst = this.parseAndGeneralize(expectedSql);
+        const currentSchema = this.buildSchema(currentAst);
+        const expectedSchema = this.buildSchema(expectedAst);
+        const diffAsts = [];
+        // Compare Tables
+        for (const [tableName, expectedTable] of expectedSchema.tables) {
+            const currentTable = currentSchema.tables.get(tableName);
+            if (!currentTable) {
+                // Table missing in current -> Create it
+                // We reconstruct the CreateTableQuery from columns
+                const columns = Array.from(expectedTable.columns.values()).map(c => c.definition);
+                const tableNameStr = expectedTable.qualifiedName.name instanceof ValueComponent_1.RawString
+                    ? expectedTable.qualifiedName.name.value
+                    : expectedTable.qualifiedName.name.name;
+                const namespaces = expectedTable.qualifiedName.namespaces
+                    ? expectedTable.qualifiedName.namespaces.map(ns => ns.name)
+                    : null;
+                const createTable = new CreateTableQuery_1.CreateTableQuery({
+                    tableName: tableNameStr,
+                    namespaces: namespaces,
+                    columns: columns
+                });
+                diffAsts.push(createTable);
+                // And add constraints
+                for (const constraint of expectedTable.constraints) {
+                    diffAsts.push(new DDLStatements_1.AlterTableStatement({
+                        table: expectedTable.qualifiedName,
+                        actions: [new DDLStatements_1.AlterTableAddConstraint({ constraint: constraint.definition })]
+                    }));
+                }
+                // And add indexes
+                for (const index of expectedTable.indexes) {
+                    diffAsts.push(index.definition);
+                }
+            }
+            else {
+                // Table exists -> Compare columns and constraints
+                this.compareColumns(currentTable, expectedTable, diffAsts, options);
+                this.compareConstraints(currentTable, expectedTable, diffAsts, options);
+                this.compareIndexes(currentTable, expectedTable, diffAsts, options);
+            }
+        }
+        // Drop Tables (if enabled)
+        if (options.dropTables) {
+            for (const [tableName, currentTable] of currentSchema.tables) {
+                if (!expectedSchema.tables.has(tableName)) {
+                    // Table exists in current but not in expected -> Drop it
+                    // We need a DropTableStatement. For now, we can manually construct the SQL or add DropTableStatement model.
+                    // Since we return string[], we can just push a raw SQL string if we don't have the AST model yet,
+                    // OR better, let's use a simple object that formats to DROP TABLE.
+                    // But wait, the return type is string[] derived from ASTs.
+                    // Let's assume we can use a raw SQL component or similar.
+                    // Actually, let's just use a simple custom AST node or formatted string injection if possible.
+                    // Looking at imports, we don't have DropTableStatement.
+                    // Let's add a temporary workaround or just return the string directly?
+                    // The method returns string[] by mapping diffAsts.
+                    // We should add a DropTableStatement class or similar.
+                    // For now, let's just push a dummy component that formats to DROP TABLE.
+                    // Actually, let's check if we can import DropTableStatement.
+                    // It seems it's not imported. Let's check DDLStatements.ts.
+                    // If we can't easily add the AST, we might need to hack it or add the class.
+                    // Let's try to add a simple DropTableStatement to DDLStatements.ts first if needed.
+                    // But for now, let's assume we can just append the string at the end?
+                    // No, the return is `diffAsts.map(...)`.
+                    // Let's create a simple ad-hoc object that satisfies SqlComponent and formats correctly.
+                    diffAsts.push(new DDLStatements_1.DropTableStatement({
+                        tables: [currentTable.qualifiedName],
+                        ifExists: false
+                    }));
+                }
+            }
+        }
+        // Format output
+        const formatter = new SqlFormatter_1.SqlFormatter(options.formatOptions || { keywordCase: 'upper' });
+        return diffAsts.map(ast => formatter.format(ast).formattedSql + ';');
+    }
+    static parseAndGeneralize(sql) {
+        const split = MultiQuerySplitter_1.MultiQuerySplitter.split(sql);
+        const asts = [];
+        for (const q of split.queries) {
+            if (q.isEmpty)
+                continue;
+            try {
+                const ast = SqlParser_1.SqlParser.parse(q.sql);
+                asts.push(ast);
+            }
+            catch (e) {
+                // Ignore parse errors? Or throw?
+                // For diffing, we probably want to know if input is invalid.
+                console.warn("Failed to parse SQL for diff:", q.sql, e);
+            }
+        }
+        return DDLGeneralizer_1.DDLGeneralizer.generalize(asts);
+    }
+    static buildSchema(asts) {
+        var _a;
+        const tables = new Map();
+        const formatter = new SqlFormatter_1.SqlFormatter({ keywordCase: 'none' });
+        for (const ast of asts) {
+            if (ast instanceof CreateTableQuery_1.CreateTableQuery) {
+                const qName = new ValueComponent_1.QualifiedName(ast.namespaces || [], ast.tableName);
+                const key = this.getQualifiedNameKey(qName);
+                const tableModel = {
+                    name: key,
+                    qualifiedName: qName,
+                    columns: new Map(),
+                    constraints: [],
+                    indexes: []
+                };
+                for (const col of ast.columns) {
+                    tableModel.columns.set(col.name.name, {
+                        name: col.name.name,
+                        definition: col
+                    });
+                }
+                // Generalized CreateTable shouldn't have tableConstraints, but if it did, we'd handle them.
+                tables.set(key, tableModel);
+            }
+            else if (ast instanceof DDLStatements_1.AlterTableStatement) {
+                const key = this.getQualifiedNameKey(ast.table);
+                const tableModel = tables.get(key);
+                if (tableModel) {
+                    for (const action of ast.actions) {
+                        if (action instanceof DDLStatements_1.AlterTableAddConstraint) {
+                            const formatted = formatter.format(action.constraint).formattedSql;
+                            tableModel.constraints.push({
+                                name: (_a = action.constraint.constraintName) === null || _a === void 0 ? void 0 : _a.name,
+                                kind: action.constraint.kind,
+                                definition: action.constraint,
+                                formatted: formatted
+                            });
+                        }
+                        else if (action instanceof DDLStatements_1.AlterTableAddColumn) {
+                            tableModel.columns.set(action.column.name.name, {
+                                name: action.column.name.name,
+                                definition: action.column
+                            });
+                        }
+                    }
+                }
+            }
+            else if (ast instanceof DDLStatements_1.CreateIndexStatement) {
+                const key = this.getQualifiedNameKey(ast.tableName);
+                const tableModel = tables.get(key);
+                if (tableModel) {
+                    const formatted = formatter.format(ast).formattedSql;
+                    tableModel.indexes.push({
+                        name: ast.indexName.toString(),
+                        definition: ast,
+                        formatted: formatted
+                    });
+                }
+            }
+        }
+        return { tables };
+    }
+    static compareColumns(current, expected, diffs, options) {
+        // Add missing columns
+        for (const [name, col] of expected.columns) {
+            if (!current.columns.has(name)) {
+                diffs.push(new DDLStatements_1.AlterTableStatement({
+                    table: expected.qualifiedName,
+                    actions: [new DDLStatements_1.AlterTableAddColumn({ column: col.definition })]
+                }));
+            }
+        }
+        // Drop extra columns
+        if (options.dropColumns) {
+            for (const [name, col] of current.columns) {
+                if (!expected.columns.has(name)) {
+                    diffs.push(new DDLStatements_1.AlterTableStatement({
+                        table: expected.qualifiedName,
+                        actions: [new DDLStatements_1.AlterTableDropColumn({ columnName: col.definition.name })]
+                    }));
+                }
+            }
+        }
+    }
+    static compareConstraints(current, expected, diffs, options) {
+        // We need to match constraints.
+        // If checkConstraintNames is true, match by name.
+        // Else, match by formatted definition (ignoring name in formatting if possible, but SqlFormatter prints name).
+        // To compare by definition ignoring name, we might need to strip name from definition before formatting.
+        const formatter = new SqlFormatter_1.SqlFormatter({ keywordCase: 'none' });
+        const getConstraintSignature = (c) => {
+            if (options.checkConstraintNames) {
+                // Special handling for PRIMARY KEY: ignore name difference
+                if (c.kind === 'primary-key') {
+                    // Strip "CONSTRAINT name" prefix, handling both quoted and unquoted names
+                    // Match: CONSTRAINT "name" or CONSTRAINT name (case-insensitive)
+                    const sig = c.formatted.replace(/^constraint\s+("[^"]+"|[^\s]+)\s+/i, '').trim();
+                    return sig;
+                }
+                return c.name || c.formatted; // Fallback if no name?
+            }
+            // Remove name from definition for comparison
+            // "CONSTRAINT name PRIMARY KEY ..." vs "PRIMARY KEY ..."
+            // If we format it, it might include CONSTRAINT name.
+            // We can regex remove it?
+            return c.formatted.replace(/^constraint\s+("[^"]+"|[^\s]+)\s+/i, '').trim();
+        };
+        const currentSignatures = new Set(current.constraints.map(getConstraintSignature));
+        // Add missing constraints
+        for (const expectedC of expected.constraints) {
+            const sig = getConstraintSignature(expectedC);
+            if (!currentSignatures.has(sig)) {
+                diffs.push(new DDLStatements_1.AlterTableStatement({
+                    table: expected.qualifiedName,
+                    actions: [new DDLStatements_1.AlterTableAddConstraint({ constraint: expectedC.definition })]
+                }));
+            }
+        }
+        // Drop extra constraints
+        if (options.dropConstraints) {
+            const expectedSignatures = new Set(expected.constraints.map(getConstraintSignature));
+            for (const currentC of current.constraints) {
+                const sig = getConstraintSignature(currentC);
+                if (!expectedSignatures.has(sig)) {
+                    // To drop, we need a name. If no name, we can't drop easily (DBs usually auto-name).
+                    if (currentC.name) {
+                        diffs.push(new DDLStatements_1.AlterTableStatement({
+                            table: expected.qualifiedName,
+                            actions: [new DDLStatements_1.AlterTableDropConstraint({ constraintName: new ValueComponent_1.IdentifierString(currentC.name) })]
+                        }));
+                    }
+                    else {
+                        console.warn("Cannot drop unnamed constraint:", currentC.formatted);
+                    }
+                }
+            }
+        }
+    }
+    static compareIndexes(current, expected, diffs, options) {
+        const getIndexSignature = (idx) => {
+            if (options.checkConstraintNames) {
+                // When Check Names is enabled, index name matters
+                return idx.name;
+            }
+            // When Check Names is disabled, compare by structural properties from AST
+            // Compare: table name, columns (expressions), unique flag, using method, where clause
+            const def = idx.definition;
+            const parts = [];
+            // Table name
+            parts.push(def.tableName.toString());
+            // Unique flag
+            if (def.unique) {
+                parts.push('UNIQUE');
+            }
+            // Using method (e.g., BTREE, HASH)
+            if (def.usingMethod) {
+                parts.push(`USING:${def.usingMethod.toString()}`);
+            }
+            // Columns (expressions and sort orders)
+            const columnSigs = def.columns.map(col => {
+                const expr = col.expression.toString();
+                const sort = col.sortOrder || '';
+                const nulls = col.nullsOrder || '';
+                return `${expr}${sort}${nulls}`;
+            });
+            parts.push(`COLS:${columnSigs.join(',')}`);
+            // Include columns
+            if (def.include && def.include.length > 0) {
+                parts.push(`INCLUDE:${def.include.map(i => i.toString()).join(',')}`);
+            }
+            // Where clause
+            if (def.where) {
+                parts.push(`WHERE:${def.where.toString()}`);
+            }
+            return parts.join('|');
+        };
+        const currentSignatures = new Set(current.indexes.map(getIndexSignature));
+        // Add missing indexes
+        for (const expectedIdx of expected.indexes) {
+            const sig = getIndexSignature(expectedIdx);
+            if (!currentSignatures.has(sig)) {
+                diffs.push(expectedIdx.definition);
+            }
+        }
+        // Drop extra indexes
+        // When checkConstraintNames is enabled, we should drop indexes with different names
+        // When dropIndexes is enabled, we should drop all extra indexes
+        if (options.checkConstraintNames || options.dropIndexes) {
+            const expectedSignatures = new Set(expected.indexes.map(getIndexSignature));
+            for (const currentIdx of current.indexes) {
+                const sig = getIndexSignature(currentIdx);
+                if (!expectedSignatures.has(sig)) {
+                    diffs.push(new DDLStatements_1.DropIndexStatement({
+                        indexNames: [currentIdx.definition.indexName],
+                        ifExists: false
+                    }));
+                }
+            }
+        }
+    }
+    static getQualifiedNameKey(qName) {
+        return qName.toString();
+    }
+}
+exports.DDLDiffGenerator = DDLDiffGenerator;
+//# sourceMappingURL=DDLDiffGenerator.js.map

package/dist/transformers/DDLDiffGenerator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"DDLDiffGenerator.js","sourceRoot":"","sources":["../../src/transformers/DDLDiffGenerator.ts"],"names":[],"mappings":";;;AAAA,oDAAiD;AACjD,iDAA8C;AAC9C,qDAAkD;AAClD,oEAAiE;AACjE,iEAAgH;AAChH,2DAA0N;AAE1N,6DAAsF;AAuCtF,MAAa,gBAAgB;IAClB,MAAM,CAAC,YAAY,CAAC,UAAkB,EAAE,WAAmB,EAAE,UAA0B,EAAE;QAC5F,MAAM,UAAU,GAAG,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC;QACvD,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC;QAEzD,MAAM,aAAa,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;QACnD,MAAM,cAAc,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;QAErD,MAAM,QAAQ,GAAmB,EAAE,CAAC;QAEpC,iBAAiB;QACjB,KAAK,MAAM,CAAC,SAAS,EAAE,aAAa,CAAC,IAAI,cAAc,CAAC,MAAM,EAAE,CAAC;YAC7D,MAAM,YAAY,GAAG,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAEzD,IAAI,CAAC,YAAY,EAAE,CAAC;gBAChB,wCAAwC;gBACxC,mDAAmD;gBACnD,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;gBAClF,MAAM,YAAY,GAAG,aAAa,CAAC,aAAa,CAAC,IAAI,YAAY,0BAAS;oBACtE,CAAC,CAAC,aAAa,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK;oBACxC,CAAC,CAAC,aAAa,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC;gBAC5C,MAAM,UAAU,GAAG,aAAa,CAAC,aAAa,CAAC,UAAU;oBACrD,CAAC,CAAC,aAAa,CAAC,aAAa,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;oBAC3D,CAAC,CAAC,IAAI,CAAC;gBAEX,MAAM,WAAW,GAAG,IAAI,mCAAgB,CAAC;oBACrC,SAAS,EAAE,YAAY;oBACvB,UAAU,EAAE,UAAU;oBACtB,OAAO,EAAE,OAAO;iBACnB,CAAC,CAAC;gBACH,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;gBAE3B,sBAAsB;gBACtB,KAAK,MAAM,UAAU,IAAI,aAAa,CAAC,WAAW,EAAE,CAAC;oBACjD,QAAQ,CAAC,IAAI,CAAC,IAAI,mCAAmB,CAAC;wBAClC,KAAK,EAAE,aAAa,CAAC,aAAa;wBAClC,OAAO,EAAE,CAAC,IAAI,uCAAuB,CAAC,EAAE,UAAU,EAAE,UAAU,CAAC,UAAU,EAAE,CAAC,CAAC;qBAChF,CAAC,CAAC,CAAC;gBACR,CAAC;gBAED,kBAAkB;gBAClB,KAAK,MAAM,KAAK,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC;oBACxC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;gBACpC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACJ,kDAAkD;gBAClD,IAAI,CAAC,cAAc,CAAC,YAAY,EAAE,aAAa,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;gBACpE,IAAI,CAAC,kBAAkB,CAAC,YAAY,EAAE,aAAa,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;gBACxE,IAAI,CAAC,cAAc,CAAC,YAAY,EAAE,aAAa,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;YACxE,CAAC;QACL,CAAC;QAED,2BAA2B;QAC3B,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACrB,KAAK,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;gBAC3D,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;oBACxC,yDAAyD;oBACzD,4GAA4G;oBAC5G,kGAAkG;oBAClG,mEAAmE;oBACnE,2DAA2D;oBAC3D,0DAA0D;oBAC1D,+FAA+F;oBAC/F,wDAAwD;oBACxD,uEAAuE;oBACvE,mDAAmD;oBACnD,uDAAuD;oBACvD,yEAAyE;oBAEzE,6DAA6D;oBAC7D,4DAA4D;oBAE5D,6EAA6E;oBAC7E,oFAAoF;oBACpF,sEAAsE;oBACtE,yCAAyC;oBAEzC,yFAAyF;oBACzF,QAAQ,CAAC,IAAI,CAAC,IAAI,kCAAkB,CAAC;wBACjC,MAAM,EAAE,CAAC,YAAY,CAAC,aAAa,CAAC;wBACpC,QAAQ,EAAE,KAAK;qBAClB,CAAC,CAAC,CAAC;gBACR,CAAC;YACL,CAAC;QACL,CAAC;QAED,gBAAgB;QAChB,MAAM,SAAS,GAAG,IAAI,2BAAY,CAAC,OAAO,CAAC,aAAa,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;QACtF,OAAO,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,YAAY,GAAG,GAAG,CAAC,CAAC;IACzE,CAAC;IAEO,MAAM,CAAC,kBAAkB,CAAC,GAAW;QACzC,MAAM,KAAK,GAAG,uCAAkB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC5C,MAAM,IAAI,GAAmB,EAAE,CAAC;QAChC,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAC5B,IAAI,CAAC,CAAC,OAAO;gBAAE,SAAS;YACxB,IAAI,CAAC;gBACD,MAAM,GAAG,GAAG,qBAAS,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBACnC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACT,iCAAiC;gBACjC,6DAA6D;gBAC7D,OAAO,CAAC,IAAI,CAAC,+BAA+B,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;YAC5D,CAAC;QACL,CAAC;QACD,OAAO,+BAAc,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IAC3C,CAAC;IAEO,MAAM,CAAC,WAAW,CAAC,IAAoB;;QAC3C,MAAM,MAAM,GAAG,IAAI,GAAG,EAAsB,CAAC;QAC7C,MAAM,SAAS,GAAG,IAAI,2BAAY,CAAC,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,CAAC;QAE5D,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACrB,IAAI,GAAG,YAAY,mCAAgB,EAAE,CAAC;gBAClC,MAAM,KAAK,GAAG,IAAI,8BAAa,CAAC,GAAG,CAAC,UAAU,IAAI,EAAE,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC;gBACrE,MAAM,GAAG,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;gBAE5C,MAAM,UAAU,GAAe;oBAC3B,IAAI,EAAE,GAAG;oBACT,aAAa,EAAE,KAAK;oBACpB,OAAO,EAAE,IAAI,GAAG,EAAE;oBAClB,WAAW,EAAE,EAAE;oBACf,OAAO,EAAE,EAAE;iBACd,CAAC;gBAEF,KAAK,MAAM,GAAG,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;oBAC5B,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE;wBAClC,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,IAAI;wBACnB,UAAU,EAAE,GAAG;qBAClB,CAAC,CAAC;gBACP,CAAC;gBACD,4FAA4F;gBAE5F,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;YAChC,CAAC;iBAAM,IAAI,GAAG,YAAY,mCAAmB,EAAE,CAAC;gBAC5C,MAAM,GAAG,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBAChD,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACnC,IAAI,UAAU,EAAE,CAAC;oBACb,KAAK,MAAM,MAAM,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;wBAC/B,IAAI,MAAM,YAAY,uCAAuB,EAAE,CAAC;4BAC5C,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC;4BACnE,UAAU,CAAC,WAAW,CAAC,IAAI,CAAC;gCACxB,IAAI,EAAE,MAAA,MAAM,CAAC,UAAU,CAAC,cAAc,0CAAE,IAAI;gCAC5C,IAAI,EAAE,MAAM,CAAC,UAAU,CAAC,IAAI;gCAC5B,UAAU,EAAE,MAAM,CAAC,UAAU;gCAC7B,SAAS,EAAE,SAAS;6BACvB,CAAC,CAAC;wBACP,CAAC;6BAAM,IAAI,MAAM,YAAY,mCAAmB,EAAE,CAAC;4BAC/C,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE;gCAC5C,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI;gCAC7B,UAAU,EAAE,MAAM,CAAC,MAAM;6BAC5B,CAAC,CAAC;wBACP,CAAC;oBACL,CAAC;gBACL,CAAC;YACL,CAAC;iBAAM,IAAI,GAAG,YAAY,oCAAoB,EAAE,CAAC;gBAC7C,MAAM,GAAG,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBACpD,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACnC,IAAI,UAAU,EAAE,CAAC;oBACb,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,YAAY,CAAC;oBACrD,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC;wBACpB,IAAI,EAAE,GAAG,CAAC,SAAS,CAAC,QAAQ,EAAE;wBAC9B,UAAU,EAAE,GAAG;wBACf,SAAS,EAAE,SAAS;qBACvB,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;QACL,CAAC;QACD,OAAO,EAAE,MAAM,EAAE,CAAC;IACtB,CAAC;IAEO,MAAM,CAAC,cAAc,CAAC,OAAmB,EAAE,QAAoB,EAAE,KAAqB,EAAE,OAAuB;QACnH,sBAAsB;QACtB,KAAK,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACzC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC7B,KAAK,CAAC,IAAI,CAAC,IAAI,mCAAmB,CAAC;oBAC/B,KAAK,EAAE,QAAQ,CAAC,aAAa;oBAC7B,OAAO,EAAE,CAAC,IAAI,mCAAmB,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC;iBACjE,CAAC,CAAC,CAAC;YACR,CAAC;QACL,CAAC;QAED,qBAAqB;QACrB,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YACtB,KAAK,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBACxC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC9B,KAAK,CAAC,IAAI,CAAC,IAAI,mCAAmB,CAAC;wBAC/B,KAAK,EAAE,QAAQ,CAAC,aAAa;wBAC7B,OAAO,EAAE,CAAC,IAAI,oCAAoB,CAAC,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC;qBAC3E,CAAC,CAAC,CAAC;gBACR,CAAC;YACL,CAAC;QACL,CAAC;IACL,CAAC;IAEO,MAAM,CAAC,kBAAkB,CAAC,OAAmB,EAAE,QAAoB,EAAE,KAAqB,EAAE,OAAuB;QACvH,gCAAgC;QAChC,kDAAkD;QAClD,+GAA+G;QAC/G,yGAAyG;QAEzG,MAAM,SAAS,GAAG,IAAI,2BAAY,CAAC,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,CAAC;QAE5D,MAAM,sBAAsB,GAAG,CAAC,CAAkB,EAAE,EAAE;YAClD,IAAI,OAAO,CAAC,oBAAoB,EAAE,CAAC;gBAC/B,2DAA2D;gBAC3D,IAAI,CAAC,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;oBAC3B,0EAA0E;oBAC1E,iEAAiE;oBACjE,MAAM,GAAG,GAAG,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,oCAAoC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;oBACjF,OAAO,GAAG,CAAC;gBACf,CAAC;gBACD,OAAO,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,SAAS,CAAC,CAAC,uBAAuB;YACzD,CAAC;YACD,6CAA6C;YAC7C,yDAAyD;YACzD,qDAAqD;YACrD,0BAA0B;YAC1B,OAAO,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,oCAAoC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAChF,CAAC,CAAC;QAEF,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC,CAAC;QAEnF,0BAA0B;QAC1B,KAAK,MAAM,SAAS,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC3C,MAAM,GAAG,GAAG,sBAAsB,CAAC,SAAS,CAAC,CAAC;YAC9C,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC9B,KAAK,CAAC,IAAI,CAAC,IAAI,mCAAmB,CAAC;oBAC/B,KAAK,EAAE,QAAQ,CAAC,aAAa;oBAC7B,OAAO,EAAE,CAAC,IAAI,uCAAuB,CAAC,EAAE,UAAU,EAAE,SAAS,CAAC,UAAU,EAAE,CAAC,CAAC;iBAC/E,CAAC,CAAC,CAAC;YACR,CAAC;QACL,CAAC;QAED,yBAAyB;QACzB,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;YAC1B,MAAM,kBAAkB,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC,CAAC;YACrF,KAAK,MAAM,QAAQ,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;gBACzC,MAAM,GAAG,GAAG,sBAAsB,CAAC,QAAQ,CAAC,CAAC;gBAC7C,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;oBAC/B,qFAAqF;oBACrF,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;wBAChB,KAAK,CAAC,IAAI,CAAC,IAAI,mCAAmB,CAAC;4BAC/B,KAAK,EAAE,QAAQ,CAAC,aAAa;4BAC7B,OAAO,EAAE,CAAC,IAAI,wCAAwB,CAAC,EAAE,cAAc,EAAE,IAAI,iCAAgB,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;yBACnG,CAAC,CAAC,CAAC;oBACR,CAAC;yBAAM,CAAC;wBACJ,OAAO,CAAC,IAAI,CAAC,iCAAiC,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;oBACxE,CAAC;gBACL,CAAC;YACL,CAAC;QACL,CAAC;IACL,CAAC;IAEO,MAAM,CAAC,cAAc,CAAC,OAAmB,EAAE,QAAoB,EAAE,KAAqB,EAAE,OAAuB;QACnH,MAAM,iBAAiB,GAAG,CAAC,GAAe,EAAE,EAAE;YAC1C,IAAI,OAAO,CAAC,oBAAoB,EAAE,CAAC;gBAC/B,kDAAkD;gBAClD,OAAO,GAAG,CAAC,IAAI,CAAC;YACpB,CAAC;YACD,0EAA0E;YAC1E,sFAAsF;YACtF,MAAM,GAAG,GAAG,GAAG,CAAC,UAAU,CAAC;YAC3B,MAAM,KAAK,GAAa,EAAE,CAAC;YAE3B,aAAa;YACb,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC;YAErC,cAAc;YACd,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;gBACb,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACzB,CAAC;YAED,mCAAmC;YACnC,IAAI,GAAG,CAAC,WAAW,EAAE,CAAC;gBAClB,KAAK,CAAC,IAAI,CAAC,SAAS,GAAG,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YACtD,CAAC;YAED,wCAAwC;YACxC,MAAM,UAAU,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;gBACrC,MAAM,IAAI,GAAG,GAAG,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC;gBACvC,MAAM,IAAI,GAAG,GAAG,CAAC,SAAS,IAAI,EAAE,CAAC;gBACjC,MAAM,KAAK,GAAG,GAAG,CAAC,UAAU,IAAI,EAAE,CAAC;gBACnC,OAAO,GAAG,IAAI,GAAG,IAAI,GAAG,KAAK,EAAE,CAAC;YACpC,CAAC,CAAC,CAAC;YACH,KAAK,CAAC,IAAI,CAAC,QAAQ,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAE3C,kBAAkB;YAClB,IAAI,GAAG,CAAC,OAAO,IAAI,GAAG,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxC,KAAK,CAAC,IAAI,CAAC,WAAW,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC1E,CAAC;YAED,eAAe;YACf,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;gBACZ,KAAK,CAAC,IAAI,CAAC,SAAS,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YAChD,CAAC;YAED,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC3B,CAAC,CAAC;QAEF,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC;QAE1E,sBAAsB;QACtB,KAAK,MAAM,WAAW,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACzC,MAAM,GAAG,GAAG,iBAAiB,CAAC,WAAW,CAAC,CAAC;YAC3C,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC9B,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;YACvC,CAAC;QACL,CAAC;QAED,qBAAqB;QACrB,oFAAoF;QACpF,gEAAgE;QAChE,IAAI,OAAO,CAAC,oBAAoB,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YACtD,MAAM,kBAAkB,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC;YAC5E,KAAK,MAAM,UAAU,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBACvC,MAAM,GAAG,GAAG,iBAAiB,CAAC,UAAU,CAAC,CAAC;gBAC1C,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;oBAC/B,KAAK,CAAC,IAAI,CAAC,IAAI,kCAAkB,CAAC;wBAC9B,UAAU,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC;wBAC7C,QAAQ,EAAE,KAAK;qBAClB,CAAC,CAAC,CAAC;gBACR,CAAC;YACL,CAAC;QACL,CAAC;IACL,CAAC;IAEO,MAAM,CAAC,mBAAmB,CAAC,KAAoB;QACnD,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC;IAC5B,CAAC;CACJ;AA1UD,4CA0UC"}

package/dist/transformers/DDLGeneralizer.js

@@ -0,0 +1,119 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DDLGeneralizer = void 0;
+const CreateTableQuery_1 = require("../models/CreateTableQuery");
+const DDLStatements_1 = require("../models/DDLStatements");
+const ValueComponent_1 = require("../models/ValueComponent");
+class DDLGeneralizer {
+    /**
+     * Generalizes DDL statements by moving constraints from CREATE TABLE to ALTER TABLE statements.
+     * This normalizes the DDL for easier comparison.
+     *
+     * @param ast List of SQL components (DDL statements)
+     * @returns Generalized list of SQL components
+     */
+    static generalize(ast) {
+        const result = [];
+        for (const component of ast) {
+            if (component instanceof CreateTableQuery_1.CreateTableQuery) {
+                const { createTable, alterTables } = this.splitCreateTable(component);
+                result.push(createTable);
+                result.push(...alterTables);
+            }
+            else {
+                result.push(component);
+            }
+        }
+        return result;
+    }
+    static splitCreateTable(query) {
+        const newColumns = [];
+        const alterTables = [];
+        // Construct QualifiedName for the table
+        const tableQualifiedName = new ValueComponent_1.QualifiedName(query.namespaces || [], query.tableName.name);
+        // Process columns
+        for (const col of query.columns) {
+            const newConstraints = [];
+            for (const constraint of col.constraints) {
+                if (['primary-key', 'unique', 'references', 'check'].includes(constraint.kind)) {
+                    // Move to Alter Table
+                    const tableConstraint = this.columnToTableConstraint(col.name, constraint);
+                    alterTables.push(new DDLStatements_1.AlterTableStatement({
+                        table: tableQualifiedName,
+                        actions: [new DDLStatements_1.AlterTableAddConstraint({ constraint: tableConstraint })]
+                    }));
+                }
+                else {
+                    // Keep (not-null, default, etc.)
+                    newConstraints.push(constraint);
+                }
+            }
+            newColumns.push(new CreateTableQuery_1.TableColumnDefinition({
+                name: col.name,
+                dataType: col.dataType,
+                constraints: newConstraints
+            }));
+        }
+        // Process table constraints
+        if (query.tableConstraints) {
+            for (const constraint of query.tableConstraints) {
+                alterTables.push(new DDLStatements_1.AlterTableStatement({
+                    table: tableQualifiedName,
+                    actions: [new DDLStatements_1.AlterTableAddConstraint({ constraint })]
+                }));
+            }
+        }
+        const newCreateTable = new CreateTableQuery_1.CreateTableQuery({
+            tableName: query.tableName.name,
+            namespaces: query.namespaces,
+            columns: newColumns,
+            ifNotExists: query.ifNotExists,
+            isTemporary: query.isTemporary,
+            tableOptions: query.tableOptions,
+            asSelectQuery: query.asSelectQuery,
+            withDataOption: query.withDataOption,
+            // tableConstraints is empty
+            tableConstraints: []
+        });
+        return { createTable: newCreateTable, alterTables };
+    }
+    static columnToTableConstraint(columnName, constraint) {
+        var _a, _b;
+        const baseParams = {
+            constraintName: constraint.constraintName,
+            deferrable: (_a = constraint.reference) === null || _a === void 0 ? void 0 : _a.deferrable,
+            initially: (_b = constraint.reference) === null || _b === void 0 ? void 0 : _b.initially
+        };
+        switch (constraint.kind) {
+            case 'primary-key':
+                return new CreateTableQuery_1.TableConstraintDefinition({
+                    kind: 'primary-key',
+                    columns: [columnName],
+                    ...baseParams
+                });
+            case 'unique':
+                return new CreateTableQuery_1.TableConstraintDefinition({
+                    kind: 'unique',
+                    columns: [columnName],
+                    ...baseParams
+                });
+            case 'references':
+                return new CreateTableQuery_1.TableConstraintDefinition({
+                    kind: 'foreign-key',
+                    columns: [columnName],
+                    reference: constraint.reference,
+                    ...baseParams
+                });
+            case 'check':
+                return new CreateTableQuery_1.TableConstraintDefinition({
+                    kind: 'check',
+                    checkExpression: constraint.checkExpression,
+                    ...baseParams
+                });
+            default:
+                throw new Error(`Unsupported constraint kind for generalization: ${constraint.kind}`);
+        }
+    }
+}
+exports.DDLGeneralizer = DDLGeneralizer;
+//# sourceMappingURL=DDLGeneralizer.js.map

package/dist/transformers/DDLGeneralizer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"DDLGeneralizer.js","sourceRoot":"","sources":["../../src/transformers/DDLGeneralizer.ts"],"names":[],"mappings":";;;AACA,iEAA4I;AAC5I,2DAAuF;AACvF,6DAA2E;AAE3E,MAAa,cAAc;IACvB;;;;;;OAMG;IACI,MAAM,CAAC,UAAU,CAAC,GAAmB;QACxC,MAAM,MAAM,GAAmB,EAAE,CAAC;QAElC,KAAK,MAAM,SAAS,IAAI,GAAG,EAAE,CAAC;YAC1B,IAAI,SAAS,YAAY,mCAAgB,EAAE,CAAC;gBACxC,MAAM,EAAE,WAAW,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;gBACtE,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;gBACzB,MAAM,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,CAAC;YAChC,CAAC;iBAAM,CAAC;gBACJ,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC3B,CAAC;QACL,CAAC;QAED,OAAO,MAAM,CAAC;IAClB,CAAC;IAEO,MAAM,CAAC,gBAAgB,CAAC,KAAuB;QACnD,MAAM,UAAU,GAA4B,EAAE,CAAC;QAC/C,MAAM,WAAW,GAA0B,EAAE,CAAC;QAE9C,wCAAwC;QACxC,MAAM,kBAAkB,GAAG,IAAI,8BAAa,CAAC,KAAK,CAAC,UAAU,IAAI,EAAE,EAAE,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAE3F,kBAAkB;QAClB,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAC9B,MAAM,cAAc,GAAiC,EAAE,CAAC;YACxD,KAAK,MAAM,UAAU,IAAI,GAAG,CAAC,WAAW,EAAE,CAAC;gBACvC,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC7E,sBAAsB;oBACtB,MAAM,eAAe,GAAG,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;oBAC3E,WAAW,CAAC,IAAI,CAAC,IAAI,mCAAmB,CAAC;wBACrC,KAAK,EAAE,kBAAkB;wBACzB,OAAO,EAAE,CAAC,IAAI,uCAAuB,CAAC,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;qBAC1E,CAAC,CAAC,CAAC;gBACR,CAAC;qBAAM,CAAC;oBACJ,iCAAiC;oBACjC,cAAc,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACpC,CAAC;YACL,CAAC;YACD,UAAU,CAAC,IAAI,CAAC,IAAI,wCAAqB,CAAC;gBACtC,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,WAAW,EAAE,cAAc;aAC9B,CAAC,CAAC,CAAC;QACR,CAAC;QAED,4BAA4B;QAC5B,IAAI,KAAK,CAAC,gBAAgB,EAAE,CAAC;YACzB,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,gBAAgB,EAAE,CAAC;gBAC9C,WAAW,CAAC,IAAI,CAAC,IAAI,mCAAmB,CAAC;oBACrC,KAAK,EAAE,kBAAkB;oBACzB,OAAO,EAAE,CAAC,IAAI,uCAAuB,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC;iBACzD,CAAC,CAAC,CAAC;YACR,CAAC;QACL,CAAC;QAED,MAAM,cAAc,GAAG,IAAI,mCAAgB,CAAC;YACxC,SAAS,EAAE,KAAK,CAAC,SAAS,CAAC,IAAI;YAC/B,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,OAAO,EAAE,UAAU;YACnB,WAAW,EAAE,KAAK,CAAC,WAAW;YAC9B,WAAW,EAAE,KAAK,CAAC,WAAW;YAC9B,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,aAAa,EAAE,KAAK,CAAC,aAAa;YAClC,cAAc,EAAE,KAAK,CAAC,cAAc;YACpC,4BAA4B;YAC5B,gBAAgB,EAAE,EAAE;SACvB,CAAC,CAAC;QAEH,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,WAAW,EAAE,CAAC;IACxD,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAAC,UAA4B,EAAE,UAAsC;;QACvG,MAAM,UAAU,GAAG;YACf,cAAc,EAAE,UAAU,CAAC,cAAc;YACzC,UAAU,EAAE,MAAA,UAAU,CAAC,SAAS,0CAAE,UAAU;YAC5C,SAAS,EAAE,MAAA,UAAU,CAAC,SAAS,0CAAE,SAAS;SAC7C,CAAC;QAEF,QAAQ,UAAU,CAAC,IAAI,EAAE,CAAC;YACtB,KAAK,aAAa;gBACd,OAAO,IAAI,4CAAyB,CAAC;oBACjC,IAAI,EAAE,aAAa;oBACnB,OAAO,EAAE,CAAC,UAAU,CAAC;oBACrB,GAAG,UAAU;iBAChB,CAAC,CAAC;YACP,KAAK,QAAQ;gBACT,OAAO,IAAI,4CAAyB,CAAC;oBACjC,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,CAAC,UAAU,CAAC;oBACrB,GAAG,UAAU;iBAChB,CAAC,CAAC;YACP,KAAK,YAAY;gBACb,OAAO,IAAI,4CAAyB,CAAC;oBACjC,IAAI,EAAE,aAAa;oBACnB,OAAO,EAAE,CAAC,UAAU,CAAC;oBACrB,SAAS,EAAE,UAAU,CAAC,SAAS;oBAC/B,GAAG,UAAU;iBAChB,CAAC,CAAC;YACP,KAAK,OAAO;gBACR,OAAO,IAAI,4CAAyB,CAAC;oBACjC,IAAI,EAAE,OAAO;oBACb,eAAe,EAAE,UAAU,CAAC,eAAe;oBAC3C,GAAG,UAAU;iBAChB,CAAC,CAAC;YACP;gBACI,MAAM,IAAI,KAAK,CAAC,mDAAmD,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC;QAC9F,CAAC;IACL,CAAC;CACJ;AArHD,wCAqHC"}

package/dist/transformers/DDLToFixtureConverter.js

@@ -0,0 +1,206 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DDLToFixtureConverter = void 0;
+const MultiQuerySplitter_1 = require("../utils/MultiQuerySplitter");
+const SqlParser_1 = require("../parsers/SqlParser");
+const CreateTableQuery_1 = require("../models/CreateTableQuery");
+const InsertQuery_1 = require("../models/InsertQuery");
+const ValuesQuery_1 = require("../models/ValuesQuery");
+const TableDefinitionModel_1 = require("../models/TableDefinitionModel");
+const ValueComponent_1 = require("../models/ValueComponent");
+const SqlFormatter_1 = require("./SqlFormatter");
+class DDLToFixtureConverter {
+    /**
+     * Converts DDL statements (CREATE TABLE) in the provided SQL text to a Fixture JSON object.
+     * Ignores non-DDL statements and parse errors.
+     *
+     * @param ddlSql The SQL text containing CREATE TABLE statements.
+     * @returns A Record representing the Fixture JSON.
+     */
+    /**
+     * Converts DDL statements (CREATE TABLE) in the provided SQL text to a Fixture JSON object.
+     * Ignores non-DDL statements and parse errors.
+     *
+     * @param ddlSql The SQL text containing CREATE TABLE statements.
+     * @returns A Record representing the Fixture JSON.
+     */
+    static convert(ddlSql) {
+        const splitResult = MultiQuerySplitter_1.MultiQuerySplitter.split(ddlSql);
+        const createTableQueries = [];
+        const insertQueries = [];
+        for (const query of splitResult.queries) {
+            if (query.isEmpty)
+                continue;
+            try {
+                const ast = SqlParser_1.SqlParser.parse(query.sql);
+                if (ast instanceof CreateTableQuery_1.CreateTableQuery) {
+                    createTableQueries.push(ast);
+                }
+                else if (ast instanceof InsertQuery_1.InsertQuery) {
+                    insertQueries.push(ast);
+                }
+            }
+            catch (e) {
+                // Ignore parse errors for non-DDL or invalid SQL
+            }
+        }
+        const registry = (0, TableDefinitionModel_1.createTableDefinitionRegistryFromCreateTableQueries)(createTableQueries);
+        const fixtureJson = {};
+        // Initialize fixtureJson with empty rows
+        for (const [tableName, def] of Object.entries(registry)) {
+            fixtureJson[tableName] = {
+                columns: def.columns.map(col => ({
+                    name: col.name,
+                    type: col.typeName,
+                    default: this.formatDefaultValue(col.defaultValue)
+                })),
+                rows: []
+            };
+        }
+        // Sequence counters for nextval defaults: table -> column -> counter
+        const sequences = {};
+        // Process INSERT statements
+        for (const insert of insertQueries) {
+            // Only support INSERT ... VALUES
+            if (!(insert.selectQuery instanceof ValuesQuery_1.ValuesQuery)) {
+                continue;
+            }
+            const tableName = insert.insertClause.source.getAliasName();
+            if (!tableName || !fixtureJson[tableName]) {
+                // Ignore INSERTs for unknown tables
+                continue;
+            }
+            const tableDef = registry[tableName];
+            const targetColumns = insert.insertClause.columns;
+            const valuesQuery = insert.selectQuery;
+            // If columns are not specified, use all columns in order
+            const columnNames = targetColumns
+                ? targetColumns.map(c => c.name)
+                : tableDef.columns.map(c => c.name);
+            for (const tuple of valuesQuery.tuples) {
+                const row = {};
+                // Map provided values
+                for (let i = 0; i < columnNames.length; i++) {
+                    const colName = columnNames[i];
+                    if (i < tuple.values.length) {
+                        const val = tuple.values[i];
+                        row[colName] = this.extractValue(val);
+                    }
+                }
+                // Fill missing columns with defaults
+                for (const colDef of tableDef.columns) {
+                    if (row[colDef.name] !== undefined) {
+                        continue;
+                    }
+                    // Check for NOT NULL constraint without default
+                    // TableColumnDefinitionModel.required is true if NOT NULL and no default/identity
+                    if (colDef.required) {
+                        throw new Error(`Column '${colDef.name}' in table '${tableName}' cannot be null and has no default value.`);
+                    }
+                    const hasDefault = colDef.defaultValue !== null && colDef.defaultValue !== undefined;
+                    if (hasDefault) {
+                        const defaultValStr = this.formatDefaultValue(colDef.defaultValue);
+                        if (defaultValStr) {
+                            const trimmedDefault = defaultValStr.trim();
+                            const lowerTrimmed = trimmedDefault.toLowerCase();
+                            const stringLiteralValue = trimmedDefault.startsWith("'") && trimmedDefault.endsWith("'")
+                                ? trimmedDefault.slice(1, -1).replace(/''/g, "'")
+                                : trimmedDefault;
+                            // Normalize literal NULLs (even when quoted) into JS null
+                            if (stringLiteralValue.toLowerCase() === 'null') {
+                                row[colDef.name] = null;
+                            }
+                            else if (lowerTrimmed.includes('nextval')) {
+                                // Handle nextval sequence
+                                if (!sequences[tableName])
+                                    sequences[tableName] = {};
+                                if (!sequences[tableName][colDef.name])
+                                    sequences[tableName][colDef.name] = 0;
+                                sequences[tableName][colDef.name]++;
+                                row[colDef.name] = sequences[tableName][colDef.name];
+                            }
+                            else if (lowerTrimmed.includes('now') || lowerTrimmed.includes('current_timestamp')) {
+                                // Handle timestamp defaults - use a fixed date for consistency or current date
+                                // Using a fixed date makes tests deterministic
+                                row[colDef.name] = "2023-01-01 00:00:00";
+                            }
+                            else {
+                                // Use the literal default value
+                                // Try to unquote if it's a string literal
+                                if (trimmedDefault.startsWith("'") && trimmedDefault.endsWith("'")) {
+                                    row[colDef.name] = stringLiteralValue;
+                                }
+                                else {
+                                    // Try to parse number/boolean/null
+                                    const num = Number(trimmedDefault);
+                                    if (!isNaN(num)) {
+                                        row[colDef.name] = num;
+                                    }
+                                    else {
+                                        const lower = trimmedDefault.toLowerCase();
+                                        if (lower === 'true') {
+                                            row[colDef.name] = true;
+                                        }
+                                        else if (lower === 'false') {
+                                            row[colDef.name] = false;
+                                        }
+                                        else if (lower === 'null') {
+                                            row[colDef.name] = null;
+                                        }
+                                        else {
+                                            row[colDef.name] = trimmedDefault;
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    }
+                    else {
+                        // Default to null
+                        row[colDef.name] = null;
+                    }
+                }
+                fixtureJson[tableName].rows.push(row);
+            }
+        }
+        return fixtureJson;
+    }
+    static extractValue(value) {
+        if (value instanceof ValueComponent_1.LiteralValue) {
+            return value.value;
+        }
+        // For other types, try to format to string
+        try {
+            const formatter = new SqlFormatter_1.SqlFormatter({ keywordCase: 'none' });
+            const { formattedSql } = formatter.format(value);
+            // Remove quotes if it looks like a string literal
+            if (formattedSql.startsWith("'") && formattedSql.endsWith("'")) {
+                return formattedSql.slice(1, -1).replace(/''/g, "'");
+            }
+            return formattedSql;
+        }
+        catch (e) {
+            return String(value);
+        }
+    }
+    static formatDefaultValue(value) {
+        if (value === null || value === undefined) {
+            return undefined;
+        }
+        if (typeof value === 'string') {
+            return value;
+        }
+        try {
+            // Use SqlFormatter to print the AST node
+            const formatter = new SqlFormatter_1.SqlFormatter({ keywordCase: 'none' });
+            const { formattedSql } = formatter.format(value);
+            return formattedSql;
+        }
+        catch (e) {
+            // Fallback if formatter fails
+            return String(value);
+        }
+    }
+}
+exports.DDLToFixtureConverter = DDLToFixtureConverter;
+//# sourceMappingURL=DDLToFixtureConverter.js.map