homeio

1import { existsSync, readFileSync } from "node:fs";
2import { join } from "node:path";
3import { Pool } from "pg";
4import { drizzle } from "drizzle-orm/node-postgres";
5import { migrate } from "drizzle-orm/node-postgres/migrator";
6import { parseEnvFileContent } from "@/lib/shared/env-file";
7
8const MIGRATIONS_FOLDER = join(process.cwd(), "drizzle");
9
10function getDatabaseUrl(): string {
11  if (process.env.DATABASE_URL) return process.env.DATABASE_URL;
12
13  for (const envFile of [".env.local", ".env"]) {
14    const envPath = join(process.cwd(), envFile);
15    if (!existsSync(envPath)) continue;
16    const env = parseEnvFileContent(readFileSync(envPath, "utf8"));
17    if (env.DATABASE_URL) return env.DATABASE_URL;
18  }
19
20  return "postgresql://postgres:postgres@127.0.0.1:5432/home_server";
21}
22
23async function run() {
24  const pool = new Pool({ connectionString: getDatabaseUrl() });
25  const db = drizzle(pool);
26
27  // Reset migration tracking so every migration reruns from scratch.
28  // All SQL files use IF NOT EXISTS / DO $$ EXCEPTION blocks, so rerunning
29  // is always safe — existing tables and data are never touched.
30  await pool.query(`CREATE SCHEMA IF NOT EXISTS drizzle`);
31  await pool.query(`DROP TABLE IF EXISTS drizzle.__drizzle_migrations`);
32
33  await migrate(db, { migrationsFolder: MIGRATIONS_FOLDER });
34
35  await pool.end();
36  console.log("[db] Migrations applied successfully.");
37}
38
39run().catch((err) => {
40  console.error("[db] Migration failed:", err);
41  process.exit(1);
42});
43