refactor: use esm-safe database migrations

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
MythEclipse
2026-05-14 23:23:07 +07:00
co-authored by Claude Opus 4.7
parent 954faf6c5a
commit e5aa398e5c
2 changed files with 45 additions and 10 deletions
+24 -10
View File
@@ -1,28 +1,43 @@
import "dotenv/config";
import Database from "better-sqlite3";
import { drizzle as drizzleSqlite } from "drizzle-orm/better-sqlite3";
import { migrate as migrateSqlite } from "drizzle-orm/better-sqlite3/migrator";
import { migrate } from "drizzle-orm/node-postgres/migrator";
import { migrate as migratePostgres } from "drizzle-orm/node-postgres/migrator";
import { config } from "../config";
import { createChildLogger } from "../logger";
import { initializeDatabase } from "./drizzle";
import { closeDatabase, initializeDatabase } from "./drizzle";
const logger = createChildLogger("migrate");
export function initializeMigrationSqliteDatabase(path = ".muxer-queue.db") {
const sqlite = new Database(path);
sqlite.pragma("journal_mode = WAL");
return { sqlite, db: drizzleSqlite(sqlite) };
}
export async function runMigrations(): Promise<void> {
try {
logger.info("Starting database migrations");
if (config.DATABASE_TYPE === "postgres") {
logger.info("Running PostgreSQL migrations");
const db = await initializeDatabase();
await migrate(db as any, { migrationsFolder: "./drizzle/migrations" });
const db = (await initializeDatabase()) as Parameters<
typeof migratePostgres
>[0];
try {
await migratePostgres(db, { migrationsFolder: "./drizzle/migrations" });
} finally {
await closeDatabase();
}
logger.info("PostgreSQL migrations completed successfully");
} else {
logger.info("Running SQLite migrations");
const sqlite = new Database(".muxer-queue.db");
sqlite.pragma("journal_mode = WAL");
const db = require("drizzle-orm/better-sqlite3").drizzle(sqlite);
migrateSqlite(db, { migrationsFolder: "./drizzle/migrations" });
const { sqlite, db } = initializeMigrationSqliteDatabase();
try {
migrateSqlite(db, { migrationsFolder: "./drizzle/migrations" });
} finally {
sqlite.close();
}
logger.info("SQLite migrations completed successfully");
}
} catch (error) {
@@ -34,8 +49,7 @@ export async function runMigrations(): Promise<void> {
}
}
// Run migrations if called directly
if (require.main === module) {
if (import.meta.url === `file://${process.argv[1]}`) {
runMigrations()
.then(() => {
logger.info("Migrations completed");