ironmount/app/server/db/db.ts

import "dotenv/config";
import { Database } from "bun:sqlite";
import path from "node:path";
import { drizzle } from "drizzle-orm/bun-sqlite";
import { migrate } from "drizzle-orm/bun-sqlite/migrator";
import { DATABASE_URL } from "../core/constants";
import * as schema from "./schema";
import fs from "node:fs/promises";
import { logger } from "../utils/logger";

await fs.mkdir(path.dirname(DATABASE_URL), { recursive: true });

const sqlite = new Database(DATABASE_URL);
export const db = drizzle({
	client: sqlite,
	schema,
	logger: {
		logQuery(query, params) {
			logger.debug(`[Drizzle] ${query} -- [${params.join(",")}]`);
		},
	},
});

export const runDbMigrations = () => {
	let migrationsFolder = path.join("/app", "assets", "migrations");

	const { NODE_ENV } = process.env;
	if (NODE_ENV !== "production") {
		migrationsFolder = path.join("/app", "app", "drizzle");
	}

	migrate(db, { migrationsFolder });

	sqlite.run("PRAGMA foreign_keys = ON;");
};