rake-db 2.4.2 → 2.4.3

package/dist/index.mjs

@@ -1,5 +1,5 @@
-import { quote, getRaw, EnumColumn, columnTypes, getColumnTypes, getTableData, ColumnType, resetTableData, TransactionAdapter, logParamToLogObject, createDb as createDb$1, Adapter, columnsByType, instantiateColumn, codeToString, primaryKeyToCode, indexToCode, foreignKeyToCode, TimestampColumn, foreignKeyArgsToCode } from 'pqb';
-import { singleQuote, isRaw, toArray, snakeCaseKey, nameKey, emptyObject, addCode, quoteObjectKey } from 'orchid-core';
+import { quote, getRaw, EnumColumn, columnTypes, getColumnTypes, getTableData, ColumnType, resetTableData, TransactionAdapter, logParamToLogObject, createDb as createDb$1, Adapter, columnsByType, instantiateColumn, ArrayColumn, columnCode, codeToString, primaryKeyToCode, indexToCode, foreignKeyToCode, TimestampColumn, foreignKeyArgsToCode } from 'pqb';
+import { singleQuote, isRaw, toArray, snakeCaseKey, nameKey, emptyObject, pathToLog, addCode, quoteObjectKey } from 'orchid-core';
 import path from 'path';
 import { readdir, mkdir, writeFile } from 'fs/promises';
 import prompts from 'prompts';
@@ -37,10 +37,15 @@ const migrationConfigDefaults = {
 const processRakeDbConfig = (config) => {
   var _a;
   const result = __spreadValues$6(__spreadValues$6({}, migrationConfigDefaults), config);
+  if (!result.log) {
+    delete result.logger;
+  }
   if (!result.basePath) {
     let stack;
+    const original = Error.prepareStackTrace;
     Error.prepareStackTrace = (_, s) => stack = s;
     new Error().stack;
+    Error.prepareStackTrace = original;
     if (stack) {
       const thisFile = (_a = stack[0]) == null ? void 0 : _a.getFileName();
       const thisDir = thisFile && path.dirname(thisFile);
@@ -137,16 +142,17 @@ const setAdminCredentialsToOptions = async (options, create) => {
   }));
 };
 const createSchemaMigrations = async (db, config) => {
+  var _a, _b;
   try {
     await db.query(
       `CREATE TABLE ${quoteWithSchema({
         name: config.migrationsTable
       })} ( version TEXT NOT NULL )`
     );
-    console.log("Created versions table");
+    (_a = config.logger) == null ? void 0 : _a.log("Created versions table");
   } catch (err) {
     if (err.code === "42P07") {
-      console.log("Versions table exists");
+      (_b = config.logger) == null ? void 0 : _b.log("Versions table exists");
     } else {
       throw err;
     }
@@ -500,7 +506,7 @@ const createTable$1 = async (migration, up, tableName, options, fn) => {
     const result = await migration.adapter.arrays(query);
     then == null ? void 0 : then(result);
   }
-  await runCodeUpdater(migration, ast);
+  migration.migratedAsts.push(ast);
 };
 const makeAst$1 = (up, tableName, shape, tableData, options, noPrimaryKey) => {
   const shapePKeys = [];
@@ -758,7 +764,7 @@ const changeTable = async (migration, up, tableName, options, fn) => {
     const result = await migration.adapter.arrays(query);
     (_a = query.then) == null ? void 0 : _a.call(query, result);
   }
-  await runCodeUpdater(migration, ast);
+  migration.migratedAsts.push(ast);
 };
 const makeAst = (up, name, changeData, changeTableData2, options) => {
   const { comment } = options;
@@ -996,7 +1002,7 @@ var __spreadValues$2 = (a, b) => {
   return a;
 };
 var __spreadProps$1 = (a, b) => __defProps$1(a, __getOwnPropDescs$1(b));
-const createMigrationInterface = (tx, up, options, adapterOptions, appCodeUpdaterCache) => {
+const createMigrationInterface = (tx, up, options) => {
   const adapter = new TransactionAdapter(tx, tx.client, tx.types);
   const { query, arrays } = adapter;
   const log = logParamToLogObject(options.logger || console, options.log);
@@ -1011,13 +1017,12 @@ const createMigrationInterface = (tx, up, options, adapterOptions, appCodeUpdate
   for (const key of Object.getOwnPropertyNames(proto)) {
     db[key] = proto[key];
   }
+  db.migratedAsts = [];
   return Object.assign(db, {
     adapter,
     log,
     up,
-    options,
-    adapterOptions,
-    appCodeUpdaterCache
+    options
   });
 };
 class MigrationBase {
@@ -1054,7 +1059,7 @@ class MigrationBase {
         name: ast.to
       })}`
     );
-    await runCodeUpdater(this, ast);
+    this.migratedAsts.push(ast);
   }
   addColumn(tableName, columnName, fn) {
     return addColumn(this, this.up, tableName, columnName, fn);
@@ -1177,7 +1182,7 @@ const createSchema$1 = async (migration, up, name) => {
   await migration.adapter.query(
     `${ast.action === "create" ? "CREATE" : "DROP"} SCHEMA "${name}"`
   );
-  await runCodeUpdater(migration, ast);
+  migration.migratedAsts.push(ast);
 };
 const createExtension$1 = async (migration, up, name, options) => {
   const ast = __spreadValues$2({
@@ -1192,7 +1197,7 @@ const createExtension$1 = async (migration, up, name, options) => {
     query = `CREATE EXTENSION${ast.createIfNotExists ? " IF NOT EXISTS" : ""} "${ast.name}"${ast.schema ? ` SCHEMA "${ast.schema}"` : ""}${ast.version ? ` VERSION '${ast.version}'` : ""}${ast.cascade ? " CASCADE" : ""}`;
   }
   await migration.adapter.query(query);
-  await runCodeUpdater(migration, ast);
+  migration.migratedAsts.push(ast);
 };
 const createEnum$1 = async (migration, up, name, values, options = {}) => {
   const [schema, enumName] = getSchemaAndTableFromName(name);
@@ -1211,20 +1216,11 @@ const createEnum$1 = async (migration, up, name, values, options = {}) => {
     query = `DROP TYPE${ast.dropIfExists ? " IF EXISTS" : ""} ${quotedName}${ast.cascade ? " CASCADE" : ""}`;
   }
   await migration.adapter.query(query);
-  await runCodeUpdater(migration, ast);
+  migration.migratedAsts.push(ast);
 };
 const queryExists = (db, sql) => {
   return db.adapter.query(sql).then(({ rowCount }) => rowCount > 0);
 };
-const runCodeUpdater = (migration, ast) => {
-  var _a, _b;
-  return (_b = (_a = migration.options).appCodeUpdater) == null ? void 0 : _b.call(_a, {
-    ast,
-    options: migration.adapterOptions,
-    basePath: migration.options.basePath,
-    cache: migration.appCodeUpdaterCache
-  });
-};
 
 var __defProp$1 = Object.defineProperty;
 var __getOwnPropSymbols$1 = Object.getOwnPropertySymbols;
@@ -1285,7 +1281,9 @@ const migrateOrRollback = async (options, config, args, up) => {
           opts,
           appCodeUpdaterCache
         );
-        (_c = config.logger) == null ? void 0 : _c.log(`${file.path} ${up ? "migrated" : "rolled back"}`);
+        (_c = config.logger) == null ? void 0 : _c.log(
+          `${up ? "Migrated" : "Rolled back"} ${pathToLog(file.path)}`
+        );
       }
       if (up) {
         await ((_d = config.afterMigrate) == null ? void 0 : _d.call(config, db != null ? db : db = getDb(adapter)));
@@ -1300,14 +1298,9 @@ const migrateOrRollback = async (options, config, args, up) => {
 };
 const changeCache = {};
 const processMigration = async (db, up, file, config, options, appCodeUpdaterCache) => {
-  await db.transaction(async (tx) => {
-    const db2 = createMigrationInterface(
-      tx,
-      up,
-      config,
-      options,
-      appCodeUpdaterCache
-    );
+  var _a;
+  const asts = await db.transaction(async (tx) => {
+    const db2 = createMigrationInterface(tx, up, config);
     clearChanges();
     let changes = changeCache[file.path];
     if (!changes) {
@@ -1329,7 +1322,17 @@ const processMigration = async (db, up, file, config, options, appCodeUpdaterCac
       file.version,
       config
     );
+    return db2.migratedAsts;
   });
+  for (const ast of asts) {
+    await ((_a = config.appCodeUpdater) == null ? void 0 : _a.call(config, {
+      ast,
+      options,
+      basePath: config.basePath,
+      cache: appCodeUpdaterCache,
+      logger: config.logger
+    }));
+  }
 };
 const saveMigratedVersion = async (db, version, config) => {
   await db.query(
@@ -1384,17 +1387,20 @@ const execute = async (options, sql) => {
   }
 };
 const createOrDrop = async (options, adminOptions, config, args) => {
+  var _a, _b, _c, _d;
   const params = getDatabaseAndUserFromOptions(options);
   const result = await execute(
     setAdapterOptions(adminOptions, { database: "postgres" }),
     args.sql(params)
   );
   if (result === "ok") {
-    console.log(args.successMessage(params));
+    (_a = config.logger) == null ? void 0 : _a.log(args.successMessage(params));
   } else if (result === "already") {
-    console.log(args.alreadyMessage(params));
+    (_b = config.logger) == null ? void 0 : _b.log(args.alreadyMessage(params));
   } else if (result === "ssl required") {
-    console.log("SSL is required: append ?ssl=true to the database url string");
+    (_c = config.logger) == null ? void 0 : _c.log(
+      "SSL is required: append ?ssl=true to the database url string"
+    );
     return;
   } else if (result === "forbidden") {
     let message = `Permission denied to ${args.create ? "create" : "drop"} database.`;
@@ -1404,7 +1410,7 @@ const createOrDrop = async (options, adminOptions, config, args) => {
       message += `
 Don't use this command for database service providers, only for a local db.`;
     }
-    console.log(message);
+    (_d = config.logger) == null ? void 0 : _d.log(message);
     const updatedOptions = await setAdminCredentialsToOptions(
       options,
       args.create
@@ -1460,13 +1466,14 @@ const resetDb = async (arg, config) => {
 };
 
 const writeMigrationFile = async (config, name, content) => {
+  var _a;
   await mkdir(config.migrationsPath, { recursive: true });
   const filePath = path.resolve(
     config.migrationsPath,
     `${makeFileTimeStamp()}_${name}.ts`
   );
   await writeFile(filePath, content);
-  console.log(`Created ${filePath}`);
+  (_a = config.logger) == null ? void 0 : _a.log(`Created ${pathToLog(filePath)}`);
 };
 const generate = async (config, args) => {
   const name = args[0];
@@ -1604,6 +1611,8 @@ WHERE ${filterSchema("n.nspname")}`
   table_name "tableName",
   column_name "name",
   udt_name "type",
+  udt_schema "typeSchema",
+  data_type "dataType",
   character_maximum_length AS "maxChars",
   numeric_precision AS "numericPrecision",
   numeric_scale AS "numericScale",
@@ -1843,6 +1852,11 @@ var __spreadValues = (a, b) => {
   return a;
 };
 var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
+class RakeDbEnumColumn extends EnumColumn {
+  toCode(t) {
+    return columnCode(this, t, `enum('${this.enumName}')`);
+  }
+}
 const matchMap = {
   s: void 0,
   f: "FULL",
@@ -1973,11 +1987,10 @@ const getIsSerial = (item) => {
   }
   return false;
 };
-const getColumnType = (item, isSerial) => {
-  if (isSerial) {
-    return item.type === "int2" ? "smallserial" : item.type === "int4" ? "serial" : "bigserial";
-  }
-  return item.type;
+const getColumnType = (type, isSerial) => {
+  if (!isSerial)
+    return type;
+  return type === "int2" ? "smallserial" : type === "int4" ? "serial" : "bigserial";
 };
 const pushTableAst = (ast, data, table, pendingTables, innerFKeys = data.foreignKeys) => {
   const { schemaName, name } = table;
@@ -1996,11 +2009,26 @@ const pushTableAst = (ast, data, table, pendingTables, innerFKeys = data.foreign
     if (isSerial) {
       item = __spreadProps(__spreadValues({}, item), { default: void 0 });
     }
-    const klass = columnsByType[getColumnType(item, isSerial)];
-    if (!klass) {
-      throw new Error(`Column type \`${item.type}\` is not supported`);
+    let column;
+    const isArray = item.dataType === "ARRAY";
+    const type = isArray ? item.type.slice(1) : item.type;
+    const klass = columnsByType[getColumnType(type, isSerial)];
+    if (klass) {
+      column = instantiateColumn(klass, item);
+    } else {
+      const { type: type2, typeSchema } = item;
+      const enumType = data.enums.find(
+        (item2) => item2.name === type2 && item2.schemaName === typeSchema
+      );
+      if (!enumType) {
+        throw new Error(
+          `Cannot handle column ${item.schemaName}.${item.tableName}.${item.name}: column type \`${item.type}\` is not supported`
+        );
+      }
+      column = new RakeDbEnumColumn({}, type2, enumType.values);
     }
-    let column = instantiateColumn(klass, item);
+    if (isArray)
+      column = new ArrayColumn({}, column);
     if ((primaryKey == null ? void 0 : primaryKey.columnNames.length) === 1 && (primaryKey == null ? void 0 : primaryKey.columnNames[0]) === item.name) {
       column = column.primaryKey();
     }
@@ -2240,7 +2268,7 @@ const pullDbStructure = async (options, config) => {
 };
 
 const rakeDb = async (options, partialConfig = {}, args = process.argv.slice(2)) => {
-  var _a;
+  var _a, _b, _c;
   const config = processRakeDbConfig(partialConfig);
   const command = (_a = args[0]) == null ? void 0 : _a.split(":")[0];
   try {
@@ -2261,18 +2289,17 @@ const rakeDb = async (options, partialConfig = {}, args = process.argv.slice(2))
     } else if (config.commands[command]) {
       await config.commands[command](toArray(options), config, args.slice(1));
     } else {
-      printHelp();
+      (_b = config.logger) == null ? void 0 : _b.log(help);
     }
   } catch (err) {
     if (err instanceof RakeDbError) {
-      console.error(err.message);
+      (_c = config.logger) == null ? void 0 : _c.error(err.message);
       process.exit(1);
     }
     throw err;
   }
 };
-const printHelp = () => console.log(
-  `Usage: rake-db [command] [arguments]
+const help = `Usage: rake-db [command] [arguments]
 
 Commands:
   create                  create databases
@@ -2306,8 +2333,7 @@ Generate arguments:
 
 - other arguments considered as columns with types and optional methods:
   rake-db g createTable id:serial.primaryKey name:text.nullable
-`
-);
+`;
 
-export { MigrationBase, change, changeCache, createDb, createMigrationInterface, dropDb, generate, migrate, migrateOrRollback, rakeDb, resetDb, rollback, runCodeUpdater, writeMigrationFile };
+export { MigrationBase, change, changeCache, createDb, createMigrationInterface, dropDb, generate, migrate, migrateOrRollback, rakeDb, resetDb, rollback, writeMigrationFile };
 //# sourceMappingURL=index.mjs.map