package com.walker.web.token;

import com.walker.infrastructure.utils.StringUtils;
import com.walker.web.TokenException;
import com.walker.web.TokenGenerator;
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.ExpiredJwtException;
import io.jsonwebtoken.JwtBuilder;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import io.jsonwebtoken.SignatureException;

import java.util.Date;
import java.util.HashMap;
import java.util.Map;

public class JwtTokenGenerator implements TokenGenerator {

    private static final long ONE_MINUTE_MILLS = 1 * 60 * 1000;

    @Override
    public String createToken(String userId, long expiredMinutes, String magicKey) {
//        if(StringUtils.isEmpty(claims)){
//            throw new IllegalArgumentException("createToken error: claims is required!");
//        }
//
//        JwtBuilder builder = Jwts.builder().setIssuer(NAME_OWNER)
//                .setIssuedAt(new Date())
//                .setSubject(claims)
//                .signWith(SignatureAlgorithm.HS512, magicKey);
//
//        long expiredTime = 0;
//        if(expiredMinutes > 0){
//            expiredTime = System.currentTimeMillis() + expiredMinutes * ONE_MINUTE_MILLS;
//            builder.setExpiration(new Date(expiredTime));
//            logger.debug("token 过期时间:" + expiredTime);
//        }
//        return builder.compact();
        return this.createToken(null, userId, expiredMinutes, magicKey);
    }

    @Override
    public String createToken(String userKey, String userId, long expiredMinutes, String magicKey){
        if(StringUtils.isEmpty(userId)){
            throw new IllegalArgumentException("createToken error: userId is required!");
        }

        JwtBuilder builder = Jwts.builder().setIssuer(NAME_OWNER)
                .setIssuedAt(new Date());
//                .setSubject(userId);
        // 如果存在，设置用户缓存唯一标识(UUID),2022-11-01
        Map<String, Object> claims = new HashMap<>(2);
        if(StringUtils.isNotEmpty(userKey)){
            claims.put(NAME_lOGIN_USER_KEY, userKey);
            claims.put(NAME_USER_ID_KEY, userId);
        } else {
            claims.put(NAME_USER_ID_KEY, userId);
        }
        builder.addClaims(claims);
        builder.signWith(SignatureAlgorithm.HS512, magicKey);

        long expiredTime = 0;
        if(expiredMinutes > 0){
            expiredTime = System.currentTimeMillis() + expiredMinutes * ONE_MINUTE_MILLS;
            builder.setExpiration(new Date(expiredTime));
            logger.debug("token 过期时间:" + expiredTime);
        }
        return builder.compact();
    }

    @Override
    public String validateToken(String token, String secretKey) throws TokenException {
        try{
            Claims c = Jwts.parser().setSigningKey(secretKey).parseClaimsJws(token).getBody();
            if(c == null){
                throw new TokenException("token claims is null!");
            }
            StringBuilder sb = new StringBuilder(c.get(NAME_USER_ID_KEY).toString());
            if(c.get(NAME_lOGIN_USER_KEY) != null){
                sb.append(StringUtils.DEFAULT_SPLIT_SEPARATOR);
                sb.append(c.get(NAME_lOGIN_USER_KEY).toString());
            }
            return sb.toString();
//            return c.getSubject();
        } catch (Exception ex){
            if(ex instanceof SignatureException){
                throw new TokenException("token签名解析错误", ex);
            } else if(ex instanceof ExpiredJwtException){
                throw new TokenException("token过期", ex, true);
            } else {
                throw new TokenException(ex.getMessage(), ex);
            }
        }
    }

//    public String getDataFromToken(String token, String secretKey){
//        try{
//            Claims c = Jwts.parser().setSigningKey(secretKey).parseClaimsJws(token).getBody();
//            if(c == null){
////                throw new TokenException("token claims is null!");
//                System.out.println("解析token业务数据时，返回null");
//                return null;
//            }
//        } catch (Exception ex){
//        }
//    }

    @Override
    public String acquireSecretKey() {
        return null;
    }

//    private String getData(Claims c){
//        StringBuilder sb = new StringBuilder(c.get(NAME_USER_ID_KEY).toString());
//        if(c.get(NAME_lOGIN_USER_KEY) != null){
//            sb.append(StringUtils.DEFAULT_SPLIT_SEPARATOR);
//            sb.append(c.get(NAME_lOGIN_USER_KEY).toString());
//        }
//        return sb.toString();
//    }
}