package com.jysemel;

import io.jsonwebtoken.*;
import io.jsonwebtoken.security.Keys;

import javax.crypto.SecretKey;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;

public class JwtExample {

    // 生成一个足够安全的签名密钥（实际应用中应从配置文件读取或使用密钥库）
    private static final SecretKey SECRET_KEY = Keys.secretKeyFor(SignatureAlgorithm.HS256); // HS256 需要 256位密钥

    // 过期时间：1小时（毫秒）
    private static final long EXPIRATION_TIME = 3600000;

    public static void main(String[] args) {
        // 1. 生成 JWT
        String token = generateToken("user123", "admin");
        System.out.println("生成的 JWT：\n" + token);

        // 2. 验证并解析 JWT
        parseToken(token);
    }

    /**
     * 生成 JWT
     * @param subject  用户标识（如用户名、用户ID）
     * @param role     用户角色（自定义声明示例）
     * @return JWT 字符串
     */
    public static String generateToken(String subject, String role) {
        // 设置头部（可选，默认即可）
        Map<String, Object> header = new HashMap<>();
        header.put("typ", "JWT");

        // 设置载荷（Claims）
        JwtBuilder builder = Jwts.builder()
                .setHeader(header)                 // 头部
                .setSubject(subject)                // 主题（通常是用户标识）
                .setIssuedAt(new Date())            // 签发时间
                .setExpiration(new Date(System.currentTimeMillis() + EXPIRATION_TIME)) // 过期时间
                .setIssuer("MyApp")                  // 签发者
                .setAudience("MyClient")             // 接收方
                .setId(java.util.UUID.randomUUID().toString()) // JWT ID，用于防止重放攻击
                .claim("role", role)                 // 自定义声明：角色
                .claim("dept", "engineering")        // 自定义声明：部门
                .signWith(SECRET_KEY);                // 签名

        return builder.compact();
    }

    /**
     * 解析 JWT，验证签名并提取数据
     * @param token JWT 字符串
     */
    public static void parseToken(String token) {
        try {
            // 创建解析器，设置签名密钥
            JwtParser parser = Jwts.parserBuilder()
                    .setSigningKey(SECRET_KEY)
                    .build();

            // 解析 JWT，得到 Jws<Claims> 对象
            Jws<Claims> jws = parser.parseClaimsJws(token);

            // 获取头部
            JwsHeader header = jws.getHeader();
            System.out.println("\n--- 解析结果 ---");
            System.out.println("算法：" + header.getAlgorithm());
            System.out.println("类型：" + header.getType());

            // 获取载荷
            Claims claims = jws.getBody();
            System.out.println("主题（subject）：" + claims.getSubject());
            System.out.println("签发时间：" + claims.getIssuedAt());
            System.out.println("过期时间：" + claims.getExpiration());
            System.out.println("签发者：" + claims.getIssuer());
            System.out.println("接收方：" + claims.getAudience());
            System.out.println("JWT ID：" + claims.getId());
            System.out.println("角色（自定义声明）：" + claims.get("role"));
            System.out.println("部门（自定义声明）：" + claims.get("dept"));

        } catch (ExpiredJwtException e) {
            System.err.println("JWT 已过期：" + e.getMessage());
        } catch (UnsupportedJwtException e) {
            System.err.println("不支持的 JWT：" + e.getMessage());
        } catch (MalformedJwtException e) {
            System.err.println("JWT 格式错误：" + e.getMessage());
        } catch (SignatureException e) {
            System.err.println("JWT 签名验证失败：" + e.getMessage());
        } catch (IllegalArgumentException e) {
            System.err.println("JWT 参数非法：" + e.getMessage());
        }
    }
}