drizzle-seed 0.3.1-dc3b366 → 0.3.1-dfa4db3

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');
@@ -133097,7 +133098,7 @@ class GeneratePoint extends AbstractGenerator {
         }
         const x = this.state.xCoordinateGen.generate();
         const y = this.state.yCoordinateGen.generate();
-        if (this.dataType === 'json') {
+        if (this.dataType === 'object') {
             return { x, y };
         }
         else if (this.dataType === 'string') {
@@ -133134,7 +133135,7 @@ class GenerateUniquePoint extends AbstractGenerator {
         }
         const x = this.state.xCoordinateGen.generate();
         const y = this.state.yCoordinateGen.generate();
-        if (this.dataType === 'json') {
+        if (this.dataType === 'object') {
             return { x, y };
         }
         else if (this.dataType === 'string') {
@@ -133183,7 +133184,7 @@ class GenerateLine extends AbstractGenerator {
             b = this.state.bCoefficientGen.generate();
         }
         const c = this.state.cCoefficientGen.generate();
-        if (this.dataType === 'json') {
+        if (this.dataType === 'object') {
             return { a, b, c };
         }
         else if (this.dataType === 'string') {
@@ -133231,7 +133232,7 @@ class GenerateUniqueLine extends AbstractGenerator {
             b = this.state.bCoefficientGen.generate();
         }
         const c = this.state.cCoefficientGen.generate();
-        if (this.dataType === 'json') {
+        if (this.dataType === 'object') {
             return { a, b, c };
         }
         else if (this.dataType === 'string') {
@@ -134945,7 +134946,7 @@ class SeedService {
                 const generator = new generatorsMap.GenerateBoolean[0]();
                 return generator;
             }
-            if ((col.columnType === 'integer' && col.dataType === 'date')) {
+            if ((col.columnType === 'integer' && col.dataType === 'object')) {
                 const generator = new generatorsMap.GenerateTimestamp[0]();
                 return generator;
             }
@@ -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 };
 };
@@ -135570,20 +135571,22 @@ const getPostgresInfo = (pgSchema, pgTables) => {
             return dbToTsColumnNamesMap;
         }
         const tableConfig = pgCore.getTableConfig(table);
-        for (const [tsCol, col] of Object.entries(tableConfig.columns[0].table)) {
-            dbToTsColumnNamesMap[col.name] = tsCol;
+        for (const [tsCol, col] of Object.entries(drizzleOrm.getColumnTable(tableConfig.columns[0]))) {
+            // dbToTsColumnNamesMap[col.name] = tsCol;
+            if (drizzleOrm.is(col, drizzleOrm.Column))
+                dbToTsColumnNamesMap[col.name] = tsCol;
         }
         dbToTsColumnNamesMapGlobal[tableName] = dbToTsColumnNamesMap;
         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)
                 continue;
             for (const drizzleRel of Object.values(table.relations)) {
-                if (!drizzleOrm.is(drizzleRel, drizzleOrm.One))
+                if (!drizzleOrm.is(drizzleRel, _relations.One))
                     continue;
                 const tableConfig = pgCore.getTableConfig(drizzleRel.sourceTable);
                 const tableDbSchema = tableConfig.schema ?? 'public';
@@ -135627,10 +135630,7 @@ const getPostgresInfo = (pgSchema, pgTables) => {
     };
     for (const table of Object.values(pgTables)) {
         tableConfig = pgCore.getTableConfig(table);
-        dbToTsColumnNamesMap = {};
-        for (const [tsCol, col] of Object.entries(tableConfig.columns[0].table)) {
-            dbToTsColumnNamesMap[col.name] = tsCol;
-        }
+        dbToTsColumnNamesMap = getDbToTsColumnNamesMap(table);
         // might be empty list
         const newRelations = tableConfig.foreignKeys.map((fk) => {
             const table = dbToTsTableNamesMap[tableConfig.name];
@@ -135661,8 +135661,8 @@ const getPostgresInfo = (pgSchema, pgTables) => {
                 name: baseColumn.name,
                 columnType: baseColumn.getSQLType(),
                 typeParams: getTypeParams(baseColumn.getSQLType()),
-                dataType: baseColumn.dataType,
-                size: baseColumn.size,
+                dataType: baseColumn.dataType.split(' ')[0],
+                size: baseColumn.length,
                 hasDefault: baseColumn.hasDefault,
                 enumValues: baseColumn.enumValues,
                 default: baseColumn.default,
@@ -135714,8 +135714,8 @@ const getPostgresInfo = (pgSchema, pgTables) => {
                 name: dbToTsColumnNamesMap[column.name],
                 columnType: column.getSQLType(),
                 typeParams: getTypeParams(column.getSQLType()),
-                dataType: column.dataType,
-                size: column.size,
+                dataType: column.dataType.split(' ')[0],
+                size: column.length,
                 hasDefault: column.hasDefault,
                 default: column.default,
                 enumValues: column.enumValues,
@@ -135789,7 +135789,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 };
 };
@@ -135820,20 +135820,21 @@ const getMySqlInfo = (mysqlSchema, mysqlTables) => {
             return dbToTsColumnNamesMap;
         }
         const tableConfig = mysqlCore.getTableConfig(table);
-        for (const [tsCol, col] of Object.entries(tableConfig.columns[0].table)) {
-            dbToTsColumnNamesMap[col.name] = tsCol;
+        for (const [tsCol, col] of Object.entries(drizzleOrm.getColumnTable(tableConfig.columns[0]))) {
+            if (drizzleOrm.is(col, drizzleOrm.Column))
+                dbToTsColumnNamesMap[col.name] = tsCol;
         }
         dbToTsColumnNamesMapGlobal[tableName] = dbToTsColumnNamesMap;
         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)
                 continue;
             for (const drizzleRel of Object.values(table.relations)) {
-                if (!drizzleOrm.is(drizzleRel, drizzleOrm.One))
+                if (!drizzleOrm.is(drizzleRel, _relations.One))
                     continue;
                 const tableConfig = mysqlCore.getTableConfig(drizzleRel.sourceTable);
                 const tableDbSchema = tableConfig.schema ?? 'public';
@@ -135877,10 +135878,7 @@ const getMySqlInfo = (mysqlSchema, mysqlTables) => {
     };
     for (const table of Object.values(mysqlTables)) {
         tableConfig = mysqlCore.getTableConfig(table);
-        dbToTsColumnNamesMap = {};
-        for (const [tsCol, col] of Object.entries(tableConfig.columns[0].table)) {
-            dbToTsColumnNamesMap[col.name] = tsCol;
-        }
+        dbToTsColumnNamesMap = getDbToTsColumnNamesMap(table);
         const newRelations = tableConfig.foreignKeys.map((fk) => {
             const table = dbToTsTableNamesMap[tableConfig.name];
             const refTable = dbToTsTableNamesMap[drizzleOrm.getTableName(fk.reference().foreignTable)];
@@ -135935,7 +135933,7 @@ const getMySqlInfo = (mysqlSchema, mysqlTables) => {
                 name: dbToTsColumnNamesMap[column.name],
                 columnType: column.getSQLType(),
                 typeParams: getTypeParams(column.getSQLType()),
-                dataType: column.dataType,
+                dataType: column.dataType.split(' ')[0],
                 hasDefault: column.hasDefault,
                 default: column.default,
                 enumValues: column.enumValues,
@@ -135975,7 +135973,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 };
 };
@@ -136006,20 +136004,21 @@ const getSqliteInfo = (sqliteSchema, sqliteTables) => {
             return dbToTsColumnNamesMap;
         }
         const tableConfig = sqliteCore.getTableConfig(table);
-        for (const [tsCol, col] of Object.entries(tableConfig.columns[0].table)) {
-            dbToTsColumnNamesMap[col.name] = tsCol;
+        for (const [tsCol, col] of Object.entries(drizzleOrm.getColumnTable(tableConfig.columns[0]))) {
+            if (drizzleOrm.is(col, drizzleOrm.Column))
+                dbToTsColumnNamesMap[col.name] = tsCol;
         }
         dbToTsColumnNamesMapGlobal[tableName] = dbToTsColumnNamesMap;
         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)
                 continue;
             for (const drizzleRel of Object.values(table.relations)) {
-                if (!drizzleOrm.is(drizzleRel, drizzleOrm.One))
+                if (!drizzleOrm.is(drizzleRel, _relations.One))
                     continue;
                 const tableConfig = sqliteCore.getTableConfig(drizzleRel.sourceTable);
                 const tableDbName = tableConfig.name;
@@ -136062,10 +136061,7 @@ const getSqliteInfo = (sqliteSchema, sqliteTables) => {
     };
     for (const table of Object.values(sqliteTables)) {
         tableConfig = sqliteCore.getTableConfig(table);
-        dbToTsColumnNamesMap = {};
-        for (const [tsCol, col] of Object.entries(tableConfig.columns[0].table)) {
-            dbToTsColumnNamesMap[col.name] = tsCol;
-        }
+        dbToTsColumnNamesMap = getDbToTsColumnNamesMap(table);
         const newRelations = tableConfig.foreignKeys.map((fk) => {
             const table = dbToTsTableNamesMap[tableConfig.name];
             const refTable = dbToTsTableNamesMap[drizzleOrm.getTableName(fk.reference().foreignTable)];
@@ -136116,7 +136112,7 @@ const getSqliteInfo = (sqliteSchema, sqliteTables) => {
                 name: dbToTsColumnNamesMap[column.name],
                 columnType: column.getSQLType(),
                 typeParams: getTypeParams(column.getSQLType()),
-                dataType: column.dataType,
+                dataType: column.dataType.split(' ')[0],
                 hasDefault: column.hasDefault,
                 default: column.default,
                 enumValues: column.enumValues,