drizzle-seed 0.3.0 → 0.3.1-00df263

package/index.cjs

@@ -1,6 +1,7 @@
 'use strict';
 
 var drizzleOrm = require('drizzle-orm');
+var _relations = require('drizzle-orm/_relations');
 var mysqlCore = require('drizzle-orm/mysql-core');
 var pgCore = require('drizzle-orm/pg-core');
 var sqliteCore = require('drizzle-orm/sqlite-core');
@@ -134483,7 +134484,7 @@ class SeedService {
                 orderedTablesNames.push(parent);
             }
             else {
-                leafTablesNames.push(parent);
+                leafTablesNames.push(...tablesInOutRelations[parent].requiredTableNames, parent);
                 continue;
             }
             children = [...tablesInOutRelations[parent].dependantTableNames];
@@ -135214,7 +135215,7 @@ class SeedService {
             maxParametersNumber = this.mysqlMaxParametersNumber;
         }
         else {
-            // is(db, BaseSQLiteDatabase<any, any>)
+            // is(db, BaseSQLiteDatabase<any, any, any, any, any ,any>)
             maxParametersNumber = this.sqliteMaxParametersNumber;
         }
         const maxBatchSize = Math.floor(maxParametersNumber / columnsNumber);
@@ -135539,7 +135540,7 @@ const resetPostgres = async (db, pgTables) => {
     await db.execute(drizzleOrm.sql.raw(`truncate ${tablesToTruncate.join(',')} cascade;`));
 };
 const filterPgSchema = (schema) => {
-    const pgSchema = Object.fromEntries(Object.entries(schema).filter((keyValue) => drizzleOrm.is(keyValue[1], pgCore.PgTable) || drizzleOrm.is(keyValue[1], drizzleOrm.Relations)));
+    const pgSchema = Object.fromEntries(Object.entries(schema).filter((keyValue) => drizzleOrm.is(keyValue[1], pgCore.PgTable) || drizzleOrm.is(keyValue[1], _relations.Relations)));
     const pgTables = Object.fromEntries(Object.entries(schema).filter((keyValue) => drizzleOrm.is(keyValue[1], pgCore.PgTable)));
     return { pgSchema, pgTables };
 };
@@ -135577,42 +135578,50 @@ const getPostgresInfo = (pgSchema, pgTables) => {
         return dbToTsColumnNamesMap;
     };
     const transformFromDrizzleRelation = (schema, getDbToTsColumnNamesMap, tableRelations) => {
-        const schemaConfig = drizzleOrm.extractTablesRelationalConfig(schema, drizzleOrm.createTableRelationsHelpers);
+        const schemaConfig = _relations.extractTablesRelationalConfig(schema, _relations.createTableRelationsHelpers);
         const relations = [];
         for (const table of Object.values(schemaConfig.tables)) {
-            if (table.relations !== undefined) {
-                for (const drizzleRel of Object.values(table.relations)) {
-                    if (drizzleOrm.is(drizzleRel, drizzleOrm.One)) {
-                        const tableConfig = pgCore.getTableConfig(drizzleRel.sourceTable);
-                        const tableDbSchema = tableConfig.schema ?? 'public';
-                        const tableDbName = tableConfig.name;
-                        const tableTsName = schemaConfig.tableNamesMap[`${tableDbSchema}.${tableDbName}`] ?? tableDbName;
-                        const dbToTsColumnNamesMap = getDbToTsColumnNamesMap(drizzleRel.sourceTable);
-                        const columns = drizzleRel.config?.fields.map((field) => dbToTsColumnNamesMap[field.name])
-                            ?? [];
-                        const refTableConfig = pgCore.getTableConfig(drizzleRel.referencedTable);
-                        const refTableDbSchema = refTableConfig.schema ?? 'public';
-                        const refTableDbName = refTableConfig.name;
-                        const refTableTsName = schemaConfig.tableNamesMap[`${refTableDbSchema}.${refTableDbName}`]
-                            ?? refTableDbName;
-                        const dbToTsColumnNamesMapForRefTable = getDbToTsColumnNamesMap(drizzleRel.referencedTable);
-                        const refColumns = drizzleRel.config?.references.map((ref) => dbToTsColumnNamesMapForRefTable[ref.name])
-                            ?? [];
-                        if (tableRelations[refTableTsName] === undefined) {
-                            tableRelations[refTableTsName] = [];
-                        }
-                        const relation = {
-                            table: tableTsName,
-                            columns,
-                            refTable: refTableTsName,
-                            refColumns,
-                            refTableRels: tableRelations[refTableTsName],
-                            type: 'one',
-                        };
-                        relations.push(relation);
-                        tableRelations[tableTsName].push(relation);
-                    }
+            if (table.relations === undefined)
+                continue;
+            for (const drizzleRel of Object.values(table.relations)) {
+                if (!drizzleOrm.is(drizzleRel, _relations.One))
+                    continue;
+                const tableConfig = pgCore.getTableConfig(drizzleRel.sourceTable);
+                const tableDbSchema = tableConfig.schema ?? 'public';
+                const tableDbName = tableConfig.name;
+                const tableTsName = schemaConfig.tableNamesMap[`${tableDbSchema}.${tableDbName}`] ?? tableDbName;
+                const dbToTsColumnNamesMap = getDbToTsColumnNamesMap(drizzleRel.sourceTable);
+                const columns = drizzleRel.config?.fields.map((field) => dbToTsColumnNamesMap[field.name])
+                    ?? [];
+                const refTableConfig = pgCore.getTableConfig(drizzleRel.referencedTable);
+                const refTableDbSchema = refTableConfig.schema ?? 'public';
+                const refTableDbName = refTableConfig.name;
+                const refTableTsName = schemaConfig.tableNamesMap[`${refTableDbSchema}.${refTableDbName}`]
+                    ?? refTableDbName;
+                const dbToTsColumnNamesMapForRefTable = getDbToTsColumnNamesMap(drizzleRel.referencedTable);
+                const refColumns = drizzleRel.config?.references.map((ref) => dbToTsColumnNamesMapForRefTable[ref.name])
+                    ?? [];
+                if (tableRelations[refTableTsName] === undefined) {
+                    tableRelations[refTableTsName] = [];
+                }
+                const relation = {
+                    table: tableTsName,
+                    columns,
+                    refTable: refTableTsName,
+                    refColumns,
+                    refTableRels: tableRelations[refTableTsName],
+                    type: 'one',
+                };
+                // do not add duplicate relation
+                if (tableRelations[tableTsName]?.some((rel) => rel.table === relation.table
+                    && rel.refTable === relation.refTable)) {
+                    console.warn(`You are providing a one-to-many relation between the '${relation.refTable}' and '${relation.table}' tables,\n`
+                        + `while the '${relation.table}' table object already has foreign key constraint in the schema referencing '${relation.refTable}' table.\n`
+                        + `In this case, the foreign key constraint will be used.\n`);
+                    continue;
                 }
+                relations.push(relation);
+                tableRelations[tableTsName].push(relation);
             }
         }
         return relations;
@@ -135781,7 +135790,7 @@ const resetMySql = async (db, schema) => {
     await db.execute(drizzleOrm.sql.raw('SET FOREIGN_KEY_CHECKS = 1;'));
 };
 const filterMysqlTables = (schema) => {
-    const mysqlSchema = Object.fromEntries(Object.entries(schema).filter((keyValue) => drizzleOrm.is(keyValue[1], mysqlCore.MySqlTable) || drizzleOrm.is(keyValue[1], drizzleOrm.Relations)));
+    const mysqlSchema = Object.fromEntries(Object.entries(schema).filter((keyValue) => drizzleOrm.is(keyValue[1], mysqlCore.MySqlTable) || drizzleOrm.is(keyValue[1], _relations.Relations)));
     const mysqlTables = Object.fromEntries(Object.entries(schema).filter((keyValue) => drizzleOrm.is(keyValue[1], mysqlCore.MySqlTable)));
     return { mysqlSchema, mysqlTables };
 };
@@ -135819,42 +135828,50 @@ const getMySqlInfo = (mysqlSchema, mysqlTables) => {
         return dbToTsColumnNamesMap;
     };
     const transformFromDrizzleRelation = (schema, getDbToTsColumnNamesMap, tableRelations) => {
-        const schemaConfig = drizzleOrm.extractTablesRelationalConfig(schema, drizzleOrm.createTableRelationsHelpers);
+        const schemaConfig = _relations.extractTablesRelationalConfig(schema, _relations.createTableRelationsHelpers);
         const relations = [];
         for (const table of Object.values(schemaConfig.tables)) {
-            if (table.relations !== undefined) {
-                for (const drizzleRel of Object.values(table.relations)) {
-                    if (drizzleOrm.is(drizzleRel, drizzleOrm.One)) {
-                        const tableConfig = mysqlCore.getTableConfig(drizzleRel.sourceTable);
-                        const tableDbSchema = tableConfig.schema ?? 'public';
-                        const tableDbName = tableConfig.name;
-                        const tableTsName = schemaConfig.tableNamesMap[`${tableDbSchema}.${tableDbName}`] ?? tableDbName;
-                        const dbToTsColumnNamesMap = getDbToTsColumnNamesMap(drizzleRel.sourceTable);
-                        const columns = drizzleRel.config?.fields.map((field) => dbToTsColumnNamesMap[field.name])
-                            ?? [];
-                        const refTableConfig = mysqlCore.getTableConfig(drizzleRel.referencedTable);
-                        const refTableDbSchema = refTableConfig.schema ?? 'public';
-                        const refTableDbName = refTableConfig.name;
-                        const refTableTsName = schemaConfig.tableNamesMap[`${refTableDbSchema}.${refTableDbName}`]
-                            ?? refTableDbName;
-                        const dbToTsColumnNamesMapForRefTable = getDbToTsColumnNamesMap(drizzleRel.referencedTable);
-                        const refColumns = drizzleRel.config?.references.map((ref) => dbToTsColumnNamesMapForRefTable[ref.name])
-                            ?? [];
-                        if (tableRelations[refTableTsName] === undefined) {
-                            tableRelations[refTableTsName] = [];
-                        }
-                        const relation = {
-                            table: tableTsName,
-                            columns,
-                            refTable: refTableTsName,
-                            refColumns,
-                            refTableRels: tableRelations[refTableTsName],
-                            type: 'one',
-                        };
-                        relations.push(relation);
-                        tableRelations[tableTsName].push(relation);
-                    }
+            if (table.relations === undefined)
+                continue;
+            for (const drizzleRel of Object.values(table.relations)) {
+                if (!drizzleOrm.is(drizzleRel, _relations.One))
+                    continue;
+                const tableConfig = mysqlCore.getTableConfig(drizzleRel.sourceTable);
+                const tableDbSchema = tableConfig.schema ?? 'public';
+                const tableDbName = tableConfig.name;
+                const tableTsName = schemaConfig.tableNamesMap[`${tableDbSchema}.${tableDbName}`] ?? tableDbName;
+                const dbToTsColumnNamesMap = getDbToTsColumnNamesMap(drizzleRel.sourceTable);
+                const columns = drizzleRel.config?.fields.map((field) => dbToTsColumnNamesMap[field.name])
+                    ?? [];
+                const refTableConfig = mysqlCore.getTableConfig(drizzleRel.referencedTable);
+                const refTableDbSchema = refTableConfig.schema ?? 'public';
+                const refTableDbName = refTableConfig.name;
+                const refTableTsName = schemaConfig.tableNamesMap[`${refTableDbSchema}.${refTableDbName}`]
+                    ?? refTableDbName;
+                const dbToTsColumnNamesMapForRefTable = getDbToTsColumnNamesMap(drizzleRel.referencedTable);
+                const refColumns = drizzleRel.config?.references.map((ref) => dbToTsColumnNamesMapForRefTable[ref.name])
+                    ?? [];
+                if (tableRelations[refTableTsName] === undefined) {
+                    tableRelations[refTableTsName] = [];
+                }
+                const relation = {
+                    table: tableTsName,
+                    columns,
+                    refTable: refTableTsName,
+                    refColumns,
+                    refTableRels: tableRelations[refTableTsName],
+                    type: 'one',
+                };
+                // do not add duplicate relation
+                if (tableRelations[tableTsName]?.some((rel) => rel.table === relation.table
+                    && rel.refTable === relation.refTable)) {
+                    console.warn(`You are providing a one-to-many relation between the '${relation.refTable}' and '${relation.table}' tables,\n`
+                        + `while the '${relation.table}' table object already has foreign key constraint in the schema referencing '${relation.refTable}' table.\n`
+                        + `In this case, the foreign key constraint will be used.\n`);
+                    continue;
                 }
+                relations.push(relation);
+                tableRelations[tableTsName].push(relation);
             }
         }
         return relations;
@@ -135959,7 +135976,7 @@ const resetSqlite = async (db, schema) => {
     await db.run(drizzleOrm.sql.raw('PRAGMA foreign_keys = ON'));
 };
 const filterSqliteTables = (schema) => {
-    const sqliteSchema = Object.fromEntries(Object.entries(schema).filter((keyValue) => drizzleOrm.is(keyValue[1], sqliteCore.SQLiteTable) || drizzleOrm.is(keyValue[1], drizzleOrm.Relations)));
+    const sqliteSchema = Object.fromEntries(Object.entries(schema).filter((keyValue) => drizzleOrm.is(keyValue[1], sqliteCore.SQLiteTable) || drizzleOrm.is(keyValue[1], _relations.Relations)));
     const sqliteTables = Object.fromEntries(Object.entries(schema).filter((keyValue) => drizzleOrm.is(keyValue[1], sqliteCore.SQLiteTable)));
     return { sqliteSchema, sqliteTables };
 };
@@ -135997,41 +136014,49 @@ const getSqliteInfo = (sqliteSchema, sqliteTables) => {
         return dbToTsColumnNamesMap;
     };
     const transformFromDrizzleRelation = (schema, getDbToTsColumnNamesMap, tableRelations) => {
-        const schemaConfig = drizzleOrm.extractTablesRelationalConfig(schema, drizzleOrm.createTableRelationsHelpers);
+        const schemaConfig = _relations.extractTablesRelationalConfig(schema, _relations.createTableRelationsHelpers);
         const relations = [];
         for (const table of Object.values(schemaConfig.tables)) {
-            if (table.relations !== undefined) {
-                for (const drizzleRel of Object.values(table.relations)) {
-                    if (drizzleOrm.is(drizzleRel, drizzleOrm.One)) {
-                        const tableConfig = sqliteCore.getTableConfig(drizzleRel.sourceTable);
-                        const tableDbName = tableConfig.name;
-                        // TODO: tableNamesMap: have {public.customer: 'customer'} structure in sqlite
-                        const tableTsName = schemaConfig.tableNamesMap[`public.${tableDbName}`] ?? tableDbName;
-                        const dbToTsColumnNamesMap = getDbToTsColumnNamesMap(drizzleRel.sourceTable);
-                        const columns = drizzleRel.config?.fields.map((field) => dbToTsColumnNamesMap[field.name])
-                            ?? [];
-                        const refTableConfig = sqliteCore.getTableConfig(drizzleRel.referencedTable);
-                        const refTableDbName = refTableConfig.name;
-                        const refTableTsName = schemaConfig.tableNamesMap[`public.${refTableDbName}`]
-                            ?? refTableDbName;
-                        const dbToTsColumnNamesMapForRefTable = getDbToTsColumnNamesMap(drizzleRel.referencedTable);
-                        const refColumns = drizzleRel.config?.references.map((ref) => dbToTsColumnNamesMapForRefTable[ref.name])
-                            ?? [];
-                        if (tableRelations[refTableTsName] === undefined) {
-                            tableRelations[refTableTsName] = [];
-                        }
-                        const relation = {
-                            table: tableTsName,
-                            columns,
-                            refTable: refTableTsName,
-                            refColumns,
-                            refTableRels: tableRelations[refTableTsName],
-                            type: 'one',
-                        };
-                        relations.push(relation);
-                        tableRelations[tableTsName].push(relation);
-                    }
+            if (table.relations === undefined)
+                continue;
+            for (const drizzleRel of Object.values(table.relations)) {
+                if (!drizzleOrm.is(drizzleRel, _relations.One))
+                    continue;
+                const tableConfig = sqliteCore.getTableConfig(drizzleRel.sourceTable);
+                const tableDbName = tableConfig.name;
+                // TODO: tableNamesMap: have {public.customer: 'customer'} structure in sqlite
+                const tableTsName = schemaConfig.tableNamesMap[`public.${tableDbName}`] ?? tableDbName;
+                const dbToTsColumnNamesMap = getDbToTsColumnNamesMap(drizzleRel.sourceTable);
+                const columns = drizzleRel.config?.fields.map((field) => dbToTsColumnNamesMap[field.name])
+                    ?? [];
+                const refTableConfig = sqliteCore.getTableConfig(drizzleRel.referencedTable);
+                const refTableDbName = refTableConfig.name;
+                const refTableTsName = schemaConfig.tableNamesMap[`public.${refTableDbName}`]
+                    ?? refTableDbName;
+                const dbToTsColumnNamesMapForRefTable = getDbToTsColumnNamesMap(drizzleRel.referencedTable);
+                const refColumns = drizzleRel.config?.references.map((ref) => dbToTsColumnNamesMapForRefTable[ref.name])
+                    ?? [];
+                if (tableRelations[refTableTsName] === undefined) {
+                    tableRelations[refTableTsName] = [];
+                }
+                const relation = {
+                    table: tableTsName,
+                    columns,
+                    refTable: refTableTsName,
+                    refColumns,
+                    refTableRels: tableRelations[refTableTsName],
+                    type: 'one',
+                };
+                // do not add duplicate relation
+                if (tableRelations[tableTsName]?.some((rel) => rel.table === relation.table
+                    && rel.refTable === relation.refTable)) {
+                    console.warn(`You are providing a one-to-many relation between the '${relation.refTable}' and '${relation.table}' tables,\n`
+                        + `while the '${relation.table}' table object already has foreign key constraint in the schema referencing '${relation.refTable}' table.\n`
+                        + `In this case, the foreign key constraint will be used.\n`);
+                    continue;
                 }
+                relations.push(relation);
+                tableRelations[tableTsName].push(relation);
             }
         }
         return relations;