refactor: to module

Signed-off-by: Innei <[email protected]>

apps/core/src/app.module.ts

@@ -24,6 +24,7 @@ import { GatewayModule } from './processors/gateway/gateway.module'
 import { HelperModule } from './processors/helper/helper.module'
 import { RequestContextMiddleware } from './common/middlewares/request-context.middleware'
 import { UserModule } from './modules/user/user.module'
+import { authConfig } from './modules/auth/auth.config'
 
 // Request ----->
 // Response <-----
@@ -44,7 +45,7 @@ const appInterceptors: Type<any>[] = [
     GatewayModule,
 
     // BIZ
-    AuthModule,
+    AuthModule.forRoot(authConfig),
     UserModule,
   ],
   controllers: [AppController],

apps/core/src/common/guards/auth.guard.ts

@@ -1,23 +1,29 @@
 /* eslint-disable dot-notation */
 import { isTest } from '@core/global/env.global'
-import { getSessionUser } from '@core/modules/auth/auth.util'
+import { AuthService } from '@core/modules/auth/auth.service'
+
 import { getNestExecutionContextRequest } from '@core/transformers/get-req.transformer'
 import {
   CanActivate,
   ExecutionContext,
+  Inject,
   Injectable,
   UnauthorizedException,
 } from '@nestjs/common'
 
 @Injectable()
 export class AuthGuard implements CanActivate {
+  constructor(
+    @Inject(AuthService)
+    private readonly authService: AuthService,
+  ) {}
   async canActivate(context: ExecutionContext): Promise<any> {
     if (isTest) {
       return true
     }
 
     const req = this.getRequest(context)
-    const session = await getSessionUser(req)
+    const session = await this.authService.getSessionUser(req.raw)
 
     req.raw['session'] = session
     req.raw['isAuthenticated'] = !!session

apps/core/src/modules/auth/auth.config.ts

@@ -1,6 +1,6 @@
-import { DrizzleAdapter, authjs } from '@meta-muse/complied'
-import { isDev } from '@core/global/env.global'
 import { API_VERSION, AUTH } from '@core/app.config'
+import { isDev } from '@core/global/env.global'
+import { DrizzleAdapter, authjs } from '@meta-muse/complied'
 
 import { db } from '@core/processors/database/database.service'
 import {
@@ -10,7 +10,7 @@ import {
   users,
   verificationTokens,
 } from '@meta-muse/drizzle/schema'
-import { CreateAuth, type ServerAuthConfig } from './auth.implement'
+import { type ServerAuthConfig } from './auth.implement'
 
 const {
   providers: { github: GitHub },
@@ -38,4 +38,3 @@ export const authConfig: ServerAuthConfig = {
     authenticatorsTable: authenticators,
   }),
 }
-export const authHandler = CreateAuth(authConfig)

apps/core/src/modules/auth/auth.constant.ts

@@ -0,0 +1 @@
+export const AuthConfigInjectKey = Symbol()

apps/core/src/modules/auth/auth.middleware.ts

@@ -1,15 +1,20 @@
-import type { NestMiddleware } from '@nestjs/common'
+import { Inject, type NestMiddleware } from '@nestjs/common'
 import type { IncomingMessage, ServerResponse } from 'http'
-import { authHandler } from './auth.config'
+import { AuthConfigInjectKey } from './auth.constant'
+import { CreateAuth, ServerAuthConfig } from './auth.implement'
 
 export class AuthMiddleware implements NestMiddleware {
+  constructor(
+    @Inject(AuthConfigInjectKey) private readonly config: ServerAuthConfig,
+  ) {}
+  authHandler = CreateAuth(this.config)
   async use(req: IncomingMessage, res: ServerResponse, next: () => void) {
     if (req.method !== 'GET' && req.method !== 'POST') {
       next()
       return
     }
 
-    await authHandler(req, res)
+    await this.authHandler(req, res)
 
     next()
   }

apps/core/src/modules/auth/auth.module.ts

@@ -1,22 +1,48 @@
 import {
   Global,
+  Inject,
   Module,
+  type DynamicModule,
   type MiddlewareConsumer,
   type NestModule,
 } from '@nestjs/common'
 
-import { AuthService } from './auth.service'
+import { AuthConfigInjectKey } from './auth.constant'
+import type { ServerAuthConfig } from './auth.implement'
 import { AuthMiddleware } from './auth.middleware'
+import { AuthService } from './auth.service'
 
-@Module({
-  imports: [],
-  controllers: [],
-  providers: [AuthService],
-  exports: [],
-})
+@Module({})
 @Global()
 export class AuthModule implements NestModule {
+  constructor(
+    @Inject(AuthConfigInjectKey) private readonly config: ServerAuthConfig,
+  ) {}
+  static forRoot(config: ServerAuthConfig): DynamicModule {
+    return {
+      module: AuthModule,
+      global: true,
+      exports: [AuthService],
+      providers: [
+        {
+          provide: AuthService,
+          useFactory() {
+            return new AuthService(config)
+          },
+        },
+        {
+          provide: AuthConfigInjectKey,
+          useValue: config,
+        },
+      ],
+    }
+  }
+
   configure(consumer: MiddlewareConsumer) {
-    consumer.apply(AuthMiddleware).forRoutes('/auth/(.*)')
+    const config = this.config
+
+    consumer
+      .apply(AuthMiddleware)
+      .forRoutes(`${config.basePath || '/auth'}/(.*)`)
   }
 }

apps/core/src/modules/auth/auth.service.ts

@@ -1,7 +1,40 @@
-import { DatabaseService } from '@core/processors/database/database.service'
 import { Injectable } from '@nestjs/common'
+import { getSessionBase, ServerAuthConfig } from './auth.implement'
 
+import type { Session } from '@meta-muse/complied'
+import { IncomingMessage } from 'http'
+
+export interface SessionUser {
+  sessionToken: string
+  userId: string
+  expires: string
+}
 @Injectable()
 export class AuthService {
-  constructor(private readonly db: DatabaseService) {}
+  constructor(private readonly authConfig: ServerAuthConfig) {}
+
+  getSessionUser(req: IncomingMessage) {
+    const { authConfig } = this
+    return new Promise<SessionUser | null>((resolve) => {
+      getSessionBase(req, {
+        ...authConfig,
+        callbacks: {
+          ...authConfig.callbacks,
+          async session(...args) {
+            resolve(args[0].session as SessionUser)
+
+            const session =
+              (await authConfig.callbacks?.session?.(...args)) ??
+              args[0].session
+            const user = args[0].user ?? args[0].token
+            return { user, ...session } satisfies Session
+          },
+        },
+      }).then((session) => {
+        if (!session) {
+          resolve(null)
+        }
+      })
+    })
+  }
 }

apps/core/src/processors/gateway/gateway.module.ts

@@ -1,12 +1,11 @@
 import { Global, Module } from '@nestjs/common'
 
-import { AuthModule } from '../../modules/auth/auth.module'
 import { SharedGateway } from './shared/events.gateway'
 import { WebEventsGateway } from './web/events.gateway'
 
 @Global()
 @Module({
-  imports: [AuthModule],
+  imports: [],
   providers: [WebEventsGateway, SharedGateway],
   exports: [WebEventsGateway, SharedGateway],
 })