wafrn / packages / backend / migrate.ts

import { Sequelize } from "sequelize";
import { Umzug, SequelizeStorage } from "umzug";
import process from "process";
import { realpathSync } from "fs";
import { pathToFileURL } from "url";
import { completeEnvironment } from "./utils/backendOptions.js";
import { wait } from "./utils/wait.js";

function wasCalledAsScript() {
  const realPath = realpathSync(process.argv[1]);
  const realPathAsUrl = pathToFileURL(realPath).href;
  return import.meta.url === realPathAsUrl;
}

await wait(5000);

const sequelize = new Sequelize(completeEnvironment.databaseConnectionString);

const umzug = new Umzug({
  migrations: { glob: "migrations/*.ts" },
  context: sequelize.getQueryInterface(),
  storage: new SequelizeStorage({ sequelize }),
  logger: console,
});

// export the type helper exposed by umzug, which will have the `context` argument typed correctly
export type Migration = typeof umzug._types.migration;

if (wasCalledAsScript()) {
  if (process.argv[2] == "init-container") {
    await umzug.up();
    console.log("Migrations run successfully");
    process.exit(0);
  } else {
    await umzug.runAsCLI();
  }
}