add deleteClient endpoint

This commit is contained in:
miloschwartz 2025-02-21 15:26:48 -05:00
parent 255e29d9c8
commit 11920ca997
No known key found for this signature in database
6 changed files with 161 additions and 45 deletions

View file

@ -1,23 +1,21 @@
import { Request, Response, NextFunction } from "express";
import { z } from "zod";
import { db } from "@server/db";
import { newts, newtSessions, sites } from "@server/db/schema";
import { clients, sites } from "@server/db/schema";
import { eq } from "drizzle-orm";
import response from "@server/lib/response";
import HttpCode from "@server/types/HttpCode";
import createHttpError from "http-errors";
import logger from "@server/logger";
import { deletePeer } from "../gerbil/peers";
import { fromError } from "zod-validation-error";
import { sendToClient } from "../ws";
const deleteClientSchema = z
.object({
siteId: z.string().transform(Number).pipe(z.number().int().positive())
clientId: z.string().transform(Number).pipe(z.number().int().positive())
})
.strict();
export async function deleteSite(
export async function deleteClient(
req: Request,
res: Response,
next: NextFunction
@ -33,56 +31,30 @@ export async function deleteSite(
);
}
const { siteId } = parsedParams.data;
const { clientId } = parsedParams.data;
const [site] = await db
const [client] = await db
.select()
.from(sites)
.where(eq(sites.siteId, siteId))
.from(clients)
.where(eq(clients.clientId, clientId))
.limit(1);
if (!site) {
if (!client) {
return next(
createHttpError(
HttpCode.NOT_FOUND,
`Site with ID ${siteId} not found`
`Site with ID ${clientId} not found`
)
);
}
await db.transaction(async (trx) => {
if (site.pubKey) {
if (site.type == "wireguard") {
await deletePeer(site.exitNodeId!, site.pubKey);
} else if (site.type == "newt") {
// 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) {
const payload = {
type: `newt/terminate`,
data: {}
};
sendToClient(deletedNewt.newtId, payload);
// delete all of the sessions for the newt
await trx
.delete(newtSessions)
.where(eq(newtSessions.newtId, deletedNewt.newtId));
}
}
}
await trx.delete(sites).where(eq(sites.siteId, siteId));
});
await db.delete(sites).where(eq(sites.siteId, clientId));
return response(res, {
data: null,
success: true,
error: false,
message: "Site deleted successfully",
message: "Client deleted successfully",
status: HttpCode.OK
});
} catch (error) {