Add an environment variable to disable new user creation (close #35)

heyform · Apr 10, 2024 · c4d508e · c4d508e
1 parent 23e9269
commit c4d508e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/packages/server/src/environments/index.ts b/packages/server/src/environments/index.ts
@@ -17,6 +17,7 @@ export const APP_LISTEN_PORT: number = +process.env.APP_LISTEN_PORT || 8000
 export const APP_LISTEN_HOSTNAME: string = process.env.APP_LISTEN_HOSTNAME || '0.0.0.0'
 export const APP_HOMEPAGE_URL: string =
   process.env.APP_HOMEPAGE_URL || `http://${APP_LISTEN_HOSTNAME}:${APP_LISTEN_PORT}`
+export const APP_DISABLE_REGISTRATION: boolean = helper.isTrue(process.env.APP_DISABLE_REGISTRATION)
 
 // Cookie
 export const COOKIE_MAX_AGE: string = process.env.COOKIE_MAX_AGE || '1y'

diff --git a/packages/server/src/resolver/auth/sign-up.resolver.ts b/packages/server/src/resolver/auth/sign-up.resolver.ts
@@ -4,7 +4,7 @@ import { Args, Query, Resolver } from '@nestjs/graphql'
 import { helper } from '@heyform-inc/utils'
 
 import { GraphqlRequest, GraphqlResponse } from '@decorator'
-import { BCRYPT_SALT, VERIFY_USER_EMAIL } from '@environments'
+import { APP_DISABLE_REGISTRATION, BCRYPT_SALT, VERIFY_USER_EMAIL } from '@environments'
 import { SignUpInput } from '@graphql'
 import { BrowserIdGuard } from '@guard'
 import { AuthService, UserService } from '@service'
@@ -27,6 +27,10 @@ export class SignUpResolver {
     @GraphqlResponse() res: any,
     @Args('input') input: SignUpInput
   ): Promise<boolean> {
+    if (APP_DISABLE_REGISTRATION) {
+      throw new BadRequestException('Error: Registration is disabled')
+    }
+
     if (isDisposableEmail(input.email)) {
       throw new BadRequestException(
         'Error: Disposable email address detected, please use a work email to create the account'

diff --git a/packages/server/src/service/social-login.service.ts b/packages/server/src/service/social-login.service.ts
@@ -12,7 +12,8 @@ import {
   APPLE_LOGIN_WEB_CLIENT_ID,
   APP_HOMEPAGE_URL,
   GOOGLE_LOGIN_CLIENT_ID,
-  GOOGLE_LOGIN_CLIENT_SECRET
+  GOOGLE_LOGIN_CLIENT_SECRET,
+  APP_DISABLE_REGISTRATION
 } from '@environments'
 import { UserSocialAccountModel } from '@model'
 import { UserInfo, appleLoginUrl, appleUserInfo, googleLoginUrl, googleUserInfo } from '@utils'
@@ -139,6 +140,10 @@ export class SocialLoginService {
 
       // Create new user
       if (!userId) {
+        if (APP_DISABLE_REGISTRATION) {
+          throw new BadRequestException('Error: Registration is disabled')
+        }
+
         if (userInfo!.user.email) {
           // @ts-ignore
           userInfo!.user.isEmailVerified = true