fj-dc/.svn/pristine/48/4890d92700da514d1f486c79592fed7cd1023420.svn-base

package cn.com.goldenwater.dcproj.utils.gwxy;

import org.apache.commons.codec.binary.Base64;
import org.apache.commons.codec.digest.DigestUtils;

import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.security.SignatureException;
import java.util.*;

public class MeetUtil {

    protected static String computeStringToSign(String requestPath, Map<String, String> reqParams, String reqJsonEntity, String reqMethod, String requestUriPrefix) throws Exception {
        //1. request method
        StringBuffer strToSign = new StringBuffer(reqMethod);
        strToSign.append("\n");
        //2. request path
        strToSign.append(requestPath.substring(requestUriPrefix.length()));
        strToSign.append("\n");
        //3. sorted request param and value
        List<String> params = new ArrayList<>(reqParams.keySet());
        Collections.sort(params);
        for (String param : params) {
            strToSign.append(param);
            strToSign.append("=");
            try {
                strToSign.append(URLEncoder.encode(reqParams.get(param), "UTF-8"));
            } catch (UnsupportedEncodingException e) {
            }
            strToSign.append("&");
        }
        strToSign.deleteCharAt(strToSign.length() - 1);
        strToSign.append("\n");
        //4. request entity
        byte[] reqEntity = reqJsonEntity.getBytes("utf-8");
        if (reqEntity.length == 0) {
            byte[] entity = DigestUtils.sha256("");
            strToSign.append(Base64.encodeBase64String(entity));
        } else {
            byte[] data = null;
            if (reqEntity.length <= 100) {
                data = reqEntity;
            } else {
                data = Arrays.copyOf(reqEntity, 100);
            }
            byte[] entity = DigestUtils.sha256(data);
            strToSign.append(Base64.encodeBase64String(entity));
        }

        String ret = strToSign.toString();
        System.out.println(ret);
        System.out.println("------------------");
        return ret;
    }

    private static String calculateHMAC(String data, String key) throws SignatureException {
        try {
            SecretKeySpec e = new SecretKeySpec(key.getBytes("UTF8"), "HmacSHA256");
            Mac mac = Mac.getInstance("HmacSHA256");
            mac.init(e);
            byte[] rawHmac = mac.doFinal(data.getBytes("UTF8"));
            String result = Base64.encodeBase64String(rawHmac);
            return result;
        } catch (Exception var6) {
            throw new SignatureException("Failed to generate HMAC : " + var6.getMessage());
        }
    }

    /**
     * 根据参数获取 请求签名
     *
     * @param jsonEntity       json 参数对象
     * @param method           请求方式
     * @param token            企业token
     * @param reqPath          请求地址
     * @param requestUriPrefix 请求前缀
     * @return 签名
     */
    public static String computeSignature(String jsonEntity, String method, String token, String reqPath, String requestUriPrefix) {
        try {
            Map<String, String> reqParams = new HashMap<>();
            int idx = reqPath.indexOf("?");
            String[] params = reqPath.substring(idx + 1).split("&");
            for (String param : params) {
                String[] pair = param.split("=");
                reqParams.put(pair[0], pair[1]);
            }
            reqPath = reqPath.substring(0, idx);
            String strToSign = computeStringToSign(reqPath, reqParams, jsonEntity, method, requestUriPrefix);
            String mySignature = calculateHMAC(strToSign, token);
            mySignature = mySignature.replace(" ", "+");
            return URLEncoder.encode(mySignature, "utf-8");
        } catch (Exception e) {
            return null;
        }
    }


}