Refactor SAML user registration to include company validation and remove unused imports

Author: Artuomka

backend/src/microservices/saas-microservice/use-cases/register-user-with-saml-use.case.ts

@@ -1,17 +1,12 @@
-import { Inject, Injectable } from '@nestjs/common';
+import { Inject, Injectable, NotFoundException } from '@nestjs/common';
 import AbstractUseCase from '../../../common/abstract-use.case.js';
 import { IGlobalDatabaseContext } from '../../../common/application/global-database-context.interface.js';
 import { BaseType } from '../../../common/data-injection.tokens.js';
-import { ConnectionEntity } from '../../../entities/connection/connection.entity.js';
-import { GroupEntity } from '../../../entities/group/group.entity.js';
-import { PermissionEntity } from '../../../entities/permission/permission.entity.js';
 import { RegisterUserDs } from '../../../entities/user/application/data-structures/register-user-ds.js';
 import { ExternalRegistrationProviderEnum } from '../../../entities/user/enums/external-registration-provider.enum.js';
+import { UserRoleEnum } from '../../../entities/user/enums/user-role.enum.js';
 import { UserEntity } from '../../../entities/user/user.entity.js';
-import { buildConnectionEntitiesFromTestDtos } from '../../../entities/user/utils/build-connection-entities-from-test-dtos.js';
-import { buildDefaultAdminGroups } from '../../../entities/user/utils/build-default-admin-groups.js';
-import { buildDefaultAdminPermissions } from '../../../entities/user/utils/build-default-admin-permissions.js';
-import { Constants } from '../../../helpers/constants/constants.js';
+import { Messages } from '../../../exceptions/text/messages.js';
 import { SaasSAMLUserRegisterDS } from '../data-structures/saas-saml-user-register.ds.js';
 
 @Injectable()
@@ -24,7 +19,7 @@ export class SaaSRegisterUserWIthSamlUseCase extends AbstractUseCase<SaasSAMLUse
   }
 
   public async implementation(inputData: SaasSAMLUserRegisterDS): Promise<UserEntity> {
-    const { email, name, samlNameId } = inputData;
+    const { email, name, samlNameId, companyId } = inputData;
     const foundUser = await this._dbContext.userRepository.findOneUserByEmail(
       email,
       ExternalRegistrationProviderEnum.SAML,
@@ -47,27 +42,15 @@ export class SaaSRegisterUserWIthSamlUseCase extends AbstractUseCase<SaasSAMLUse
       ExternalRegistrationProviderEnum.SAML,
     );
 
-    const testConnections = Constants.getTestConnectionsArr();
-    const testConnectionsEntities = buildConnectionEntitiesFromTestDtos(testConnections);
-    const createdTestConnections = await Promise.all(
-      testConnectionsEntities.map(async (connection): Promise<ConnectionEntity> => {
-        connection.author = savedUser;
-        return await this._dbContext.connectionRepository.saveNewConnection(connection);
-      }),
-    );
-    const testGroupsEntities = buildDefaultAdminGroups(savedUser, createdTestConnections);
-    const createdTestGroups = await Promise.all(
-      testGroupsEntities.map(async (group: GroupEntity) => {
-        return await this._dbContext.groupRepository.saveNewOrUpdatedGroup(group);
-      }),
-    );
-    const testPermissionsEntities = buildDefaultAdminPermissions(createdTestGroups);
-    await Promise.all(
-      testPermissionsEntities.map(async (permission: PermissionEntity) => {
-        await this._dbContext.permissionRepository.saveNewOrUpdatedPermission(permission);
-      }),
-    );
+    const foundCompanyInfo = await this._dbContext.companyInfoRepository.findOne({ where: { id: companyId } });
+    if (!foundCompanyInfo) {
+      throw new NotFoundException(Messages.COMPANY_NOT_FOUND);
+    }
+
+    savedUser.company = foundCompanyInfo;
+    savedUser.samlNameId = samlNameId;
+    savedUser.role = UserRoleEnum.USER;
 
-    return savedUser;
+    return await this._dbContext.userRepository.saveUserEntity(savedUser);
   }
 }