package com.xuqm.tenant.service;

import com.xuqm.common.exception.BusinessException;
import com.xuqm.tenant.entity.FeatureServiceEntity;
import com.xuqm.tenant.repository.FeatureServiceRepository;
import org.springframework.stereotype.Service;

import java.security.SecureRandom;
import java.time.LocalDateTime;
import java.util.Base64;
import java.util.List;
import java.util.UUID;

@Service
public class FeatureServiceManager {

    private final FeatureServiceRepository repository;
    private static final SecureRandom random = new SecureRandom();

    public FeatureServiceManager(FeatureServiceRepository repository) {
        this.repository = repository;
    }

    public List<FeatureServiceEntity> listByApp(String appId) {
        return repository.findByAppId(appId);
    }

    public FeatureServiceEntity toggle(String appId, FeatureServiceEntity.Platform platform,
                                       FeatureServiceEntity.ServiceType serviceType, boolean enable) {
        FeatureServiceEntity entity = repository
                .findByAppIdAndPlatformAndServiceType(appId, platform, serviceType)
                .orElseGet(() -> {
                    FeatureServiceEntity e = new FeatureServiceEntity();
                    e.setId(UUID.randomUUID().toString());
                    e.setAppId(appId);
                    e.setPlatform(platform);
                    e.setServiceType(serviceType);
                    e.setSecretKey(generateSecretKey());
                    e.setCreatedAt(LocalDateTime.now());
                    return e;
                });
        entity.setEnabled(enable);
        return repository.save(entity);
    }

    public FeatureServiceEntity getOrFail(String appId, FeatureServiceEntity.Platform platform,
                                          FeatureServiceEntity.ServiceType serviceType) {
        return repository.findByAppIdAndPlatformAndServiceType(appId, platform, serviceType)
                .orElseThrow(() -> new BusinessException(404, "服务未配置"));
    }

    public FeatureServiceEntity regenerateKey(String id) {
        FeatureServiceEntity entity = repository.findById(id)
                .orElseThrow(() -> new BusinessException(404, "服务不存在"));
        entity.setSecretKey(generateSecretKey());
        return repository.save(entity);
    }

    private String generateSecretKey() {
        byte[] bytes = new byte[32];
        random.nextBytes(bytes);
        return "sk_" + Base64.getUrlEncoder().withoutPadding().encodeToString(bytes);
    }
}