Fix to allow first user(admin) to sign up

Author: aq-ikhwa-tech

server/api-service/lowcoder-domain/src/main/java/org/lowcoder/domain/authentication/AuthenticationServiceImpl.java

@@ -1,6 +1,7 @@
 package org.lowcoder.domain.authentication;
 
 import lombok.extern.slf4j.Slf4j;
+import org.lowcoder.domain.organization.service.OrgMemberService;
 import org.lowcoder.domain.organization.service.OrganizationService;
 import org.lowcoder.sdk.auth.AbstractAuthConfig;
 import org.lowcoder.sdk.auth.EmailAuthConfig;
@@ -26,6 +27,10 @@ public class AuthenticationServiceImpl implements AuthenticationService {
 
     @Autowired
     private OrganizationService organizationService;
+
+    @Autowired
+    private OrgMemberService orgMemberService;
+
     @Autowired
     private CommonConfig commonConfig;
     @Autowired
@@ -51,18 +56,27 @@ private Mono<FindAuthConfig> findAuthConfig(String orgId, Function<AbstractAuthC
 
     @Override
     public Flux<FindAuthConfig> findAllAuthConfigs(String orgId, boolean enableOnly) {
-        return findAllAuthConfigsByDomain()
+
+        Mono<FindAuthConfig> emailAuthConfigMono = orgMemberService.doesAtleastOneAdminExist()
+                .map(doesAtleastOneAdminExist -> {
+                    boolean shouldEnableRegister = !doesAtleastOneAdminExist && authProperties.getEmail().isEnableRegister();
+                    return new FindAuthConfig
+                            (new EmailAuthConfig(AuthSourceConstants.EMAIL, authProperties.getEmail().isEnable(), shouldEnableRegister), null);
+                });
+
+
+        Flux<FindAuthConfig> findAuthConfigFlux = findAllAuthConfigsByDomain()
                 .switchIfEmpty(findAllAuthConfigsForEnterpriseMode())
                 .switchIfEmpty(findAllAuthConfigsForSaasMode(orgId))
                 .filter(findAuthConfig -> {
                     if (enableOnly) {
                         return findAuthConfig.authConfig().isEnable();
                     }
                     return true;
-                })
-                .concatWithValues(new FindAuthConfig
-                        (new EmailAuthConfig(AuthSourceConstants.EMAIL, authProperties.getEmail().isEnable(),
-                                authProperties.getEmail().isEnableRegister()), null));
+                });
+
+        return Flux.concat(findAuthConfigFlux, emailAuthConfigMono);
+
     }
 
     private Flux<FindAuthConfig> findAllAuthConfigsByDomain() {

server/api-service/lowcoder-domain/src/main/java/org/lowcoder/domain/organization/service/OrgMemberService.java

@@ -24,6 +24,8 @@ public interface OrgMemberService {
 
     Mono<Long> getOrgMemberCount(String orgId);
 
+    Mono<Boolean> doesAtleastOneAdminExist();
+
     Mono<Long> countAllActiveOrgs(String userId);
 
     Mono<OrgMember> getOrgMember(String orgId, String userId);

server/api-service/lowcoder-domain/src/main/java/org/lowcoder/domain/organization/service/OrgMemberServiceImpl.java

@@ -125,6 +125,13 @@ public Mono<Long> getOrgMemberCount(String orgId) {
         return biRelationService.countBySourceId(ORG_MEMBER, orgId);
     }
 
+    @Override
+    public Mono<Boolean> doesAtleastOneAdminExist() {
+        return biRelationService.countByRelation(ORG_MEMBER, MemberRole.ADMIN.getValue())
+                .single()
+                .map(count -> count != 0);
+    }
+
     @Override
     public Mono<Boolean> addMember(String orgId, String userId, MemberRole memberRole) {
         return biRelationService.addBiRelation(ORG_MEMBER, orgId,

server/api-service/lowcoder-infra/src/main/java/org/lowcoder/infra/birelation/BiRelationRepository.java

@@ -26,6 +26,8 @@ public interface BiRelationRepository extends ReactiveMongoRepository<BiRelation
 
     Flux<BiRelation> findByBizTypeAndSourceIdAndRelation(BiRelationBizType bizType, String sourceId, String relation);
 
+    Mono<Long> countByBizTypeAndRelation(BiRelationBizType bizType, String relation);
+
     Mono<Long> countByBizTypeAndSourceId(BiRelationBizType bizType, String sourceId);
 
     Mono<Long> countByBizTypeAndTargetId(BiRelationBizType bizType, String targetId);

server/api-service/lowcoder-infra/src/main/java/org/lowcoder/infra/birelation/BiRelationService.java

@@ -166,6 +166,10 @@ public Flux<BiRelation> getBySourceIdAndRelation(BiRelationBizType bizType, Stri
         return biRelationRepository.findByBizTypeAndSourceIdAndRelation(bizType, sourceId, relation);
     }
 
+    public Mono<Long> countByRelation(BiRelationBizType bizType, String relation) {
+        return biRelationRepository.countByBizTypeAndRelation(bizType, relation);
+    }
+
     public Mono<Long> countBySourceId(BiRelationBizType bizType, String sourceId) {
         return biRelationRepository.countByBizTypeAndSourceId(bizType, sourceId);
     }