/* eslint-disable no-console */
require("dotenv").config();
const { PrismaClient } = require("@prisma/client");
const chalk = require("chalk");
const { execSync } = require("child_process");
const semver = require("semver");

if (process.env.SKIP_DB_CHECK === "true") {
  console.log("Skipping database check.");
  process.exit(0);
}

function getDatabaseType(url = process.env.DATABASE_URL) {
  const type = url && url.split(":")[0];

  if (type === "postgres") {
    return "postgresql";
  }

  return type;
}

const prisma = new PrismaClient();

function success(msg) {
  console.log(chalk.greenBright(`✓ ${msg}`));
}

function error(msg) {
  console.log(chalk.redBright(`✗ ${msg}`));
}

async function checkEnv() {
  if (!process.env.DATABASE_URL) {
    throw new Error("DATABASE_URL is not defined.");
  } else {
    success("DATABASE_URL is defined.");
  }
}

async function checkConnection() {
  try {
    await prisma.$connect();

    success("Database connection successful.");
  } catch (e) {
    throw new Error("Unable to connect to the database: " + e.message);
  }
}

async function checkDatabaseVersion() {
  const query = await prisma.$queryRaw`select version() as version`;
  const version = semver.valid(semver.coerce(query[0].version));

  const databaseType = getDatabaseType();
  const minVersion = databaseType === "postgresql" ? "9.4.0" : "5.7.0";

  if (semver.lt(version, minVersion)) {
    throw new Error(
      `Database version is not compatible. Please upgrade ${databaseType} version to ${minVersion} or greater`,
    );
  }

  success("Database version check successful.");
}

async function applyMigration() {
  if (process.env.SKIP_DB_MIGRATION === "false") {
    console.log(execSync("prisma generate").toString());
    console.log(execSync("prisma migrate deploy").toString());

    success("Database is up to date.");
  }
}

(async () => {
  let err = false;
  for (let fn of [
    checkEnv,
    checkConnection,
    checkDatabaseVersion,
    applyMigration,
  ]) {
    try {
      await fn();
    } catch (e) {
      error(e.message);
      err = true;
    } finally {
      await prisma.$disconnect();
      if (err) {
        process.exit(1);
      }
    }
  }
})();