dev add rest for company, add link for users , update reg users

src/main/java/ru/soune/nocopy/service/register/AuthService.java

@@ -5,14 +5,17 @@ import org.springframework.security.crypto.password.PasswordEncoder;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 import ru.soune.nocopy.dto.*;
+import ru.soune.nocopy.dto.register.AccountType;
 import ru.soune.nocopy.dto.register.LoginAnswer;
 import ru.soune.nocopy.dto.register.LoginRequest;
 import ru.soune.nocopy.dto.register.RegRequest;
+import ru.soune.nocopy.entity.company.Company;
 import ru.soune.nocopy.entity.user.AuthToken;
 import ru.soune.nocopy.entity.user.User;
 import ru.soune.nocopy.exception.NotFoundAuthToken;
 import ru.soune.nocopy.exception.NotValidFieldException;
 import ru.soune.nocopy.repository.AuthTokenRepository;
+import ru.soune.nocopy.repository.CompanyRepository;
 import ru.soune.nocopy.repository.UserRepository;
 import ru.soune.nocopy.service.file.CheckCounterService;
 
@@ -30,6 +33,8 @@ public class AuthService {
 
     private final PasswordEncoder passwordEncoder;
 
+    private final CompanyRepository companyRepository;
+
     private final CheckCounterService checkCounterService;
 
     private final SecureRandom secureRandom = new SecureRandom();
@@ -43,8 +48,14 @@ public class AuthService {
         user.setActive(true);
         user.setEmailVerified(true);
 
-        if (registerRequest.getCompanyName() != null) {
-            user.setCompany(registerRequest.getCompanyName());
+        if (registerRequest.getCompanyName() != null &&
+                AccountType.valueOf(registerRequest.getAccountType().toUpperCase()).equals(AccountType.B2B)) {
+            Company company = new Company();
+            company.setCompanyName(registerRequest.getCompanyName());
+
+            companyRepository.save(company);
+
+            user.setCompany(company);
         }
 
         if (registerRequest.getPhone() != null) {