twenty/packages/twenty-server/src/engine/integrations/captcha/captcha.module.ts

import { DynamicModule, Global } from '@nestjs/common';

import { CAPTCHA_DRIVER } from 'src/engine/integrations/captcha/captcha.constants';
import { CaptchaService } from 'src/engine/integrations/captcha/captcha.service';
import { GoogleRecaptchaDriver } from 'src/engine/integrations/captcha/drivers/google-recaptcha.driver';
import { TurnstileDriver } from 'src/engine/integrations/captcha/drivers/turnstile.driver';
import {
  CaptchaDriverType,
  CaptchaModuleAsyncOptions,
} from 'src/engine/integrations/captcha/interfaces';

@Global()
export class CaptchaModule {
  static forRoot(options: CaptchaModuleAsyncOptions): DynamicModule {
    const provider = {
      provide: CAPTCHA_DRIVER,
      useFactory: async (...args: any[]) => {
        const config = await options.useFactory(...args);

        if (!config) {
          return;
        }

        switch (config.type) {
          case CaptchaDriverType.GoogleRecaptcha:
            return new GoogleRecaptchaDriver(config.options);
          case CaptchaDriverType.Turnstile:
            return new TurnstileDriver(config.options);
          default:
            return;
        }
      },
      inject: options.inject || [],
    };

    return {
      module: CaptchaModule,
      providers: [CaptchaService, provider],
      exports: [CaptchaService],
    };
  }
}