com.seejoke.wechat.util.http.MediaDownloadGetRequestExecutor Maven / Gradle / Ivy

Go to download
package com.seejoke.wechat.util.http;

import com.seejoke.wechat.entity.result.WxError;
import com.seejoke.wechat.exception.WxErrorException;
import com.seejoke.wechat.util.StringUtils;
import com.seejoke.wechat.util.FileUtils;
import org.apache.http.Header;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.entity.ContentType;
import org.apache.http.impl.client.CloseableHttpClient;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class MediaDownloadGetRequestExecutor implements RequestExecutor> {

	private File tmpDirFile;

	public MediaDownloadGetRequestExecutor() {
		super();
	}

	public MediaDownloadGetRequestExecutor(File tmpDirFile) {
		super();
		this.tmpDirFile = tmpDirFile;
	}

	@Override
	public File execute(CloseableHttpClient httpclient, String uri, Map params)
			throws WxErrorException, ClientProtocolException, IOException {
		if (params != null) {
			uri += '?';
			for(String key : params.keySet()){
				uri+=key+"="+params.get(key)+"&";
			}
			uri = uri.substring(0, uri.length()-1);
		}
		
		HttpGet httpGet = new HttpGet(uri);

		try (CloseableHttpResponse response = httpclient.execute(httpGet)) {
			System.out.println(response);
			Header[] contentTypeHeader = response.getHeaders("Content-Type");
			for(Header h : contentTypeHeader) {
                System.out.println(h);
            }
			if (contentTypeHeader != null && contentTypeHeader.length > 0) {
				// 下载媒体文件出错
				if (ContentType.TEXT_PLAIN.getMimeType().equals(contentTypeHeader[0].getValue())) {
					String responseContent = Utf8ResponseHandler.INSTANCE.handleResponse(response);
					throw new WxErrorException(WxError.fromJson(responseContent));
				}
			}
			InputStream inputStream = InputStreamResponseHandler.INSTANCE.handleResponse(response);
			String fileName = getFileName(response);
			if (StringUtils.isBlank(fileName)) {
				return null;
			}
			String[] name_ext = fileName.split("\\.");
			File localFile = FileUtils.createTmpFile(inputStream, name_ext[0], name_ext[name_ext.length-1], tmpDirFile);
			return localFile;
		}
	}

	protected String getFileName(CloseableHttpResponse response) {
		Header[] contentDispositionHeader = response.getHeaders("Content-disposition");
		Pattern p = Pattern.compile(".*filename=\"(.*)\"");
		Matcher m = p.matcher(contentDispositionHeader[0].getValue());
		m.matches();
		String fileName = m.group(m.groupCount());
		return fileName;
	}

}