node-server-engine 1.1.4 → 1.1.6

package/src/utils/dbMigration/dbMigration.ts

@@ -1,199 +1,199 @@
-/* eslint-disable @typescript-eslint/no-require-imports */
-// Register ts-node at the top of runMigrations.ts
-require('ts-node/register');
-
-import { existsSync, lstatSync } from 'fs';
-import { join } from 'path';
-import { Sequelize, Transaction } from 'sequelize';
-import { Umzug, SequelizeStorage } from 'umzug';
-import { ClearDbOptions, RecreateDbOptions } from './dbMigration.types';
-import { migrateMetadataTable } from './migrateMetadataTable';
-import { SequelizeMeta } from 'db/models';
-import { EngineError } from 'entities/EngineError';
-import { sequelize } from 'entities/Sequelize';
-import { reportDebug } from 'utils/report';
-import { semverCompare } from 'utils/semverCompare';
-
-const namespace = 'engine:utils:dbMigration';
-
-function getMigrationPaths(): Array<string> | undefined {
-  // We look between possibilities for the first directory that exists and matches our expectations
-  // First a user specified one, then the production default, then the development default
-  const prodDefaultPath = join(process.cwd(), './dist/db/migrations');
-  const devDefaultPath = join(process.cwd(), './src/db/migrations');
-  const path = [
-    process.env.DB_MIGRATIONS_DIR,
-    process.env.NODE_ENV === 'prod' ? prodDefaultPath : devDefaultPath
-  ].filter((path) => path && existsSync(path) && lstatSync(path).isDirectory());
-  reportDebug({
-    namespace,
-    message: 'Obtaining Umzug migration options',
-    data: {
-      path,
-      prodPath: prodDefaultPath,
-      devPath: devDefaultPath,
-      customPath: process.env.DB_MIGRATIONS_DIR
-    }
-  });
-  return path.map((p) =>
-    process.env.NODE_ENV === 'prod' ? `${p}/*.js` : `${p}/*.ts`
-  );
-}
-
-/** Run all the non-executed migrations one by one */
-export async function runPendingMigrations(): Promise<void> {
-  if (!sequelize)
-    throw new EngineError({ message: 'Sequelize is not initialized' });
-  // First make sure that the metadata table is up to date
-  await migrateMetadataTable(sequelize);
-  const migrationPaths = getMigrationPaths();
-
-  if (!migrationPaths || migrationPaths.length <= 0) {
-    reportDebug({
-      namespace,
-      message: 'No valid migration paths found.',
-      data: { migrationPaths }
-    });
-    return;
-  }
-
-  migrationPaths?.forEach(async (path) => {
-    const migrator = new Umzug({
-      migrations: {
-        glob: path,
-        resolve: ({ name, path, context }) => {
-          // Use require to import the migration file
-          const migration = require(path || '');
-          return {
-            name,
-            up: async () => migration.up(context),
-            down: async () => migration.down(context)
-          };
-        }
-      },
-      context: sequelize.getQueryInterface(),
-      storage: new SequelizeStorage({ sequelize }),
-      logger: console
-    });
-
-    reportDebug({
-      namespace,
-      message: 'Starting pending database migrations',
-      data: { path }
-    });
-    await migrator.up();
-    console.log('Db Migration Completed');
-  });
-}
-
-/** Rollback all migrations up to a specified target version */
-export async function rollbackMigrations(): Promise<void> {
-  if (!sequelize)
-    throw new EngineError({ message: 'Sequelize is not initialized' });
-  if (!process.env.BUILD_ID)
-    throw new EngineError({ message: 'BUILD_ID is not defined' });
-
-  // Fetch executed migrations in ascending order
-  const migrations = await SequelizeMeta.findAll({
-    order: [['name', 'ASC']],
-    raw: true
-  });
-  const rollbackUntil = migrations.find(
-    (migration) =>
-      migration.version &&
-      semverCompare(migration.version, process.env.BUILD_ID as string) === 1
-  );
-
-  if (!rollbackUntil) {
-    reportDebug({
-      namespace,
-      message: 'No rollback target found for the current BUILD_ID.',
-      data: { buildId: process.env.BUILD_ID }
-    });
-    return;
-  }
-
-  const migrationPaths = getMigrationPaths();
-
-  if (!migrationPaths || migrationPaths.length <= 0) {
-    reportDebug({
-      namespace,
-      message: 'No valid migration paths found for rollback.',
-      data: { migrationPaths }
-    });
-    return;
-  }
-
-  migrationPaths.forEach(async (path) => {
-    const migrator = new Umzug({
-      migrations: {
-        glob: path,
-        resolve: ({ name, path, context }) => {
-          const migration = require(path || '');
-          return {
-            name,
-            up: async () => migration.up(context),
-            down: async () => migration.down(context)
-          };
-        }
-      },
-      context: sequelize.getQueryInterface(),
-      storage: new SequelizeStorage({ sequelize }),
-      logger: console
-    });
-
-    reportDebug({
-      namespace,
-      message: `Rolling back migrations up to: ${rollbackUntil.name}`,
-      data: { path }
-    });
-
-    await migrator.down({ to: rollbackUntil.name });
-  });
-}
-
-/**
- * Recreate the whole database
- * Quick way to make sure that we are working on a clean database
- * /!\ Should only be used when running tests
- */
-export async function testRecreateDB(
-  options: RecreateDbOptions = {}
-): Promise<void> {
-  if (!sequelize)
-    throw new EngineError({ message: 'Sequelize is not initialized' });
-  await sequelize.query(
-    `DROP SCHEMA public${process.env.SQL_TYPE === 'postgres' ? ' CASCADE' : ''}`
-  );
-  await sequelize.query(`CREATE SCHEMA public`);
-  if (!options.clean) await runPendingMigrations();
-}
-
-/**
- * Empty all the schemas in the database
- * Quick way to flush the database data while keep the structure
- * /!\ Should only be used when running tests
- */
-export async function testClearDB(options: ClearDbOptions = {}): Promise<void> {
-  if (!sequelize)
-    throw new EngineError({ message: 'Sequelize is not initialized' });
-  await sequelize.transaction(clearDbTransaction(sequelize, options));
-}
-
-/** DB clearing transaction */
-function clearDbTransaction(sequelize: Sequelize, options: ClearDbOptions) {
-  return async (t: Transaction): Promise<void> => {
-    for (const model of Object.keys(sequelize.models)) {
-      // We ignore the migration metadata schema
-      if (!options.includeMetadata && model === 'sequelizeMeta') continue;
-      const table = sequelize.models[model].tableName;
-      await sequelize.query(
-        `TRUNCATE ${table}${process.env.SQL_TYPE === 'postgres' ? ' CASCADE' : ''};`,
-        {
-          raw: true,
-          transaction: t
-        }
-      );
-    }
-  };
-}
+/* eslint-disable @typescript-eslint/no-require-imports */
+// Register ts-node at the top of runMigrations.ts
+require('ts-node/register');
+
+import { existsSync, lstatSync } from 'fs';
+import { join } from 'path';
+import { Sequelize, Transaction } from 'sequelize';
+import { Umzug, SequelizeStorage } from 'umzug';
+import { ClearDbOptions, RecreateDbOptions } from './dbMigration.types';
+import { migrateMetadataTable } from './migrateMetadataTable';
+import { SequelizeMeta } from 'db/models';
+import { EngineError } from 'entities/EngineError';
+import { sequelize } from 'entities/Sequelize';
+import { reportDebug } from 'utils/report';
+import { semverCompare } from 'utils/semverCompare';
+
+const namespace = 'engine:utils:dbMigration';
+
+function getMigrationPaths(): Array<string> | undefined {
+  // We look between possibilities for the first directory that exists and matches our expectations
+  // First a user specified one, then the production default, then the development default
+  const prodDefaultPath = join(process.cwd(), './dist/db/migrations');
+  const devDefaultPath = join(process.cwd(), './src/db/migrations');
+  const path = [
+    process.env.DB_MIGRATIONS_DIR,
+    process.env.NODE_ENV === 'prod' ? prodDefaultPath : devDefaultPath
+  ].filter((path) => path && existsSync(path) && lstatSync(path).isDirectory());
+  reportDebug({
+    namespace,
+    message: 'Obtaining Umzug migration options',
+    data: {
+      path,
+      prodPath: prodDefaultPath,
+      devPath: devDefaultPath,
+      customPath: process.env.DB_MIGRATIONS_DIR
+    }
+  });
+  return path.map((p) =>
+    process.env.NODE_ENV === 'prod' ? `${p}/*.js` : `${p}/*.ts`
+  );
+}
+
+/** Run all the non-executed migrations one by one */
+export async function runPendingMigrations(): Promise<void> {
+  if (!sequelize)
+    throw new EngineError({ message: 'Sequelize is not initialized' });
+  // First make sure that the metadata table is up to date
+  await migrateMetadataTable(sequelize);
+  const migrationPaths = getMigrationPaths();
+
+  if (!migrationPaths || migrationPaths.length <= 0) {
+    reportDebug({
+      namespace,
+      message: 'No valid migration paths found.',
+      data: { migrationPaths }
+    });
+    return;
+  }
+
+  migrationPaths?.forEach(async (path) => {
+    const migrator = new Umzug({
+      migrations: {
+        glob: path,
+        resolve: ({ name, path, context }) => {
+          // Use require to import the migration file
+          const migration = require(path || '');
+          return {
+            name,
+            up: async () => migration.up(context),
+            down: async () => migration.down(context)
+          };
+        }
+      },
+      context: sequelize.getQueryInterface(),
+      storage: new SequelizeStorage({ sequelize }),
+      logger: console
+    });
+
+    reportDebug({
+      namespace,
+      message: 'Starting pending database migrations',
+      data: { path }
+    });
+    await migrator.up();
+    console.log('Db Migration Completed');
+  });
+}
+
+/** Rollback all migrations up to a specified target version */
+export async function rollbackMigrations(): Promise<void> {
+  if (!sequelize)
+    throw new EngineError({ message: 'Sequelize is not initialized' });
+  if (!process.env.BUILD_ID)
+    throw new EngineError({ message: 'BUILD_ID is not defined' });
+
+  // Fetch executed migrations in ascending order
+  const migrations = await SequelizeMeta.findAll({
+    order: [['name', 'ASC']],
+    raw: true
+  });
+  const rollbackUntil = migrations.find(
+    (migration) =>
+      migration.version &&
+      semverCompare(migration.version, process.env.BUILD_ID as string) === 1
+  );
+
+  if (!rollbackUntil) {
+    reportDebug({
+      namespace,
+      message: 'No rollback target found for the current BUILD_ID.',
+      data: { buildId: process.env.BUILD_ID }
+    });
+    return;
+  }
+
+  const migrationPaths = getMigrationPaths();
+
+  if (!migrationPaths || migrationPaths.length <= 0) {
+    reportDebug({
+      namespace,
+      message: 'No valid migration paths found for rollback.',
+      data: { migrationPaths }
+    });
+    return;
+  }
+
+  migrationPaths.forEach(async (path) => {
+    const migrator = new Umzug({
+      migrations: {
+        glob: path,
+        resolve: ({ name, path, context }) => {
+          const migration = require(path || '');
+          return {
+            name,
+            up: async () => migration.up(context),
+            down: async () => migration.down(context)
+          };
+        }
+      },
+      context: sequelize.getQueryInterface(),
+      storage: new SequelizeStorage({ sequelize }),
+      logger: console
+    });
+
+    reportDebug({
+      namespace,
+      message: `Rolling back migrations up to: ${rollbackUntil.name}`,
+      data: { path }
+    });
+
+    await migrator.down({ to: rollbackUntil.name });
+  });
+}
+
+/**
+ * Recreate the whole database
+ * Quick way to make sure that we are working on a clean database
+ * /!\ Should only be used when running tests
+ */
+export async function testRecreateDB(
+  options: RecreateDbOptions = {}
+): Promise<void> {
+  if (!sequelize)
+    throw new EngineError({ message: 'Sequelize is not initialized' });
+  await sequelize.query(
+    `DROP SCHEMA public${process.env.SQL_TYPE === 'postgres' ? ' CASCADE' : ''}`
+  );
+  await sequelize.query(`CREATE SCHEMA public`);
+  if (!options.clean) await runPendingMigrations();
+}
+
+/**
+ * Empty all the schemas in the database
+ * Quick way to flush the database data while keep the structure
+ * /!\ Should only be used when running tests
+ */
+export async function testClearDB(options: ClearDbOptions = {}): Promise<void> {
+  if (!sequelize)
+    throw new EngineError({ message: 'Sequelize is not initialized' });
+  await sequelize.transaction(clearDbTransaction(sequelize, options));
+}
+
+/** DB clearing transaction */
+function clearDbTransaction(sequelize: Sequelize, options: ClearDbOptions) {
+  return async (t: Transaction): Promise<void> => {
+    for (const model of Object.keys(sequelize.models)) {
+      // We ignore the migration metadata schema
+      if (!options.includeMetadata && model === 'sequelizeMeta') continue;
+      const table = sequelize.models[model].tableName;
+      await sequelize.query(
+        `TRUNCATE ${table}${process.env.SQL_TYPE === 'postgres' ? ' CASCADE' : ''};`,
+        {
+          raw: true,
+          transaction: t
+        }
+      );
+    }
+  };
+}

package/src/utils/dbMigration/dbMigration.types.ts

@@ -1,11 +1,11 @@
-/** Options to recreate the database */
-export interface RecreateDbOptions {
-  /** Do not run the migrations, leave the DB empty */
-  clean?: boolean;
-}
-
-/** Options to clear the database */
-export interface ClearDbOptions {
-  /** Also truncate the migration metadata table. This should only be used when testing on the metadata table itself */
-  includeMetadata?: boolean;
-}
+/** Options to recreate the database */
+export interface RecreateDbOptions {
+  /** Do not run the migrations, leave the DB empty */
+  clean?: boolean;
+}
+
+/** Options to clear the database */
+export interface ClearDbOptions {
+  /** Also truncate the migration metadata table. This should only be used when testing on the metadata table itself */
+  includeMetadata?: boolean;
+}

package/src/utils/dbMigration/index.ts

@@ -1,2 +1,2 @@
-export * from './dbMigration';
-export * from './dbMigration.types';
+export * from './dbMigration';
+export * from './dbMigration.types';

package/src/utils/dbMigration/migrateMetadataTable/index.ts

@@ -1 +1 @@
-export * from './migrateMetadataTable';
+export * from './migrateMetadataTable';

package/src/utils/dbMigration/migrateMetadataTable/migrateMetadataTable.test.ts

@@ -1,59 +1,59 @@
-// import {expect} from 'chai';
-// import { faker } from '@faker-js/faker';
-// import {QueryTypes} from 'sequelize';
-// import {testRecreateDB} from '../dbMigration';
-// import {migrateMetadataTable} from './migrateMetadataTable';
-// import {sequelize} from 'entities/Sequelize';
-
-// describe('Utils - DB Migration - Migrate Metadata Table', function () {
-//   before(async () => {
-//     sequelize.init();
-//     await testRecreateDB({clean: true});
-//   });
-
-//   afterEach(async () => {
-//     await testRecreateDB({clean: true});
-//   });
-
-//   // it('should create the metadata table if it does not exist', async () => {
-//   //   await migrateMetadataTable(sequelize);
-
-//   //   const tables = await sequelize.query<[string]>(
-//   //     "SELECT table_name FROM information_schema.tables WHERE table_schema = 'public'",
-//   //     {type: QueryTypes.SELECT},
-//   //   );
-//   //   const hasTable = tables.some((table) => table[0] === 'sequelize_meta');
-//   //   expect(hasTable).to.be.true;
-//   // });
-
-//   it('should copy legacy table data to the new table', async () => {
-//     await sequelize.query('CREATE TABLE SequelizeMeta (name varchar(100) PRIMARY KEY)', {
-//       type: QueryTypes.RAW,
-//     });
-
-//     const migrations = [faker.lorem.word(), faker.lorem.word(), faker.lorem.word()];
-//     await sequelize.query(
-//       `INSERT INTO "SequelizeMeta" ("name") VALUES ('${migrations[0]}'), ('${migrations[1]}'), ('${migrations[2]}')`,
-//       {
-//         type: QueryTypes.RAW,
-//       },
-//     );
-
-//     await migrateMetadataTable(sequelize);
-
-//     const results = await sequelize.query<{
-//       /** Name of the migration */
-//       name: string;
-//       /** Version at which the migration was executed */
-//       version: string;
-//     }>('SELECT * FROM sequelize_meta', {
-//       type: QueryTypes.SELECT,
-//     });
-
-//     expect(results).to.have.length(3);
-//     results.forEach((result) => {
-//       expect(migrations).to.include(result.name);
-//       expect(result.version).to.be.null;
-//     });
-//   });
-// });
+// import {expect} from 'chai';
+// import { faker } from '@faker-js/faker';
+// import {QueryTypes} from 'sequelize';
+// import {testRecreateDB} from '../dbMigration';
+// import {migrateMetadataTable} from './migrateMetadataTable';
+// import {sequelize} from 'entities/Sequelize';
+
+// describe('Utils - DB Migration - Migrate Metadata Table', function () {
+//   before(async () => {
+//     sequelize.init();
+//     await testRecreateDB({clean: true});
+//   });
+
+//   afterEach(async () => {
+//     await testRecreateDB({clean: true});
+//   });
+
+//   // it('should create the metadata table if it does not exist', async () => {
+//   //   await migrateMetadataTable(sequelize);
+
+//   //   const tables = await sequelize.query<[string]>(
+//   //     "SELECT table_name FROM information_schema.tables WHERE table_schema = 'public'",
+//   //     {type: QueryTypes.SELECT},
+//   //   );
+//   //   const hasTable = tables.some((table) => table[0] === 'sequelize_meta');
+//   //   expect(hasTable).to.be.true;
+//   // });
+
+//   it('should copy legacy table data to the new table', async () => {
+//     await sequelize.query('CREATE TABLE SequelizeMeta (name varchar(100) PRIMARY KEY)', {
+//       type: QueryTypes.RAW,
+//     });
+
+//     const migrations = [faker.lorem.word(), faker.lorem.word(), faker.lorem.word()];
+//     await sequelize.query(
+//       `INSERT INTO "SequelizeMeta" ("name") VALUES ('${migrations[0]}'), ('${migrations[1]}'), ('${migrations[2]}')`,
+//       {
+//         type: QueryTypes.RAW,
+//       },
+//     );
+
+//     await migrateMetadataTable(sequelize);
+
+//     const results = await sequelize.query<{
+//       /** Name of the migration */
+//       name: string;
+//       /** Version at which the migration was executed */
+//       version: string;
+//     }>('SELECT * FROM sequelize_meta', {
+//       type: QueryTypes.SELECT,
+//     });
+
+//     expect(results).to.have.length(3);
+//     results.forEach((result) => {
+//       expect(migrations).to.include(result.name);
+//       expect(result.version).to.be.null;
+//     });
+//   });
+// });