From 838e3efbcac7dbc6f712dd5b1407cd54bcc475a9 Mon Sep 17 00:00:00 2001
From: Owen <owen@txv.io>
Date: Thu, 4 Sep 2025 17:59:59 -0700
Subject: [PATCH] Pass in db to pickPort

---
 server/routers/target/createTarget.ts | 2 +-
 server/routers/target/helpers.ts      | 6 +++---
 server/routers/target/updateTarget.ts | 2 +-
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/server/routers/target/createTarget.ts b/server/routers/target/createTarget.ts
index 7a3acd55..f58c236e 100644
--- a/server/routers/target/createTarget.ts
+++ b/server/routers/target/createTarget.ts
@@ -161,7 +161,7 @@ export async function createTarget(
                 );
             }
 
-            const { internalPort, targetIps } = await pickPort(site.siteId!);
+            const { internalPort, targetIps } = await pickPort(site.siteId!, db);
 
             if (!internalPort) {
                 return next(
diff --git a/server/routers/target/helpers.ts b/server/routers/target/helpers.ts
index 4935d28a..13b2ee46 100644
--- a/server/routers/target/helpers.ts
+++ b/server/routers/target/helpers.ts
@@ -1,10 +1,10 @@
-import { db } from "@server/db";
+import { db, Transaction } from "@server/db";
 import { resources, targets } from "@server/db";
 import { eq } from "drizzle-orm";
 
 const currentBannedPorts: number[] = [];
 
-export async function pickPort(siteId: number): Promise<{
+export async function pickPort(siteId: number, trx: Transaction | typeof db): Promise<{
     internalPort: number;
     targetIps: string[];
 }> {
@@ -12,7 +12,7 @@ export async function pickPort(siteId: number): Promise<{
     const targetIps: string[] = [];
     const targetInternalPorts: number[] = [];
 
-    const targetsRes = await db
+    const targetsRes = await trx
         .select()
         .from(targets)
         .where(eq(targets.siteId, siteId));
diff --git a/server/routers/target/updateTarget.ts b/server/routers/target/updateTarget.ts
index 67d9a8df..47300619 100644
--- a/server/routers/target/updateTarget.ts
+++ b/server/routers/target/updateTarget.ts
@@ -153,7 +153,7 @@ export async function updateTarget(
             );
         }
 
-        const { internalPort, targetIps } = await pickPort(site.siteId!);
+        const { internalPort, targetIps } = await pickPort(site.siteId!, db);
 
         if (!internalPort) {
             return next(