dev add referral

src/main/java/ru/soune/nocopy/configuration/ReferralConfig.java

@@ -0,0 +1,16 @@
+package ru.soune.nocopy.configuration;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import ru.soune.ReferralLevelProvider;
+import ru.soune.ReferralRepo;
+import ru.soune.ReferralService;
+
+@Configuration
+public class ReferralConfig {
+
+    @Bean
+    public ReferralService referralService(ReferralLevelProvider referralLevelProvider, ReferralRepo referralRepo) {
+        return new ReferralService(referralLevelProvider, referralRepo);
+    }
+}

src/main/java/ru/soune/nocopy/dto/referral/ReferralRequest.java

@@ -0,0 +1,19 @@
+package ru.soune.nocopy.dto.referral;
+
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+import ru.soune.nocopy.entity.user.AuthToken;
+
+@Data
+@AllArgsConstructor
+@NoArgsConstructor
+@Builder
+public class ReferralRequest {
+    private String token;
+    private Integer amount;
+    private Integer pageSize;
+    private Integer pageNumber;
+    private String action;
+}

src/main/java/ru/soune/nocopy/dto/register/RegRequest.java

@@ -29,4 +29,6 @@ public class RegRequest {
     private String password;
 
     private String authToken;
+
+    private String referralLink;
 }

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

@@ -0,0 +1,91 @@
+package ru.soune.nocopy.handler;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import lombok.RequiredArgsConstructor;
+import org.springframework.stereotype.Component;
+import ru.soune.ReferralInvitee;
+import ru.soune.ReferralLevel;
+import ru.soune.ReferralService;
+import ru.soune.ReferralStat;
+import ru.soune.nocopy.dto.BaseRequest;
+import ru.soune.nocopy.dto.BaseResponse;
+import ru.soune.nocopy.dto.MessageCode;
+import ru.soune.nocopy.dto.referral.ReferralRequest;
+import ru.soune.nocopy.entity.user.AuthToken;
+import ru.soune.nocopy.repository.AuthTokenRepository;
+
+import java.util.List;
+import java.util.Map;
+
+
+@Component
+@RequiredArgsConstructor
+public class ReferralHandler implements RequestHandler {
+
+    private final ReferralService referralService;
+
+    private final AuthTokenRepository authTokenRepository;
+
+    private final ObjectMapper mapper = new ObjectMapper();
+
+    @Override
+    public BaseResponse handle(BaseRequest request) throws Exception {
+        ReferralRequest referralRequest = mapper.convertValue(request.getMessageBody(), ReferralRequest.class);
+        String action = referralRequest.getAction();
+        String token = referralRequest.getToken();
+        AuthToken authToken = authTokenRepository.findByToken(token).orElseThrow();
+
+        switch (action) {
+            case "refill":
+                return handleRefill(request, authToken.getUser().getId(), referralRequest.getAmount());
+            case "levels":
+                return handleReferralLevels(request);
+            case "level":
+                return handleUserLevel(request, authToken.getUser().getId());
+            case "userStats" :
+                return handleUserStats(request, authToken.getUser().getId());
+            case "invitees":
+                return handleInvitees(referralRequest, request, authToken.getUser().getId());
+        }
+
+        return null;
+    }
+
+
+    private BaseResponse handleRefill(BaseRequest request, Long userId, Integer amount) {
+        referralService.onUserAccountRefill(userId, amount);
+
+        return new BaseResponse(request.getMsgId(), MessageCode.SUCCESS.getCode(),
+                MessageCode.SUCCESS.getDescription(), Map.of("account", "refill"));
+    }
+
+    private BaseResponse handleReferralLevels(BaseRequest request) {
+        List<ReferralLevel> referralLevels = referralService.getReferralLevels();
+
+        return new BaseResponse(request.getMsgId(), MessageCode.SUCCESS.getCode(),
+                MessageCode.SUCCESS.getDescription(), referralLevels);
+
+    }
+
+    private BaseResponse handleUserLevel(BaseRequest request, Long userId) {
+        ReferralLevel referralLevelForUser = referralService.getReferralLevelForUser(userId);
+
+        return new BaseResponse(request.getMsgId(), MessageCode.SUCCESS.getCode(),
+                MessageCode.SUCCESS.getDescription(), referralLevelForUser);
+    }
+
+    private BaseResponse handleUserStats(BaseRequest request, Long userId) {
+        ReferralStat referralStatForUser = referralService.getReferralStatForUser(userId);
+
+        return new BaseResponse(request.getMsgId(), MessageCode.SUCCESS.getCode(),
+                MessageCode.SUCCESS.getDescription(), referralStatForUser);
+    }
+
+    private BaseResponse handleInvitees(ReferralRequest referralRequest, BaseRequest request, Long userId) {
+        List<ReferralInvitee> inviteeForUser = referralService.getInviteeForUser(userId, referralRequest.getPageSize(),
+                referralRequest.getPageNumber());
+
+        return new BaseResponse(request.getMsgId(), MessageCode.SUCCESS.getCode(),
+                MessageCode.SUCCESS.getDescription(), inviteeForUser);
+    }
+}

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

@@ -6,6 +6,7 @@ import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Component;
 import org.springframework.validation.BeanPropertyBindingResult;
 import org.springframework.validation.BindingResult;
+import ru.soune.ReferralService;
 import ru.soune.nocopy.dto.*;
 import ru.soune.nocopy.dto.register.RegAnswer;
 import ru.soune.nocopy.dto.register.RegRequest;
@@ -34,6 +35,9 @@ public class RegRequestHandler implements RequestHandler {
 
     private final EmailService emailService;
 
+    //TODO add when add all referall
+//    private final ReferralService referralService;
+
     @Override
     public BaseResponse handle(BaseRequest request) throws ValidationException {
         RegRequest regRequest = objectMapper.convertValue(request.getMessageBody(), RegRequest.class);
@@ -81,6 +85,8 @@ public class RegRequestHandler implements RequestHandler {
         regAnswer.setToken(token);
         regAnswer.setVerified(true);
         regAnswer.setActive(true);
+        //TODO add when add all referall
+//        referralService.onRegister(authToken.getUser().getId(), regRequest.getReferralLink());
 
         return new BaseResponse(request.getMsgId(), MessageCode.SUCCESS.getCode(),
                 MessageCode.SUCCESS.getDescription(), regAnswer);

src/main/java/ru/soune/nocopy/service/referral/ReferralLevelProviderImpl.java

@@ -0,0 +1,31 @@
+package ru.soune.nocopy.service.referral;
+
+import org.springframework.stereotype.Component;
+import ru.soune.ReferralLevel;
+import ru.soune.ReferralLevelProvider;
+
+import java.util.ArrayList;
+import java.util.List;
+
+@Component
+public class ReferralLevelProviderImpl implements ReferralLevelProvider {
+
+    @Override
+    public List<ReferralLevel> getAvailableReferralLevels() {
+        // TODO
+
+        return new ArrayList<ReferralLevel>();
+    }
+
+    @Override
+    public String getInitialReferralLevelId() {
+        return "bronze";
+    }
+
+    @Override
+    public ReferralLevel getReferralLevelById(String levelId) {
+        // TODO
+
+        return null;
+    }
+}

src/main/java/ru/soune/nocopy/service/referral/ReferralRepoImpl.java

@@ -0,0 +1,75 @@
+package ru.soune.nocopy.service.referral;
+
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+import org.springframework.stereotype.Repository;
+import ru.soune.Referral;
+import ru.soune.ReferralInvitee;
+import ru.soune.ReferralRepo;
+
+import java.util.List;
+
+@Repository
+public class ReferralRepoImpl implements ReferralRepo {
+
+    @Override
+    public String getUserIdByLink(String link) {
+        // TODO
+        return "";
+    }
+
+    @Override
+    public @NotNull String getUserReferralLink(long userId) {
+        return "";
+    }
+
+    @Override
+    public @Nullable Long getInviterIdForUser(long userId) {
+        return 0L;
+    }
+
+    @Override
+    public @NotNull String getReferralLevelForUser(long userId) {
+        return "";
+    }
+
+    @Override
+    public @NotNull Referral getReferralByUserId(long userId) {
+        return null;
+    }
+
+    @Override
+    public @NotNull List<ReferralInvitee> getReferralInvitees(long userId, int pageSize, int pageNumber) {
+        return List.of();
+    }
+
+    @Override
+    public void createReferralEntity(@NotNull Referral entity) {
+
+    }
+
+    @Override
+    public void increaseIncome(long userId, int transferAmount) {
+
+    }
+
+    @Override
+    public boolean activateUser(long userId) {
+        return false;
+    }
+
+    @Override
+    public int getActiveInviteeForUser(long userId) {
+        return 0;
+    }
+
+    @Override
+    public int getTotalInviteeForUser(long userId) {
+        return 0;
+    }
+
+    @Override
+    public void upgradeReferralLevelForUser(long userId, @NotNull String newLevelId) {
+
+    }
+}