rake-db 2.4.13 → 2.4.15

package/dist/index.d.ts

@@ -40,6 +40,7 @@ declare type TextColumnCreator = () => TextColumn;
 declare type MigrationColumnTypes = Omit<ColumnTypes, 'text' | 'string' | 'enum'> & {
     text: TextColumnCreator;
     string: TextColumnCreator;
+    citext: TextColumnCreator;
     enum: (name: string) => EnumColumn;
 };
 declare type ColumnsShapeCallback = (t: MigrationColumnTypes & {

package/dist/index.js

@@ -2051,7 +2051,7 @@ const fkeyActionMap = {
   n: "SET NULL",
   d: "SET DEFAULT"
 };
-const structureToAst = async (config, db) => {
+const structureToAst = async (unsupportedTypes, db) => {
   const ast = [];
   const data = await getData(db);
   for (const name of data.schemas) {
@@ -2079,19 +2079,24 @@ const structureToAst = async (config, db) => {
   }
   const domains = {};
   for (const it of data.domains) {
-    domains[`${it.schemaName}.${it.name}`] = getColumn(config, data, domains, {
-      schemaName: it.schemaName,
-      name: it.name,
-      type: it.type,
-      typeSchema: it.typeSchema,
-      isArray: it.isArray,
-      isSerial: false
-    });
+    domains[`${it.schemaName}.${it.name}`] = getColumn(
+      unsupportedTypes,
+      data,
+      domains,
+      {
+        schemaName: it.schemaName,
+        name: it.name,
+        type: it.type,
+        typeSchema: it.typeSchema,
+        isArray: it.isArray,
+        isSerial: false
+      }
+    );
   }
   for (const key in pendingTables) {
     const { table, dependsOn } = pendingTables[key];
     if (!dependsOn.size) {
-      pushTableAst(config, ast, data, domains, table, pendingTables);
+      pushTableAst(unsupportedTypes, ast, data, domains, table, pendingTables);
     }
   }
   const outerFKeys = [];
@@ -2139,7 +2144,15 @@ const structureToAst = async (config, db) => {
         outerFKeys.push([fkey, table]);
       }
     }
-    pushTableAst(config, ast, data, domains, table, pendingTables, innerFKeys);
+    pushTableAst(
+      unsupportedTypes,
+      ast,
+      data,
+      domains,
+      table,
+      pendingTables,
+      innerFKeys
+    );
   }
   for (const [fkey, table] of outerFKeys) {
     ast.push(__spreadProps(__spreadValues({}, foreignKeyToAst(fkey)), {
@@ -2199,7 +2212,7 @@ const getIsSerial = (item) => {
   }
   return false;
 };
-const getColumn = (config, data, domains, _a) => {
+const getColumn = (unsupportedTypes, data, domains, _a) => {
   var _b = _a, {
     schemaName,
     tableName,
@@ -2234,8 +2247,8 @@ const getColumn = (config, data, domains, _a) => {
         column = new RakeDbEnumColumn({}, type, enumType.values);
       } else {
         column = new pqb.CustomTypeColumn({}, type);
-        (_a2 = config.logger) == null ? void 0 : _a2.warn(
-          `${tableName ? "Column" : "Domain"} ${schemaName}${tableName ? `.${tableName}` : ""}.${name} has unsupported type \`${type}\`, append \`as\` method manually to it to treat it as other column type`
+        ((_a2 = unsupportedTypes[type]) != null ? _a2 : unsupportedTypes[type] = []).push(
+          `${schemaName}${tableName ? `.${tableName}` : ""}.${name}`
         );
       }
     }
@@ -2247,7 +2260,7 @@ const getColumnType = (type, isSerial) => {
     return type;
   return type === "int2" ? "smallserial" : type === "int4" ? "serial" : "bigserial";
 };
-const pushTableAst = (config, ast, data, domains, table, pendingTables, innerFKeys = data.foreignKeys) => {
+const pushTableAst = (unsupportedTypes, ast, data, domains, table, pendingTables, innerFKeys = data.foreignKeys) => {
   const { schemaName, name } = table;
   const key = `${schemaName}.${table.name}`;
   delete pendingTables[key];
@@ -2270,7 +2283,7 @@ const pushTableAst = (config, ast, data, domains, table, pendingTables, innerFKe
     if (isSerial) {
       item = __spreadProps(__spreadValues({}, item), { default: void 0 });
     }
-    let column = getColumn(config, data, domains, __spreadProps(__spreadValues({}, item), {
+    let column = getColumn(unsupportedTypes, data, domains, __spreadProps(__spreadValues({}, item), {
       type: item.type,
       isArray: item.isArray,
       isSerial
@@ -2353,7 +2366,14 @@ const pushTableAst = (config, ast, data, domains, table, pendingTables, innerFKe
   for (const otherKey in pendingTables) {
     const item = pendingTables[otherKey];
     if (item.dependsOn.delete(key) && item.dependsOn.size === 0) {
-      pushTableAst(config, ast, data, domains, item.table, pendingTables);
+      pushTableAst(
+        unsupportedTypes,
+        ast,
+        data,
+        domains,
+        item.table,
+        pendingTables
+      );
     }
   }
 };
@@ -2403,7 +2423,7 @@ const astToMigration = (config, ast) => {
   if (first.length) {
     code += `
 change(async (db) => {
-${pqb.codeToString(first, "  ", "  ")}
+${orchidCore.codeToString(first, "  ", "  ")}
 });
 `;
   }
@@ -2411,7 +2431,7 @@ ${pqb.codeToString(first, "  ", "  ")}
     for (const table of tables) {
       code += `
 change(async (db) => {
-${pqb.codeToString(table, "  ", "  ")}
+${orchidCore.codeToString(table, "  ", "  ")}
 });
 `;
     }
@@ -2419,7 +2439,7 @@ ${pqb.codeToString(table, "  ", "  ")}
   if (foreignKeys.length) {
     code += `
 change(async (db) => {
-${pqb.codeToString(foreignKeys, "  ", "  ")}
+${orchidCore.codeToString(foreignKeys, "  ", "  ")}
 });
 `;
   }
@@ -2459,9 +2479,9 @@ const createDomain = (ast) => {
     if (ast.collation)
       props.push(`collation: ${orchidCore.singleQuote(ast.collation)},`);
     if (ast.default)
-      props.push(`default: ${pqb.rawToCode("db", ast.default)},`);
+      props.push(`default: ${orchidCore.rawToCode("db", ast.default)},`);
     if (ast.check)
-      props.push(`check: ${pqb.rawToCode("db", ast.check)},`);
+      props.push(`check: ${orchidCore.rawToCode("db", ast.check)},`);
     orchidCore.addCode(code, ", {");
     code.push(props);
     orchidCore.addCode(code, "}");
@@ -2525,10 +2545,11 @@ const createForeignKey = (item) => {
 };
 
 const pullDbStructure = async (options, config) => {
-  var _a;
+  var _a, _b, _c;
   const adapter = new pqb.Adapter(options);
   const db = new DbStructure(adapter);
-  const ast = await structureToAst(config, db);
+  const unsupportedTypes = {};
+  const ast = await structureToAst(unsupportedTypes, db);
   await adapter.close();
   const result = astToMigration(config, ast);
   if (!result)
@@ -2546,6 +2567,21 @@ const pullDbStructure = async (options, config) => {
       logger: config.logger
     }));
   }
+  const unsupportedEntries = Object.entries(unsupportedTypes);
+  const len = unsupportedEntries.length;
+  if (len) {
+    let count = 0;
+    (_b = config.logger) == null ? void 0 : _b.warn(
+      `Found unsupported types:
+${unsupportedEntries.map(([type, columns]) => {
+        count += columns.length;
+        return `- ${type} is used for column${columns.length > 1 ? "s" : ""} ${columns.join(", ")}`;
+      }).join("\n")}
+Append \`as\` method manually to ${count > 1 ? "these" : "this"} column${count > 1 ? "s" : ""} to treat ${count > 1 ? "them" : "it"} as other column type`
+    );
+  }
+  (_c = config.logger) == null ? void 0 : _c.log("Database pulled successfully");
+  adapter.close();
 };
 
 const rakeDb = async (options, partialConfig = {}, args = process.argv.slice(2)) => {