orchid-orm 1.50.2 → 1.50.4

package/dist/migrations.mjs

@@ -442,43 +442,14 @@ const changeColumns = async (adapter, config, structureToAstCtx, dbStructure, do
   }
 };
 const compareColumns = async (adapter, domainsMap, ast, currentSchema, compareSql, changeTableData, typeCastsCache, verifying, key, dbName, dbColumn, codeColumn) => {
-  const dbType = getColumnDbType(dbColumn, currentSchema);
-  const codeType = getColumnDbType(codeColumn, currentSchema);
   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);
   if (dbType !== codeType) {
-    let typeCasts = typeCastsCache.value;
-    if (!typeCasts) {
-      const { rows } = await adapter.arrays(`SELECT s.typname, t.typname
-FROM pg_cast
-JOIN pg_type AS s ON s.oid = castsource
-JOIN pg_type AS t ON t.oid = casttarget`);
-      const directTypeCasts = /* @__PURE__ */ new Map();
-      for (const [source, target] of rows) {
-        const set = directTypeCasts.get(source);
-        if (set) {
-          set.add(target);
-        } else {
-          directTypeCasts.set(source, /* @__PURE__ */ new Set([target]));
-        }
-      }
-      typeCasts = /* @__PURE__ */ new Map();
-      for (const [type, directSet] of directTypeCasts.entries()) {
-        const set = new Set(directSet);
-        typeCasts.set(type, set);
-        for (const subtype of directSet) {
-          const subset = directTypeCasts.get(subtype);
-          if (subset) {
-            for (const type2 of subset) {
-              set.add(type2);
-            }
-          }
-        }
-      }
-      typeCastsCache.value = typeCasts;
-    }
+    const typeCasts = await getTypeCasts(adapter, typeCastsCache);
     const dbBaseType = getColumnBaseType(dbColumn, domainsMap, dbType);
     const codeBaseType = getColumnBaseType(codeColumn, domainsMap, codeType);
     if (!typeCasts.get(dbBaseType)?.has(codeBaseType)) {
@@ -524,6 +495,15 @@ JOIN pg_type AS t ON t.oid = casttarget`);
       return "change";
     }
   }
+  if (dbColumn.data.isOfCustomType) {
+    const { typmod } = dbColumn.data;
+    if (typmod !== void 0 && typmod !== -1) {
+      const i = codeColumn.dataType.indexOf("(");
+      if (i === -1 || codeColumn.dataType.slice(i + 1, -1) !== `${typmod}`) {
+        return "change";
+      }
+    }
+  }
   if (!deepCompare(
     dbData.identity,
     codeData.identity ? {
@@ -570,6 +550,39 @@ JOIN pg_type AS t ON t.oid = casttarget`);
   }
   return;
 };
+const getTypeCasts = async (adapter, typeCastsCache) => {
+  let typeCasts = typeCastsCache.value;
+  if (!typeCasts) {
+    const { rows } = await adapter.arrays(`SELECT s.typname, t.typname
+FROM pg_cast
+JOIN pg_type AS s ON s.oid = castsource
+JOIN pg_type AS t ON t.oid = casttarget`);
+    const directTypeCasts = /* @__PURE__ */ new Map();
+    for (const [source, target] of rows) {
+      const set = directTypeCasts.get(source);
+      if (set) {
+        set.add(target);
+      } else {
+        directTypeCasts.set(source, /* @__PURE__ */ new Set([target]));
+      }
+    }
+    typeCasts = /* @__PURE__ */ new Map();
+    for (const [type, directSet] of directTypeCasts.entries()) {
+      const set = new Set(directSet);
+      typeCasts.set(type, set);
+      for (const subtype of directSet) {
+        const subset = directTypeCasts.get(subtype);
+        if (subset) {
+          for (const type2 of subset) {
+            set.add(type2);
+          }
+        }
+      }
+    }
+    typeCastsCache.value = typeCasts;
+  }
+  return typeCasts;
+};
 const changeColumn = (changeTableData, key, dbName, dbColumn, codeColumn) => {
   dbColumn.data.as = codeColumn.data.as = void 0;
   const simpleCodeColumn = Object.create(codeColumn);
@@ -598,7 +611,17 @@ const getColumnDbType = (column, currentSchema) => {
     );
     return column.enumName = `${schema}.${name}`;
   } else if (column instanceof ArrayColumn) {
-    return column.data.item.dataType + "[]".repeat(column.data.arrayDims);
+    const { item } = column.data;
+    let type = item instanceof EnumColumn ? item.enumName : item.dataType;
+    type = type.startsWith(currentSchema + ".") ? type.slice(currentSchema.length + 1) : type;
+    return type + "[]".repeat(column.data.arrayDims);
+  } else if (column.data.isOfCustomType) {
+    let type = column.dataType;
+    const i = type.indexOf("(");
+    if (i !== -1) {
+      type = type.slice(0, i);
+    }
+    return type.includes(".") ? type : currentSchema + "." + type;
   } else {
     return column.dataType;
   }
@@ -2234,6 +2257,7 @@ const report = (ast, config, currentSchema) => {
         const toCodeCtx = {
           t: "t",
           table: a.name,
+          currentSchema,
           migration: true,
           snakeCase: config.snakeCase
         };
@@ -2694,8 +2718,11 @@ const getActualItems = async (db, currentSchema, internal, columnTypes) => {
           name: name2,
           column: column.data.as ?? new UnknownColumn(defaultSchemaConfig)
         });
-      } else if (column.dataType === "enum") {
-        processEnumColumn(column, currentSchema, codeItems);
+      } 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);
+        }
       }
     }
   }
@@ -2811,7 +2838,7 @@ const getTableInfosAndFKeys = (asts, config) => {
   }
   return { tableInfos, fkeys };
 };
-const appCodeGenTable = (tableInfos, fkeys, ast, baseTablePath, baseTableExportedAs) => {
+const appCodeGenTable = (tableInfos, fkeys, ast, baseTablePath, baseTableExportedAs, currentSchema) => {
   const tableInfo = tableInfos[ast.schema ? `${ast.schema}.${ast.name}` : ast.name];
   const imports = {
     "orchid-orm": "Selectable, Insertable, Updatable",
@@ -2831,7 +2858,10 @@ const appCodeGenTable = (tableInfos, fkeys, ast, baseTablePath, baseTableExporte
   const hasTableData = Boolean(
     ast.primaryKey || ast.indexes?.length || ast.excludes?.length || ast.constraints?.length
   );
-  const shapeCode = columnsShapeToCode({ t: "t", table: ast.name }, ast.shape);
+  const shapeCode = columnsShapeToCode(
+    { t: "t", table: ast.name, currentSchema },
+    ast.shape
+  );
   props.push(
     `columns = this.setColumns(${hasTableData ? "\n    " : ""}(t) => ({`,
     hasTableData ? [shapeCode] : shapeCode,
@@ -2895,7 +2925,7 @@ function importsToCode(imports) {
 }
 
 const { createSourceFile, ScriptTarget, SyntaxKind } = typescript;
-const appCodeGenUpdateDbFile = async (dbPath, tables, extensions, domains) => {
+const appCodeGenUpdateDbFile = async (dbPath, tables, extensions, domains, currentSchema) => {
   const content = await fs.readFile(dbPath, "utf-8");
   const statements = getTsStatements(content);
   const importName = getOrchidOrmImportName(statements);
@@ -2923,7 +2953,7 @@ const appCodeGenUpdateDbFile = async (dbPath, tables, extensions, domains) => {
         (ast) => `${quoteObjectKey(
           ast.schema ? `${ast.schema}.${ast.name}` : ast.name
         )}: (t) => ${ast.baseType.toCode(
-          { t: "t", table: ast.name },
+          { t: "t", table: ast.name, currentSchema },
           ast.baseType.data.name ?? ""
         )},`
       ).join("\n    ")}
@@ -3083,7 +3113,8 @@ const pull = async (options, config) => {
           fkeys,
           ast,
           baseTablePath,
-          baseTableExportedAs
+          baseTableExportedAs,
+          currentSchema
         );
         tables[table.key] = table;
         if (!firstTable) firstTable = table;
@@ -3117,7 +3148,8 @@ const pull = async (options, config) => {
     dbPath,
     tables,
     extensions,
-    domains
+    domains,
+    currentSchema
   );
   if (content) pendingFileWrites.push([dbPath, content]);
   if (firstTable) {