diff --git a/src/auth/auth.guard.ts b/src/auth/auth.guard.ts index 20201f9..defebc3 100644 --- a/src/auth/auth.guard.ts +++ b/src/auth/auth.guard.ts @@ -4,13 +4,18 @@ import { CredentialsService } from "src/credentials/credentials.service"; import { DrizzleService } from "src/drizzle/drizzle.service"; import { UsersTable } from "src/schema"; import { eq } from "drizzle-orm"; +import { Reflector } from "@nestjs/core"; @Injectable() export class UserGuard implements CanActivate { - constructor( - private readonly credentialService: CredentialsService, - private readonly databaseService: DrizzleService - ) {} + private readonly credentialService: CredentialsService; + private readonly databaseService: DrizzleService; + + constructor() { + const reflector = new Reflector(); + this.credentialService = reflector.get('CredentialsService', UserGuard); + this.databaseService = reflector.get('DrizzleService', UserGuard); + } async canActivate( context: ExecutionContext, @@ -44,10 +49,14 @@ export class UserGuard implements CanActivate { @Injectable() export class AdminGuard implements CanActivate { - constructor( - private readonly credentialService: CredentialsService, - private readonly databaseService: DrizzleService - ) {} + private readonly credentialService: CredentialsService; + private readonly databaseService: DrizzleService; + + constructor() { + const reflector = new Reflector(); + this.credentialService = reflector.get('CredentialsService', UserGuard); + this.databaseService = reflector.get('DrizzleService', UserGuard); + } async canActivate( context: ExecutionContext,