package top.cenze.utils.crypt;

import cn.hutool.core.util.StrUtil;
import lombok.SneakyThrows;
import org.apache.commons.codec.binary.Base64;
import top.cenze.utils.pojo.AppidSecret;

import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import javax.crypto.SecretKey;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;
import java.util.UUID;

/**
 * AES对称加密工具
 * 密钥加密/解密
 * @author chengze
 * @date 2023-11-14 22:05
 */
public class AESUtil {
    public static String AES = "AES";//指定算法类型
    public static int KEY_LEN = 128;//指定密钥长度
    public static String UTF_8 = "UTF-8";//编码格式

    // CBC模式
    private static final String key = "aesQiuChengZeKey";   // 必需为16位
    private static final String initVector = "aesQiuChengZeVec";    // 必需为16位

    /*
     * @Author boy
     * @Description 生成AES密钥
     * @Date 2019/8/22 10:49 AM
     * @Param []
     * @return java.lang.String
     */
    public static String genAESKey() throws Exception{
        //构造密钥生成器，指定为AES算法
        KeyGenerator keyGenerator = KeyGenerator.getInstance(AES);
        //生成一个指定位数的随机源，KEY_LEN=128就是128位
        keyGenerator.init(KEY_LEN);
        //生成对称密钥
        SecretKey sKey = keyGenerator.generateKey();
        return Base64.encodeBase64String(sKey.getEncoded());
    }

    public static String encrypt(String data) throws Exception {
        return encrypt(key, data);
    }

    /*
     * @Author boy
     * @Description 使用AES密钥加密数据
     * @Date 2019/8/22 10:51 AM
     * @Param [key, data]
     * @return java.lang.String
     */
    public static String encrypt(String key, String data) throws Exception {
        if (StrUtil.isEmpty(key) || key.length() != 16) {
            throw new RuntimeException("无效的密钥key");
        }

        //获取key
        IvParameterSpec iv = new IvParameterSpec(initVector.getBytes("UTF-8"));
        SecretKeySpec skeySpec = new SecretKeySpec(key.getBytes("UTF-8"), AES);
        //根据指定算法生成密码器
        Cipher cipher=Cipher.getInstance("AES/CBC/PKCS5PADDING");
        //初始化密码器，第一个参数为加密或者解密解密操作，第二个参数为使用的KEY
        cipher.init(Cipher.ENCRYPT_MODE, skeySpec, iv);
        //将加密内容转化为字节数组
        byte [] byte_data=data.getBytes(UTF_8);
        //将字节数组加密
        byte [] AES_data=cipher.doFinal(byte_data);
        return Base64.encodeBase64String(AES_data);
    }

    public static String decrypt(String data) throws Exception {
        return decrypt(key, data);
    }

    /*
     * @Author boy
     * @Description 使用AES密钥解密数据
     * @Date 2019/8/22 8:06 PM
     * @Param [key, data]
     * @return java.lang.String
     */
    public static String decrypt(String key, String data) throws Exception {
        if (StrUtil.isEmpty(key) || key.length() != 16) {
            throw new RuntimeException("无效的密钥key");
        }

        //获取key
        IvParameterSpec iv = new IvParameterSpec(initVector.getBytes("UTF-8"));
        SecretKeySpec skeySpec = new SecretKeySpec(key.getBytes("UTF-8"), AES);
        //根据指定算法生成密码器
        Cipher cipher=Cipher.getInstance("AES/CBC/PKCS5PADDING");
        //初始化密码器，第一个参数为加密或者解密解密操作，第二个参数为使用的KEY
        cipher.init(Cipher.DECRYPT_MODE, skeySpec, iv);
        //将加密内容转化为字节数组，因为数据是用Base64转换过的，所以需要使用base64解密
        byte [] dataByte = Base64.decodeBase64(data.getBytes(UTF_8));
        //解密字节数组
        byte [] decryptData=cipher.doFinal(dataByte);
        return new String(decryptData);
    }

    @SneakyThrows
    public static AppidSecret genAppidSecret() {
        AppidSecret appidSecret = new AppidSecret();

        UUID uuid = UUID.randomUUID();
        String str = uuid.toString();
        // 将UUID的字符串形式中的破折号去掉
        appidSecret.setAppId(str.replace("-", ""));
        appidSecret.setSecret(encrypt(key, appidSecret.getAppId()));

        return appidSecret;
    }

//    private static final String key = "aeswwwLkd56InKey";   // 必需为16位
//    private static final String initVector = "aeswwwLkd56InVec";    // 必需为16位
//
//    //加密
//    public static String encrypt(String value) {
//        try {
//            IvParameterSpec iv = new IvParameterSpec(initVector.getBytes("UTF-8"));
//            SecretKeySpec skeySpec = new SecretKeySpec(key.getBytes("UTF-8"), "AES");
//
//            Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5PADDING");
//            cipher.init(Cipher.ENCRYPT_MODE, skeySpec, iv);
//
//            byte[] encrypted = cipher.doFinal(value.getBytes());
//            return Base64.encodeBase64String(encrypted);
//        } catch (Exception ex) {
//            ex.printStackTrace();
//        }
//
//        return null;
//    }
//
//    //    解密
//    public static String decrypt(String encrypted) {
//        try {
//            IvParameterSpec iv = new IvParameterSpec(initVector.getBytes("UTF-8"));
//            SecretKeySpec skeySpec = new SecretKeySpec(key.getBytes("UTF-8"), "AES");
//            Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5PADDING");
//            cipher.init(Cipher.DECRYPT_MODE, skeySpec, iv);
//            byte[] original = cipher.doFinal(Base64.decodeBase64(encrypted));
//            return new String(original);
//        } catch (Exception ex) {
//            ex.printStackTrace();
//        }
//
//        return null;
//    }
}