97 lines
2.2 KiB
JavaScript
97 lines
2.2 KiB
JavaScript
/* 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);
|
|
}
|
|
}
|
|
}
|
|
})();
|