package jnpf.util;

import cn.hutool.cache.CacheUtil;
import cn.hutool.cache.impl.TimedCache;
import cn.hutool.crypto.digest.DigestUtil;
import jnpf.base.UserInfo;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import java.util.Date;
import java.util.LinkedHashMap;
import java.util.Map;

/**
 * 网关工具类
 */
@Component
@Slf4j
@RequiredArgsConstructor
public class GatewayUtil {

    private static final String PENDING_PREFIX = "jnpf:login-session:pending:";
    private static final String TOKEN_MAPPING_PREFIX = "jnpf:login-session:token:";
    private static final String TENANT_REGISTRY_KEY = "jnpf:login-session:tenants";
    private static final long PENDING_TTL_SECONDS = 7 * 24 * 60 * 60L;
    private static final TimedCache<String, Integer> renewLimit;
    private static final long TIMEOUT = 1000L * 60;

    private final RedisUtil redisUtil;
    private final RedisTemplate<String, Object> redisTemplate;

    static {
        renewLimit = CacheUtil.newTimedCache(TIMEOUT);
        // 一分钟清理一次无用数据
        renewLimit.schedulePrune(TIMEOUT);
    }

    /**
     * 同一个TOKEN一分钟内只会续期一次
     */
    public void renewToken(String token){
        if(renewLimit.containsKey(token)) {
            return;
        }
        UserProvider.renewTimeout(token);
        renewLimit.put(token, 0);
        enqueueSessionTouch(token);
    }

    private void enqueueSessionTouch(String token) {
        try {
            UserInfo userInfo = UserProvider.getUser();
            if (userInfo == null || userInfo.getTokenTimeout() == null) {
                return;
            }
            Date now = DateUtil.getNowDate();
            Date expireTime = DateUtil.dateAddMinutes(now, userInfo.getTokenTimeout());
            String normalizedToken = token == null ? "" : token.trim();
            if (normalizedToken.regionMatches(true, 0, "bearer", 0, "bearer".length())) {
                normalizedToken = normalizedToken.substring("bearer".length()).trim();
            }
            String sessionId = DigestUtil.sha256Hex(normalizedToken);
            String eventId = RandomUtil.uuId();
            Map<String, Object> event = new LinkedHashMap<>();
            event.put("eventId", eventId);
            event.put("transitionId", eventId);
            event.put("eventType", "TOUCH");
            event.put("sessionId", sessionId);
            event.put("userInfo", userInfo);
            event.put("eventTime", now);
            event.put("expireTime", expireTime);
            long ttl = Math.max(1, userInfo.getTokenTimeout() * 60L + 120L);
            redisUtil.insert(TOKEN_MAPPING_PREFIX + sessionId, "1", ttl);
            if (StringUtil.isNotEmpty(userInfo.getTenantId())) {
                redisTemplate.opsForSet().add(TENANT_REGISTRY_KEY, userInfo.getTenantId());
            }
            redisUtil.insert(PENDING_PREFIX + eventId,
                    JsonUtil.getObjectToStringAsDate(event), PENDING_TTL_SECONDS);
        } catch (Exception e) {
            log.warn("同步登录会话续期失败，不影响当前请求", e);
        }
    }

}