rake-db 2.25.9 → 2.25.10

package/dist/index.mjs

@@ -1,4 +1,4 @@
-import { DomainColumn, escapeForMigration, EnumColumn, defaultSchemaConfig, getColumnTypes, parseTableData, escapeString, tableDataMethods, ColumnType, parseTableDataInput, UnknownColumn, raw, TransactionAdapter, logParamToLogObject, createDb as createDb$1, Adapter, makeColumnTypes, makeColumnsByType, RawSQL, CustomTypeColumn, assignDbDataToColumn, ArrayColumn, PostgisGeographyPointColumn, pushTableDataCode, primaryKeyInnerToCode, indexInnerToCode, excludeInnerToCode, constraintInnerToCode, referencesArgsToCode, TimestampTZColumn, TimestampColumn } from 'pqb';
+import { DomainColumn, escapeForMigration, ArrayColumn, EnumColumn, defaultSchemaConfig, getColumnTypes, parseTableData, escapeString, tableDataMethods, ColumnType, parseTableDataInput, UnknownColumn, raw, TransactionAdapter, logParamToLogObject, createDb as createDb$1, Adapter, makeColumnTypes, makeColumnsByType, RawSQL, CustomTypeColumn, assignDbDataToColumn, PostgisGeographyPointColumn, pushTableDataCode, primaryKeyInnerToCode, indexInnerToCode, excludeInnerToCode, constraintInnerToCode, referencesArgsToCode, TimestampTZColumn, TimestampColumn } from 'pqb';
 import { singleQuote, toSnakeCase, isRawSQL, toCamelCase, toArray, snakeCaseKey, emptyObject, setCurrentColumnName, consumeColumnName, ColumnTypeBase, setDefaultLanguage, deepCompare, getImportPath, pathToLog, emptyArray, getStackTrace, codeToString, addCode, quoteObjectKey, backtickQuote } from 'orchid-core';
 import path, { join } from 'path';
 import { pathToFileURL, fileURLToPath } from 'node:url';
@@ -160,7 +160,7 @@ const encodeColumnDefault = (def, values, column) => {
       return def.toSQL({ values });
     } else {
       return escapeForMigration(
-        column?.data.encode ? column.data.encode(def) : def
+        column instanceof ArrayColumn && Array.isArray(def) ? "{" + (column.data.item.data.encode ? def.map((x) => column.data.item.data.encode(x)) : def).join(",") + "}" : column?.data.encode ? column.data.encode(def) : def
       );
     }
   }
@@ -1177,8 +1177,9 @@ const handleTableItemChange = (key, item, ast, alterTable, renameItems, values,
     if (to.type && (from.type !== to.type || from.collate !== to.collate)) {
       changeType = true;
       const type = !to.column || to.column.data.isOfCustomType ? to.column && to.column instanceof DomainColumn ? quoteNameFromString(to.type) : quoteCustomType(to.type) : to.type;
+      const using = item.using?.usingUp ? ` USING ${item.using.usingUp.toSQL({ values })}` : to.column instanceof EnumColumn ? ` USING "${name}"::text::${type}` : to.column instanceof ArrayColumn ? ` USING "${name}"::text[]::${type}` : "";
       alterTable.push(
-        `ALTER COLUMN "${name}" TYPE ${type}${to.collate ? ` COLLATE ${quoteNameFromString(to.collate)}` : ""}${item.using?.usingUp ? ` USING ${item.using.usingUp.toSQL({ values })}` : to.column instanceof EnumColumn ? ` USING "${name}"::text::${type}` : ""}`
+        `ALTER COLUMN "${name}" TYPE ${type}${to.collate ? ` COLLATE ${quoteNameFromString(to.collate)}` : ""}${using}`
       );
     }
     if (typeof from.identity !== typeof to.identity || !deepCompare(from.identity, to.identity)) {
@@ -2481,11 +2482,15 @@ const recreateEnum = async (migration, { schema, name }, values, errorMessage) =
   const { rows: tables } = await migration.adapter.query(
     `SELECT n.nspname AS "schema",
   c.relname AS "table",
-  json_agg(a.attname ORDER BY a.attnum) AS "columns"
+  json_agg(
+    json_build_object('name', a.attname, 'arrayDims', a.attndims)
+    ORDER BY a.attnum
+  ) AS "columns"
 FROM pg_class c
 JOIN pg_catalog.pg_namespace n ON n.oid = relnamespace
-JOIN pg_attribute a ON a.attrelid = c.oid
-JOIN pg_type t ON a.atttypid = t.oid AND t.typname = ${singleQuote(name)}
+JOIN pg_type bt ON bt.typname = ${singleQuote(name)}
+JOIN pg_type t ON t.oid = bt.oid OR t.typelem = bt.oid
+JOIN pg_attribute a ON a.attrelid = c.oid AND a.atttypid = t.oid
 JOIN pg_namespace tn ON tn.oid = t.typnamespace AND tn.nspname = ${singleQuote(
       schema ?? defaultSchema
     )}
@@ -2494,7 +2499,9 @@ GROUP BY n.nspname, c.relname`
   );
   const sql = tables.map(
     (t) => `ALTER TABLE ${quoteTable(t.schema, t.table)}
-        ${t.columns.map((c) => `  ALTER COLUMN "${c}" TYPE text`).join(",\n")}`
+        ${t.columns.map(
+      (c) => `  ALTER COLUMN "${c.name}" TYPE text${"[]".repeat(c.arrayDims)}`
+    ).join(",\n")}`
   );
   sql.push(
     `DROP TYPE ${quotedName}`,
@@ -2504,14 +2511,17 @@ GROUP BY n.nspname, c.relname`
   for (const t of tables) {
     const table = quoteTable(t.schema, t.table);
     for (const c of t.columns) {
+      const type = quotedName + "[]".repeat(c.arrayDims);
       try {
         await migration.adapter.query(
           `ALTER TABLE ${table}
-  ALTER COLUMN "${c}" TYPE ${quotedName} USING "${c}"::${quotedName}`
+  ALTER COLUMN "${c.name}" TYPE ${type} USING "${c.name}"::${type}`
         );
       } catch (err) {
         if (err.code === "22P02") {
-          throw new Error(errorMessage(quotedName, table, c), { cause: err });
+          throw new Error(errorMessage(quotedName, table, c.name), {
+            cause: err
+          });
         }
         throw err;
       }
@@ -4238,7 +4248,8 @@ const instantiateDbColumn = (ctx, data, domains, dbColumn) => {
     if (domainColumn) {
       column = new DomainColumn(
         ctx.columnSchemaConfig,
-        typeId,
+        typeName,
+        typeSchema,
         dbColumn.extension
       ).as(domainColumn);
     } else {
@@ -4255,7 +4266,8 @@ const instantiateDbColumn = (ctx, data, domains, dbColumn) => {
       } else {
         column = new CustomTypeColumn(
           ctx.columnSchemaConfig,
-          typeId,
+          typeName,
+          typeSchema === "pg_catalog" ? void 0 : typeSchema,
           dbColumn.extension
         );
         ((_a = ctx.unsupportedTypes)[_b = dbColumn.type] ?? (_a[_b] = [])).push(
@@ -4939,7 +4951,7 @@ ${group.map(
   return code;
 };
 const astEncoders = {
-  table(ast, config) {
+  table(ast, config, currentSchema) {
     let code = [];
     const result = code;
     const hasOptions = Boolean(ast.comment || ast.noPrimaryKey === "ignore");
@@ -4973,6 +4985,7 @@ const astEncoders = {
     const toCodeCtx = {
       t: "t",
       table: ast.name,
+      currentSchema,
       migration: true,
       snakeCase: config.snakeCase
     };
@@ -5028,6 +5041,7 @@ const astEncoders = {
     const toCodeCtx = {
       t: "t",
       table: ast.name,
+      currentSchema,
       migration: true,
       snakeCase: config.snakeCase
     };
@@ -5060,6 +5074,7 @@ const astEncoders = {
             {
               t: "t",
               table: ast.name,
+              currentSchema,
               migration: true,
               snakeCase: config.snakeCase
             },
@@ -5196,11 +5211,11 @@ const astEncoders = {
       ast
     )}, [${ast.fromValues.map(singleQuote).join(", ")}], [${ast.toValues.map(singleQuote).join(", ")}]);`;
   },
-  domain(ast) {
+  domain(ast, _, currentSchema) {
     return `await db.${ast.action}Domain(${quoteSchemaTable(
       ast
     )}, (t) => ${ast.baseType.toCode(
-      { t: "t", table: ast.name },
+      { t: "t", table: ast.name, currentSchema },
       ast.baseType.data.name ?? ""
     )});`;
   },