|
1 | | -import { Injectable, NestMiddleware } from "@nestjs/common"; |
| 1 | +import { Injectable, Logger, NestMiddleware } from "@nestjs/common"; |
2 | 2 | import { Request, Response, NextFunction } from "express"; |
3 | 3 | import * as express from "express"; |
| 4 | +import { HasuraService } from "src/hasura/hasura.service"; |
| 5 | +import { CacheService } from "src/cache/cache.service"; |
4 | 6 |
|
5 | 7 | @Injectable() |
6 | 8 | export class RawBodyMiddleware implements NestMiddleware { |
7 | | - use(req: Request, res: Response, next: NextFunction) { |
| 9 | + constructor( |
| 10 | + private readonly hasura: HasuraService, |
| 11 | + private readonly logger: Logger, |
| 12 | + private readonly cache: CacheService, |
| 13 | + ) {} |
| 14 | + |
| 15 | + async use(request: Request, response: Response, next: NextFunction) { |
| 16 | + try { |
| 17 | + const [matchId, apiPassword] = ( |
| 18 | + request.headers["x-origin-auth"] as string |
| 19 | + )?.split(":"); |
| 20 | + |
| 21 | + const { matches_by_pk: match } = await this.hasura.query({ |
| 22 | + matches_by_pk: { |
| 23 | + __args: { |
| 24 | + id: matchId, |
| 25 | + }, |
| 26 | + password: true, |
| 27 | + }, |
| 28 | + }); |
| 29 | + |
| 30 | + if (match?.password !== apiPassword) { |
| 31 | + this.logger.warn("invalid api password", { |
| 32 | + matchId, |
| 33 | + apiPassword, |
| 34 | + }); |
| 35 | + return response.status(401).end(); |
| 36 | + } |
| 37 | + } catch (error) { |
| 38 | + this.logger.warn("unable to fetch server", error.message); |
| 39 | + return response.status(401).end(); |
| 40 | + } |
| 41 | + |
8 | 42 | express.raw({ |
9 | 43 | type: "*/*", |
10 | 44 | limit: "50mb", |
11 | | - verify: (req: any, res, buf) => { |
12 | | - (req as any).rawBody = buf; |
13 | | - req.body = buf; |
| 45 | + verify: (_request: any, _response: any, buf: Buffer) => { |
| 46 | + (_request as any).rawBody = buf; |
| 47 | + _request.body = buf; |
14 | 48 | }, |
15 | | - })(req, res, next); |
| 49 | + })(request, response, next); |
16 | 50 | } |
17 | 51 | } |
0 commit comments