package com.dw.task.utils;

import cn.hutool.crypto.SmUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import java.lang.reflect.Field;
import java.util.Objects;

@Slf4j
@Component
public class Sm3UtilHua {

    private static Integer httpCheckSignTimeOut = 1;//請求簽名有效時間 默認(rèn)1分鐘

    /**
     * 獲取實體類拼成的加密字段
     * @param checkSign  前端傳入的簽名（從請求報文頭獲?。?
     * @param signModel  查詢的DTO模型類
     * @param privateKey  簽名加密私鑰
     * @param timestamp  時間戳（從請求報文頭獲?。?
     * @return  比對結(jié)果
     */
    public boolean checkSign(String checkSign , Object signModel , String privateKey,Long timestamp) throws Exception {
        Long thisTime = System.currentTimeMillis() - timestamp;
        Integer checkSignTimeOut = httpCheckSignTimeOut;
        if (!(Objects.isNull(checkSignTimeOut) || checkSignTimeOut.intValue() == 0)){
            //時間為0或者未配置簽名超時時間，默認(rèn)不驗證時間戳
            if(thisTime >= 60*1000*checkSignTimeOut || thisTime<=0){
                //checkSignTimeOut分鐘內(nèi)的時間戳才處理
                log.error("時間戳異常,非"+checkSignTimeOut+"分鐘內(nèi)請求,當(dāng)前時間戳:"+System.currentTimeMillis());
                return false;
            }
        }
        String signValue = getSignValue(signModel) + "&timestamp=" + timestamp +"&privateKey=" +privateKey;
        String sign = getSign(signValue);
        log.info("【本地加密后 sm3 簽名】" + sign);//生產(chǎn)上建議注釋此行,防止泄露
        return sign.toUpperCase().equals(checkSign.toUpperCase())? true : false;
    }
    /**
     * 加密簽名
     * @param signValue  待加密簽名字符串
     * @return  加密后簽名字符串
     */
    public String getSign(String signValue){
        return SmUtil.sm3(signValue);
    }
    
    /**
     * 獲取實體類拼成的加密字段
     * @param classA  傳入?yún)?shù)實體類
     * @return  待加密字符串
     */
    public  String getSignValue(Object classA) {
        Field[] fs = classA.getClass().getDeclaredFields();//獲取所有屬性
        String[][] temp = new String[fs.length][2]; //用二維數(shù)組保存  參數(shù)名和參數(shù)值
        for (int i=0; i<fs.length;  i++) {
            fs[i].setAccessible(true);
            temp[i][0] = fs[i].getName().toLowerCase(); //獲取屬性名
            try {
                temp[i][1] = String.valueOf(fs[i].get(classA)) ;//把屬性值放進(jìn)數(shù)組
            }  catch (Exception e) {
                log.error("【簽名字段："+fs[i].getName()+"添加失敗】");
            }
        }
        temp = doChooseSort(temp); //對參數(shù)實體類按照字母順序排續(xù)
        String result = "";
        for (int i = 0; i < temp.length; i++) {//按照簽名規(guī)則生成待加密字符串
            result = result + temp[i][0]+"="+temp[i][1]+"&";
        }
        result = result.substring(0, result.length()-1);//消除掉最后的“&”
        log.info("【簽名信息】{}" ,result);
        return result;
    }

    /**
     * 對二維數(shù)組里面的數(shù)據(jù)進(jìn)行選擇排序，按字段名按abcd順序排列
     * @param data 未按照字母順序排序的二維數(shù)組
     * @return
     */
    private  String[][] doChooseSort(String[][] data) {//排序方式為選擇排序
        String[][] temp = new String[data.length][2];
        temp = data;
        int n = temp.length;
        for (int i = 0; i < n-1; i++) {
            int k = i;// 初始化最小值的小標(biāo)
            for (int j = i+1; j<n; j++) {
                if (temp[k][0].compareTo(temp[j][0]) > 0) {    //下標(biāo)k字段名大于當(dāng)前字段名
                    k = j;// 修改最大值的小標(biāo)
                }
            }
            // 將最小值放到排序序列末尾
            if (k > i) {  //用相加相減法交換data[i] 和 data[k]
                String tempValue ;
                tempValue = temp[k][0];
                temp[k][0] = temp[i][0];
                temp[i][0] = tempValue;
                tempValue = temp[k][1];
                temp[k][1] = temp[i][1];
                temp[i][1] = tempValue;
            }
        }
        return temp;
    }

}

    public static void main(String[] args) throws Exception {
        //模擬請求參數(shù)
        QueryDTO dto = new QueryDTO();
        dto.setName("張三");
        dto.setAge(18);
        dto.setIdCard("666666666666666");
        //模擬請求報文頭時間戳
        Long nowTime = System.currentTimeMillis();
        //簽名加密私鑰（不要在互聯(lián)網(wǎng)上傳輸,調(diào)用方和提供方私下物理確認(rèn)和設(shè)置）
        String privateKey = "48d95af20fa1bc438db42e280085707b60841c";

        Sm3UtilHua sm3UtilHua = new Sm3UtilHua();

        System.out.println("1：請求簽名錯誤的案例");
        //模擬請求錯誤的簽名
        String errorSign = "2222222222222222222";
        System.out.println("驗簽校驗結(jié)果：" + sm3UtilHua.checkSign(errorSign,dto, privateKey,nowTime));
        System.out.println();

        System.out.println("2：請求簽名正確的案例");
        String signValue = sm3UtilHua.getSignValue(dto) + "&timestamp=" + nowTime +"&privateKey=" +privateKey;//復(fù)制上面1請求的加密后簽名
        String trueSign = sm3UtilHua.getSign(signValue);
        System.out.println("驗簽校驗結(jié)果：" + sm3UtilHua.checkSign(trueSign,dto, privateKey,nowTime));
        System.out.println();

        System.out.println("3：請求簽名正確，但是時間非一分鐘內(nèi)請求的案例");
        Long oldTime = 1688036145560L;//這個時間戳大概是2023年06月29號的時間戳，所以必不是當(dāng)前一分鐘內(nèi)時間
        System.out.println("驗簽校驗結(jié)果：" + sm3UtilHua.checkSign(trueSign,dto, privateKey,oldTime));

    }