mirror of
https://github.com/fosrl/pangolin.git
synced 2026-05-20 15:55:31 +00:00
Working on new target type
This commit is contained in:
@@ -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));
|
||||||
|
}
|
||||||
|
|||||||
@@ -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
|
||||||
|
}))
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
|
|||||||
@@ -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") }
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user