package com.lingyue.common.util;

import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import lombok.extern.slf4j.Slf4j;

import javax.crypto.SecretKey;
import javax.crypto.spec.SecretKeySpec;
import java.nio.charset.StandardCharsets;
import java.util.Date;

/**
 * JWT 工具类
 */
@Slf4j
public class JwtUtil {
    
    /**
     * 默认密钥（生产环境应从配置文件读取）
     */
    private static final String DEFAULT_SECRET = "lingyue-zhibao-secret-key-2024-please-change-in-production";
    
    /**
     * 默认过期时间（7天）
     */
    private static final long DEFAULT_EXPIRATION = 7 * 24 * 60 * 60 * 1000L;
    
    /**
     * 刷新Token过期时间（30天）
     */
    private static final long REFRESH_EXPIRATION = 30 * 24 * 60 * 60 * 1000L;
    
    /**
     * 生成密钥
     */
    private static SecretKey getSecretKey(String secret) {
        byte[] keyBytes = secret.getBytes(StandardCharsets.UTF_8);
        return new SecretKeySpec(keyBytes, SignatureAlgorithm.HS256.getJcaName());
    }
    
    /**
     * 生成Token
     */
    public static String generateToken(String userId, String username, String secret, long expiration) {
        Date now = new Date();
        Date expiryDate = new Date(now.getTime() + expiration);
        
        return Jwts.builder()
            .setSubject(userId)
            .claim("username", username)
            .setIssuedAt(now)
            .setExpiration(expiryDate)
            .signWith(getSecretKey(secret), SignatureAlgorithm.HS256)
            .compact();
    }
    
    /**
     * 生成Token（使用默认配置）
     */
    public static String generateToken(String userId, String username) {
        return generateToken(userId, username, DEFAULT_SECRET, DEFAULT_EXPIRATION);
    }
    
    /**
     * 生成刷新Token
     */
    public static String generateRefreshToken(String userId, String username, String secret) {
        return generateToken(userId, username, secret, REFRESH_EXPIRATION);
    }
    
    /**
     * 生成刷新Token（使用默认配置）
     */
    public static String generateRefreshToken(String userId, String username) {
        return generateRefreshToken(userId, username, DEFAULT_SECRET);
    }
    
    /**
     * 解析Token
     */
    public static Claims parseToken(String token, String secret) {
        try {
            return Jwts.parserBuilder()
                .setSigningKey(getSecretKey(secret))
                .build()
                .parseClaimsJws(token)
                .getBody();
        } catch (Exception e) {
            log.error("Failed to parse token: {}", e.getMessage());
            return null;
        }
    }
    
    /**
     * 解析Token（使用默认密钥）
     */
    public static Claims parseToken(String token) {
        return parseToken(token, DEFAULT_SECRET);
    }
    
    /**
     * 验证Token是否有效
     */
    public static boolean validateToken(String token, String secret) {
        Claims claims = parseToken(token, secret);
        return claims != null && !claims.getExpiration().before(new Date());
    }
    
    /**
     * 验证Token是否有效（使用默认密钥）
     */
    public static boolean validateToken(String token) {
        return validateToken(token, DEFAULT_SECRET);
    }
    
    /**
     * 从Token中获取用户ID
     */
    public static String getUserIdFromToken(String token, String secret) {
        Claims claims = parseToken(token, secret);
        return claims != null ? claims.getSubject() : null;
    }
    
    /**
     * 从Token中获取用户ID（使用默认密钥）
     */
    public static String getUserIdFromToken(String token) {
        return getUserIdFromToken(token, DEFAULT_SECRET);
    }
}
