From f1e88d35ba00cd578b85e5035dfefed7c182f9b2 Mon Sep 17 00:00:00 2001
From: Ryan Vogel <ryan@mandarin3d.com>
Date: Sat, 28 Mar 2026 14:28:44 -0400
Subject: [PATCH] update for the db.ts

---
 packages/apn-relay/package.json |  1 +
 packages/apn-relay/src/check.ts | 28 ++++++++++++++++++++++++++++
 packages/apn-relay/src/db.ts    |  2 +-
 3 files changed, 30 insertions(+), 1 deletion(-)
 create mode 100644 packages/apn-relay/src/check.ts

diff --git a/packages/apn-relay/package.json b/packages/apn-relay/package.json
index d6c3d11b79..516be53563 100644
--- a/packages/apn-relay/package.json
+++ b/packages/apn-relay/package.json
@@ -7,6 +7,7 @@
   "license": "MIT",
   "scripts": {
     "dev": "bun run src/index.ts",
+    "db:check": "bun run --env-file .env src/check.ts",
     "typecheck": "tsgo --noEmit"
   },
   "dependencies": {
diff --git a/packages/apn-relay/src/check.ts b/packages/apn-relay/src/check.ts
new file mode 100644
index 0000000000..acd0a72ed2
--- /dev/null
+++ b/packages/apn-relay/src/check.ts
@@ -0,0 +1,28 @@
+import { sql } from "drizzle-orm"
+import { db } from "./db"
+import { env } from "./env"
+import { delivery_log, device_registration } from "./schema.sql"
+import { setup } from "./setup"
+
+async function run() {
+  console.log(`[apn-relay] DB host: ${env.DATABASE_HOST}`)
+
+  await db.execute(sql`SELECT 1`)
+  console.log("[apn-relay] DB connection OK")
+
+  await setup()
+  console.log("[apn-relay] Setup migration OK")
+
+  const [a] = await db.select({ value: sql<number>`count(*)` }).from(device_registration)
+  const [b] = await db.select({ value: sql<number>`count(*)` }).from(delivery_log)
+
+  console.log(`[apn-relay] device_registration rows: ${Number(a?.value ?? 0)}`)
+  console.log(`[apn-relay] delivery_log rows: ${Number(b?.value ?? 0)}`)
+  console.log("[apn-relay] DB check passed")
+}
+
+run().catch((err) => {
+  console.error("[apn-relay] DB check failed")
+  console.error(err)
+  process.exit(1)
+})
diff --git a/packages/apn-relay/src/db.ts b/packages/apn-relay/src/db.ts
index 0035abf818..527bf17228 100644
--- a/packages/apn-relay/src/db.ts
+++ b/packages/apn-relay/src/db.ts
@@ -8,4 +8,4 @@ const client = new Client({
   password: env.DATABASE_PASSWORD,
 })
 
-export const db = drizzle(client)
+export const db = drizzle({ client })