rake-db 2.8.0 → 2.8.1

package/dist/index.d.ts

@@ -293,8 +293,7 @@ type RakeDbConfig<CT extends ColumnTypesBase = DefaultColumnTypes> = {
     beforeRollback?(db: Db): Promise<void>;
     afterRollback?(db: Db): Promise<void>;
 } & QueryLogOptions;
-type AppCodeUpdater = (params: {
-    ast: RakeDbAst;
+type AppCodeUpdaterParams = {
     options: AdapterOptions;
     basePath: string;
     cache: object;
@@ -303,7 +302,13 @@ type AppCodeUpdater = (params: {
         filePath: string;
         name: string;
     };
-}) => Promise<void>;
+};
+type AppCodeUpdater = {
+    process(params: AppCodeUpdaterParams & {
+        ast: RakeDbAst;
+    }): Promise<void>;
+    afterAll(params: AppCodeUpdaterParams): Promise<void>;
+};
 
 declare const createDb: <CT extends ColumnTypesBase>(arg: MaybeArray<AdapterOptions>, config: RakeDbConfig<CT>) => Promise<void>;
 declare const dropDb: (arg: MaybeArray<AdapterOptions>) => Promise<void>;

package/dist/index.js

@@ -1629,7 +1629,7 @@ var __spreadValues$1 = (a, b) => {
 };
 const getDb = (adapter) => pqb.createDb({ adapter });
 const migrateOrRollback = async (options, config, args, up) => {
-  var _a, _b, _c, _d, _e;
+  var _a, _b, _c, _d, _e, _f;
   config = __spreadValues$1({}, config);
   const files = await getMigrationFiles(config, up);
   let count = up ? Infinity : 1;
@@ -1680,6 +1680,14 @@ const migrateOrRollback = async (options, config, args, up) => {
         await ((_e = config.afterRollback) == null ? void 0 : _e.call(config, db != null ? db : db = getDb(adapter)));
       }
     } finally {
+      await ((_f = config.appCodeUpdater) == null ? void 0 : _f.afterAll({
+        options: opts,
+        basePath: config.basePath,
+        cache: appCodeUpdaterCache,
+        logger: config.logger,
+        // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+        baseTable: config.baseTable
+      }));
       await adapter.close();
     }
     delete config.appCodeUpdater;
@@ -1718,7 +1726,7 @@ const processMigration = async (db, up, file, config, options, appCodeUpdaterCac
     return db2.migratedAsts;
   });
   for (const ast of asts) {
-    await ((_a = config.appCodeUpdater) == null ? void 0 : _a.call(config, {
+    await ((_a = config.appCodeUpdater) == null ? void 0 : _a.process({
       ast,
       options,
       basePath: config.basePath,
@@ -2971,7 +2979,7 @@ const createView = (ast) => {
 };
 
 const pullDbStructure = async (options, config) => {
-  var _a, _b, _c;
+  var _a, _b, _c, _d;
   const adapter = new pqb.Adapter(options);
   const currentSchema = adapter.schema || "public";
   const db = new DbStructure(adapter);
@@ -2994,7 +3002,7 @@ const pullDbStructure = async (options, config) => {
   await saveMigratedVersion(silentQueries, version, config);
   const cache = {};
   for (const item of ast) {
-    await ((_a = config == null ? void 0 : config.appCodeUpdater) == null ? void 0 : _a.call(config, {
+    await ((_a = config == null ? void 0 : config.appCodeUpdater) == null ? void 0 : _a.process({
       ast: item,
       options,
       basePath: config.basePath,
@@ -3004,11 +3012,19 @@ const pullDbStructure = async (options, config) => {
       baseTable: config.baseTable
     }));
   }
+  await ((_b = config == null ? void 0 : config.appCodeUpdater) == null ? void 0 : _b.afterAll({
+    options,
+    basePath: config.basePath,
+    cache,
+    logger: config.logger,
+    // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+    baseTable: config.baseTable
+  }));
   const unsupportedEntries = Object.entries(ctx.unsupportedTypes);
   const len = unsupportedEntries.length;
   if (len) {
     let count = 0;
-    (_b = config.logger) == null ? void 0 : _b.warn(
+    (_c = config.logger) == null ? void 0 : _c.warn(
       `Found unsupported types:
 ${unsupportedEntries.map(([type, columns]) => {
         count += columns.length;
@@ -3017,7 +3033,7 @@ ${unsupportedEntries.map(([type, columns]) => {
 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");
+  (_d = config.logger) == null ? void 0 : _d.log("Database pulled successfully");
   adapter.close();
 };