Ryan/drop
1
0
Fork 0
mirror of https://github.com/Drop-OSS/drop.git synced 2025-11-13 00:02:37 +10:00
Code Issues Packages Projects Releases Wiki Activity

fix(ws): fixed websocket authentication

Browse Source
This commit is contained in:
DecDuck
2024-11-16 18:28:38 +11:00
parent f7d767d73e
commit 62ea9a116b
4 changed files with 42 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
server/api/v1/task/index.get.ts
View File

@ -2,26 +2,30 @@ import { H3Event } from "h3";
import session from "~/server/internal/session";
import { v4 as uuidv4 } from "uuid";
import taskHandler, { TaskMessage } from "~/server/internal/tasks";
import { parse as parseCookies } from "cookie-es";
// TODO add web socket sessions for horizontal scaling
// ID to admin
const adminSocketSessions: { [key: string]: boolean } = {};
export default defineWebSocketHandler({
open(peer) {
const dummyEvent = {
node: {
req: {
headers: peer.request?.headers,
},
},
} as unknown as H3Event;
const userId = session.getUserId(dummyEvent);
async open(peer) {
const cookies = peer.request?.headers?.get("Cookie");
if (!cookies) {
peer.send("unauthenticated");
return;
}
const parsedCookies = parseCookies(cookies);
const token = parsedCookies[session.getDropTokenCookie()];
const userId = await session.getUserIdRaw(token);
if (!userId) {
peer.send("unauthenticated");
return;
}
const admin = session.getAdminUser(dummyEvent);
const admin = session.getAdminUser(token);
adminSocketSessions[peer.id] = admin !== undefined;
const rtMsg: TaskMessage = {