fosrl.pangolin/server/routers/newt/peers.ts

import { db } from "@server/db";
import { newts, sites } from "@server/db";
import { eq } from "drizzle-orm";
import { sendToClient } from "../ws";
import logger from "@server/logger";

export async function addPeer(
    siteId: number,
    peer: {
        publicKey: string;
        allowedIps: string[];
        endpoint: string;
    }
) {
    const [site] = await db
        .select()
        .from(sites)
        .where(eq(sites.siteId, siteId))
        .limit(1);
    if (!site) {
        throw new Error(`Exit node with ID ${siteId} not found`);
    }

    // get the newt on the site
    const [newt] = await db
        .select()
        .from(newts)
        .where(eq(newts.siteId, siteId))
        .limit(1);
    if (!newt) {
        throw new Error(`Site found for site ${siteId}`);
    }

    sendToClient(newt.newtId, {
        type: "newt/wg/peer/add",
        data: peer
    });

    logger.info(`Added peer ${peer.publicKey} to newt ${newt.newtId}`);

    return site;
}

export async function deletePeer(siteId: number, publicKey: string) {
    const [site] = await db
        .select()
        .from(sites)
        .where(eq(sites.siteId, siteId))
        .limit(1);
    if (!site) {
        throw new Error(`Site with ID ${siteId} not found`);
    }

    // get the newt on the site
    const [newt] = await db
        .select()
        .from(newts)
        .where(eq(newts.siteId, siteId))
        .limit(1);
    if (!newt) {
        throw new Error(`Newt not found for site ${siteId}`);
    }

    sendToClient(newt.newtId, {
        type: "newt/wg/peer/remove",
        data: {
            publicKey
        }
    });

    logger.info(`Deleted peer ${publicKey} from newt ${newt.newtId}`);

    return site;
}

export async function updatePeer(
    siteId: number,
    publicKey: string,
    peer: {
        allowedIps?: string[];
        endpoint?: string;
    }
) {
    const [site] = await db
        .select()
        .from(sites)
        .where(eq(sites.siteId, siteId))
        .limit(1);
    if (!site) {
        throw new Error(`Site with ID ${siteId} not found`);
    }

    // get the newt on the site
    const [newt] = await db
        .select()
        .from(newts)
        .where(eq(newts.siteId, siteId))
        .limit(1);
    if (!newt) {
        throw new Error(`Newt not found for site ${siteId}`);
    }

    sendToClient(newt.newtId, {
        type: "newt/wg/peer/update",
        data: {
            publicKey,
            ...peer
        }
    });

    logger.info(`Updated peer ${publicKey} on newt ${newt.newtId}`);

    return site;
}