dev add api for payment

src/main/java/ru/soune/nocopy/service/payment/PaymentService.java

@@ -15,6 +15,7 @@ import ru.soune.nocopy.exception.PaymentNotFoundException;
 import ru.soune.nocopy.exception.TariffNotFoundException;
 import ru.soune.nocopy.exception.UserNotFoundException;
 import ru.soune.nocopy.repository.PaymentRepository;
+import ru.soune.nocopy.repository.TariffInfoRepository;
 import ru.soune.nocopy.repository.TariffRepository;
 import ru.soune.nocopy.repository.UserRepository;
 import ru.soune.nocopy.service.tariff.TariffInfoService;
@@ -33,6 +34,8 @@ public class PaymentService {
 
     private final TariffRepository tariffRepository;
 
+    private final TariffInfoRepository tariffInfoRepository;
+
     private final TariffInfoService tariffInfoService;
 
     @Transactional
@@ -59,13 +62,10 @@ public class PaymentService {
 
     @Transactional
     public void handlePaymentNotification(Map<String, Object> notification) {
-        log.info("INFO FROM YOKASSA:" + notification.toString());
-
-
         String eventType = (String) notification.get("event");
         Map<String, Object> object = (Map<String, Object>) notification.get("object");
         String paymentUuid = (String) object.get("id");
-        log.info("METADATA:" +  (String) object.get("metadata"));
+
         Payment payment = paymentRepository.findByPaymentUuid(paymentUuid)
                 .orElseThrow(() -> new RuntimeException("Payment not found"));
 
@@ -73,7 +73,6 @@ public class PaymentService {
             payment.setStatus(PaymentStatus.SUCCEEDED);
             payment.setCancellationReason(null);
             activateTariffForUser(payment.getUser(), payment.getTariff());
-
         } else if ("payment.canceled".equals(eventType)) {
             payment.setStatus(PaymentStatus.CANCELED);
 
@@ -116,13 +115,23 @@ public class PaymentService {
     }
 
     private void activateTariffForUser(User user, Tariff tariff) {
-        TariffInfo tariffInfo = new TariffInfo();
-        tariffInfo.setStatus(TariffStatus.ACTIVE);
-        tariffInfo.setStartTariff(LocalDateTime.now());
-        tariffInfo.setEndTariff(LocalDateTime.now().plusMonths(1));
-        tariffInfo.setTokens(tariff.getTokens());
-        tariffInfo.setTariff(tariff);
+        String type = tariff.getType();
 
-        tariffInfoService.linkUserTariffInfo(user, tariffInfoService.addTariffInfo(tariffInfo));
+        if (type.equals("token")) {
+            TariffInfo activeTariffInfo = user.getActiveTariffInfo();
+            Integer tokens = activeTariffInfo.getTokens() + tariff.getTokens();
+            activeTariffInfo.setTokens(tokens);
+
+            tariffInfoRepository.save(activeTariffInfo);
+        } else {
+            TariffInfo tariffInfo = new TariffInfo();
+            tariffInfo.setStatus(TariffStatus.ACTIVE);
+            tariffInfo.setStartTariff(LocalDateTime.now());
+            tariffInfo.setEndTariff(LocalDateTime.now().plusMonths(1));
+            tariffInfo.setTokens(tariff.getTokens());
+            tariffInfo.setTariff(tariff);
+
+            tariffInfoService.linkUserTariffInfo(user, tariffInfoService.addTariffInfo(tariffInfo));
+        }
     }
 }

src/main/java/ru/soune/nocopy/service/search/SearchImageService.java

@@ -34,8 +34,8 @@ public class SearchImageService {
         this.httpClient = new OkHttpClient.Builder()
                 .connectTimeout(30, TimeUnit.SECONDS)
                 .writeTimeout(30, TimeUnit.SECONDS)
-                .readTimeout(120, TimeUnit.SECONDS)
-                .callTimeout(180, TimeUnit.SECONDS)
+                .readTimeout(45, TimeUnit.SECONDS)
+                .callTimeout(90, TimeUnit.SECONDS)
                 .retryOnConnectionFailure(true)
                 .build();
 

src/main/java/ru/soune/nocopy/service/tariff/TariffInitializationService.java

@@ -180,6 +180,46 @@ public class TariffInitializationService {
                 tariff.setMaxUsers(10L);
                 tariff.setTariffAccountType("b2b");
                 break;
+
+            case TOKEN_100:
+                tariff.setName("token_100");
+                tariff.setPrice(299);
+                tariff.setTokens(100);
+                tariff.setMaxFilesCount(0);
+                tariff.setDiskSize(0L);
+                tariff.setMaxUsers(0L);
+                tariff.setTariffAccountType("token");
+                break;
+
+            case TOKEN_500:
+                tariff.setName("token_500");
+                tariff.setPrice(1299);
+                tariff.setTokens(500);
+                tariff.setMaxFilesCount(0);
+                tariff.setDiskSize(0L);
+                tariff.setMaxUsers(0L);
+                tariff.setTariffAccountType("token");
+                break;
+
+            case TOKEN_1000:
+                tariff.setName("token_1000");
+                tariff.setPrice(2399);
+                tariff.setTokens(1000);
+                tariff.setMaxFilesCount(0);
+                tariff.setDiskSize(0L);
+                tariff.setMaxUsers(0L);
+                tariff.setTariffAccountType("token");
+                break;
+
+            case TOKEN_5000:
+                tariff.setName("token_5000");
+                tariff.setPrice(9999);
+                tariff.setTokens(5000);
+                tariff.setMaxFilesCount(0);
+                tariff.setDiskSize(0L);
+                tariff.setMaxUsers(0L);
+                tariff.setTariffAccountType("token");
+                break;
         }
 
         tariffRepository.save(tariff);

src/main/java/ru/soune/nocopy/service/tariff/TariffService.java

@@ -42,6 +42,13 @@ public class TariffService {
                 .collect(Collectors.toList());
     }
 
+    public List<TariffDTO> getTariffByType(String type) {
+        return tariffRepository.findByTariffAccountType(type)
+                .stream()
+                .map(this::convertToDTO)
+                .collect(Collectors.toList());
+    }
+
     public List<TariffDTO> getAllTariffs() {
         return tariffRepository.findAllByOrderByPriceAsc()
                 .stream()