rake-db 2.2.1 → 2.2.4

package/dist/index.js

@@ -13,21 +13,21 @@ function _interopDefaultLegacy (e) { return e && typeof e === 'object' && 'defau
 var Enquirer__default = /*#__PURE__*/_interopDefaultLegacy(Enquirer);
 var path__default = /*#__PURE__*/_interopDefaultLegacy(path);
 
-var __defProp$5 = Object.defineProperty;
+var __defProp$6 = Object.defineProperty;
 var __defProps$5 = Object.defineProperties;
 var __getOwnPropDescs$5 = Object.getOwnPropertyDescriptors;
-var __getOwnPropSymbols$5 = Object.getOwnPropertySymbols;
-var __hasOwnProp$5 = Object.prototype.hasOwnProperty;
-var __propIsEnum$5 = Object.prototype.propertyIsEnumerable;
-var __defNormalProp$5 = (obj, key, value) => key in obj ? __defProp$5(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __spreadValues$5 = (a, b) => {
+var __getOwnPropSymbols$6 = Object.getOwnPropertySymbols;
+var __hasOwnProp$6 = Object.prototype.hasOwnProperty;
+var __propIsEnum$6 = Object.prototype.propertyIsEnumerable;
+var __defNormalProp$6 = (obj, key, value) => key in obj ? __defProp$6(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues$6 = (a, b) => {
   for (var prop in b || (b = {}))
-    if (__hasOwnProp$5.call(b, prop))
-      __defNormalProp$5(a, prop, b[prop]);
-  if (__getOwnPropSymbols$5)
-    for (var prop of __getOwnPropSymbols$5(b)) {
-      if (__propIsEnum$5.call(b, prop))
-        __defNormalProp$5(a, prop, b[prop]);
+    if (__hasOwnProp$6.call(b, prop))
+      __defNormalProp$6(a, prop, b[prop]);
+  if (__getOwnPropSymbols$6)
+    for (var prop of __getOwnPropSymbols$6(b)) {
+      if (__propIsEnum$6.call(b, prop))
+        __defNormalProp$6(a, prop, b[prop]);
     }
   return a;
 };
@@ -37,10 +37,11 @@ const migrationConfigDefaults = {
   migrationsTable: "schemaMigrations",
   requireTs: require,
   log: true,
-  logger: console
+  logger: console,
+  useCodeUpdaterByDefault: true
 };
 const getMigrationConfigWithDefaults = (config) => {
-  return __spreadValues$5(__spreadValues$5({}, migrationConfigDefaults), config);
+  return __spreadValues$6(__spreadValues$6({}, migrationConfigDefaults), config);
 };
 const getDatabaseAndUserFromOptions = (options) => {
   if (options.databaseURL) {
@@ -68,9 +69,9 @@ const setAdapterOptions = (options, set) => {
     if (set.password !== void 0) {
       url.password = set.password;
     }
-    return __spreadProps$5(__spreadValues$5({}, options), { databaseURL: url.toString() });
+    return __spreadProps$5(__spreadValues$6({}, options), { databaseURL: url.toString() });
   } else {
-    return __spreadValues$5(__spreadValues$5({}, options), set);
+    return __spreadValues$6(__spreadValues$6({}, options), set);
   }
 };
 const askAdminCredentials = async () => {
@@ -209,21 +210,21 @@ const setCurrentMigrationUp = (up) => {
 const getCurrentPromise = () => currentPromise;
 const getCurrentChangeCallback = () => currentChangeCallback;
 
-var __defProp$4 = Object.defineProperty;
+var __defProp$5 = Object.defineProperty;
 var __defProps$4 = Object.defineProperties;
 var __getOwnPropDescs$4 = Object.getOwnPropertyDescriptors;
-var __getOwnPropSymbols$4 = Object.getOwnPropertySymbols;
-var __hasOwnProp$4 = Object.prototype.hasOwnProperty;
-var __propIsEnum$4 = Object.prototype.propertyIsEnumerable;
-var __defNormalProp$4 = (obj, key, value) => key in obj ? __defProp$4(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __spreadValues$4 = (a, b) => {
+var __getOwnPropSymbols$5 = Object.getOwnPropertySymbols;
+var __hasOwnProp$5 = Object.prototype.hasOwnProperty;
+var __propIsEnum$5 = Object.prototype.propertyIsEnumerable;
+var __defNormalProp$5 = (obj, key, value) => key in obj ? __defProp$5(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues$5 = (a, b) => {
   for (var prop in b || (b = {}))
-    if (__hasOwnProp$4.call(b, prop))
-      __defNormalProp$4(a, prop, b[prop]);
-  if (__getOwnPropSymbols$4)
-    for (var prop of __getOwnPropSymbols$4(b)) {
-      if (__propIsEnum$4.call(b, prop))
-        __defNormalProp$4(a, prop, b[prop]);
+    if (__hasOwnProp$5.call(b, prop))
+      __defNormalProp$5(a, prop, b[prop]);
+  if (__getOwnPropSymbols$5)
+    for (var prop of __getOwnPropSymbols$5(b)) {
+      if (__propIsEnum$5.call(b, prop))
+        __defNormalProp$5(a, prop, b[prop]);
     }
   return a;
 };
@@ -264,7 +265,7 @@ const addColumnIndex = (indexes, key, item) => {
   if (item.data) {
     if (item.data.index) {
       indexes.push({
-        columns: [__spreadProps$4(__spreadValues$4({}, item.data.index), { column: key })],
+        columns: [__spreadProps$4(__spreadValues$5({}, item.data.index), { column: key })],
         options: item.data.index
       });
     }
@@ -400,21 +401,21 @@ WHERE
   return rows;
 };
 
-var __defProp$3 = Object.defineProperty;
+var __defProp$4 = Object.defineProperty;
 var __defProps$3 = Object.defineProperties;
 var __getOwnPropDescs$3 = Object.getOwnPropertyDescriptors;
-var __getOwnPropSymbols$3 = Object.getOwnPropertySymbols;
-var __hasOwnProp$3 = Object.prototype.hasOwnProperty;
-var __propIsEnum$3 = Object.prototype.propertyIsEnumerable;
-var __defNormalProp$3 = (obj, key, value) => key in obj ? __defProp$3(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __spreadValues$3 = (a, b) => {
+var __getOwnPropSymbols$4 = Object.getOwnPropertySymbols;
+var __hasOwnProp$4 = Object.prototype.hasOwnProperty;
+var __propIsEnum$4 = Object.prototype.propertyIsEnumerable;
+var __defNormalProp$4 = (obj, key, value) => key in obj ? __defProp$4(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues$4 = (a, b) => {
   for (var prop in b || (b = {}))
-    if (__hasOwnProp$3.call(b, prop))
-      __defNormalProp$3(a, prop, b[prop]);
-  if (__getOwnPropSymbols$3)
-    for (var prop of __getOwnPropSymbols$3(b)) {
-      if (__propIsEnum$3.call(b, prop))
-        __defNormalProp$3(a, prop, b[prop]);
+    if (__hasOwnProp$4.call(b, prop))
+      __defNormalProp$4(a, prop, b[prop]);
+  if (__getOwnPropSymbols$4)
+    for (var prop of __getOwnPropSymbols$4(b)) {
+      if (__propIsEnum$4.call(b, prop))
+        __defNormalProp$4(a, prop, b[prop]);
     }
   return a;
 };
@@ -423,7 +424,6 @@ const types = Object.assign(Object.create(pqb.columnTypes), {
   raw: pqb.raw
 });
 const createTable = async (migration, up, tableName, options, fn) => {
-  var _a, _b;
   const shape = pqb.getColumnTypes(types, fn);
   const tableData = pqb.getTableData();
   const ast = makeAst$1(
@@ -439,7 +439,7 @@ const createTable = async (migration, up, tableName, options, fn) => {
   for (const query of queries) {
     await migration.query(query);
   }
-  await ((_b = (_a = migration.options).appCodeUpdater) == null ? void 0 : _b.call(_a, ast));
+  await runCodeUpdater(migration, ast);
 };
 const makeAst$1 = (up, tableName, shape, tableData, options, noPrimaryKey) => {
   const shapePKeys = [];
@@ -449,13 +449,13 @@ const makeAst$1 = (up, tableName, shape, tableData, options, noPrimaryKey) => {
     }
   }
   const primaryKey = tableData.primaryKey;
-  return __spreadProps$3(__spreadValues$3(__spreadProps$3(__spreadValues$3({
+  return __spreadProps$3(__spreadValues$4(__spreadProps$3(__spreadValues$4({
     type: "table",
     action: up ? "create" : "drop",
     name: tableName,
     shape
   }, tableData), {
-    primaryKey: shapePKeys.length <= 1 ? primaryKey : primaryKey ? __spreadProps$3(__spreadValues$3({}, primaryKey), { columns: [...shapePKeys, ...primaryKey.columns] }) : { columns: shapePKeys }
+    primaryKey: shapePKeys.length <= 1 ? primaryKey : primaryKey ? __spreadProps$3(__spreadValues$4({}, primaryKey), { columns: [...shapePKeys, ...primaryKey.columns] }) : { columns: shapePKeys }
   }), options), {
     noPrimaryKey: options.noPrimaryKey ? "ignore" : noPrimaryKey || "error"
   });
@@ -529,21 +529,21 @@ const astToQueries$1 = (ast) => {
   return result;
 };
 
-var __defProp$2 = Object.defineProperty;
+var __defProp$3 = Object.defineProperty;
 var __defProps$2 = Object.defineProperties;
 var __getOwnPropDescs$2 = Object.getOwnPropertyDescriptors;
-var __getOwnPropSymbols$2 = Object.getOwnPropertySymbols;
-var __hasOwnProp$2 = Object.prototype.hasOwnProperty;
-var __propIsEnum$2 = Object.prototype.propertyIsEnumerable;
-var __defNormalProp$2 = (obj, key, value) => key in obj ? __defProp$2(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __spreadValues$2 = (a, b) => {
+var __getOwnPropSymbols$3 = Object.getOwnPropertySymbols;
+var __hasOwnProp$3 = Object.prototype.hasOwnProperty;
+var __propIsEnum$3 = Object.prototype.propertyIsEnumerable;
+var __defNormalProp$3 = (obj, key, value) => key in obj ? __defProp$3(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues$3 = (a, b) => {
   for (var prop in b || (b = {}))
-    if (__hasOwnProp$2.call(b, prop))
-      __defNormalProp$2(a, prop, b[prop]);
-  if (__getOwnPropSymbols$2)
-    for (var prop of __getOwnPropSymbols$2(b)) {
-      if (__propIsEnum$2.call(b, prop))
-        __defNormalProp$2(a, prop, b[prop]);
+    if (__hasOwnProp$3.call(b, prop))
+      __defNormalProp$3(a, prop, b[prop]);
+  if (__getOwnPropSymbols$3)
+    for (var prop of __getOwnPropSymbols$3(b)) {
+      if (__propIsEnum$3.call(b, prop))
+        __defNormalProp$3(a, prop, b[prop]);
     }
   return a;
 };
@@ -563,7 +563,7 @@ const mergeTableData = (a, b) => {
     } else {
       a.primaryKey = {
         columns: [...a.primaryKey.columns, ...b.primaryKey.columns],
-        options: __spreadValues$2(__spreadValues$2({}, a.primaryKey.options), b.primaryKey.options)
+        options: __spreadValues$3(__spreadValues$3({}, a.primaryKey.options), b.primaryKey.options)
       };
     }
   }
@@ -614,7 +614,7 @@ const columnTypeToColumnChange = (item) => {
     if (foreignKey && "fn" in foreignKey) {
       throw new Error("Callback in foreignKey is not allowed in migration");
     }
-    return __spreadProps$2(__spreadValues$2({
+    return __spreadProps$2(__spreadValues$3({
       column: item,
       type: item.toSQL(),
       nullable: item.isNullable,
@@ -630,7 +630,7 @@ const tableChangeMethods = {
   add,
   drop,
   change(from, to, options) {
-    return __spreadValues$2({
+    return __spreadValues$3({
       type: "change",
       from: columnTypeToColumnChange(from),
       to: columnTypeToColumnChange(to)
@@ -661,7 +661,6 @@ const tableChangeMethods = {
   }
 };
 const changeTable = async (migration, up, tableName, options, fn) => {
-  var _a, _b;
   pqb.resetTableData();
   resetChangeTableData();
   const tableChanger = Object.create(pqb.columnTypes);
@@ -672,7 +671,7 @@ const changeTable = async (migration, up, tableName, options, fn) => {
   for (const query of queries) {
     await migration.query(query);
   }
-  await ((_b = (_a = migration.options).appCodeUpdater) == null ? void 0 : _b.call(_a, ast));
+  await runCodeUpdater(migration, ast);
 };
 const makeAst = (up, name, changeData, changeTableData2, options) => {
   const { comment } = options;
@@ -681,17 +680,17 @@ const makeAst = (up, name, changeData, changeTableData2, options) => {
     const item = changeData[key];
     if ("type" in item) {
       if (up) {
-        shape[key] = item.type === "change" && item.usingUp ? __spreadProps$2(__spreadValues$2({}, item), { using: item.usingUp }) : item;
+        shape[key] = item.type === "change" && item.usingUp ? __spreadProps$2(__spreadValues$3({}, item), { using: item.usingUp }) : item;
       } else {
         if (item.type === "rename") {
-          shape[item.name] = __spreadProps$2(__spreadValues$2({}, item), { name: key });
+          shape[item.name] = __spreadProps$2(__spreadValues$3({}, item), { name: key });
         } else {
-          shape[key] = item.type === "add" ? __spreadProps$2(__spreadValues$2({}, item), { type: "drop" }) : item.type === "drop" ? __spreadProps$2(__spreadValues$2({}, item), { type: "add" }) : item.type === "change" ? __spreadProps$2(__spreadValues$2({}, item), { from: item.to, to: item.from, using: item.usingDown }) : item;
+          shape[key] = item.type === "add" ? __spreadProps$2(__spreadValues$3({}, item), { type: "drop" }) : item.type === "drop" ? __spreadProps$2(__spreadValues$3({}, item), { type: "add" }) : item.type === "change" ? __spreadProps$2(__spreadValues$3({}, item), { from: item.to, to: item.from, using: item.usingDown }) : item;
         }
       }
     }
   }
-  return __spreadValues$2({
+  return __spreadValues$3({
     type: "changeTable",
     name,
     comment: comment ? up ? Array.isArray(comment) ? comment[1] : comment : Array.isArray(comment) ? comment[0] : null : void 0,
@@ -707,10 +706,10 @@ const astToQueries = (ast) => {
       values: []
     });
   }
-  const addPrimaryKeys = ast.add.primaryKey ? __spreadValues$2({}, ast.add.primaryKey) : {
+  const addPrimaryKeys = ast.add.primaryKey ? __spreadValues$3({}, ast.add.primaryKey) : {
     columns: []
   };
-  const dropPrimaryKeys = ast.drop.primaryKey ? __spreadValues$2({}, ast.drop.primaryKey) : {
+  const dropPrimaryKeys = ast.drop.primaryKey ? __spreadValues$3({}, ast.drop.primaryKey) : {
     columns: []
   };
   for (const key in ast.shape) {
@@ -807,7 +806,7 @@ const astToQueries = (ast) => {
         if (fromIndex) {
           dropIndexes.push({
             columns: [
-              __spreadValues$2({
+              __spreadValues$3({
                 column: key
               }, fromIndex)
             ],
@@ -817,7 +816,7 @@ const astToQueries = (ast) => {
         if (toIndex) {
           addIndexes.push({
             columns: [
-              __spreadValues$2({
+              __spreadValues$3({
                 column: key
               }, toIndex)
             ],
@@ -866,21 +865,21 @@ const astToQueries = (ast) => {
   return result;
 };
 
-var __defProp$1 = Object.defineProperty;
+var __defProp$2 = Object.defineProperty;
 var __defProps$1 = Object.defineProperties;
 var __getOwnPropDescs$1 = Object.getOwnPropertyDescriptors;
-var __getOwnPropSymbols$1 = Object.getOwnPropertySymbols;
-var __hasOwnProp$1 = Object.prototype.hasOwnProperty;
-var __propIsEnum$1 = Object.prototype.propertyIsEnumerable;
-var __defNormalProp$1 = (obj, key, value) => key in obj ? __defProp$1(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __spreadValues$1 = (a, b) => {
+var __getOwnPropSymbols$2 = Object.getOwnPropertySymbols;
+var __hasOwnProp$2 = Object.prototype.hasOwnProperty;
+var __propIsEnum$2 = Object.prototype.propertyIsEnumerable;
+var __defNormalProp$2 = (obj, key, value) => key in obj ? __defProp$2(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues$2 = (a, b) => {
   for (var prop in b || (b = {}))
-    if (__hasOwnProp$1.call(b, prop))
-      __defNormalProp$1(a, prop, b[prop]);
-  if (__getOwnPropSymbols$1)
-    for (var prop of __getOwnPropSymbols$1(b)) {
-      if (__propIsEnum$1.call(b, prop))
-        __defNormalProp$1(a, prop, b[prop]);
+    if (__hasOwnProp$2.call(b, prop))
+      __defNormalProp$2(a, prop, b[prop]);
+  if (__getOwnPropSymbols$2)
+    for (var prop of __getOwnPropSymbols$2(b)) {
+      if (__propIsEnum$2.call(b, prop))
+        __defNormalProp$2(a, prop, b[prop]);
     }
   return a;
 };
@@ -891,6 +890,9 @@ class UnknownColumn extends pqb.ColumnType {
     this.dataType = dataType;
     this.operators = pqb.Operators.any;
   }
+  toCode() {
+    return "unknown";
+  }
 }
 const createJoinTable = async (migration, up, tables, options, fn) => {
   const tableName = options.tableName || joinWords(...tables);
@@ -899,14 +901,14 @@ const createJoinTable = async (migration, up, tables, options, fn) => {
       migration,
       up,
       tableName,
-      __spreadProps$1(__spreadValues$1({}, options), { noPrimaryKey: true }),
+      __spreadProps$1(__spreadValues$2({}, options), { noPrimaryKey: true }),
       () => ({})
     );
   }
   const tablesWithPrimaryKeys = await Promise.all(
     tables.map(async (table) => {
       const primaryKeys = await getPrimaryKeysOfTable(migration, table).then(
-        (items) => items.map((item) => __spreadProps$1(__spreadValues$1({}, item), {
+        (items) => items.map((item) => __spreadProps$1(__spreadValues$2({}, item), {
           joinedName: joinWords(pluralize.singular(table), item.name)
         }))
       );
@@ -948,30 +950,32 @@ const createJoinTable = async (migration, up, tables, options, fn) => {
   });
 };
 
-var __defProp = Object.defineProperty;
+var __defProp$1 = Object.defineProperty;
 var __defProps = Object.defineProperties;
 var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
-var __getOwnPropSymbols = Object.getOwnPropertySymbols;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __propIsEnum = Object.prototype.propertyIsEnumerable;
-var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __spreadValues = (a, b) => {
+var __getOwnPropSymbols$1 = Object.getOwnPropertySymbols;
+var __hasOwnProp$1 = Object.prototype.hasOwnProperty;
+var __propIsEnum$1 = Object.prototype.propertyIsEnumerable;
+var __defNormalProp$1 = (obj, key, value) => key in obj ? __defProp$1(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues$1 = (a, b) => {
   for (var prop in b || (b = {}))
-    if (__hasOwnProp.call(b, prop))
-      __defNormalProp(a, prop, b[prop]);
-  if (__getOwnPropSymbols)
-    for (var prop of __getOwnPropSymbols(b)) {
-      if (__propIsEnum.call(b, prop))
-        __defNormalProp(a, prop, b[prop]);
+    if (__hasOwnProp$1.call(b, prop))
+      __defNormalProp$1(a, prop, b[prop]);
+  if (__getOwnPropSymbols$1)
+    for (var prop of __getOwnPropSymbols$1(b)) {
+      if (__propIsEnum$1.call(b, prop))
+        __defNormalProp$1(a, prop, b[prop]);
     }
   return a;
 };
 var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
 class Migration extends pqb.TransactionAdapter {
-  constructor(tx, up, options) {
+  constructor(tx, up, options, adapterOptions, appCodeUpdaterCache) {
     super(tx, tx.client, tx.types);
     this.up = up;
     this.options = options;
+    this.adapterOptions = adapterOptions;
+    this.appCodeUpdaterCache = appCodeUpdaterCache;
     this.log = pqb.logParamToLogObject(options.logger || console, options.log);
   }
   async query(query, types = this.types, log = this.log) {
@@ -1005,7 +1009,6 @@ class Migration extends pqb.TransactionAdapter {
     return changeTable(this, this.up, tableName, options, fn);
   }
   async renameTable(from, to) {
-    var _a, _b;
     const ast = {
       type: "renameTable",
       from: this.up ? from : to,
@@ -1014,7 +1017,7 @@ class Migration extends pqb.TransactionAdapter {
     await this.query(
       `ALTER TABLE ${quoteTable(ast.from)} RENAME TO "${ast.to}"`
     );
-    await ((_b = (_a = this.options).appCodeUpdater) == null ? void 0 : _b.call(_a, ast));
+    await runCodeUpdater(this, ast);
   }
   addColumn(tableName, columnName, fn) {
     return addColumn(this, this.up, tableName, columnName, fn);
@@ -1096,7 +1099,7 @@ const wrapWithLog = async (log, query, fn) => {
   if (!log) {
     return fn();
   } else {
-    const sql = typeof query === "string" ? { text: query, values: [] } : query.values ? query : __spreadProps(__spreadValues({}, query), { values: [] });
+    const sql = typeof query === "string" ? { text: query, values: [] } : query.values ? query : __spreadProps(__spreadValues$1({}, query), { values: [] });
     const logData = log.beforeQuery(sql);
     try {
       const result = await fn();
@@ -1114,16 +1117,15 @@ const addColumn = (migration, up, tableName, columnName, fn) => {
   }));
 };
 const addIndex = (migration, up, tableName, columns, options) => {
-  return changeTable(migration, up, tableName, {}, (t) => __spreadValues({}, t.add(t.index(columns, options))));
+  return changeTable(migration, up, tableName, {}, (t) => __spreadValues$1({}, t.add(t.index(columns, options))));
 };
 const addForeignKey = (migration, up, tableName, columns, foreignTable, foreignColumns, options) => {
-  return changeTable(migration, up, tableName, {}, (t) => __spreadValues({}, t.add(t.foreignKey(columns, foreignTable, foreignColumns, options))));
+  return changeTable(migration, up, tableName, {}, (t) => __spreadValues$1({}, t.add(t.foreignKey(columns, foreignTable, foreignColumns, options))));
 };
 const addPrimaryKey = (migration, up, tableName, columns, options) => {
-  return changeTable(migration, up, tableName, {}, (t) => __spreadValues({}, t.add(t.primaryKey(columns, options))));
+  return changeTable(migration, up, tableName, {}, (t) => __spreadValues$1({}, t.add(t.primaryKey(columns, options))));
 };
 const createSchema = async (migration, up, name) => {
-  var _a, _b;
   const ast = {
     type: "schema",
     action: up ? "create" : "drop",
@@ -1132,11 +1134,10 @@ const createSchema = async (migration, up, name) => {
   await migration.query(
     `${ast.action === "create" ? "CREATE" : "DROP"} SCHEMA "${name}"`
   );
-  await ((_b = (_a = migration.options).appCodeUpdater) == null ? void 0 : _b.call(_a, ast));
+  await runCodeUpdater(migration, ast);
 };
 const createExtension = async (migration, up, name, options) => {
-  var _a, _b;
-  const ast = __spreadValues({
+  const ast = __spreadValues$1({
     type: "extension",
     action: up ? "create" : "drop",
     name
@@ -1148,17 +1149,54 @@ const createExtension = async (migration, up, name, options) => {
     query = `CREATE EXTENSION${ast.ifExists ? " IF NOT EXISTS" : ""} "${ast.name}"${ast.schema ? ` SCHEMA "${ast.schema}"` : ""}${ast.version ? ` VERSION '${ast.version}'` : ""}${ast.cascade ? " CASCADE" : ""}`;
   }
   await migration.query(query);
-  await ((_b = (_a = migration.options).appCodeUpdater) == null ? void 0 : _b.call(_a, ast));
+  await runCodeUpdater(migration, ast);
 };
 const queryExists = (db, sql) => {
   return db.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,
+    cache: migration.appCodeUpdaterCache
+  });
+};
 
+var __defProp = Object.defineProperty;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
 const migrateOrRollback = async (options, config, args, up) => {
   var _a;
+  config = __spreadValues({}, config);
   const files = await getMigrationFiles(config, up);
-  const argCount = args[0] === "all" ? Infinity : parseInt(args[0]);
-  let count = isNaN(argCount) ? up ? Infinity : 1 : argCount;
+  let count = up ? Infinity : 1;
+  let argI = 0;
+  const num = args[0] === "all" ? Infinity : parseInt(args[0]);
+  if (!isNaN(num)) {
+    argI++;
+    count = num;
+  }
+  const arg = args[argI];
+  if (arg === "--code") {
+    config.useCodeUpdater = args[argI + 1] !== "false";
+  }
+  if (!config.useCodeUpdater)
+    delete config.appCodeUpdater;
+  const appCodeUpdaterCache = {};
   for (const opts of pqb.toArray(options)) {
     const db = new pqb.Adapter(opts);
     const migratedVersions = await getMigratedVersionsMap(db, config);
@@ -1169,18 +1207,19 @@ const migrateOrRollback = async (options, config, args, up) => {
         }
         if (count-- <= 0)
           break;
-        await processMigration(db, up, file, config);
+        await processMigration(db, up, file, config, opts, appCodeUpdaterCache);
         (_a = config.logger) == null ? void 0 : _a.log(`${file.path} ${up ? "migrated" : "rolled back"}`);
       }
     } finally {
       await db.close();
     }
+    delete config.appCodeUpdater;
   }
 };
 const changeCache = {};
-const processMigration = async (db, up, file, config) => {
+const processMigration = async (db, up, file, config, options, appCodeUpdaterCache) => {
   await db.transaction(async (tx) => {
-    const db2 = new Migration(tx, up, config);
+    const db2 = new Migration(tx, up, config, options, appCodeUpdaterCache);
     setCurrentMigration(db2);
     setCurrentMigrationUp(up);
     const callback = changeCache[file.path];
@@ -1403,17 +1442,30 @@ Commands:
   drop                    drop databases
   reset                   drop, create and migrate databases
   g, generate             generate migration file, see below
-  migrate                 migrate all pending migrations
+  migrate                 migrate pending migrations
   rollback                rollback the last migrated
   no or unknown command   prints this message
   
+Migrate arguments:
+  no arguments            run all pending migrations
+  number                  run specific number of pending migrations
+
+Rollback arguments:
+  no arguments            rollback one last applied migration
+  number                  rollback specific number of applied migrations
+  all                     rollback all applied migrations
+
+Migrate and rollback common arguments:
+  --code                  run code updater, overrides \`useCodeUpdater\` option
+  --code false            do not run code updater
+  
 Generate arguments:
 - (required) first argument is migration name
-  * create*      template for create table
-  * change*      template for change table
-  * add*To*      template for add columns
-  * remove*From* template for remove columns
-  * drop*        template for drop table
+  * create*               template for create table
+  * change*               template for change table
+  * add*To*               template for add columns
+  * remove*From*          template for remove columns
+  * drop*                 template for drop table
 
 - other arguments considered as columns with types and optional methods:
   rake-db g createTable id:serial.primaryKey name:text.nullable
@@ -1429,4 +1481,5 @@ exports.migrate = migrate;
 exports.rakeDb = rakeDb;
 exports.resetDb = resetDb;
 exports.rollback = rollback;
+exports.runCodeUpdater = runCodeUpdater;
 //# sourceMappingURL=index.js.map