35773a6969
* refactor: move migrations to new structure * refactor: convert all findMany to new structure * fix(backups-schedule): missing null matching for last backup status * chore: move root lib to server
75 lines
2.1 KiB
TypeScript
75 lines
2.1 KiB
TypeScript
import { Database } from "bun:sqlite";
|
|
import { relations } from "./relations";
|
|
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 fs from "node:fs";
|
|
import { config } from "../core/config";
|
|
import type * as schemaTypes from "./schema";
|
|
|
|
/**
|
|
* TODO: try to remove this if moving away from react-router.
|
|
* The rr vite plugin doesn't let us customize the chunk names
|
|
* to isolate the db initialization code from the rest of the server code.
|
|
*/
|
|
let _sqlite: Database | undefined;
|
|
let _db: ReturnType<typeof initDb> | undefined;
|
|
let _schema: typeof schemaTypes | undefined;
|
|
|
|
/**
|
|
* Sets the database schema. This must be called before any database operations.
|
|
*/
|
|
export const setSchema = (schema: typeof schemaTypes) => {
|
|
_schema = schema;
|
|
};
|
|
|
|
const initDb = () => {
|
|
if (!_schema) {
|
|
throw new Error("Database schema not set. Call setSchema() before accessing the database.");
|
|
}
|
|
|
|
fs.mkdirSync(path.dirname(DATABASE_URL), { recursive: true });
|
|
|
|
if (fs.existsSync(path.join(path.dirname(DATABASE_URL), "ironmount.db")) && !fs.existsSync(DATABASE_URL)) {
|
|
fs.renameSync(path.join(path.dirname(DATABASE_URL), "ironmount.db"), DATABASE_URL);
|
|
}
|
|
|
|
_sqlite = new Database(DATABASE_URL);
|
|
return drizzle({ client: _sqlite, relations, schema: _schema });
|
|
};
|
|
|
|
/**
|
|
* Database instance (Proxy for lazy initialization)
|
|
*/
|
|
export const db = new Proxy(
|
|
{},
|
|
{
|
|
get(_, prop, receiver) {
|
|
if (!_db) {
|
|
_db = initDb();
|
|
}
|
|
return Reflect.get(_db, prop, receiver);
|
|
},
|
|
},
|
|
) as ReturnType<typeof initDb>;
|
|
|
|
export const runDbMigrations = () => {
|
|
let migrationsFolder: string;
|
|
|
|
if (config.migrationsPath) {
|
|
migrationsFolder = config.migrationsPath;
|
|
} else if (config.__prod__) {
|
|
migrationsFolder = path.join("/app", "assets", "migrations");
|
|
} else {
|
|
migrationsFolder = path.join(process.cwd(), "app", "drizzle");
|
|
}
|
|
|
|
migrate(db, { migrationsFolder });
|
|
|
|
if (!_sqlite) {
|
|
throw new Error("Database not initialized");
|
|
}
|
|
|
|
_sqlite.run("PRAGMA foreign_keys = ON;");
|
|
};
|