pangolin/server/routers/newt/sync.ts

import { ExitNode, exitNodes, Newt, Site, db } from "@server/db";
import { eq } from "drizzle-orm";
import { sendToClient } from "#dynamic/routers/ws";
import logger from "@server/logger";
import {
    buildClientConfigurationForNewtClient,
    buildTargetConfigurationForNewtClient
} from "./buildConfiguration";

export async function sendNewtSyncMessage(newt: Newt, site: Site) {
    const { tcpTargets, udpTargets, validHealthCheckTargets } =
        await buildTargetConfigurationForNewtClient(site.siteId);

    let exitNode: ExitNode | undefined;
    if (site.exitNodeId) {
        [exitNode] = await db
            .select()
            .from(exitNodes)
            .where(eq(exitNodes.exitNodeId, site.exitNodeId))
            .limit(1);
    }
    const { peers, targets } = await buildClientConfigurationForNewtClient(
        site,
        exitNode
    );

    await sendToClient(newt.newtId, {
        type: "newt/sync",
        data: {
            proxyTargets: {
                udp: udpTargets,
                tcp: tcpTargets
            },
            healthCheckTargets: validHealthCheckTargets,
            peers: peers,
            clientTargets: targets
        }
    }).catch((error) => {
        logger.warn(`Error sending newt sync message:`, error);
    });
}