rake-db 2.2.3 → 2.2.4

package/CHANGELOG.md

@@ -1,5 +1,15 @@
 # rake-db
 
+## 2.2.4
+
+### Patch Changes
+
+- Add --code cli argument to rake-db
+- Improve codegen
+- Updated dependencies
+- Updated dependencies
+  - pqb@0.8.4
+
 ## 2.2.3
 
 ### Patch Changes

package/db.ts

@@ -2,6 +2,7 @@ import { config } from 'dotenv';
 import path from 'path';
 import { rakeDb } from './src/rakeDb';
 import { AdapterOptions } from 'pqb';
+import { appCodeUpdater } from '../orm/src';
 
 config({ path: path.resolve(process.cwd(), '.env.local') });
 config();
@@ -22,4 +23,11 @@ if (databaseURLTest) {
 
 rakeDb(options, {
   migrationsPath: path.resolve(process.cwd(), 'migrations'),
+  appCodeUpdater: appCodeUpdater({
+    tablePath: (tableName) => `app/tables/${tableName}.ts`,
+    baseTablePath: 'app/lib/baseTable.ts',
+    baseTableName: 'BaseTable',
+    mainFilePath: 'app/db.ts',
+  }),
+  useCodeUpdaterByDefault: false,
 });

package/dist/index.d.ts

@@ -1,5 +1,5 @@
 import * as pqb from 'pqb';
-import { EmptyObject, RawExpression, ColumnType, ColumnTypes, raw, ColumnsShape, TransactionAdapter, QueryLogObject, QueryResultRow, QueryInput, TypeParsers, QueryResult, QueryArraysResult, MaybeArray, IndexColumnOptions, IndexOptions, ForeignKeyOptions, TextColumn, NoPrimaryKeyOption, TableData, SingleColumnIndexOptions, QueryLogOptions, AdapterOptions } from 'pqb';
+import { EmptyObject, RawExpression, ColumnType, ColumnTypes, raw, ColumnsShape, TransactionAdapter, AdapterOptions, QueryLogObject, QueryResultRow, QueryInput, TypeParsers, QueryResult, QueryArraysResult, MaybeArray, IndexColumnOptions, IndexOptions, ForeignKeyOptions, TextColumn, NoPrimaryKeyOption, TableData, SingleColumnIndexOptions, QueryLogOptions } from 'pqb';
 
 declare function add(item: ColumnType, options?: {
     dropMode?: DropMode;
@@ -63,8 +63,10 @@ declare type ExtensionOptions = {
 declare class Migration extends TransactionAdapter {
     up: boolean;
     options: RakeDbConfig;
+    adapterOptions: AdapterOptions;
+    appCodeUpdaterCache: object;
     log?: QueryLogObject;
-    constructor(tx: TransactionAdapter, up: boolean, options: RakeDbConfig);
+    constructor(tx: TransactionAdapter, up: boolean, options: RakeDbConfig, adapterOptions: AdapterOptions, appCodeUpdaterCache: object);
     query<T extends QueryResultRow = any>(query: QueryInput, types?: TypeParsers, log?: QueryLogObject | undefined): Promise<QueryResult<T>>;
     arrays<R extends any[] = any[]>(query: QueryInput, types?: TypeParsers, log?: QueryLogObject | undefined): Promise<QueryArraysResult<R>>;
     createTable(tableName: string, options: TableOptions, fn: ColumnsShapeCallback): Promise<void>;
@@ -104,6 +106,7 @@ declare class Migration extends TransactionAdapter {
     columnExists(tableName: string, columnName: string): Promise<boolean>;
     constraintExists(constraintName: string): Promise<boolean>;
 }
+declare const runCodeUpdater: (migration: Migration, ast: RakeDbAst) => Promise<void> | undefined;
 
 declare type RakeDbAst = RakeDbAst.Table | RakeDbAst.ChangeTable | RakeDbAst.RenameTable | RakeDbAst.Schema | RakeDbAst.Extension;
 declare namespace RakeDbAst {
@@ -185,8 +188,13 @@ declare type RakeDbConfig = {
     requireTs(path: string): void;
     noPrimaryKey?: NoPrimaryKeyOption;
     appCodeUpdater?: AppCodeUpdater;
+    useCodeUpdater?: boolean;
 } & QueryLogOptions;
-declare type AppCodeUpdater = (ast: RakeDbAst) => Promise<void>;
+declare type AppCodeUpdater = (params: {
+    ast: RakeDbAst;
+    options: AdapterOptions;
+    cache: object;
+}) => Promise<void>;
 
 declare const createDb: (arg: MaybeArray<AdapterOptions>, config: RakeDbConfig) => Promise<void>;
 declare const dropDb: (arg: MaybeArray<AdapterOptions>) => Promise<void>;
@@ -202,4 +210,4 @@ declare const change: (fn: ChangeCallback) => void;
 
 declare const rakeDb: (options: MaybeArray<AdapterOptions>, partialConfig?: Partial<RakeDbConfig>, args?: string[]) => Promise<void>;
 
-export { AppCodeUpdater, ChangeTableCallback, ChangeTableOptions, ColumnComment, ColumnsShapeCallback, DropMode, ExtensionOptions, JoinTableOptions, Migration, MigrationColumnTypes, RakeDbAst, RakeDbConfig, TableOptions, change, createDb, dropDb, generate, migrate, rakeDb, resetDb, rollback };
+export { AppCodeUpdater, ChangeTableCallback, ChangeTableOptions, ColumnComment, ColumnsShapeCallback, DropMode, ExtensionOptions, JoinTableOptions, Migration, MigrationColumnTypes, RakeDbAst, RakeDbConfig, TableOptions, change, createDb, dropDb, generate, migrate, rakeDb, resetDb, rollback, runCodeUpdater };

package/dist/index.esm.js

@@ -4,21 +4,21 @@ import path from 'path';
 import { readdir, mkdir, writeFile } from 'fs/promises';
 import { singular } from 'pluralize';
 
-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;
 };
@@ -28,10 +28,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) {
@@ -59,9 +60,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 () => {
@@ -200,21 +201,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;
 };
@@ -255,7 +256,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
       });
     }
@@ -391,21 +392,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;
 };
@@ -414,7 +415,6 @@ const types = Object.assign(Object.create(columnTypes), {
   raw
 });
 const createTable = async (migration, up, tableName, options, fn) => {
-  var _a, _b;
   const shape = getColumnTypes(types, fn);
   const tableData = getTableData();
   const ast = makeAst$1(
@@ -430,7 +430,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 = [];
@@ -440,13 +440,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"
   });
@@ -520,21 +520,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;
 };
@@ -554,7 +554,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)
       };
     }
   }
@@ -605,7 +605,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,
@@ -621,7 +621,7 @@ const tableChangeMethods = {
   add,
   drop,
   change(from, to, options) {
-    return __spreadValues$2({
+    return __spreadValues$3({
       type: "change",
       from: columnTypeToColumnChange(from),
       to: columnTypeToColumnChange(to)
@@ -652,7 +652,6 @@ const tableChangeMethods = {
   }
 };
 const changeTable = async (migration, up, tableName, options, fn) => {
-  var _a, _b;
   resetTableData();
   resetChangeTableData();
   const tableChanger = Object.create(columnTypes);
@@ -663,7 +662,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;
@@ -672,17 +671,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,
@@ -698,10 +697,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) {
@@ -798,7 +797,7 @@ const astToQueries = (ast) => {
         if (fromIndex) {
           dropIndexes.push({
             columns: [
-              __spreadValues$2({
+              __spreadValues$3({
                 column: key
               }, fromIndex)
             ],
@@ -808,7 +807,7 @@ const astToQueries = (ast) => {
         if (toIndex) {
           addIndexes.push({
             columns: [
-              __spreadValues$2({
+              __spreadValues$3({
                 column: key
               }, toIndex)
             ],
@@ -857,21 +856,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;
 };
@@ -893,14 +892,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(singular(table), item.name)
         }))
       );
@@ -942,30 +941,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 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 = logParamToLogObject(options.logger || console, options.log);
   }
   async query(query, types = this.types, log = this.log) {
@@ -999,7 +1000,6 @@ class Migration extends 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,
@@ -1008,7 +1008,7 @@ class Migration extends 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);
@@ -1090,7 +1090,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();
@@ -1108,16 +1108,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",
@@ -1126,11 +1125,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
@@ -1142,17 +1140,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 toArray(options)) {
     const db = new Adapter(opts);
     const migratedVersions = await getMigratedVersionsMap(db, config);
@@ -1163,18 +1198,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];
@@ -1397,22 +1433,35 @@ 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
 `
 );
 
-export { Migration, change, createDb, dropDb, generate, migrate, rakeDb, resetDb, rollback };
+export { Migration, change, createDb, dropDb, generate, migrate, rakeDb, resetDb, rollback, runCodeUpdater };
 //# sourceMappingURL=index.esm.js.map