package org.jeecg.common.aspect.baiduAddress;

import lombok.SneakyThrows;

import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.util.LinkedHashMap;
import java.util.Map.Entry;

public class BaiduSnUtil {
//    public static void main(String[] args) throws UnsupportedEncodingException,
//            NoSuchAlgorithmException {
//        BaiduSnUtil snCal = new BaiduSnUtil();
//
//// 计算sn跟参数对出现顺序有关，get请求请使用LinkedHashMap保存<key,value>，该方法根据key的插入顺序排序；post请使用TreeMap保存<key,value>，该方法会自动将key按照字母a-z顺序排序。所以get请求可自定义参数顺序（sn参数必须在最后）发送请求，但是post请求必须按照字母a-z顺序填充body（sn参数必须在最后）。以get请求为例：http://api.map.baidu.com/geocoder/v2/?address=百度大厦&output=json&ak=yourak，paramsMap中先放入address，再放output，然后放ak，放入顺序必须跟get请求中对应参数的出现顺序保持一致。
//
//        LinkedHashMap paramsMap = new LinkedHashMap<String, String>();
//        paramsMap.put("address", "百度大厦");
//        paramsMap.put("output", "json");
//        paramsMap.put("ak", "yourak");
//
//        // 调用下面的toQueryString方法，对LinkedHashMap内所有value作utf8编码，拼接返回结果address=%E7%99%BE%E5%BA%A6%E5%A4%A7%E5%8E%A6&output=json&ak=yourak
//        String paramsStr = snCal.toQueryString(paramsMap);
//
//        // 对paramsStr前面拼接上/geocoder/v2/?，后面直接拼接yoursk得到/geocoder/v2/?address=%E7%99%BE%E5%BA%A6%E5%A4%A7%E5%8E%A6&output=json&ak=yourakyoursk
//        String wholeStr = new String("/geocoder/v2/?" + paramsStr + "yoursk");
//
//        // 对上面wholeStr再作utf8编码
//        String tempStr = URLEncoder.encode(wholeStr, "UTF-8");
//
//        // 调用下面的MD5方法得到最后的sn签名7de5a22212ffaa9e326444c75a58f9a0
//        System.out.println(snCal.md5(tempStr));
//        System.out.println(sn(paramsMap, "yoursk"));
//    }

    // 对Map内所有value作utf8编码，拼接返回结果
    @SneakyThrows
    private static String toQueryString(LinkedHashMap<?, ?> data) {
        StringBuffer queryString = new StringBuffer();
        for (Entry<?, ?> pair : data.entrySet()) {
            queryString.append(pair.getKey() + "=");
            queryString.append(URLEncoder.encode((String) pair.getValue(), StandardCharsets.UTF_8) + "&");
        }
        if (queryString.length() > 0) {
            queryString.deleteCharAt(queryString.length() - 1);
        }
        return queryString.toString();
    }

    // 来自stackoverflow的MD5计算方法，调用了MessageDigest库函数，并把byte数组结果转换成16进制
    @SneakyThrows
    private static String md5(String md5) {
        java.security.MessageDigest md = java.security.MessageDigest.getInstance("MD5");
        byte[] array = md.digest(md5.getBytes());
        StringBuffer sb = new StringBuffer();
        for (int i = 0; i < array.length; ++i) {
            sb.append(Integer.toHexString((array[i] & 0xFF) | 0x100).substring(1, 3));
        }
        return sb.toString();
    }


    @SneakyThrows
    public static String sn(String uri, LinkedHashMap<?, ?> paramsMap, String sk) {
        String paramsStr = toQueryString(paramsMap);
        String wholeStr = uri + paramsStr + sk;
        String tempStr = URLEncoder.encode(wholeStr, StandardCharsets.UTF_8);
        return md5(tempStr);
    }


}