package com.myproject.util;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.myproject.exception.ApiException;
import org.apache.commons.httpclient.*;
import org.apache.commons.httpclient.cookie.CookiePolicy;
import org.apache.commons.httpclient.methods.GetMethod;
import org.apache.commons.httpclient.methods.PostMethod;
import org.apache.commons.httpclient.methods.StringRequestEntity;
import org.apache.commons.httpclient.methods.multipart.FilePart;
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 java.io.*;
import java.net.URL;
import java.net.URLConnection;
import java.util.*;
import java.util.Map.Entry;

/**
 * HTTP网络请求
 */
public class HttpHelper {
	// 编码方式
	private static final String CONTENT_CHARSET = "UTF-8";

	// 连接超时时间
	private static final int CONNECTION_TIMEOUT = 3000;

	// 读数据超时时间
	private static final int READ_DATA_TIMEOUT = 3000;

	// 设置User-Agent
	private static final String USER_AGENT = "SMD-BOX";

	public static String postJson(String url, Map<String, Object> params) throws ApiException {
		return postJson(url,params,null, "http");
	}

	public static String get(String url,HashMap<String, String> params) throws ApiException {
		return get(url,params,null, "http");
	}


	/**
	 * 向指定URL发送POST请求
	 * @param url
	 * @param paramMap
	 * @return 响应结果
	 */
	public static String postParam(String url, Map<String, Object> paramMap) throws ApiException {
		PrintWriter out = null;
		BufferedReader in = null;
		String result = "";
		try {
			URL realUrl = new URL(url);
			// 打开和URL之间的连接
			URLConnection conn = realUrl.openConnection();
			// 设置通用的请求属性
			conn.setRequestProperty("accept", "*/*");
			conn.setRequestProperty("connection", "Keep-Alive");
			conn.setRequestProperty("user-agent","Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)");
			// conn.setRequestProperty("Charset", "UTF-8");
			// 发送POST请求必须设置如下两行
			conn.setDoOutput(true);
			conn.setDoInput(true);
			// 获取URLConnection对象对应的输出流
			out = new PrintWriter(conn.getOutputStream());

			// 设置请求属性
			String param = "";
			if (paramMap != null && paramMap.size() > 0) {
				Iterator<String> ite = paramMap.keySet().iterator();
				while (ite.hasNext()) {
					String key = ite.next();// key
					Object valueObj = paramMap.get(key);
					String value = "";
					if(valueObj != null){
						if(valueObj instanceof Date){
							DateUtil.toDateString((Date)valueObj,"yyyyMMdd");
						}else{
							value = valueObj.toString();
						}
					}
					param += key + "=" + value + "&";
				}
				param = param.substring(0, param.length() - 1);
			}

			// 发送请求参数
			out.print(param);
			// flush输出流的缓冲
			out.flush();
			// 定义BufferedReader输入流来读取URL的响应
			in = new BufferedReader(
					new InputStreamReader(conn.getInputStream()));
			String line;
			while ((line = in.readLine()) != null) {
				result += line;
			}
		} catch (Exception e) {
			throw new ApiException("Request [" + url + "] failed:" + e.getMessage());
		}
		// 使用finally块来关闭输出流、输入流
		finally {
			try {
				if (out != null) {
					out.close();
				}
				if (in != null) {
					in.close();
				}
			} catch (IOException ex) {
				throw new ApiException("Request [" + url + "] close instream failed:" + ex.getMessage());
			}
		}
		return result;
	}

	/**
	 * 发送POST请求
	 * 
	 * @param url
	 *            请求URL地址
	 * @param params
	 *            请求参数
	 * @param protocol
	 *            请求协议 "http" / "https"
	 * @return 服务器响应的请求结果
	 */
	private static String postInBody(String url, HashMap<String, String> params,
			HashMap<String, String> cookies, String protocol) throws ApiException {
//		if (protocol.equalsIgnoreCase("https")) {
//			Protocol httpsProtocol = new Protocol("https", new SecureProtocolSocketFactoryImpl(), 443);
//			Protocol.registerProtocol("https", httpsProtocol);
//		}

		HttpClient httpClient = new HttpClient();
		PostMethod postMethod = new PostMethod(url);

		// 设置请求参数
		if (params != null && !params.isEmpty()) {
			NameValuePair[] data = new NameValuePair[params.size()];
			Iterator<Entry<String, String>> iterator = params.entrySet().iterator();
			int i = 0;
			while (iterator.hasNext()) {
				Entry<String, String> entry = iterator.next();
				data[i] = new NameValuePair(entry.getKey(), entry.getValue());
				++i;
			}
			postMethod.setRequestBody(data);
		}

		// 设置cookie
		if (cookies != null && !cookies.isEmpty()) {
			Iterator<Entry<String, String>> iterator = cookies.entrySet().iterator();
			StringBuilder buffer = new StringBuilder(128);
			while (iterator.hasNext()) {
				Entry<String, String> entry = iterator.next();
				buffer.append(entry.getKey()).append("=").append(entry.getValue()).append("; ");
			}
			// 设置cookie策略
			postMethod.getParams().setCookiePolicy(CookiePolicy.IGNORE_COOKIES);
			// 设置cookie内容
			postMethod.setRequestHeader("Cookie", buffer.toString());
		}

		// 设置User-Agent
		postMethod.setRequestHeader("User-Agent", USER_AGENT);
		// 设置建立连接超时时间
		httpClient.getHttpConnectionManager().getParams().setConnectionTimeout(CONNECTION_TIMEOUT);
		// 设置读数据超时时间
		httpClient.getHttpConnectionManager().getParams().setSoTimeout(READ_DATA_TIMEOUT);
		// 设置编码
		postMethod.getParams().setParameter(HttpMethodParams.HTTP_CONTENT_CHARSET, CONTENT_CHARSET);
		// 使用系统提供的默认的恢复策略
		postMethod.getParams().setParameter(HttpMethodParams.RETRY_HANDLER, new DefaultHttpMethodRetryHandler());

		try {
			try {
				int statusCode = httpClient.executeMethod(postMethod);
				if (statusCode != HttpStatus.SC_OK) {
					throw new ApiException("Request [" + url + "] failed:" + postMethod.getStatusLine());
				}
				// 读取内容
				byte[] responseBody = postMethod.getResponseBody();
				return new String(responseBody, CONTENT_CHARSET);
			} finally {
				// 释放链接
				postMethod.releaseConnection();
			}
		} catch (HttpException e) {
			// 发生致命的异常，可能是协议不对或者返回的内容有问题
			throw new ApiException("Request [" + url + "] failed:" + e.getMessage());
		} catch (IOException e) {
			// 发生网络异常
			throw new ApiException("Request [" + url + "] failed:" + e.getMessage());
		}
	}
	
	/**
	 * 发送Get请求
	 * 
	 * @param url
	 *            请求URL地址
	 * @param params
	 *            请求参数
	 * @param protocol
	 *            请求协议 "http" / "https"
	 * @return 服务器响应的请求结果
	 */
	public static String get(String url, HashMap<String, String> params,
			HashMap<String, String> cookies, String protocol) throws ApiException {
//		if (protocol.equalsIgnoreCase("https")) {
//			Protocol httpsProtocol = new Protocol("https", new SecureProtocolSocketFactoryImpl(), 443);
//			Protocol.registerProtocol("https", httpsProtocol);
//		}

		HttpClient httpClient = new HttpClient();
		GetMethod getMethod = new GetMethod(url);
//		PostMethod postMethod = new PostMethod(url);

		// 设置请求参数
		if (params != null && !params.isEmpty()) {
			NameValuePair[] data = new NameValuePair[params.size()];
			Iterator<Entry<String, String>> iterator = params.entrySet().iterator();
			int i = 0;
			while (iterator.hasNext()) {
				Entry<String, String> entry = iterator.next();
				data[i] = new NameValuePair(entry.getKey(), entry.getValue());
				++i;
			}
			getMethod.setQueryString(data);
		}

		// 设置cookie
		if (cookies != null && !cookies.isEmpty()) {
			Iterator<Entry<String, String>> iterator = cookies.entrySet().iterator();
			StringBuilder buffer = new StringBuilder(128);
			while (iterator.hasNext()) {
				Entry<String, String> entry = iterator.next();
				buffer.append(entry.getKey()).append("=").append(entry.getValue()).append("; ");
			}
			// 设置cookie策略
			getMethod.getParams().setCookiePolicy(CookiePolicy.IGNORE_COOKIES);
			// 设置cookie内容
			getMethod.setRequestHeader("Cookie", buffer.toString());
		}

		// 设置User-Agent
		getMethod.setRequestHeader("User-Agent", USER_AGENT);
		// 设置建立连接超时时间
		httpClient.getHttpConnectionManager().getParams().setConnectionTimeout(CONNECTION_TIMEOUT);
		// 设置读数据超时时间
		httpClient.getHttpConnectionManager().getParams().setSoTimeout(READ_DATA_TIMEOUT);
		// 设置编码
		getMethod.getParams().setParameter(HttpMethodParams.HTTP_CONTENT_CHARSET, CONTENT_CHARSET);
		// 使用系统提供的默认的恢复策略
		getMethod.getParams().setParameter(HttpMethodParams.RETRY_HANDLER, new DefaultHttpMethodRetryHandler());

		try {
			try {
				int statusCode = httpClient.executeMethod(getMethod);
				if (statusCode != HttpStatus.SC_OK) {
					throw new ApiException("Request [" + url + "] failed:" + getMethod.getStatusLine());
				}
				// 读取内容
				byte[] responseBody = getMethod.getResponseBody();
				return new String(responseBody, CONTENT_CHARSET);
			} finally {
				// 释放链接
				getMethod.releaseConnection();
			}
		} catch (HttpException e) {
			// 发生致命的异常，可能是协议不对或者返回的内容有问题
			throw new ApiException("Request [" + url + "] failed:" + e.getMessage());
		} catch (IOException e) {
			// 发生网络异常
			throw new ApiException("Request [" + url + "] failed:" + e.getMessage());
		}
	}

	/**
	 * POST提交JSON格式的内容
	 */
	private static String postJson(String url, Map<String, Object> params,
			HashMap<String, String> cookies, String protocol) throws ApiException {
//		if (protocol.equalsIgnoreCase("https")) {
//			Protocol httpsProtocol = new Protocol("https", new SecureProtocolSocketFactoryImpl(), 443);
//			Protocol.registerProtocol("https", httpsProtocol);
//		}

		HttpClient httpClient = new HttpClient();
		PostMethod postMethod = new PostMethod(url);

		// 设置请求参数
		if (params != null && !params.isEmpty()) {
			ObjectMapper mapper = new ObjectMapper();
			try {
				String requestBody = mapper.writeValueAsString(params);
				StringRequestEntity requestEntity = new StringRequestEntity(
						requestBody, "text/json", CONTENT_CHARSET);
				postMethod.setRequestEntity(requestEntity);
			} catch (JsonProcessingException e) {
				throw new ApiException("Request [" + url + "] failed:" + e.getMessage());
			} catch (UnsupportedEncodingException e) {
				throw new ApiException("Request [" + url + "] failed:" + e.getMessage());
			}
		}

		// 设置cookie
		if (cookies != null && !cookies.isEmpty()) {
			Iterator<Entry<String, String>> iterator = cookies.entrySet().iterator();
			StringBuilder buffer = new StringBuilder(128);
			while (iterator.hasNext()) {
				Entry<String, String> entry = iterator.next();
				buffer.append(entry.getKey()).append("=").append(entry.getValue()).append("; ");
			}
			// 设置cookie策略
			postMethod.getParams().setCookiePolicy(CookiePolicy.IGNORE_COOKIES);
			// 设置cookie内容
			postMethod.setRequestHeader("Cookie", buffer.toString());
		}

		// 设置User-Agent
		postMethod.setRequestHeader("User-Agent", USER_AGENT);
		// 设置建立连接超时时间
		httpClient.getHttpConnectionManager().getParams().setConnectionTimeout(CONNECTION_TIMEOUT);
		// 设置读数据超时时间
		httpClient.getHttpConnectionManager().getParams().setSoTimeout(READ_DATA_TIMEOUT);
		// 设置编码
		postMethod.getParams().setParameter(HttpMethodParams.HTTP_CONTENT_CHARSET, CONTENT_CHARSET);
		// 使用系统提供的默认的恢复策略
		postMethod.getParams().setParameter(HttpMethodParams.RETRY_HANDLER, new DefaultHttpMethodRetryHandler());

		try {
			try {
				int statusCode = httpClient.executeMethod(postMethod);
				if (statusCode != HttpStatus.SC_OK) {
					throw new ApiException("Request [" + url + "] failed:" + postMethod.getStatusLine());
				}
				// 读取内容
				byte[] responseBody = postMethod.getResponseBody();
				return new String(responseBody, CONTENT_CHARSET);
			} finally {
				// 释放链接
				postMethod.releaseConnection();
			}
		} catch (HttpException e) {
			// 发生致命的异常，可能是协议不对或者返回的内容有问题
			throw new ApiException("Request [" + url + "] failed:" + e.getMessage());
		} catch (IOException e) {
			// 发生网络异常
			throw new ApiException("Request [" + url + "] failed:" + e.getMessage());
		}
	}

	/**
	 * 发送POST请求（上传文件）
	 * 
	 * @param url
	 *            请求URL地址
	 * @param params
	 *            请求参数
	 * @param protocol
	 *            请求协议 "http" / "https"
	 * @param fp
	 *            上传的文件
	 * @return 服务器响应的请求结果
	 */
	private static String postWithFile(String url,
			HashMap<String, String> params, HashMap<String, String> cookies,
			FilePart fp, String protocol) throws ApiException {
//		if (protocol.equalsIgnoreCase("https")) {
//			Protocol httpsProtocol = new Protocol("https", new SecureProtocolSocketFactoryImpl(), 443);
//			Protocol.registerProtocol("https", httpsProtocol);
//		}

		HttpClient httpClient = new HttpClient();
		PostMethod postMethod = new PostMethod(url);

		List<Part> parts = new ArrayList<Part>();
		parts.add(fp);

		// 设置请求参数
		if (params != null && !params.isEmpty()) {
			Iterator<Entry<String, String>> iterator = params.entrySet().iterator();
			while (iterator.hasNext()) {
				Entry<String, String> entry = iterator.next();
				// 设置参数为UTF-8编码集，解决中文乱码问题，并且将参数加到post中
				parts.add(new StringPart(entry.getKey(), entry.getValue(), CONTENT_CHARSET));
			}
		}
		MultipartRequestEntity requestEntity = new MultipartRequestEntity(
				(Part[]) parts.toArray(), postMethod.getParams());
		postMethod.setRequestEntity(requestEntity);

		// 设置cookie
		if (cookies != null && !cookies.isEmpty()) {
			Iterator<Entry<String, String>> iterator = cookies.entrySet().iterator();
			StringBuilder buffer = new StringBuilder(128);
			while (iterator.hasNext()) {
				Entry<String, String> entry = iterator.next();
				buffer.append(entry.getKey()).append("=").append(entry.getValue()).append("; ");
			}
			// 设置cookie策略
			postMethod.getParams().setCookiePolicy(CookiePolicy.IGNORE_COOKIES);
			// 设置cookie内容
			postMethod.setRequestHeader("Cookie", buffer.toString());
		}

		// 设置User-Agent
		postMethod.setRequestHeader("User-Agent", USER_AGENT);
		// 设置建立连接超时时间
		httpClient.getHttpConnectionManager().getParams().setConnectionTimeout(CONNECTION_TIMEOUT);
		// 设置读数据超时时间
		httpClient.getHttpConnectionManager().getParams().setSoTimeout(READ_DATA_TIMEOUT);
		// 使用系统提供的默认的恢复策略
		postMethod.getParams().setParameter(HttpMethodParams.RETRY_HANDLER, new DefaultHttpMethodRetryHandler());

		// 发送请求
		try {
			try {
				postMethod.getParams().setContentCharset("UTF-8");
				int statusCode = httpClient.executeMethod(postMethod);
				if (statusCode != HttpStatus.SC_OK) {
					throw new ApiException("Request [" + url + "] failed:" + postMethod.getStatusLine());
				}
				// 读取内容
				byte[] responseBody = postMethod.getResponseBody();
				return new String(responseBody, CONTENT_CHARSET);
			} finally {
				// 释放链接
				postMethod.releaseConnection();
			}
		} catch (HttpException e) {
			// 发生致命的异常，可能是协议不对或者返回的内容有问题
			throw new ApiException("Request [" + url + "] failed:" + e.getMessage());
		} catch (IOException e) {
			// 发生网络异常
			throw new ApiException("Request [" + url + "] failed:" + e.getMessage());
		}
	}
}