122 lines
2.9 KiB
JavaScript
122 lines
2.9 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) {
|
|
console.log('Skipping database check.');
|
|
process.exit(0);
|
|
}
|
|
|
|
function getDatabaseType(url = process.env.DATABASE_URL) {
|
|
const type = process.env.DATABASE_TYPE || (url && url.split(':')[0]);
|
|
|
|
if (type === 'postgres') {
|
|
return 'postgresql';
|
|
} else if (type === 'mongodb+srv') {
|
|
return 'mongodb';
|
|
}
|
|
|
|
return type;
|
|
}
|
|
|
|
const databaseType = getDatabaseType();
|
|
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.');
|
|
}
|
|
}
|
|
|
|
async function checkDatabaseVersion(databaseType) {
|
|
let version;
|
|
if (databaseType === 'mongodb') {
|
|
const query = await prisma.$runCommandRaw({
|
|
buildInfo: 1,
|
|
});
|
|
version = semver.valid(query.version);
|
|
} else {
|
|
const query = await prisma.$queryRaw`select version() as version`;
|
|
version = semver.valid(semver.coerce(query[0].version));
|
|
}
|
|
|
|
const minVersion =
|
|
databaseType === 'postgresql' ? '9.4.0' : databaseType === 'mongodb' ? '3.0.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 checkV1Tables() {
|
|
if (databaseType === 'mongodb') {
|
|
// Ignore
|
|
return;
|
|
}
|
|
|
|
try {
|
|
await prisma.$queryRaw`select * from account limit 1`;
|
|
|
|
error(
|
|
'Umami v1 tables detected. For how to upgrade from v1 to v2 go to https://umami.is/docs/migrate-v1-v2.',
|
|
);
|
|
process.exit(1);
|
|
} catch (e) {
|
|
// Ignore
|
|
}
|
|
}
|
|
|
|
async function applyMigration() {
|
|
if (databaseType === 'mongodb') {
|
|
console.log(execSync('prisma db push').toString());
|
|
} else {
|
|
console.log(execSync('prisma migrate deploy').toString());
|
|
}
|
|
|
|
success('Database is up to date.');
|
|
}
|
|
|
|
(async () => {
|
|
let err = false;
|
|
for (let fn of [checkEnv, checkConnection, checkDatabaseVersion, checkV1Tables, applyMigration]) {
|
|
try {
|
|
fn.name === 'checkDatabaseVersion' ? await fn(databaseType) : await fn();
|
|
} catch (e) {
|
|
error(e.message);
|
|
err = true;
|
|
} finally {
|
|
await prisma.$disconnect();
|
|
if (err) {
|
|
process.exit(1);
|
|
}
|
|
}
|
|
}
|
|
})();
|