server/scripts/db_migrations.js

#!/usr/bin/node
const path = require("path");
const fs = require("fs");
const { stdout, exit } = require("process");
const { execIn } = require("./utils.js");
const { ask } = require("./utils/ask.js");

async function main() {
	let filename;
	if (process.argv[2]) filename = process.argv[2];
	else filename = await ask("Please enter the name of your migration: ");
	let dbconf;
	try {
		dbconf = JSON.parse(fs.readFileSync("dbconf.json"));
	} catch (e) {
		console.log("No dbconf.json found!");
		dbconf = {};
	}

	if (!dbconf["sqlite"])
		dbconf.sqlite = {
			conn_str: "migrations.db",
			migrations_dir: "sqlite",
			package: "sqlite3"
		};
	if (!dbconf["postgres"] && process.env.FC_DB_POSTGRES) {
		console.log("Found FC_DB_POSTGRES environment variable. Using it!");
		dbconf.postgres = {
			conn_str: process.env.FC_DB_POSTGRES,
			migrations_dir: "postgres",
			package: "pg"
		};
	}
	if (!dbconf["mariadb"] && process.env.FC_DB_MARIADB) {
		console.log("Found FC_DB_MARIADB environment variable. Using it!");
		dbconf.mariadb = {
			conn_str: process.env.FC_DB_MARIADB,
			migrations_dir: "mariadb",
			package: "mysql2"
		};
	}
	fs.writeFileSync("dbconf.json", JSON.stringify(dbconf, null, 4));

	//build
	execIn(`node scripts/build_new.js`, process.cwd(), { stdio: "inherit" });

	if (fs.existsSync(".env") && !fs.existsSync(".env.bak")) fs.renameSync(".env", ".env.bak");
	Object.keys(dbconf).forEach((db) => {
		console.log(`Applying migrations for ${db}`);
		if (!fs.existsSync(path.join("node_modules", dbconf[db].package))) execIn(`npm i ${dbconf[db].package}`, process.cwd());
		fs.writeFileSync(
			`.env`,
			`DATABASE=${dbconf[db].conn_str}
    THREADS=1
    DB_MIGRATE=true
    DB_VERBOSE=true`
		);
		execIn(`node dist/start.js`, process.cwd(), { stdio: "inherit" });
	});

	Object.keys(dbconf).forEach((db) => {
		console.log(`Generating new migrations for ${db}`);
		fs.writeFileSync(
			`.env`,
			`DATABASE=${dbconf[db].conn_str}
    THREADS=1
    DB_MIGRATE=true
    DB_VERBOSE=true`
		);
		execIn(
			`node node_modules/typeorm/cli.js migration:generate "src/util/migrations/${db}/${filename}" -d dist/util/util/Database.js -p`,
			process.cwd(),
			{ stdio: "inherit" }
		);
	});
	if (fs.existsSync(".env.bak")) {
		fs.rmSync(".env");
		fs.renameSync(".env.bak", ".env");
	}
	exit(0);
}
main();