dev add

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

@@ -10,13 +10,16 @@ import ru.soune.nocopy.dto.MessageCode;
 import ru.soune.nocopy.dto.tarriff.TariffInfoDTO;
 import ru.soune.nocopy.dto.tarriff.TariffInfoRequest;
 import ru.soune.nocopy.dto.tarriff.TariffInfoResponse;
+import ru.soune.nocopy.entity.tarif.Tariff;
 import ru.soune.nocopy.entity.tarif.TariffStatus;
 import ru.soune.nocopy.entity.tarif.TariffType;
+import ru.soune.nocopy.repository.TariffRepository;
 import ru.soune.nocopy.service.tariff.TariffInfoService;
 
 import java.time.LocalDateTime;
 import java.util.List;
 import java.util.Map;
+import java.util.Optional;
 import java.util.stream.Collectors;
 
 @Component
@@ -28,6 +31,8 @@ public class TariffInfoHandler implements RequestHandler {
     
     private final TariffInfoService tariffInfoService;
 
+    private final TariffRepository tariffRepository;
+
     @Override
     public BaseResponse handle(BaseRequest request) throws Exception {
         try {
@@ -68,14 +73,15 @@ public class TariffInfoHandler implements RequestHandler {
 
     private BaseResponse handleCreateTariffInfo(BaseRequest request, TariffInfoRequest tariffInfoRequest) {
         try {
-            TariffInfoDTO tariffInfoDTO = new TariffInfoDTO(
-                    null,
+            Long tariffId = tariffInfoRequest.getTariffId();
+            Tariff tariff= tariffRepository.findById(tariffId).orElseThrow();
+
+            TariffInfoDTO tariffInfoDTO = new TariffInfoDTO(null,
                     tariffInfoRequest.getStatus() != null ? tariffInfoRequest.getStatus() : TariffStatus.ACTIVE.name(),
                     tariffInfoRequest.getStartTariff() != null ? tariffInfoRequest.getStartTariff() : LocalDateTime.now(),
                     tariffInfoRequest.getEndTariff() != null ? tariffInfoRequest.getEndTariff() : LocalDateTime.now().plusDays(30),
-                    tariffInfoRequest.getTariffId(),
-                    null
-            );
+                    tariffInfoRequest.getTariffId(), tariff.getName(), tariff.getTokens(), tariff.getDiskSize(), 0L,
+                    0, tariff.getMaxFilesCount());
 
             TariffInfoDTO createdTariffInfo = tariffInfoService.createTariffInfo(tariffInfoDTO);
 
@@ -154,6 +160,8 @@ public class TariffInfoHandler implements RequestHandler {
                         null
                 );
             }
+            Long tariffId = tariffInfoRequest.getTariffId();
+            Tariff tariff = tariffRepository.findById(tariffId).orElseThrow();
 
             TariffInfoDTO tariffInfoDTO = new TariffInfoDTO(
                     tariffInfoRequest.getId(),
@@ -161,7 +169,7 @@ public class TariffInfoHandler implements RequestHandler {
                     tariffInfoRequest.getStartTariff(),
                     tariffInfoRequest.getEndTariff(),
                     tariffInfoRequest.getTariffId(),
-                    null
+                    tariff.getName(), null, tariff.getDiskSize(), null, null, tariff.getMaxFilesCount()
             );
 
             TariffInfoDTO updatedTariffInfo = tariffInfoService.updateTariffInfo(