relq 1.0.5 → 1.0.6

package/dist/cjs/cli/commands/add.cjs

@@ -32,10 +32,14 @@ var __importStar = (this && this.__importStar) || (function () {
         return result;
     };
 })();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.addCommand = addCommand;
 exports.getRelatedChanges = getRelatedChanges;
 const fs = __importStar(require("fs"));
+const strip_comments_1 = __importDefault(require("strip-comments"));
 const spinner_1 = require("../utils/spinner.cjs");
 const relqignore_1 = require("../utils/relqignore.cjs");
 const config_1 = require("../../config/config.cjs");
@@ -47,7 +51,8 @@ function parseSchemaFileForComparison(schemaPath) {
     if (!fs.existsSync(schemaPath)) {
         return null;
     }
-    const content = fs.readFileSync(schemaPath, 'utf-8');
+    const rawContent = fs.readFileSync(schemaPath, 'utf-8');
+    const content = (0, strip_comments_1.default)(rawContent);
     const tables = [];
     const tableStartRegex = /defineTable\s*\(\s*['"]([^'"]+)['"],\s*\{/g;
     let tableStartMatch;
@@ -87,15 +92,21 @@ function parseSchemaFileForComparison(schemaPath) {
         const tsToDbNameMap = new Map();
         const lines = columnsBlock.split('\n');
         let currentColDef = '';
+        let pendingJsDocComment = null;
         for (const line of lines) {
             const trimmed = line.trim();
+            const jsDocMatch = trimmed.match(/^\/\*\*\s*(.*?)\s*\*\/$/);
+            if (jsDocMatch) {
+                pendingJsDocComment = jsDocMatch[1];
+                continue;
+            }
             if (trimmed.startsWith('//') || trimmed.startsWith('/*') || trimmed.startsWith('*'))
                 continue;
             currentColDef += ' ' + trimmed;
             if (trimmed.endsWith(',') || trimmed.endsWith(')')) {
                 const colDef = currentColDef.trim();
                 currentColDef = '';
-                const typePattern = 'varchar|text|uuid|integer|bigint|boolean|timestamp|date|jsonb|json|numeric|serial|bigserial|smallserial|tsvector|smallint|real|doublePrecision|char|inet|cidr|macaddr|macaddr8|interval|time|point|line|lseg|box|path|polygon|circle|bytea|bit|varbit|money|xml|oid';
+                const typePattern = 'varchar|text|uuid|integer|bigint|boolean|timestamptz|timestamp|date|jsonb|json|numeric|serial|bigserial|smallserial|tsvector|smallint|real|doublePrecision|char|inet|cidr|macaddr|macaddr8|interval|timetz|time|point|line|lseg|box|path|polygon|circle|bytea|bit|varbit|money|xml|oid|enumType|domainType';
                 const colMatch = colDef.match(new RegExp(`^(\\w+):\\s*(${typePattern})`));
                 if (!colMatch)
                     continue;
@@ -111,13 +122,17 @@ function parseSchemaFileForComparison(schemaPath) {
                 if (bigintMatch) {
                     defaultValue = bigintMatch[1];
                 }
-                const funcDefaultMatch = !defaultValue && colDef.match(/\.default\(\s*(genRandomUuid|now|currentDate|currentTimestamp|emptyArray|emptyObject)\s*\(\s*\)\s*\)/);
+                const bigintLiteralMatch = !defaultValue && colDef.match(/\.default\(\s*(-?\d+)n\s*\)/);
+                if (bigintLiteralMatch) {
+                    defaultValue = bigintLiteralMatch[1];
+                }
+                const funcDefaultMatch = !defaultValue && colDef.match(/\.default\(\s*(?:DEFAULT\.)?(genRandomUuid|now|currentDate|currentTimestamp|emptyArray|emptyObject|emptyJsonb)\s*\(\s*\)\s*\)/);
                 if (funcDefaultMatch) {
                     const funcName = funcDefaultMatch[1];
                     if (funcName === 'emptyArray') {
                         defaultValue = isJsonbColumn ? "'[]'::jsonb" : "'{}'::text[]";
                     }
-                    else if (funcName === 'emptyObject') {
+                    else if (funcName === 'emptyObject' || funcName === 'emptyJsonb') {
                         defaultValue = "'{}'::jsonb";
                     }
                     else {
@@ -182,10 +197,22 @@ function parseSchemaFileForComparison(schemaPath) {
                 if (commentMatch) {
                     comment = commentMatch[2];
                 }
+                else if (pendingJsDocComment) {
+                    comment = pendingJsDocComment;
+                }
+                pendingJsDocComment = null;
                 if (colDef.includes('.identity()')) {
                     defaultValue = defaultValue || 'GENERATED BY DEFAULT AS IDENTITY';
                 }
-                const isArray = colDef.includes('.array()');
+                let trackingId = undefined;
+                const trackingIdMatch = colDef.match(/\.\$id\(\s*['"]([^'"]+)['"]\s*\)/);
+                if (trackingIdMatch) {
+                    trackingId = trackingIdMatch[1];
+                }
+                const hasArrayModifier = colDef.includes('.array()');
+                const hasEmptyArrayDefault = colDef.includes('emptyArray()');
+                const isJsonbType = type === 'jsonb' || type === 'json';
+                const isArray = hasArrayModifier || (hasEmptyArrayDefault && !isJsonbType);
                 columns.push({
                     name: dbColName,
                     dataType: isArray ? `${type}[]` : type,
@@ -198,6 +225,7 @@ function parseSchemaFileForComparison(schemaPath) {
                     scale: null,
                     references: null,
                     comment,
+                    trackingId,
                 });
             }
         }
@@ -210,8 +238,13 @@ function parseSchemaFileForComparison(schemaPath) {
                 const tsColName = c.trim().replace(/table\.\s*/, '');
                 return tsToDbNameMap.get(tsColName) || tsColName;
             });
-            const isUnique = optionsBlock.includes(`index('${indexName}')`) &&
-                optionsBlock.substring(optionsBlock.indexOf(`index('${indexName}')`)).split('\n')[0].includes('.unique()');
+            const indexStart = optionsBlock.indexOf(`index('${indexName}')`);
+            const indexLine = indexStart >= 0 ? optionsBlock.substring(indexStart).split('\n')[0] : '';
+            const isUnique = indexLine.includes('.unique()');
+            const commentMatch = indexLine.match(/\.comment\(\s*(['"])([^'"]*)\1\s*\)/);
+            const comment = commentMatch ? commentMatch[2] : null;
+            const trackingIdMatch = indexLine.match(/\.\$id\(\s*(['"])([^'"]+)\1\s*\)/);
+            const trackingId = trackingIdMatch ? trackingIdMatch[2] : undefined;
             indexes.push({
                 name: indexName,
                 columns: indexCols,
@@ -221,6 +254,8 @@ function parseSchemaFileForComparison(schemaPath) {
                 definition: '',
                 whereClause: null,
                 expression: null,
+                comment,
+                trackingId,
             });
         }
         const checkRegexLegacy = /check\s*\(\s*['"]([^'"]+)['"]\s*,\s*sql`([^`]+)`\s*\)/g;
@@ -271,6 +306,83 @@ function parseSchemaFileForComparison(schemaPath) {
             const dbPartitionKey = tsToDbNameMap.get(tsPartitionKey) || tsPartitionKey;
             partitionKey = [dbPartitionKey];
         }
+        const pkColumns = columns.filter(c => c.isPrimaryKey).map(c => c.name);
+        if (pkColumns.length > 0) {
+            const pkName = `${tableName}_pkey`;
+            if (!constraints.some(c => c.type === 'PRIMARY KEY')) {
+                constraints.push({
+                    name: pkName,
+                    type: 'PRIMARY KEY',
+                    columns: pkColumns,
+                    definition: '',
+                });
+            }
+        }
+        for (const col of columns) {
+            if (col.isUnique && !col.isPrimaryKey) {
+                const uniqueName = `${tableName}_${col.name}_key`;
+                col.isUnique = false;
+                if (!constraints.some(c => c.name === uniqueName)) {
+                    constraints.push({
+                        name: uniqueName,
+                        type: 'UNIQUE',
+                        columns: [col.name],
+                        definition: '',
+                    });
+                }
+            }
+        }
+        const uniqueConstraintRegex = /constraint\.unique\s*\(\s*\{\s*name:\s*['"]([^'"]+)['"]\s*,\s*columns:\s*\[([^\]]+)\]/g;
+        let uniqueMatch;
+        while ((uniqueMatch = uniqueConstraintRegex.exec(optionsBlock)) !== null) {
+            const constraintName = uniqueMatch[1];
+            const colsStr = uniqueMatch[2];
+            const uniqueCols = colsStr.split(',').map(c => {
+                const tsColName = c.trim().replace(/table\.\s*/, '');
+                return tsToDbNameMap.get(tsColName) || tsColName;
+            });
+            if (!constraints.some(c => c.name === constraintName)) {
+                constraints.push({
+                    name: constraintName,
+                    type: 'UNIQUE',
+                    columns: uniqueCols,
+                    definition: '',
+                });
+            }
+        }
+        const pkConstraintRegex = /constraint\.primaryKey\s*\(\s*\{\s*name:\s*['"]([^'"]+)['"]\s*,\s*columns:\s*\[([^\]]+)\]/g;
+        let pkMatch;
+        while ((pkMatch = pkConstraintRegex.exec(optionsBlock)) !== null) {
+            const constraintName = pkMatch[1];
+            const colsStr = pkMatch[2];
+            const pkCols = colsStr.split(',').map(c => {
+                const tsColName = c.trim().replace(/table\.\s*/, '');
+                return tsToDbNameMap.get(tsColName) || tsColName;
+            });
+            const existingPkIdx = constraints.findIndex(c => c.type === 'PRIMARY KEY');
+            if (existingPkIdx >= 0) {
+                constraints[existingPkIdx] = {
+                    name: constraintName,
+                    type: 'PRIMARY KEY',
+                    columns: pkCols,
+                    definition: '',
+                };
+            }
+            else {
+                constraints.push({
+                    name: constraintName,
+                    type: 'PRIMARY KEY',
+                    columns: pkCols,
+                    definition: '',
+                });
+            }
+            for (const dbColName of pkCols) {
+                const col = columns.find(c => c.name === dbColName);
+                if (col) {
+                    col.isPrimaryKey = true;
+                }
+            }
+        }
         tables.push({
             name: tableName,
             schema: 'public',
@@ -306,6 +418,80 @@ function parseSchemaFileForComparison(schemaPath) {
             }
         }
     }
+    const toSnakeCase = (str) => {
+        return str.replace(/[A-Z]/g, letter => `_${letter.toLowerCase()}`);
+    };
+    const pgRelationsStart = content.indexOf('pgRelations(');
+    let relationsBlock = '';
+    if (pgRelationsStart !== -1) {
+        const arrowStart = content.indexOf('=> ({', pgRelationsStart);
+        if (arrowStart !== -1) {
+            const blockStart = arrowStart + 5;
+            let depth = 1;
+            let i = blockStart;
+            while (i < content.length && depth > 0) {
+                if (content[i] === '{')
+                    depth++;
+                else if (content[i] === '}')
+                    depth--;
+                i++;
+            }
+            relationsBlock = content.slice(blockStart, i - 1);
+        }
+    }
+    if (relationsBlock) {
+        const tableEntryRegex = /(\w+):\s*tables\.\1\s*\(\s*\([^)]*\)\s*=>\s*\(\{/g;
+        let tableMatch;
+        while ((tableMatch = tableEntryRegex.exec(relationsBlock)) !== null) {
+            const tableTsName = tableMatch[1];
+            const tableDbName = toSnakeCase(tableTsName);
+            const entryStart = tableMatch.index + tableMatch[0].length;
+            let depth = 1;
+            let j = entryStart;
+            while (j < relationsBlock.length && depth > 0) {
+                if (relationsBlock[j] === '{')
+                    depth++;
+                else if (relationsBlock[j] === '}')
+                    depth--;
+                j++;
+            }
+            const tableRelationsContent = relationsBlock.slice(entryStart, j - 1);
+            const table = tables.find(t => t.name === tableDbName);
+            if (!table)
+                continue;
+            const fkRegex = /(\w+):\s*r\.referenceTo\.(\w+)\s*\(\s*\w*\s*=>\s*\(\{([^}]*)\}\)/g;
+            let fkMatch;
+            while ((fkMatch = fkRegex.exec(tableRelationsContent)) !== null) {
+                const colTsName = fkMatch[1];
+                const refTableTsName = fkMatch[2];
+                const fkOptionsStr = fkMatch[3];
+                const colDbName = toSnakeCase(colTsName);
+                const refTableDbName = toSnakeCase(refTableTsName);
+                let onDelete;
+                let onUpdate;
+                const nameMatch = fkOptionsStr.match(/name:\s*['"]([^'"]+)['"]/);
+                const onDeleteMatch = fkOptionsStr.match(/onDelete:\s*['"]([^'"]+)['"]/);
+                const onUpdateMatch = fkOptionsStr.match(/onUpdate:\s*['"]([^'"]+)['"]/);
+                if (onDeleteMatch)
+                    onDelete = onDeleteMatch[1];
+                if (onUpdateMatch)
+                    onUpdate = onUpdateMatch[1];
+                const fkName = nameMatch ? nameMatch[1] : `${tableDbName}_${colDbName}_fkey`;
+                if (!table.constraints.some(c => c.name === fkName || (c.type === 'FOREIGN KEY' && c.columns?.includes(colDbName)))) {
+                    table.constraints.push({
+                        name: fkName,
+                        type: 'FOREIGN KEY',
+                        columns: [colDbName],
+                        definition: '',
+                        referencedTable: refTableDbName,
+                        referencedColumns: ['id'],
+                        onDelete,
+                        onUpdate,
+                    });
+                }
+            }
+        }
+    }
     return {
         tables,
         enums,
@@ -338,6 +524,7 @@ function snapshotToDatabaseSchema(snapshot) {
             scale: null,
             references: null,
             comment: c.comment || null,
+            trackingId: c.trackingId,
         })),
         indexes: (t.indexes || []).map(i => ({
             name: i.name,
@@ -348,6 +535,7 @@ function snapshotToDatabaseSchema(snapshot) {
             definition: i.definition || '',
             whereClause: i.whereClause || null,
             expression: null,
+            trackingId: i.trackingId,
         })),
         constraints: (t.constraints || []).map(c => ({
             name: c.name,
@@ -376,6 +564,57 @@ function snapshotToDatabaseSchema(snapshot) {
         extensions: (snapshot.extensions || []).map(e => typeof e === 'string' ? e : e.name),
     };
 }
+let trackingIdCounter = 0;
+function generateTrackingId(prefix) {
+    trackingIdCounter++;
+    const base = (Date.now().toString(36) + trackingIdCounter.toString(36)).slice(-5);
+    return prefix + base.padStart(5, '0');
+}
+function injectTrackingIds(schemaPath) {
+    if (!fs.existsSync(schemaPath)) {
+        return 0;
+    }
+    let content = fs.readFileSync(schemaPath, 'utf-8');
+    let injectedCount = 0;
+    const columnLineRegex = /^(\s*)(\w+):\s*(varchar|text|uuid|integer|bigint|boolean|timestamptz|timestamp|date|jsonb|json|numeric|serial|bigserial|smallserial|tsvector|smallint|real|doublePrecision|char|inet|cidr|macaddr|macaddr8|interval|timetz|time|point|line|lseg|box|path|polygon|circle|bytea|bit|varbit|money|xml|oid|enumType|domainType)\s*\([^)]*\)([^,\n]*)(,?)$/gm;
+    content = content.replace(columnLineRegex, (match, indent, colName, type, modifiers, comma) => {
+        if (modifiers.includes('.$id(')) {
+            return match;
+        }
+        const commentMatch = modifiers.match(/^(.*)(\.\s*comment\s*\([^)]+\))$/);
+        if (commentMatch) {
+            const beforeComment = commentMatch[1];
+            const comment = commentMatch[2];
+            const trackingId = generateTrackingId('c');
+            injectedCount++;
+            return `${indent}${colName}: ${type}(${match.split('(')[1].split(')')[0]})${beforeComment}.$id('${trackingId}')${comment}${comma}`;
+        }
+        const trackingId = generateTrackingId('c');
+        injectedCount++;
+        return `${indent}${colName}: ${type}(${match.split('(')[1].split(')')[0]})${modifiers}.$id('${trackingId}')${comma}`;
+    });
+    const indexLineRegex = /^(\s*)(index\s*\(\s*['"][^'"]+['"]\s*\)\s*\.on\s*\([^)]+\)[^,\n]*)(,?)$/gm;
+    content = content.replace(indexLineRegex, (match, indent, indexDef, comma) => {
+        if (indexDef.includes('.$id(')) {
+            return match;
+        }
+        const commentMatch = indexDef.match(/^(.*)(\.\s*comment\s*\([^)]+\))$/);
+        if (commentMatch) {
+            const beforeComment = commentMatch[1];
+            const comment = commentMatch[2];
+            const trackingId = generateTrackingId('i');
+            injectedCount++;
+            return `${indent}${beforeComment}.$id('${trackingId}')${comment}${comma}`;
+        }
+        const trackingId = generateTrackingId('i');
+        injectedCount++;
+        return `${indent}${indexDef}.$id('${trackingId}')${comma}`;
+    });
+    if (injectedCount > 0) {
+        fs.writeFileSync(schemaPath, content, 'utf-8');
+    }
+    return injectedCount;
+}
 async function addCommand(context) {
     const { args, projectRoot } = context;
     console.log('');
@@ -386,6 +625,10 @@ async function addCommand(context) {
     const config = await (0, config_1.loadConfig)();
     const schemaPathRaw = typeof config.schema === 'string' ? config.schema : './db/schema.ts';
     const schemaPath = path.resolve(projectRoot, schemaPathRaw);
+    const injectedCount = injectTrackingIds(schemaPath);
+    if (injectedCount > 0) {
+        console.log(spinner_1.colors.muted(`Injected ${injectedCount} tracking ID(s) into schema.ts`));
+    }
     const fileChange = (0, repo_manager_1.detectFileChanges)(schemaPath, projectRoot);
     if (fileChange) {
         const currentSchema = parseSchemaFileForComparison(schemaPath);
@@ -393,16 +636,13 @@ async function addCommand(context) {
         if (currentSchema && snapshot) {
             const snapshotAsDbSchema = snapshotToDatabaseSchema(snapshot);
             const schemaChanges = (0, schema_comparator_1.compareSchemas)(snapshotAsDbSchema, currentSchema);
+            (0, repo_manager_1.cleanupStagedChanges)(schemaChanges, projectRoot);
             if (schemaChanges.length > 0) {
                 (0, repo_manager_1.clearUnstagedChanges)(projectRoot);
                 (0, repo_manager_1.addUnstagedChanges)(schemaChanges, projectRoot);
             }
             else {
-                const existingUnstaged = (0, repo_manager_1.getUnstagedChanges)(projectRoot);
-                const hasFileChange = existingUnstaged.some(c => c.objectType === 'SCHEMA_FILE');
-                if (!hasFileChange) {
-                    (0, repo_manager_1.addUnstagedChanges)([fileChange], projectRoot);
-                }
+                (0, repo_manager_1.clearUnstagedChanges)(projectRoot);
             }
         }
     }

package/dist/cjs/cli/commands/commit.cjs

@@ -75,6 +75,7 @@ async function commitCommand(context) {
     const creates = staged.filter(c => c.type === 'CREATE').length;
     const alters = staged.filter(c => c.type === 'ALTER').length;
     const drops = staged.filter(c => c.type === 'DROP').length;
+    const renames = staged.filter(c => c.type === 'RENAME').length;
     const parentHash = (0, repo_manager_1.getHead)(projectRoot);
     const hashInput = JSON.stringify({
         changes: staged.map(c => c.id),
@@ -96,6 +97,7 @@ async function commitCommand(context) {
             creates,
             alters,
             drops,
+            renames,
             total: staged.length,
         },
     };
@@ -128,7 +130,16 @@ async function commitCommand(context) {
         (0, repo_manager_1.saveFileHash)(currentHash, projectRoot);
     }
     console.log(`[${(0, repo_manager_1.shortHash)(hash)}] ${message}`);
-    console.log(` ${creates} create(s), ${alters} alter(s), ${drops} drop(s)`);
+    const statsParts = [];
+    if (creates > 0)
+        statsParts.push(`${creates} create(s)`);
+    if (alters > 0)
+        statsParts.push(`${alters} alter(s)`);
+    if (drops > 0)
+        statsParts.push(`${drops} drop(s)`);
+    if (renames > 0)
+        statsParts.push(`${renames} rename(s)`);
+    console.log(` ${statsParts.length > 0 ? statsParts.join(', ') : 'no changes'}`);
     console.log('');
     (0, cli_utils_1.hint)("run 'relq push' to apply changes to database");
     (0, cli_utils_1.hint)("run 'relq export' to export as SQL file");

package/dist/cjs/cli/commands/export.cjs

@@ -42,7 +42,6 @@ const repo_manager_1 = require("../utils/repo-manager.cjs");
 const change_tracker_1 = require("../utils/change-tracker.cjs");
 const sql_generator_1 = require("../utils/sql-generator.cjs");
 const relqignore_1 = require("../utils/relqignore.cjs");
-const config_1 = require("../../config/config.cjs");
 async function exportCommand(context) {
     const spinner = (0, spinner_1.createSpinner)();
     const { args, flags, projectRoot } = context;
@@ -64,9 +63,9 @@ async function exportCommand(context) {
     if (fromDb) {
         return exportFromDatabase(context, absoluteOutputPath, options);
     }
-    return exportFromSnapshot(projectRoot, absoluteOutputPath, options);
+    return exportFromSnapshot(projectRoot, absoluteOutputPath, options, context.config ?? undefined);
 }
-async function exportFromSnapshot(projectRoot, absoluteOutputPath, options) {
+async function exportFromSnapshot(projectRoot, absoluteOutputPath, options, config) {
     const spinner = (0, spinner_1.createSpinner)();
     if (!(0, repo_manager_1.isInitialized)(projectRoot)) {
         console.log(`${spinner_1.colors.red('error:')} relq not initialized`);
@@ -83,13 +82,13 @@ async function exportFromSnapshot(projectRoot, absoluteOutputPath, options) {
         return;
     }
     spinner.succeed('Loaded snapshot');
-    const config = await (0, config_1.loadConfig)();
+    const cfg = config || {};
     const ignorePatterns = (0, relqignore_1.loadRelqignore)(projectRoot);
     const filteredSnapshot = filterNormalizedSchema(snapshot, ignorePatterns, {
-        includeFunctions: config.includeFunctions ?? options.includeFunctions ?? false,
-        includeTriggers: config.includeTriggers ?? options.includeTriggers ?? false,
-        includeViews: config.includeViews ?? false,
-        includeFDW: config.includeFDW ?? false,
+        includeFunctions: cfg.includeFunctions ?? options.includeFunctions ?? true,
+        includeTriggers: cfg.includeTriggers ?? options.includeTriggers ?? true,
+        includeViews: cfg.includeViews ?? false,
+        includeFDW: cfg.includeFDW ?? false,
     });
     const schema = normalizedToDbSchema(filteredSnapshot);
     const ignoredCount = (snapshot.tables.length - filteredSnapshot.tables.length) +
@@ -113,7 +112,12 @@ async function exportFromSnapshot(projectRoot, absoluteOutputPath, options) {
         console.log(`  ${spinner_1.colors.muted(`${ignoredCount} object(s) filtered by .relqignore`)}`);
     }
     spinner.start('Generating SQL statements');
-    const sqlContent = generateFullSQL(schema, options);
+    const exportOptions = {
+        ...options,
+        includeFunctions: cfg.includeFunctions ?? options.includeFunctions ?? true,
+        includeTriggers: cfg.includeTriggers ?? options.includeTriggers ?? true,
+    };
+    const sqlContent = generateFullSQL(schema, exportOptions);
     spinner.succeed('Generated SQL statements');
     const outputDir = path.dirname(absoluteOutputPath);
     if (!fs.existsSync(outputDir)) {
@@ -183,15 +187,9 @@ function normalizedToDbSchema(normalized) {
                 defaultValue: c.default,
                 isPrimaryKey: c.primaryKey,
                 isUnique: c.unique,
-                maxLength: null,
-                precision: null,
-                scale: null,
-                references: c.references ? {
-                    table: c.references.table,
-                    column: c.references.column,
-                    onDelete: c.references.onDelete,
-                    onUpdate: c.references.onUpdate,
-                } : null,
+                maxLength: c.maxLength ?? null,
+                precision: c.precision ?? null,
+                scale: c.scale ?? null,
                 check: c.check ?? undefined,
                 comment: c.comment ?? undefined,
                 isGenerated: c.isGenerated || false,
@@ -207,6 +205,7 @@ function normalizedToDbSchema(normalized) {
                 definition: i.definition,
                 whereClause: i.whereClause ?? null,
                 expression: i.columnDefs?.find(cd => cd.expression)?.expression ?? null,
+                comment: i.comment ?? null,
             })),
             constraints: t.constraints.map(c => ({
                 name: c.name,
@@ -221,6 +220,7 @@ function normalizedToDbSchema(normalized) {
             isPartitioned: t.isPartitioned || t.partitionType !== undefined,
             partitionType: t.partitionType,
             partitionKey: t.partitionKey,
+            comment: t.comment ?? null,
         })),
         enums: normalized.enums.map(e => ({
             name: e.name,
@@ -269,7 +269,13 @@ function normalizedToDbSchema(normalized) {
             definition: '',
             isEnabled: t.isEnabled !== false,
         })),
-        partitions: [],
+        partitions: normalized.tables.flatMap(t => (t.partitions || []).map(p => ({
+            name: p.name,
+            parentTable: t.name,
+            partitionBound: p.bound || '',
+            partitionType: (p.boundType || t.partitionType || 'LIST'),
+            partitionKey: t.partitionKey || [],
+        }))),
         policies: [],
         foreignServers: [],
         foreignTables: [],