homeio

1import { spawn } from "node:child_process";
2import { drizzle } from "drizzle-orm/node-postgres";
3import { reset } from "drizzle-seed";
4import { Pool } from "pg";
5
6// path to a file with schema you want to reset
7import * as schema from "../lib/server/db/schema";
8
9function isMissingRelationError(error: unknown) {
10  const err = error as {
11    code?: string;
12    message?: string;
13    cause?: {
14      code?: string;
15      message?: string;
16    };
17  };
18
19  if (err.code === "42P01" || err.cause?.code === "42P01") {
20    return true;
21  }
22
23  const message = `${err.message ?? ""}\n${err.cause?.message ?? ""}`.toLowerCase();
24  return message.includes("does not exist");
25}
26
27async function runDbInit(databaseUrl: string) {
28  await new Promise<void>((resolve, reject) => {
29    const child = spawn("npm", ["run", "db:init", "--silent"], {
30      cwd: process.cwd(),
31      env: {
32        ...process.env,
33        DATABASE_URL: databaseUrl,
34      },
35      stdio: "inherit",
36      shell: process.platform === "win32",
37    });
38
39    child.once("error", reject);
40    child.once("exit", (code) => {
41      if (code === 0) {
42        resolve();
43        return;
44      }
45      reject(new Error(`db:init failed with exit code ${code ?? "unknown"}`));
46    });
47  });
48}
49
50async function main() {
51  const databaseUrl =
52    process.env.DATABASE_URL ??
53    "postgresql://postgres:postgres@127.0.0.1:5432/home_server";
54
55  if (!databaseUrl) {
56    throw new Error("DATABASE_URL is not set.");
57  }
58
59  const pool = new Pool({ connectionString: databaseUrl });
60  const db = drizzle(pool, { schema });
61
62  try {
63    let retriedAfterInit = false;
64
65    while (true) {
66      try {
67        await reset(db, schema);
68        break;
69      } catch (error) {
70        if (!retriedAfterInit && isMissingRelationError(error)) {
71          retriedAfterInit = true;
72          console.warn(
73            "Detected missing database tables. Running db:init once, then retrying db:reset...",
74          );
75          await runDbInit(databaseUrl);
76          continue;
77        }
78
79        throw error;
80      }
81    }
82
83    console.log("Database reset complete.");
84  } finally {
85    await pool.end();
86  }
87}
88
89main().catch((error) => {
90  console.error("Database reset failed.");
91  console.error(error);
92  process.exitCode = 1;
93});
94