net.sinofool.wechat.pay.WeChatPay Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of wechat-java-sdk Show documentation
WeChat open platform SDK
The newest version!
package net.sinofool.wechat.pay;

import net.sinofool.wechat.WeChatException;
import net.sinofool.wechat.base.OneLevelOnlyXML;
import net.sinofool.wechat.base.StringPair;
import net.sinofool.wechat.mp.WeChatMPEventHandler;
import net.sinofool.wechat.mp.WeChatUtils;
import net.sinofool.wechat.pay.dict.UnifedOrderRequestDict;
import org.xml.sax.SAXException;

import javax.xml.parsers.ParserConfigurationException;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.List;

public class WeChatPay {
    private static final org.slf4j.Logger LOG = org.slf4j.LoggerFactory.getLogger(WeChatPay.class);

    private final WeChatPayConfig config;
    private final WeChatPayHttpClient http;

    public WeChatPay(final WeChatPayConfig config, WeChatMPEventHandler eventHandler, final WeChatPayHttpClient http) {
        this.config = config;
        this.http = http;
        if (null != eventHandler) {
            eventHandler.setWeChatPay(this);
        }
    }

    public WeChatPayRequestData makeJSAPIPayment(int timestamp, String nonce, String prepayId) {
        WeChatPayRequestData request = new WeChatPayRequestData();
        request.setString("appId", config.getAppId());
        request.setString("timeStamp", String.valueOf(timestamp));
        request.setString("nonceStr", nonce);
        request.setString("package", "prepay_id=" + prepayId);
        request.setString("signType", "MD5");
        return request;
    }

    public String signJSAPIPayment(WeChatPayRequestData request) {
        return signMD5(request.getSortedParameters());
    }

    public boolean verifyJSAPIResponse(WeChatPayResponseData response) {
        if (!"SUCCESS".equalsIgnoreCase(response.getString("return_code"))) {
            return false;
        }
        if (!"SUCCESS".equalsIgnoreCase(response.getString("result_code"))) {
            return false;
        }
        String signFromResponse = response.getString("sign");
        String sign = signMD5(response.getSortedParameters("sign"));
        return sign.equalsIgnoreCase(signFromResponse);
    }

    public String signRaiseAppRequest(WeChatPayRequestData request) {
        return signMD5(request.getSortedParameters());
    }

    private WeChatPayRequestData makeOrder(String tradeNo, String name, double fee, String ip, String notify) {
        WeChatPayRequestData request = new WeChatPayRequestData();
        request.setString(UnifedOrderRequestDict.REQUIRED.APPID, config.getAppId());
        request.setString(UnifedOrderRequestDict.REQUIRED.MCH_ID, config.getMchId());
        request.setString(UnifedOrderRequestDict.REQUIRED.NONCE_STR, WeChatUtils.nonce());
        request.setString(UnifedOrderRequestDict.REQUIRED.OUT_TRADE_NO, tradeNo);
        request.setString(UnifedOrderRequestDict.REQUIRED.SPBILL_CREATE_IP, ip);
        request.setString(UnifedOrderRequestDict.REQUIRED.TOTAL_FEE, String.valueOf((int) (fee * 100.0f)));
        request.setString(UnifedOrderRequestDict.REQUIRED.BODY, name);
        request.setString(UnifedOrderRequestDict.REQUIRED.NOTIFY_URL, notify);
        return request;
    }

    public WeChatPayRequestData makeOrderJSAPI(String tradeNo, String name, double fee, String ip, String notify,
            String openId) {
        WeChatPayRequestData request = makeOrder(tradeNo, name, fee, ip, notify);
        request.setString(UnifedOrderRequestDict.REQUIRED.TRADE_TYPE, "JSAPI");
        request.setString(UnifedOrderRequestDict.OPTIONAL.OPEN_ID, openId);
        return request;
    }

    public WeChatPayRequestData makeOrderNATIVE(String tradeNo, String name, double fee, String ip, String notify,
            String productId) {
        WeChatPayRequestData request = makeOrder(tradeNo, name, fee, ip, notify);
        request.setString(UnifedOrderRequestDict.REQUIRED.TRADE_TYPE, "NATIVE");
        request.setString(UnifedOrderRequestDict.OPTIONAL.PRODUCT_ID, productId);
        return request;
    }

    public WeChatPayRequestData makeOrderAPP(String tradeNo, String name, double fee, String ip, String notify) {
        WeChatPayRequestData request = makeOrder(tradeNo, name, fee, ip, notify);
        request.setString(UnifedOrderRequestDict.REQUIRED.TRADE_TYPE, "APP");
        return request;
    }

    public WeChatPayRequestData makeRaiseAppRequest(String prepayId) {
        WeChatPayRequestData request = new WeChatPayRequestData();
        request.setString("appid", config.getAppId());
        request.setString("partnerid", config.getMchId());
        request.setString("prepayid", prepayId);
        request.setString("package", "Sign=WXPay");
        request.setString("noncestr", WeChatUtils.nonce());
        request.setString("timestamp", String.valueOf(WeChatUtils.now()));
        return request;
    }

    public String placeOrder(final WeChatPayRequestData request) {
        OneLevelOnlyXML xml = new OneLevelOnlyXML();
        xml.createRootElement("xml");
        for (StringPair param : sign(request.getSortedParameters())) {
            xml.createChild(param.getFirst(), param.getSecond());
        }
        String str = xml.toXMLString();
        LOG.info("Order request: {}", str);
        String ret = http.post("api.mch.weixin.qq.com", 443, "https", "/pay/unifiedorder", str);
        LOG.info("Order response: {}", ret);
        try {
            WeChatPayResponseData response = WeChatPayResponseData.parse(ret);
            String sign = signMD5(response.getAllData().getSorted("sign"));
            if (!sign.equalsIgnoreCase(response.getString("sign"))) {
                LOG.warn("Failed to verify sign");
                return null;
            }
            if (!"SUCCESS".equalsIgnoreCase(response.getString("return_code"))
                    || !"SUCCESS".equalsIgnoreCase(response.getString("result_code"))) {
                LOG.warn("Error returned {}", response.getString("return_msg"));
                return null;
            }
            return response.getString("prepay_id");
        } catch (UnsupportedEncodingException e) {
            LOG.warn("Cannot parse result", e);
            return null;
        } catch (ParserConfigurationException e) {
            LOG.warn("Cannot parse result", e);
            return null;
        } catch (SAXException e) {
            LOG.warn("Cannot parse result", e);
            return null;
        } catch (IOException e) {
            LOG.warn("Cannot parse result", e);
            return null;
        }
    }

    private List sign(final List p) {
        String sign = signMD5(p);
        p.add(new StringPair("sign", sign));
        return p;
    }

    private String signMD5(final List p) {
        String param = join(p, false) + "&key=" + config.getPayKey();
        String sign = WeChatUtils.md5HEX(param);
        LOG.trace("Signing {}={}", param, sign);
        return sign;
    }

    protected String join(final List p, boolean encode) {
        StringBuffer buff = new StringBuffer();
        for (int i = 0; i < p.size(); ++i) {
            if (i != 0) {
                buff.append("&");
            }
            buff.append(p.get(i).getFirst()).append("=");
            if (encode) {
                try {
                    buff.append(URLEncoder.encode(p.get(i).getSecond(), "utf-8"));
                } catch (UnsupportedEncodingException e) {
                    LOG.warn("Cannot encode {}", p.get(i).getSecond());
                    throw new WeChatException(e);
                }
            } else {
                buff.append(p.get(i).getSecond());
            }
        }
        return buff.toString();
    }
}