fix value

src/main/java/ru/soune/nocopy/dto/MessageCode.java

@@ -4,6 +4,7 @@ public enum MessageCode {
     SUCCESS(0, "Operation successful"),
     REG_EMAIL_EXISTS(1, "Email already registered"),
     REG_EMAIL_OR_PHONE_EXISTS(1, "Email or phone already registered"),
+    REFERRAL_LINK_IS_NOT_EXIST(1, "Refferal link is not exist"),
     INVALID_FIELD(2, "Invalid field"),
     INVALID_TOKEN(2, "Invalid token"),
     TOKEN_IS_ALIVE(2, "Token is alive"),

src/main/java/ru/soune/nocopy/handler/RegRequestHandler.java

@@ -10,10 +10,13 @@ import ru.soune.ReferralService;
 import ru.soune.nocopy.dto.*;
 import ru.soune.nocopy.dto.register.RegAnswer;
 import ru.soune.nocopy.dto.register.RegRequest;
+import ru.soune.nocopy.entity.referral.Referral;
 import ru.soune.nocopy.entity.user.AuthToken;
 import ru.soune.nocopy.exception.NotValidFieldException;
 import ru.soune.nocopy.exception.ValidationException;
 import ru.soune.nocopy.handler.validator.RegRequestValidator;
+import ru.soune.nocopy.repository.ReferralJpaRepository;
+import ru.soune.nocopy.repository.ReferralLevelRepository;
 import ru.soune.nocopy.repository.UserRepository;
 import ru.soune.nocopy.service.register.AuthService;
 import ru.soune.nocopy.service.mail.EmailService;
@@ -37,6 +40,8 @@ public class RegRequestHandler implements RequestHandler {
 
     private final ReferralService referralService;
 
+    private final ReferralJpaRepository referralJpaRepository;
+
     @Override
     public BaseResponse handle(BaseRequest request) throws ValidationException {
         RegRequest regRequest = objectMapper.convertValue(request.getMessageBody(), RegRequest.class);
@@ -52,6 +57,19 @@ public class RegRequestHandler implements RequestHandler {
                     MessageCode.REG_EMAIL_OR_PHONE_EXISTS.getDescription(), regAnswer));
         }
 
+        String referralLink = regRequest.getReferralLink();
+
+        Referral referral = referralJpaRepository.findByReferralLink(referralLink);
+
+        if (referral == null) {
+            throw new NotValidFieldException("Referral link is not found:" + referralLink +
+                    referralLink, new BaseResponse(request.getMsgId(),
+                    MessageCode.REFERRAL_LINK_IS_NOT_EXIST.getCode(),
+                    MessageCode.REFERRAL_LINK_IS_NOT_EXIST.getDescription(), Map.of(
+                            "referralLink", referralLink
+            )));
+        }
+
         BindingResult bindingResult = new BeanPropertyBindingResult(regRequest, "regRequest");
 
         regRequestValidator.validate(regRequest, bindingResult);
@@ -79,7 +97,6 @@ public class RegRequestHandler implements RequestHandler {
 
         RegAnswer regAnswer = new RegAnswer();
 
-
         regAnswer.setUserId(authToken.getUser().getId());
         regAnswer.setToken(token);
         regAnswer.setVerified(true);