@squiz/db-lib 1.2.11 → 1.2.12

package/CHANGELOG.md

@@ -3,6 +3,14 @@
 All notable changes to this project will be documented in this file.
 See [Conventional Commits](https://conventionalcommits.org) for commit guidelines.
 
+## [1.2.12](https://gitlab.squiz.net/developer-experience/cmp/compare/v1.2.2...v1.2.12) (2022-11-24)
+
+**Note:** Version bump only for package @squiz/db-lib
+
+
+
+
+
 ## [1.2.11](https://gitlab.squiz.net/developer-experience/cmp/compare/v1.2.2...v1.2.11) (2022-11-11)
 
 **Note:** Version bump only for package @squiz/db-lib

package/lib/AbstractRepository.d.ts

@@ -9,15 +9,15 @@ export interface Writer<T> {
     update(where: Partial<T>, newValue: Partial<T>): Promise<T[]>;
     delete(where: Partial<T>): Promise<number>;
 }
-export declare type Repository<T> = Reader<T> & Writer<T>;
-export declare type PageResult<T> = {
+export type Repository<T> = Reader<T> & Writer<T>;
+export type PageResult<T> = {
     items: T[];
     totalCount: number;
     pageSize: number;
 };
-export declare type SortDirection = 'desc' | 'asc';
+export type SortDirection = 'desc' | 'asc';
 export declare const DEFAULT_PAGE_SIZE = 20;
-export declare abstract class AbstractRepository<T, ObjT extends T> implements Reader<T>, Writer<T> {
+export declare abstract class AbstractRepository<T extends object, ObjT extends T> implements Reader<T>, Writer<T> {
     protected repositories: Repositories;
     protected pool: Pool;
     protected classRef: {

package/lib/ConnectionManager.d.ts

@@ -11,7 +11,7 @@ export interface DbConnection {
 export interface ConnectionStringObj {
     connectionString: string;
 }
-export declare type TransactionClient = PoolClient;
+export type TransactionClient = PoolClient;
 export declare class ConnectionManager<T extends Repositories> {
     protected applicationName: string;
     protected migrationDirectory: string;

package/lib/Migrator.d.ts

@@ -6,13 +6,15 @@ export declare class Migrator {
     constructor(migrationDir: string, migrationList: string[], pool: PoolClient);
     protected ensureMigrationTableExists(): Promise<import("pg").QueryResult<any>>;
     protected getAppliedMigrations(): Promise<any[]>;
-    protected applyMigration(migration: string, sql: string): Promise<void>;
+    protected doSqlMigration(migration: string, sql: string): Promise<void>;
     protected getPending(migrationsList: string[], appliedMigrations: string[]): Promise<string[]>;
     protected getSql(migration: string): Promise<string>;
     protected tryToObtainLock(): Promise<boolean>;
     protected releaseLock(): Promise<void>;
     migrate(): Promise<any>;
     protected runMigrations(): Promise<void>;
+    protected doScriptMigration(migration: string): Promise<void>;
     protected runMigration(migration: string): Promise<void>;
+    protected doMigrationWork(migration: string): Promise<void>;
     protected dispose(): void;
 }

package/lib/Repositories.d.ts

@@ -1,2 +1,2 @@
 import { Repository } from './AbstractRepository';
-export declare type Repositories = Record<string, Repository<any>>;
+export type Repositories = Record<string, Repository<any>>;

package/lib/index.js

@@ -31232,14 +31232,13 @@ var Migrator = class {
       return row.id;
     });
   }
-  async applyMigration(migration, sql) {
+  async doSqlMigration(migration, sql) {
     try {
       const result = await this.pool.query(sql);
       logger.info("Applying " + migration);
       if (result.rowCount !== void 0) {
         logger.info("affected rows", result.rowCount);
       }
-      await this.pool.query("insert into __migrations__ (id) values ($1)", [migration]);
     } catch (e) {
       logger.info("error occurred running migration", migration, e);
       throw e;
@@ -31309,11 +31308,24 @@ var Migrator = class {
       await this.runMigration(migration);
     }
   }
+  async doScriptMigration(migration) {
+    const migrationScript = import_path.default.join(this.migrationDir, migration);
+    const migrationFunc = require(migrationScript);
+    let callable;
+    if (migrationFunc instanceof Function) {
+      callable = migrationFunc;
+    } else if (migrationFunc.default instanceof Function) {
+      callable = migrationFunc.default;
+    } else {
+      throw new Error(`${migrationScript} isn't callable`);
+    }
+    await callable(this.pool, logger);
+  }
   async runMigration(migration) {
     try {
       await this.pool.query("BEGIN");
-      const sql = await this.getSql(migration);
-      await this.applyMigration(migration, sql);
+      await this.doMigrationWork(migration);
+      await this.pool.query("insert into __migrations__ (id) values ($1)", [migration]);
       await this.pool.query("COMMIT");
     } catch (e) {
       logger.error("migration failed", migration, e);
@@ -31321,6 +31333,16 @@ var Migrator = class {
       throw e;
     }
   }
+  async doMigrationWork(migration) {
+    if (migration.endsWith(".sql")) {
+      const sql = await this.getSql(migration);
+      await this.doSqlMigration(migration, sql);
+    } else if (migration.endsWith(".js")) {
+      await this.doScriptMigration(migration);
+    } else {
+      throw new Error(`${migration} as an invalid migration extension`);
+    }
+  }
   dispose() {
     return this.pool.release();
   }