From 1b7c6ee006b50ce29c9cb3ee42fad29b7db0f6f3 Mon Sep 17 00:00:00 2001 From: Joseph Ditton Date: Tue, 8 Mar 2022 18:18:38 -0700 Subject: finish gateway websocket example --- server/providers/guards/gatewayauth.guard.ts | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 server/providers/guards/gatewayauth.guard.ts (limited to 'server/providers/guards/gatewayauth.guard.ts') diff --git a/server/providers/guards/gatewayauth.guard.ts b/server/providers/guards/gatewayauth.guard.ts new file mode 100644 index 0000000..0843752 --- /dev/null +++ b/server/providers/guards/gatewayauth.guard.ts @@ -0,0 +1,27 @@ +import { Injectable, CanActivate, ExecutionContext } from '@nestjs/common'; +import { JwtService } from '../services/jwt.service'; +import { GuardUtil } from '../util/guard.util'; +import { Socket } from 'socket.io'; +import { WsException } from '@nestjs/websockets'; + +@Injectable() +export class GatewayAuthGuard implements CanActivate { + constructor(private guardUtil: GuardUtil, private jwtService: JwtService) {} + + canActivate(context: ExecutionContext) { + // Handlers and Controllers can both skip this guard in the event that + if (this.guardUtil.shouldSkip(this, context)) { + return true; + } + + const req = context.switchToHttp().getRequest() as Socket; + const jwt = req.handshake.auth.token; + if (!jwt) throw new WsException('Invalid auth token'); + try { + req.handshake.auth.jwtBody = this.jwtService.parseToken(jwt); + } catch (e) { + throw new WsException('Invalid auth token'); + } + return true; + } +} -- cgit v1.2.3-70-g09d2