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

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

import com.seejoke.wechat.entity.result.WxError;
import com.seejoke.wechat.entity.result.WxVideoMediaResult;
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.client.methods.HttpPost;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.codehaus.jackson.JsonNode;
import org.codehaus.jackson.map.ObjectMapper;

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

public class VideoDownloadPostRequestExecutor implements RequestExecutor {

	private File materialDirFile;

	public VideoDownloadPostRequestExecutor() {
		super();
	}

	public VideoDownloadPostRequestExecutor(File materialDirFile) {
		super();
		this.materialDirFile = materialDirFile;
	}

	@Override
	public WxVideoMediaResult execute(CloseableHttpClient httpclient, String uri, String params)
			throws WxErrorException, ClientProtocolException, IOException {
		WxVideoMediaResult result = null;
		
		HttpPost httpPost = new HttpPost(uri);
		if (params != null) {
			httpPost.setEntity(new StringEntity(params,"UTF-8"));
		}
		try (CloseableHttpResponse response = httpclient.execute(httpPost)) {
			String responseContent = Utf8ResponseHandler.INSTANCE.handleResponse(response);
			ObjectMapper mapper = new ObjectMapper();
			JsonNode node = mapper.readTree(responseContent);
			if(node.get("errcode")!=null && !(node.get("errcode").asInt()==0)){
				WxError error = WxError.fromJson(responseContent);
				throw new WxErrorException(error);
			}
			result = mapper.readValue(responseContent, WxVideoMediaResult.class);
		}
		
		HttpGet httpGet = new HttpGet(result.getDown_url());
		try (CloseableHttpResponse response = httpclient.execute(httpGet)) {
			Header[] contentTypeHeader = response.getHeaders("Content-Type");
			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("\\.");
			FileUtils.createMaterialFile(inputStream, name_ext[0], name_ext[name_ext.length-1], materialDirFile);
		}
		
		return result;
	}

	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(1);
		return fileName;
	}

}