import { BaseController } from "./BaseController"; import { HttpRequest, HttpResponse, TemplatedApp } from "uWebSockets.js"; import { ADMIN_API_TOKEN } from "../Enum/EnvironmentVariable"; import { apiClientRepository } from "../Services/ApiClientRepository"; import { AdminRoomMessage, WorldFullWarningToRoomMessage, RefreshRoomPromptMessage, } from "../Messages/generated/messages_pb"; export class AdminController extends BaseController { constructor(private App: TemplatedApp) { super(); this.App = App; this.receiveGlobalMessagePrompt(); this.receiveRoomEditionPrompt(); } receiveRoomEditionPrompt() { this.App.options("/room/refresh", (res: HttpResponse, req: HttpRequest) => { this.addCorsHeaders(res); res.end(); }); // eslint-disable-next-line @typescript-eslint/no-misused-promises this.App.post("/room/refresh", async (res: HttpResponse, req: HttpRequest) => { res.onAborted(() => { console.warn("/message request was aborted"); }); const token = req.getHeader("admin-token"); const body = await res.json(); if (token !== ADMIN_API_TOKEN) { console.error("Admin access refused for token: " + token); res.writeStatus("401 Unauthorized").end("Incorrect token"); return; } try { if (typeof body.roomId !== "string") { throw "Incorrect roomId parameter"; } const roomId: string = body.roomId; await apiClientRepository.getClient(roomId).then((roomClient) => { return new Promise((res, rej) => { const roomMessage = new RefreshRoomPromptMessage(); roomMessage.setRoomid(roomId); roomClient.sendRefreshRoomPrompt(roomMessage, (err) => { err ? rej(err) : res(); }); }); }); } catch (err) { this.errorToResponse(err, res); return; } res.writeStatus("200"); res.end("ok"); }); } receiveGlobalMessagePrompt() { this.App.options("/message", (res: HttpResponse, req: HttpRequest) => { this.addCorsHeaders(res); res.end(); }); // eslint-disable-next-line @typescript-eslint/no-misused-promises this.App.post("/message", async (res: HttpResponse, req: HttpRequest) => { res.onAborted(() => { console.warn("/message request was aborted"); }); const token = req.getHeader("admin-token"); const body = await res.json(); if (token !== ADMIN_API_TOKEN) { console.error("Admin access refused for token: " + token); res.writeStatus("401 Unauthorized").end("Incorrect token"); return; } try { if (typeof body.text !== "string") { throw "Incorrect text parameter"; } if (body.type !== "capacity" && body.type !== "message") { throw "Incorrect type parameter"; } if (!body.targets || typeof body.targets !== "object") { throw "Incorrect targets parameter"; } const text: string = body.text; const type: string = body.type; const targets: string[] = body.targets; await Promise.all( targets.map((roomId) => { return apiClientRepository.getClient(roomId).then((roomClient) => { return new Promise((res, rej) => { if (type === "message") { const roomMessage = new AdminRoomMessage(); roomMessage.setMessage(text); roomMessage.setRoomid(roomId); roomClient.sendAdminMessageToRoom(roomMessage, (err) => { err ? rej(err) : res(); }); } else if (type === "capacity") { const roomMessage = new WorldFullWarningToRoomMessage(); roomMessage.setRoomid(roomId); roomClient.sendWorldFullWarningToRoom(roomMessage, (err) => { err ? rej(err) : res(); }); } }); }); }) ); } catch (err) { this.errorToResponse(err, res); return; } res.writeStatus("200"); this.addCorsHeaders(res); res.end("ok"); }); } }