package com.honghu.cloud.tools;

import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;

import org.apache.http.Header;
import org.apache.http.NameValuePair;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.util.EntityUtils;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;

import com.alibaba.fastjson.JSON;
import com.google.common.collect.Maps;
import com.squareup.okhttp.FormEncodingBuilder;
import com.squareup.okhttp.Headers;
import com.squareup.okhttp.MediaType;
import com.squareup.okhttp.OkHttpClient;
import com.squareup.okhttp.Request;
import com.squareup.okhttp.RequestBody;
import com.squareup.okhttp.Response;

@Component
public class OkHttpUtil implements InitializingBean {
    private static int DEFAULT_READ_TIMEOUT_MILLIS = 500;
    private static int DEFAULT_CONNECT_TIMEOUT_MILLIS = 500;
    private static int DEFAULT_WRITE_TIMEOUT_MILLIS = 500;
    public static final MediaType JSONTYPE = MediaType.parse("application/json; charset=utf-8");
    private OkHttpClient client;
    /**
     * get请求
     *
     * @param url
     * @return
     */
    public static String httpGet(String url) {
        String result = null;
        OkHttpClient client = new OkHttpClient();
        Request request = new Request.Builder().url(url).build();
        try {
            Response response = client.newCall(request).execute();
            result = response.body().string();
        } catch (Exception e) {
            e.printStackTrace();
        }
        return result;
    }
     /**
     * 处理get请求，兼容原httpClient请求
     * @param baseUrl
     * @param url
     * @param nvp
     * @param headers
     * @param cookies
     * @return
     * @throws Exception
     */
    public Map<String , Object> doGet(String baseUrl, String url, List<NameValuePair> nvp, List<Header> headers)
            throws Exception {
        //Request build对象，get请求没有method方法，默认是PUT方式请求，直接在URL做参数处理
        Request.Builder builder = getBuilder(baseUrl, url + "?" + EntityUtils.toString(new UrlEncodedFormEntity(nvp, "UTF-8")));
        builder.get();
        //处理headers
        doHeaders(headers, builder);
        //创建Request
        Request request = builder.build();
        //同步请求并获取Response
        Response response = client.newCall(request).execute();
        //校验并返回
        Map<String, Object> maps = Maps.newHashMap();
         if (response.isSuccessful()) {
             maps.put("code", 200);
             maps.put("data", response.body().string());
            return maps;
        } else {
             maps.put("code", 500);
             maps.put("data", response.body().string());
           return maps;
        }
    }
     /**
     * 处理post请求，兼容原httpClient请求
     * @param baseUrl
     * @param url
     * @param nvp
     * @param headers
     * @return
     * @throws Exception
     */
    public Map<String , Object> doPost(String baseUrl, String url, List<NameValuePair> nvp , String type , Header... headers)
            throws Exception {
        //Request build对象
        Request.Builder builder = getBuilder(baseUrl, url);
        //处理headers
        doHeaders(builder, headers);
        //处理请求参数
        doPostRequestBody(builder, nvp,type);
        //创建Request
        Request request = builder.build();
        //同步请求并获取Response
        Response response = client.newCall(request).execute();
        //校验并返回
        Map<String, Object> maps = Maps.newHashMap();
        if (response.isSuccessful()) {
            maps.put("code", 200);
            maps.put("data", response.body().string());
            return maps;
        } else {
           maps.put("code", 500);
           maps.put("message", response.body().string());
           return maps;
        }
    }
    /**
     * post请求参数写入Request.Builder
     * @param builder
     * @param nvp
     */
    private void doPostRequestBody(Request.Builder builder, List<NameValuePair> nvp,String type) {
        RequestBody body =null;
        if("json".equals(type)){
            Map<String, Object> maps = Maps.newHashMap();
             for (NameValuePair bean : nvp) {
                 maps.put(bean.getName(), bean.getValue());
             }
            body = RequestBody.create(JSONTYPE, JSON.toJSONString(maps));
            builder.post(body);
        }else if("form".equals(type)){
            FormEncodingBuilder form = new FormEncodingBuilder();
                if (!CollectionUtils.isEmpty(nvp)) {
                    for (NameValuePair bean : nvp) {
                        form.add(bean.getName(), bean.getValue());
                    }
                    body= form.build();
                }
        }
        builder.post(body);
    }
    /**
     * 获取Request.Builder
     * @param baseUrl
     * @param url
     * @return
     */
    private Request.Builder getBuilder(String baseUrl, String url) {
        return new Request.Builder().url(String.format("%s%s", baseUrl, url));
    }

 
 
    /**
     * 处理请求headers
     * @param headers
     * @param builder
     */
    private void doHeaders(List<Header> headers, Request.Builder builder) {
        if (!CollectionUtils.isEmpty(headers)) {
            Headers.Builder headeBuilder = new Headers.Builder();
            for (Header header : headers) {
                headeBuilder.add(header.getName(), header.getValue());
            }
            builder.headers(headeBuilder.build());
        }
    }
 
    /**
     * 处理请求headers
     * @param headers
     * @param builder
     */
    private void doHeaders(Request.Builder builder, Header[] headers) {
        if (headers != null && headers.length > 0) {
            Headers.Builder headeBuilder = new Headers.Builder();
            for (Header header : headers) {
                headeBuilder.add(header.getName(), header.getValue());
            }
            builder.headers(headeBuilder.build());
        }
    }
    @Override
    public void afterPropertiesSet() throws Exception {
         client = new OkHttpClient();
         client.setConnectTimeout(DEFAULT_CONNECT_TIMEOUT_MILLIS, TimeUnit.MILLISECONDS);
         client.setReadTimeout(DEFAULT_READ_TIMEOUT_MILLIS, TimeUnit.MILLISECONDS);    
         client.setWriteTimeout(DEFAULT_WRITE_TIMEOUT_MILLIS, TimeUnit.MILLISECONDS);
    }
}