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; // Expiration time, in milliseconds
    /**
     * @param deviceName   The unique identifier of the device controlled by the business itself
     * @param SecretKey   The application key returned by the TencentCloud api
     * @return   The signature string required by the multi-network sdk, which is ultimately invoked via the setSign method.
     *
     */
    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)
                    .expiration(expireDate)
                    .signWith(key)
                    .compact();
        } catch (Exception e) {
            e.printStackTrace();
        }
        return "";
    }
}

// Add pod 'JWT' to the podfile to integrate the JWT library. 
// .h file:
@interface SignGenerator : NSObject
/**
* @deviceName The unique identifier of the device controlled by the business itself
* @param SecretKey The application key returned by the TencentCloud api
* @return   The signature string required by the multi-network sdk, which is ultimately invoked via the setSign method.
*/
 + (NSString *)generateSignWithDeviceName:(NSString *)deviceName secretKey:(NSString *)secretKey;
@end
// .m file
#import "SignGenerator.h"
#import <JWT/JWT.h>
@implementation SignGenerator
+ (NSString *)generateSignWithDeviceName:(NSString *)deviceName secretKey:(NSString *)secretKey {
    // 1. Set the expiration time (after 72 hours)
        NSDate *currentDate = [NSDate date];
        NSDate *expireDate = [currentDate dateByAddingTimeInterval:72 * 60 * 60];
        // 2. Construct the Claims (payload)
        NSDictionary *claims = @{
            @"deviceName": deviceName,
            @"exp": @((long)[expireDate timeIntervalSince1970])   // Expiration
        };
        NSData *keyData = [secretKey dataUsingEncoding:NSUTF8StringEncoding];
        // 3. Sign using the HS256 algorithm and a secret key
        JWTBuilder *builder = [JWTBuilder encodePayload:claims]
            .secretData(keyData)
            .algorithmName(@"HS256"); // Algorithm name
        NSString *jwt = builder.encode;
        // 4. Error handling
        if (builder.jwtError) {
            NSLog(@"JWT generation failed: %@", builder.jwtError);
            return @"";
        }  
        return jwt;
}

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                       // Expiration time, in hours
	SecretKey  = []byte("your secretkey") // APP key
)
// GenerateSign generates a signature
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
}