package com.android.tencentvpn.util;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import java.security.Key;
import java.util.Date;
import javax.crypto.spec.SecretKeySpec;

public class SignGenerate {
    private static final long EXPIRE_TIME = 72 * 60 * 60 * 1000; // 过期时间, 单位毫秒
    /**
     * @param deviceName   业务自己控制设备唯一标识
     * @param SecretKey   云api返回的 应用密钥
     * @return   多网sdk 需要传入的签名字符串   最终通过setSign 调用
     *
     */
    public static String generateSign(String deviceName, String SecretKey) {
        try {
            Date now = new Date();
            Date expireDate = new Date(now.getTime() + EXPIRE_TIME);
            Key key =
                    new SecretKeySpec(SecretKey.getBytes(), SignatureAlgorithm.HS256.getJcaName());
            return Jwts.builder()
                    .claim("deviceName", deviceName)
                    .notBefore(now)
                    .expiration(expireDate)
                    .signWith(key)
                    .compact();
        } catch (Exception e) {
            e.printStackTrace();
        }
        return "";
    }
}

//podfile中添加pod 'JWT'，集成JWT库 
// .h文件：
@interface SignGenerator : NSObject
/**
* @deviceName 业务自己控制设备唯一标识
* @param SecretKey 云api返回的 应用密钥
* @return 多网sdk 需要传入的签名字符串 最终通过setSign 调用
*/
 + (NSString *)generateSignWithDeviceName:(NSString *)deviceName secretKey:(NSString *)secretKey;
@end
// .m文件
#import "SignGenerator.h"
#import <JWT/JWT.h>
@implementation SignGenerator
+ (NSString *)generateSignWithDeviceName:(NSString *)deviceName secretKey:(NSString *)secretKey {
    // 1. 设置过期时间（72小时后）
        NSDate *currentDate = [NSDate date];
        NSDate *expireDate = [currentDate dateByAddingTimeInterval:72 * 60 * 60];
        // 2. 构建 Claims（负载）
        NSDictionary *claims = @{
            @"deviceName": deviceName,
            @"nbf": @((long)[currentDate timeIntervalSince1970]), // Not Before
            @"exp": @((long)[expireDate timeIntervalSince1970])   // Expiration
        };
        NSData *keyData = [secretKey dataUsingEncoding:NSUTF8StringEncoding];
        // 3. 使用 HS256 算法和密钥签名
        JWTBuilder *builder = [JWTBuilder encodePayload:claims]
            .secretData(keyData)
            .algorithmName(@"HS256"); //算法名
        NSString *jwt = builder.encode;
        // 4. 错误处理
        if (builder.jwtError) {
            NSLog(@"JWT 生成失败: %@", builder.jwtError);
            return @"";
        }
        return jwt;
}
@end

package main
import (
	"fmt"
	"github.com/dgrijalva/jwt-go/v4"
	"time"
)
func main() {
	sign, expireTime, err := GenerateSign("tencent-test")
	if err != nil {
		panic(err)
	}
	println(sign, expireTime)
}
// SignClaims JwtCustomClaims
type SignClaims struct {
	jwt.StandardClaims
	DeviceName string `json:"deviceName"`
}
var (
	ExpireTime = 72                                                     // 过期时间, 单位小时
	SecretKey  = []byte("shCrVLKq/Hp6UdtVpw8lBqu6ZDcLwOABf2Az67e/pjI=") // APP密钥
)
// GenerateSign 生成签名
func GenerateSign(deviceName string) (string, int64, error) {
	expireTime := time.Now().Add(time.Duration(ExpireTime) * time.Hour)
	claims := &SignClaims{
		StandardClaims: jwt.StandardClaims{
			NotBefore: jwt.Now(),
			ExpiresAt: jwt.At(expireTime),
		},
		DeviceName: deviceName,
	}
	println(fmt.Sprintf("%#v", claims))

	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
	sign, err := token.SignedString(SecretKey)
	if err != nil {
		return "", 0, err
	}
	return sign, expireTime.Unix(), nil
}