pangolin/server/setup/copyInConfig.ts

import { db } from "@server/db";
import { exitNodes, orgs, resources } from "../db/schema";
import config from "@server/lib/config";
import { eq, ne } from "drizzle-orm";
import logger from "@server/logger";

export async function copyInConfig() {
    const domain = config.getBaseDomain();
    const endpoint = config.getRawConfig().gerbil.base_endpoint;
    const listenPort = config.getRawConfig().gerbil.start_port;

    // update the domain on all of the orgs where the domain is not equal to the new domain
    // TODO: eventually each org could have a unique domain that we do not want to overwrite, so this will be unnecessary
    await db.update(orgs).set({ domain }).where(ne(orgs.domain, domain));

    // TODO: eventually each exit node could have a different endpoint
    await db.update(exitNodes).set({ endpoint }).where(ne(exitNodes.endpoint, endpoint));
    // TODO: eventually each exit node could have a different port
    await db.update(exitNodes).set({ listenPort }).where(ne(exitNodes.listenPort, listenPort));

    // update all resources fullDomain to use the new domain
    await db.transaction(async (trx) => {
        const allResources = await trx.select().from(resources);

        for (const resource of allResources) {
            let fullDomain = "";
            if (resource.isBaseDomain) {
                fullDomain = domain;
            } else {
                fullDomain = `${resource.subdomain}.${domain}`;
            }
            await trx
                .update(resources)
                .set({ fullDomain })
                .where(eq(resources.resourceId, resource.resourceId));
        }
    });

    logger.info(`Updated orgs with new domain (${domain})`);
}