Add check for if company is aleady attached to user when user is attaching company

Author: Niloy Deb Roy

api/v1/user.go

@@ -252,6 +252,7 @@ func (u userApi) registerUser(context echo.Context) error {
 	if err := formData.ResourcePermission.Validate(); err != nil {
 		return common.GenerateErrorResponse(context, err.Error(), "Please give valid user resource permission data!")
 	}
+	formData.Metadata.CompanyId = userResourcePermission.Metadata.CompanyId
 	formData.CreatedDate = time.Now().UTC()
 	formData.UpdatedDate = time.Now().UTC()
 	formData.Status=enums.ACTIVE

core/v1/logic/user.go

@@ -36,6 +36,10 @@ func (u userService) AttachCompany(company v1.Company, companyId, token string)
 	if tokenObject.Uid==""{
 		return errors.New("no token found")
 	}
+	user := u.userRepo.GetByID(tokenObject.Uid)
+	if user.Metadata.CompanyId != "" {
+		return errors.New("[ERROR]: User already got company id attached")
+	}
 	if config.ApplicationCreationEnabled {
 		marshal, marshalErr := json.Marshal(company)
 		if marshalErr != nil {

core/v1/types.go

@@ -86,6 +86,7 @@ type RoleUpdateOption struct {
 
 func GetUserFromUserRegistrationDto(u UserRegistrationDto) User {
 	user := User{
+		Metadata: UserMetadata{CompanyId: u.Metadata.CompanyId},
 		ID:          u.ID,
 		FirstName:   u.FirstName,
 		LastName:    u.LastName,