orchid-orm 1.61.2 → 1.62.0

package/dist/migrations/index.mjs

@@ -124,7 +124,8 @@ const checkForColumnAddOrDrop = (shape, key) => {
 const processSchemas = async (ast, dbStructure, {
   codeItems: { schemas },
   verifying,
-  internal: { generatorIgnore }
+  internal: { generatorIgnore },
+  currentSchema
 }) => {
   const createSchemas = [];
   const dropSchemas = [];
@@ -134,7 +135,7 @@ const processSchemas = async (ast, dbStructure, {
     }
   }
   for (const schema of dbStructure.schemas) {
-    if (!schemas.has(schema) && schema !== "public" && !generatorIgnore?.schemas?.includes(schema)) {
+    if (!schemas.has(schema) && schema !== "public" && schema !== currentSchema && !generatorIgnore?.schemas?.includes(schema)) {
       dropSchemas.push(schema);
     }
   }
@@ -195,12 +196,12 @@ const renameSchemaInStructures = (items, from, to) => {
   }
 };
 
-const processExtensions = (ast, dbStructure, {
+const processExtensions = (config, ast, dbStructure, {
   currentSchema,
   internal: { extensions, generatorIgnore }
 }) => {
   const codeExtensions = extensions?.map((ext) => {
-    const [schema, name] = getSchemaAndTableFromName(ext.name);
+    const [schema, name] = getSchemaAndTableFromName(config, ext.name);
     return { schema, name, version: ext.version };
   });
   for (const dbExt of dbStructure.extensions) {
@@ -398,6 +399,7 @@ const changeColumns = async (adapter, config, structureToAstCtx, dbStructure, do
       dbColumnStructure
     );
     const action = await compareColumns(
+      config,
       adapter,
       domainsMap,
       ast,
@@ -435,13 +437,13 @@ const changeColumns = async (adapter, config, structureToAstCtx, dbStructure, do
     }
   }
 };
-const compareColumns = async (adapter, domainsMap, ast, currentSchema, compareSql, changeTableData, typeCastsCache, verifying, key, dbName, dbColumn, codeColumn) => {
+const compareColumns = async (config, adapter, domainsMap, ast, currentSchema, compareSql, changeTableData, typeCastsCache, verifying, key, dbName, dbColumn, codeColumn) => {
   if (dbColumn instanceof ArrayColumn && codeColumn instanceof ArrayColumn) {
     dbColumn = dbColumn.data.item;
     codeColumn = codeColumn.data.item;
   }
-  const dbType = getColumnDbType(dbColumn, currentSchema);
-  const codeType = getColumnDbType(codeColumn, currentSchema);
+  const dbType = getColumnDbType(config, dbColumn, currentSchema);
+  const codeType = getColumnDbType(config, codeColumn, currentSchema);
   if (dbType !== codeType) {
     const typeCasts = await getTypeCasts(adapter, typeCastsCache);
     const dbBaseType = getColumnBaseType(dbColumn, domainsMap, dbType);
@@ -598,9 +600,10 @@ const changeColumn = (changeTableData, key, dbName, dbColumn, codeColumn) => {
     to: { column: simpleCodeColumn }
   };
 };
-const getColumnDbType = (column, currentSchema) => {
+const getColumnDbType = (config, column, currentSchema) => {
   if (column instanceof EnumColumn) {
     const [schema = currentSchema, name] = getSchemaAndTableFromName(
+      config,
       column.enumName
     );
     return `${schema}.${name}`;
@@ -628,7 +631,7 @@ const renameColumn = (columns, from, to) => {
   }
 };
 
-const processDomains = async (ast, adapter, domainsMap, dbStructure, {
+const processDomains = async (config, ast, adapter, domainsMap, dbStructure, {
   codeItems: { domains },
   structureToAstCtx,
   currentSchema,
@@ -638,7 +641,7 @@ const processDomains = async (ast, adapter, domainsMap, dbStructure, {
   if (domains) {
     for (const { schemaName, name, column } of domains) {
       codeDomains.push(
-        makeComparableDomain(currentSchema, schemaName, name, column)
+        makeComparableDomain(config, currentSchema, schemaName, name, column)
       );
     }
   }
@@ -674,6 +677,7 @@ const processDomains = async (ast, adapter, domainsMap, dbStructure, {
       }));
     }
     const dbDomain = makeComparableDomain(
+      config,
       currentSchema,
       domain.schemaName,
       domain.name,
@@ -690,6 +694,7 @@ const processDomains = async (ast, adapter, domainsMap, dbStructure, {
       pushCompareDefault(compare, domain, found);
       pushCompareChecks(compare, domain, found);
       const source = `(VALUES (NULL::${getColumnDbType(
+        config,
         dbColumn,
         currentSchema
       )})) t(value)`;
@@ -743,7 +748,7 @@ const processDomains = async (ast, adapter, domainsMap, dbStructure, {
     }
   }
 };
-const makeComparableDomain = (currentSchema, schemaName, name, column) => {
+const makeComparableDomain = (config, currentSchema, schemaName, name, column) => {
   let arrayDims = 0;
   const isNullable = column.data.isNullable ?? false;
   let inner = column;
@@ -751,8 +756,11 @@ const makeComparableDomain = (currentSchema, schemaName, name, column) => {
     inner = inner.data.item;
     arrayDims++;
   }
-  const fullType = getColumnDbType(inner, currentSchema);
-  const [typeSchema = "pg_catalog", type] = getSchemaAndTableFromName(fullType);
+  const fullType = getColumnDbType(config, inner, currentSchema);
+  const [typeSchema = "pg_catalog", type] = getSchemaAndTableFromName(
+    config,
+    fullType
+  );
   return {
     schemaName,
     name,
@@ -1383,6 +1391,7 @@ const findMatchingItemWithoutSql = (dbItem, codeComparableItems, codeItems, skip
         const {
           // eslint-disable-next-line @typescript-eslint/no-unused-vars
           columnKeys,
+          // eslint-disable-next-line @typescript-eslint/no-unused-vars
           includeKeys,
           ...codeItemWithoutKeys
         } = codeItem;
@@ -1657,7 +1666,7 @@ const fnOrTableToString = (fnOrTable) => {
 };
 const parseForeignKey = (config, codeConstraint, references, currentSchema) => {
   const { fnOrTable, columns, foreignColumns, options } = references;
-  const [schema, table] = getSchemaAndTableFromName(fnOrTable);
+  const [schema, table] = getSchemaAndTableFromName(config, fnOrTable);
   return {
     references: {
       foreignSchema: schema ?? currentSchema,
@@ -1904,7 +1913,10 @@ const collectChangeAndDropTables = (config, tables, dbStructure, currentSchema,
   const dropTables = [];
   const tableShapes = {};
   const ignoreTables = generatorIgnore?.tables?.map((name) => {
-    const [schema = currentSchema, table] = getSchemaAndTableFromName(name);
+    const [schema = currentSchema, table] = getSchemaAndTableFromName(
+      config,
+      name
+    );
     return { schema, table };
   });
   const { schema: migrationsSchema = "public", table: migrationsTable } = getMigrationsSchemaAndTable(config);
@@ -2162,9 +2174,10 @@ const composeMigration = async (adapter, config, ast, dbStructure, params) => {
   const { structureToAstCtx, currentSchema } = params;
   const domainsMap = makeDomainsMap(structureToAstCtx, dbStructure);
   await processSchemas(ast, dbStructure, params);
-  processExtensions(ast, dbStructure, params);
+  processExtensions(config, ast, dbStructure, params);
   const pendingDbTypes = new PendingDbTypes();
   await processDomains(
+    config,
     ast,
     adapter,
     domainsMap,
@@ -2307,7 +2320,7 @@ const report = (ast, config, currentSchema) => {
               const column = change.item;
               const { primaryKey, indexes, excludes, foreignKeys, checks } = column.data;
               inner.push(
-                `${change.type === "add" ? green("+ add column") : red("- drop column")} ${key} ${column.data.alias ?? getColumnDbType(column, currentSchema)}${column.data.isNullable ? " nullable" : ""}${primaryKey ? " primary key" : ""}${foreignKeys ? ` references ${foreignKeys.map((fk) => {
+                `${change.type === "add" ? green("+ add column") : red("- drop column")} ${key} ${column.data.alias ?? getColumnDbType(config, column, currentSchema)}${column.data.isNullable ? " nullable" : ""}${primaryKey ? " primary key" : ""}${foreignKeys ? ` references ${foreignKeys.map((fk) => {
                   return `${fnOrTableToString(
                     fk.fnOrTable
                   )}(${fk.foreignColumns.join(", ")})`;
@@ -2319,14 +2332,18 @@ const report = (ast, config, currentSchema) => {
               const changes2 = [];
               inner.push(`${yellow("~ change column")} ${name}:`, changes2);
               changes2.push(`${yellow("from")}: `);
-              const fromCode = change.from.column.toCode(toCodeCtx, key);
-              for (const code2 of fromCode) {
-                addCode(changes2, code2);
+              const fromCode = change.from.column?.toCode(toCodeCtx, key);
+              if (fromCode) {
+                for (const code2 of fromCode) {
+                  addCode(changes2, code2);
+                }
               }
               changes2.push(`  ${yellow("to")}: `);
-              const toCode = change.to.column.toCode(toCodeCtx, key);
-              for (const code2 of toCode) {
-                addCode(changes2, code2);
+              const toCode = change.to.column?.toCode(toCodeCtx, key);
+              if (toCode) {
+                for (const code2 of toCode) {
+                  addCode(changes2, code2);
+                }
               }
             } else if (change.type === "rename") {
               inner.push(
@@ -2364,6 +2381,7 @@ const report = (ast, config, currentSchema) => {
           for (const { references } of a.drop.constraints) {
             if (!references) continue;
             const [schema, name] = getSchemaAndTableFromName(
+              config,
               references.fnOrTable
             );
             inner.push(
@@ -2554,10 +2572,11 @@ const generate = async (adapters, config, args, afterPull) => {
     afterPull
   );
   const [adapter] = adapters;
-  const currentSchema = config.schema ?? "public";
+  const currentSchema = (typeof config.schema === "function" ? config.schema() : config.schema) ?? "public";
   const db = await getDbFromConfig(config, dbPath);
   const { columnTypes, internal } = db.$qb;
   const codeItems = await getActualItems(
+    config,
     db,
     currentSchema,
     internal,
@@ -2629,7 +2648,7 @@ ${msg}`);
   }
   if (up) {
     for (const adapter2 of adapters) {
-      await migrateAndClose({ adapter: adapter2, config });
+      await migrateAndClose(adapter2, config);
     }
   } else if (!afterPull) {
     await closeAdapters(adapters);
@@ -2669,7 +2688,7 @@ const migrateAndPullStructures = async (adapters, config, afterPull) => {
     };
   }
   for (const adapter of adapters) {
-    await migrate({ adapter, config });
+    await migrate(adapter, config);
   }
   const dbStructures = await Promise.all(
     adapters.map((adapter) => introspectDbSchema(adapter))
@@ -2714,7 +2733,7 @@ const compareDbStructures = (a, b, i, path2) => {
     throw new Error(`${path2} in the db 0 does not match db ${i}`);
   }
 };
-const getActualItems = async (db, currentSchema, internal, columnTypes) => {
+const getActualItems = async (config, db, currentSchema, internal, columnTypes) => {
   const tableNames = /* @__PURE__ */ new Set();
   const habtmTables = /* @__PURE__ */ new Map();
   const codeItems = {
@@ -2757,6 +2776,7 @@ const getActualItems = async (db, currentSchema, internal, columnTypes) => {
         delete table.shape[key2];
       } else if (column instanceof DomainColumn) {
         const [schemaName = currentSchema, name2] = getSchemaAndTableFromName(
+          config,
           column.dataType
         );
         domains.set(column.dataType, {
@@ -2767,20 +2787,23 @@ const getActualItems = async (db, currentSchema, internal, columnTypes) => {
       } else {
         const en = column.dataType === "enum" ? column : column instanceof ArrayColumn && column.data.item.dataType === "enum" ? column.data.item : void 0;
         if (en) {
-          processEnumColumn(en, currentSchema, codeItems);
+          processEnumColumn(config, en, currentSchema, codeItems);
         }
       }
     }
   }
   if (internal.extensions) {
     for (const extension of internal.extensions) {
-      const [schema] = getSchemaAndTableFromName(extension.name);
+      const [schema] = getSchemaAndTableFromName(config, extension.name);
       if (schema) codeItems.schemas.add(schema);
     }
   }
   if (internal.domains) {
     for (const key in internal.domains) {
-      const [schemaName = currentSchema, name] = getSchemaAndTableFromName(key);
+      const [schemaName = currentSchema, name] = getSchemaAndTableFromName(
+        config,
+        key
+      );
       const column = internal.domains[key](columnTypes);
       domains.set(key, {
         schemaName,
@@ -2795,9 +2818,9 @@ const getActualItems = async (db, currentSchema, internal, columnTypes) => {
   }
   return codeItems;
 };
-const processEnumColumn = (column, currentSchema, codeItems) => {
+const processEnumColumn = (config, column, currentSchema, codeItems) => {
   const { enumName, options } = column;
-  const [schema, name] = getSchemaAndTableFromName(enumName);
+  const [schema, name] = getSchemaAndTableFromName(config, enumName);
   const enumSchema = schema ?? currentSchema;
   codeItems.enums.set(`${enumSchema}.${name}`, {
     schema: enumSchema,
@@ -3141,7 +3164,7 @@ const pull = async (adapters, config) => {
   const baseTablePath = config.baseTable.getFilePath();
   const baseTableExportedAs = config.baseTable.exportAs;
   const [adapter] = adapters;
-  const currentSchema = config.schema || "public";
+  const currentSchema = (typeof config.schema === "function" ? config.schema() : config.schema) ?? "public";
   const ctx = makeStructureToAstCtx(config, currentSchema);
   const asts = await structureToAst(ctx, adapter, config);
   const { tableInfos, fkeys } = getTableInfosAndFKeys(asts, config);