dev add tariff and temp flag for files

2026-01-30 12:52:16 +07:00
parent fb5642db9e
commit 49dc19cd04
21 changed files with 612 additions and 24 deletions
@@ -0,0 +1,98 @@
+package ru.soune.nocopy.service.tariff;
+
+import lombok.RequiredArgsConstructor;
+import org.springframework.boot.context.event.ApplicationReadyEvent;
+import org.springframework.context.event.EventListener;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+import ru.soune.nocopy.entity.tarif.Tariff;
+import ru.soune.nocopy.entity.tarif.TariffType;
+import ru.soune.nocopy.repository.TariffRepository;
+
+import java.util.Arrays;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+@Service
+@RequiredArgsConstructor
+public class TariffInitializationService {
+
+    private final TariffRepository tariffRepository;
+
+    @EventListener(ApplicationReadyEvent.class)
+    @Transactional
+    public void initializeDefaultTariffs() {
+        List<Tariff> existingTariffs = tariffRepository.findAll();
+
+        if (existingTariffs.isEmpty()) {
+            createDefaultTariffs();
+        } else {
+            Map<TariffType, Boolean> existingTypes = new HashMap<>();
+            existingTariffs.forEach(tariff -> existingTypes.put(TariffType.valueOf(tariff.getType()), true));
+
+            Arrays.stream(TariffType.values())
+                    .filter(type -> !existingTypes.containsKey(type))
+                    .forEach(this::createDefaultTariff);
+        }
+    }
+
+    private void createDefaultTariffs() {
+        Arrays.stream(TariffType.values())
+                .forEach(this::createDefaultTariff);
+    }
+
+    private void createDefaultTariff(TariffType type) {
+        Tariff tariff = new Tariff();
+        tariff.setType(type.name());
+
+        switch (type) {
+            case DEMO:
+                tariff.setName("Демо");
+                tariff.setPrice(TariffConstants.DEMO_PRICE);
+                tariff.setTokens(TariffConstants.DEMO_TOKENS);
+                tariff.setMaxFilesCount(10);
+                tariff.setDiskSize(1024L * 1024 * 100);
+                tariff.setMaxUsers(1L);
+                break;
+
+            case START:
+                tariff.setName("Стартовый");
+                tariff.setPrice(TariffConstants.START_PRICE);
+                tariff.setTokens(TariffConstants.START_TOKENS);
+                tariff.setMaxFilesCount(50);
+                tariff.setDiskSize(1024L * 1024 * 500);
+                tariff.setMaxUsers(3L);
+                break;
+
+            case BASIC:
+                tariff.setName("Базовый");
+                tariff.setPrice(TariffConstants.BASIC_PRICE);
+                tariff.setTokens(TariffConstants.BASIC_TOKENS);
+                tariff.setMaxFilesCount(200);
+                tariff.setDiskSize(1024L * 1024 * 1024 * 2L);
+                tariff.setMaxUsers(10L);
+                break;
+
+            case PRO:
+                tariff.setName("Профессиональный");
+                tariff.setPrice(TariffConstants.PRO_PRICE);
+                tariff.setTokens(TariffConstants.PRO_TOKENS);
+                tariff.setMaxFilesCount(1000);
+                tariff.setDiskSize(1024L * 1024 * 1024 * 10L);
+                tariff.setMaxUsers(25L);
+                break;
+
+            case ENTERPRISE:
+                tariff.setName("Корпоративный");
+                tariff.setPrice(TariffConstants.ENTERPRISE_PRICE);
+                tariff.setTokens(TariffConstants.ENTERPRISE_TOKENS);
+                tariff.setMaxFilesCount(5000);
+                tariff.setDiskSize(1024L * 1024 * 1024 * 50L);
+                tariff.setMaxUsers(100L);
+                break;
+        }
+
+        tariffRepository.save(tariff);
+    }
+}