From 3e01bfef7d7e489dd4903f3d6692504f3ec4233e Mon Sep 17 00:00:00 2001
From: Owen <owen@txv.io>
Date: Thu, 18 Dec 2025 17:07:48 -0500
Subject: [PATCH] Move primaryDb into driver

---
 server/db/pg/driver.ts                            | 1 +
 server/db/sqlite/driver.ts                        | 2 +-
 server/routers/auditLogs/queryRequestAnalytics.ts | 7 +------
 server/routers/auditLogs/queryRequestAuditLog.ts  | 7 +------
 4 files changed, 4 insertions(+), 13 deletions(-)

diff --git a/server/db/pg/driver.ts b/server/db/pg/driver.ts
index 2ee34da6..5b357d06 100644
--- a/server/db/pg/driver.ts
+++ b/server/db/pg/driver.ts
@@ -81,6 +81,7 @@ function createDb() {
 
 export const db = createDb();
 export default db;
+export const primaryDb = db.$primary;
 export type Transaction = Parameters<
     Parameters<(typeof db)["transaction"]>[0]
 >[0];
diff --git a/server/db/sqlite/driver.ts b/server/db/sqlite/driver.ts
index 5a4aa542..9cbc8d7b 100644
--- a/server/db/sqlite/driver.ts
+++ b/server/db/sqlite/driver.ts
@@ -20,7 +20,7 @@ function createDb() {
 
 export const db = createDb();
 export default db;
-export const driver: "pg" | "sqlite" = "sqlite";
+export const primaryDb = db;
 export type Transaction = Parameters<
     Parameters<(typeof db)["transaction"]>[0]
 >[0];
diff --git a/server/routers/auditLogs/queryRequestAnalytics.ts b/server/routers/auditLogs/queryRequestAnalytics.ts
index f4b4444c..cd1218ce 100644
--- a/server/routers/auditLogs/queryRequestAnalytics.ts
+++ b/server/routers/auditLogs/queryRequestAnalytics.ts
@@ -1,4 +1,4 @@
-import { db, requestAuditLog, driver } from "@server/db";
+import { db, requestAuditLog, driver, primaryDb } from "@server/db";
 import { registry } from "@server/openApi";
 import { NextFunction } from "express";
 import { Request, Response } from "express";
@@ -12,11 +12,6 @@ import response from "@server/lib/response";
 import logger from "@server/logger";
 import { getSevenDaysAgo } from "@app/lib/getSevenDaysAgo";
 
-let primaryDb = db;
-if (driver == "pg") {
-    primaryDb = db.$primary as typeof db; // select the primary instance in a replicated setup
-}
-
 const queryAccessAuditLogsQuery = z.object({
     // iso string just validate its a parseable date
     timeStart: z
diff --git a/server/routers/auditLogs/queryRequestAuditLog.ts b/server/routers/auditLogs/queryRequestAuditLog.ts
index 73f9fc43..602b4475 100644
--- a/server/routers/auditLogs/queryRequestAuditLog.ts
+++ b/server/routers/auditLogs/queryRequestAuditLog.ts
@@ -1,4 +1,4 @@
-import { db, driver, requestAuditLog, resources } from "@server/db";
+import { db, primaryDb, requestAuditLog, resources } from "@server/db";
 import { registry } from "@server/openApi";
 import { NextFunction } from "express";
 import { Request, Response } from "express";
@@ -13,11 +13,6 @@ import response from "@server/lib/response";
 import logger from "@server/logger";
 import { getSevenDaysAgo } from "@app/lib/getSevenDaysAgo";
 
-let primaryDb = db;
-if (driver == "pg") {
-    primaryDb = db.$primary as typeof db; // select the primary instance in a replicated setup
-}
-
 export const queryAccessAuditLogsQuery = z.object({
     // iso string just validate its a parseable date
     timeStart: z