feat: add extract from cookie to refresh

LhonRafaat · Oct 14, 2024 · e355902 · e355902
1 parent bb38b59
commit e355902
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/modules/auth/strategies/refresh.strategy.ts b/src/modules/auth/strategies/refresh.strategy.ts
@@ -4,6 +4,7 @@ import { Injectable, UnauthorizedException } from '@nestjs/common';
 import { ConfigService } from '@nestjs/config';
 import { UsersService } from '../../users/users.service';
 import { EnvConfig } from '../../../config.type';
+import { IRequest } from '../../../common/helper/common-types';
 
 /**
  * Jwt Strategy Class
@@ -15,12 +16,19 @@ export class RefreshStrategy extends PassportStrategy(Strategy, 'refresh') {
     private readonly userService: UsersService,
   ) {
     super({
-      jwtFromRequest: ExtractJwt.fromAuthHeaderAsBearerToken(),
+      jwtFromRequest: ExtractJwt.fromExtractors([
+        ExtractJwt.fromAuthHeaderAsBearerToken(),
+        RefreshStrategy.extractFromCookie,
+      ]),
       ignoreExpiration: false,
       secretOrKey: configService.get('REFRESH_SECRET'),
     });
   }
 
+  private static extractFromCookie(req: IRequest): string | null {
+    return req?.cookies?.access_token || null;
+  }
+
   async validate({ iat, exp, _id }, done): Promise<boolean> {
     const timeDiff = exp - iat;
     if (timeDiff <= 0) {