Working on new target type

This commit is contained in:
Owen
2026-05-13 11:56:23 -07:00
parent 743621eb25
commit 4e07e9c52c
3 changed files with 70 additions and 2 deletions

View File

@@ -1,4 +1,6 @@
import { import {
browserGatewayTarget,
BrowserGatewayTarget,
clients, clients,
clientSiteResourcesAssociationsCache, clientSiteResourcesAssociationsCache,
clientSitesAssociationsCache, clientSitesAssociationsCache,
@@ -310,3 +312,12 @@ export async function buildTargetConfigurationForNewtClient(
udpTargets udpTargets
}; };
} }
export async function buildBrowserGatewayTargetConfigurationForNewtClient(
siteId: number
): Promise<BrowserGatewayTarget[]> {
return await db
.select()
.from(browserGatewayTarget)
.where(eq(browserGatewayTarget.siteId, siteId));
}

View File

@@ -3,6 +3,7 @@ import { eq } from "drizzle-orm";
import { sendToClient } from "#dynamic/routers/ws"; import { sendToClient } from "#dynamic/routers/ws";
import logger from "@server/logger"; import logger from "@server/logger";
import { import {
buildBrowserGatewayTargetConfigurationForNewtClient,
buildClientConfigurationForNewtClient, buildClientConfigurationForNewtClient,
buildTargetConfigurationForNewtClient buildTargetConfigurationForNewtClient
} from "./buildConfiguration"; } from "./buildConfiguration";
@@ -12,6 +13,9 @@ export async function sendNewtSyncMessage(newt: Newt, site: Site) {
const { tcpTargets, udpTargets, validHealthCheckTargets } = const { tcpTargets, udpTargets, validHealthCheckTargets } =
await buildTargetConfigurationForNewtClient(site.siteId); await buildTargetConfigurationForNewtClient(site.siteId);
const browserGatewayTargets =
await buildBrowserGatewayTargetConfigurationForNewtClient(site.siteId);
let exitNode: ExitNode | undefined; let exitNode: ExitNode | undefined;
if (site.exitNodeId) { if (site.exitNodeId) {
[exitNode] = await db [exitNode] = await db
@@ -36,7 +40,15 @@ export async function sendNewtSyncMessage(newt: Newt, site: Site) {
}, },
healthCheckTargets: validHealthCheckTargets, healthCheckTargets: validHealthCheckTargets,
peers: peers, peers: peers,
clientTargets: targets clientTargets: targets,
browserGatewayTargets: browserGatewayTargets.map((t) => ({
id: t.browserGatewayTargetId,
resourceId: t.resourceId,
siteId: t.siteId,
type: t.type,
destination: t.destination,
destinationPort: t.destinationPort
}))
} }
}, },
{ {

View File

@@ -1,4 +1,4 @@
import { Target, TargetHealthCheck } from "@server/db"; import { BrowserGatewayTarget, Target, TargetHealthCheck } from "@server/db";
import { sendToClient } from "#dynamic/routers/ws"; import { sendToClient } from "#dynamic/routers/ws";
import logger from "@server/logger"; import logger from "@server/logger";
import { canCompress } from "@server/lib/clientVersionChecks"; import { canCompress } from "@server/lib/clientVersionChecks";
@@ -239,3 +239,48 @@ export async function removeTargets(
{ incrementConfigVersion: true, compress: canCompress(version, "newt") } { incrementConfigVersion: true, compress: canCompress(version, "newt") }
); );
} }
export async function sendBrowserGatewayTargets(
newtId: string,
targets: BrowserGatewayTarget[],
version?: string | null
) {
if (targets.length === 0) return;
const payload = targets.map((t) => ({
id: t.browserGatewayTargetId,
resourceId: t.resourceId,
siteId: t.siteId,
type: t.type,
destination: t.destination,
destinationPort: t.destinationPort
}));
await sendToClient(
newtId,
{
type: "newt/browsergateway/add",
data: {
targets: payload
}
},
{ incrementConfigVersion: true, compress: canCompress(version, "newt") }
);
}
export async function removeBrowserGatewayTarget(
newtId: string,
browserGatewayTargetId: number,
version?: string | null
) {
await sendToClient(
newtId,
{
type: "newt/browsergateway/remove",
data: {
ids: [browserGatewayTargetId]
}
},
{ incrementConfigVersion: true, compress: canCompress(version, "newt") }
);
}