calvin.erfmann/pangolin
1
0
Fork 0
mirror of https://github.com/fosrl/pangolin.git synced 2026-01-29 06:10:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Pass in db to pickPort

Browse Source
This commit is contained in:
Owen
2025-09-04 17:59:59 -07:00
parent 3e353717f5
commit 838e3efbca
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
server/routers/target/createTarget.ts
View File

@@ -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(

6
server/routers/target/helpers.ts
View File

@@ -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));

2
server/routers/target/updateTarget.ts
View File

@@ -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(