speedlimit

Author: Alfagun74

src/modules/games/games.controller.ts

@@ -299,7 +299,11 @@ export class GamesController {
   ): Promise<StreamableFile> {
     response.setHeader(
       "X-Otp",
-      this.otpService.create(request.user.username, Number(params.game_id)),
+      this.otpService.create(
+        request.user.username,
+        Number(params.game_id),
+        Number(speedlimit),
+      ),
     );
     return this.filesService.download(
       response,

src/modules/otp/models/otp.model.ts

@@ -1,22 +1,31 @@
 export default class Otp {
   otp: string;
   username: string;
-  gameId?: number;
   createdAt: Date = new Date();
   expiresAt: Date = new Date(Date.now() + 5 * 60 * 1000); // 5 Minutes
+  gameId?: number;
+  xDownloadSpeedLimit?: number;
 
-  constructor(otp: string, username: string, gameId?: number) {
+  constructor(
+    otp: string,
+    username: string,
+    gameId?: number,
+    xDownloadSpeedLimit?: number,
+  ) {
     this.otp = otp;
+    this.username = username;
     this.gameId = gameId;
+    this.xDownloadSpeedLimit = xDownloadSpeedLimit;
   }
 
   getLoggableData() {
     return {
       otp: "**REDACTED**",
       username: this.username,
-      gameId: this.gameId,
       createdAt: this.createdAt,
       expiresAt: this.expiresAt,
+      gameId: this.gameId,
+      xDownloadSpeedLimit: this.xDownloadSpeedLimit,
     };
   }
 }

src/modules/otp/otp.controller.ts

@@ -22,13 +22,6 @@ export class OtpController {
   constructor(private readonly otpService: OtpService) {}
 
   @Get("game")
-  @ApiHeader({
-    name: "X-Download-Speed-Limit",
-    required: false,
-    description:
-      "This header lets you set the maximum download speed limit in kibibytes per second (kiB/s) for your request.  If the header is not present the download speed limit will be unlimited.",
-    example: "1024",
-  })
   @ApiOkResponse({ type: () => StreamableFile })
   @ApiOperation({
     summary: "returns a game download for the otp",
@@ -38,8 +31,7 @@ export class OtpController {
   async getOtpGame(
     @Query("otp") otp: string,
     @Res({ passthrough: true }) response: Response,
-    @Headers("X-Download-Speed-Limit") speedlimit?: string,
   ): Promise<StreamableFile> {
-    return this.otpService.get(otp, response, speedlimit);
+    return this.otpService.get(otp, response);
   }
 }

src/modules/otp/otp.service.ts

@@ -27,19 +27,19 @@ export class OtpService {
     });
   }
 
-  public create(username: string, gameId?: number): string {
+  public create(
+    username: string,
+    gameId?: number,
+    xDownloadSpeedLimit?: number,
+  ): string {
     const randomOtp = randomBytes(64).toString("hex");
-    const otp = new Otp(randomOtp, username, gameId);
+    const otp = new Otp(randomOtp, username, gameId, xDownloadSpeedLimit);
     this.otps.set(randomOtp, otp);
     this.logger.log("OTP Created.", otp.getLoggableData());
     return randomOtp;
   }
 
-  async get(
-    otp: string,
-    response: Response,
-    speedlimit?: string,
-  ): Promise<StreamableFile> {
+  async get(otp: string, response: Response): Promise<StreamableFile> {
     const existingOtp = this.otps.get(otp);
     if (!existingOtp) {
       throw new UnauthorizedException("Invalid OTP");
@@ -52,7 +52,7 @@ export class OtpService {
     return this.filesService.download(
       response,
       existingOtp.gameId,
-      Number(speedlimit),
+      existingOtp.xDownloadSpeedLimit,
     );
   }
 }