1 files changed, 19 insertions, 0 deletions
diff --git a/packages/queue/db.ts b/packages/queue/db.ts
new file mode 100644
index 00000000..f1412fef
--- /dev/null
+++ b/packages/queue/db.ts
@@ -0,0 +1,19 @@
+import path from "node:path";
+import Database from "better-sqlite3";
+import { BetterSQLite3Database, drizzle } from "drizzle-orm/better-sqlite3";
+import { migrate } from "drizzle-orm/better-sqlite3/migrator";
+
+import * as schema from "./schema";
+
+export function buildDBClient(dbPath: string, runMigrations = false) {
+  const sqlite = new Database(dbPath);
+  const db = drizzle(sqlite, { schema });
+  if (runMigrations) {
+    migrateDB(db);
+  }
+  return db;
+}
+
+export function migrateDB(db: BetterSQLite3Database<typeof schema>) {
+  migrate(db, { migrationsFolder: path.join(__dirname, "drizzle") });
+}