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

import db from '@server/db';
import { newts, sites } from '@server/db/schema';
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(`Newt not found for site ${siteId}`);
    }

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

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

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(`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(`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}`);
}

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(`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(`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}`);
}