org.tinygroup.weixinhttpclient3.HttpConnectorDefault Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of org.tinygroup.weixinhttpclient3 Show documentation
The newest version!
/**
 * Copyright (c) 2012-2016, www.tinygroup.org ([email protected]).
 *
 *  Licensed under the GPL, Version 3.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *       http://www.gnu.org/licenses/gpl.html
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package org.tinygroup.weixinhttpclient3;

import org.apache.commons.httpclient.HttpClient;
import org.apache.commons.httpclient.methods.PostMethod;
import org.apache.commons.httpclient.methods.multipart.MultipartRequestEntity;
import org.apache.commons.httpclient.methods.multipart.Part;
import org.apache.commons.httpclient.methods.multipart.StringPart;
import org.apache.commons.httpclient.params.HttpMethodParams;
import org.apache.commons.httpclient.protocol.Protocol;
import org.apache.commons.httpclient.protocol.ProtocolSocketFactory;
import org.tinygroup.httpvisit.HttpVisitor;
import org.tinygroup.httpvisit.impl.HttpVisitorImpl;
import org.tinygroup.vfs.FileObject;
import org.tinygroup.vfs.VFS;
import org.tinygroup.weixinhttp.WeiXinHttpConnector;
import org.tinygroup.weixinhttp.WeiXinHttpUpload;
import org.tinygroup.weixinhttp.cert.WeiXinCert;
import org.tinygroup.weixinhttpclient3.cert.AuthSSLProtocolSocketFactory;

import javax.net.ssl.KeyManagerFactory;
import javax.net.ssl.SSLContext;
import java.security.KeyStore;

/**
 * 默认采用httpvisit实现
 *
 * @author yancheng11334
 */
public class HttpConnectorDefault implements WeiXinHttpConnector {

    private static final String HTTPS_PROTOCOL = "https";
    private HttpVisitor visitor;

    public HttpVisitor getVisitor() {
        if (visitor == null) {
            visitor = new HttpVisitorImpl();
            visitor.init();
        }
        return visitor;
    }

    public void setVisitor(HttpVisitor visitor) {
        this.visitor = visitor;
    }

    public String getUrl(String url) {
        return getVisitor().getUrl(url, null);
    }

    public String postUrl(String url, String content, WeiXinCert cert) {
        if (cert != null) {
            addProtocol(cert);
        }
        return getVisitor().postXml(url, content);
    }

    private void addProtocol(WeiXinCert cert) {
        try {
            KeyStore keyStore = KeyStore.getInstance(cert.getCertType());
            FileObject certObjcet = VFS.resolveFile(cert.getCertPath());
            char[] password = cert.getPassword().toCharArray();
            try {
                keyStore.load(certObjcet.getInputStream(), password);
            } finally {
                certObjcet.clean();
            }
            KeyManagerFactory factory = KeyManagerFactory.getInstance(KeyManagerFactory.getDefaultAlgorithm());
            factory.init(keyStore, password);

            // Trust own CA and all self-signed certs
            SSLContext sslcontext = SSLContext.getInstance("TLS");
            sslcontext.init(factory.getKeyManagers(), null, null);

            ProtocolSocketFactory protocolSocketFactory = new AuthSSLProtocolSocketFactory(VFS.resolveFile(cert.getCertPath()), cert.getPassword(), null, null);
            Protocol authhttps = new Protocol(HTTPS_PROTOCOL, protocolSocketFactory, 443);
            Protocol.registerProtocol(HTTPS_PROTOCOL, authhttps);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }

    }

    public String upload(String url, WeiXinHttpUpload upload) {
        PostMethod postMethod = new PostMethod(url);
        try {
            Part[] parts = createParts(upload);
            HttpMethodParams params = postMethod.getParams();
            postMethod.setRequestEntity(new MultipartRequestEntity(parts, params));
            HttpClient client = getVisitor().getHttpClient();
            client.executeMethod(postMethod);
            return new String(postMethod.getResponseBody(), "utf-8");
        } catch (Exception e) {
            throw new RuntimeException(e);
        } finally {
            postMethod.releaseConnection();
        }
    }

    private Part[] createParts(WeiXinHttpUpload upload) {
        if (upload.getFormName() == null) {
            Part[] parts = {new FileObjectPart(upload.getFileName(), upload.getFileObject(), null, null)};
            return parts;
        } else {
            Part[] parts = {new FileObjectPart(upload.getFileName(), upload.getFileObject(), null, null), new StringPart(upload.getFormName(), upload.getContent())};
            return parts;
        }
    }

}