pangolin_mirror/pangolin
1
0
Fork 0
mirror of https://github.com/fosrl/pangolin.git synced 2026-03-07 03:06:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Delete each of the site resources and rebuild

Browse Source
This commit is contained in:
Owen
2026-01-18 14:59:49 -08:00
parent 89928c753c
commit c03a61f613
1 changed files with 31 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
server/routers/site/deleteSite.ts
View File

@@ -1,6 +1,6 @@
import { Request, Response, NextFunction } from "express"; import { Request, Response, NextFunction } from "express";
import { z } from "zod"; import { z } from "zod";
import { db } from "@server/db"; import { db, siteResources } from "@server/db";
import { newts, newtSessions, sites } from "@server/db"; import { newts, newtSessions, sites } from "@server/db";
import { eq } from "drizzle-orm"; import { eq } from "drizzle-orm";
import response from "@server/lib/response"; import response from "@server/lib/response";
@@ -11,6 +11,7 @@ import { deletePeer } from "../gerbil/peers";
import { fromError } from "zod-validation-error"; import { fromError } from "zod-validation-error";
import { sendToClient } from "#dynamic/routers/ws"; import { sendToClient } from "#dynamic/routers/ws";
import { OpenAPITags, registry } from "@server/openApi"; import { OpenAPITags, registry } from "@server/openApi";
import { rebuildClientAssociationsFromSiteResource } from "@server/lib/rebuildClientAssociations";
const deleteSiteSchema = z.strictObject({ const deleteSiteSchema = z.strictObject({
siteId: z.string().transform(Number).pipe(z.int().positive()) siteId: z.string().transform(Number).pipe(z.int().positive())
@@ -63,23 +64,37 @@ export async function deleteSite(
let deletedNewtId: string | null = null; let deletedNewtId: string | null = null;
await db.transaction(async (trx) => { await db.transaction(async (trx) => {
if (site.pubKey) { if (site.type == "wireguard") {
if (site.type == "wireguard") { if (site.pubKey) {
await deletePeer(site.exitNodeId!, site.pubKey); await deletePeer(site.exitNodeId!, site.pubKey);
} else if (site.type == "newt") { }
// get the newt on the site by querying the newt table for siteId } else if (site.type == "newt") {
const [deletedNewt] = await trx // delete all of the site resources on this site
.delete(newts) const siteResourcesOnSite = trx
.where(eq(newts.siteId, siteId)) .delete(siteResources)
.returning(); .where(eq(siteResources.siteId, siteId))
if (deletedNewt) { .returning();
deletedNewtId = deletedNewt.newtId;
// delete all of the sessions for the newt // loop through them
await trx for (const removedSiteResource of await siteResourcesOnSite) {
.delete(newtSessions) await rebuildClientAssociationsFromSiteResource(
.where(eq(newtSessions.newtId, deletedNewt.newtId)); removedSiteResource,
} trx
);
}
// get the newt on the site by querying the newt table for siteId
const [deletedNewt] = await trx
.delete(newts)
.where(eq(newts.siteId, siteId))
.returning();
if (deletedNewt) {
deletedNewtId = deletedNewt.newtId;
// delete all of the sessions for the newt
await trx
.delete(newtSessions)
.where(eq(newtSessions.newtId, deletedNewt.newtId));
} }
} }