package com.qianxunclub.ticket.util;


import com.qianxunclub.ticket.config.Config;
import org.apache.http.HttpEntity;
import org.apache.http.HttpHeaders;
import org.apache.http.HttpHost;
import org.apache.http.HttpResponse;
import org.apache.http.HttpStatus;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpRequestBase;
import org.apache.http.impl.client.BasicCookieStore;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.impl.conn.DefaultProxyRoutePlanner;
import org.apache.http.message.BasicHeader;
import org.apache.http.util.EntityUtils;

import java.io.IOException;
import java.util.concurrent.CompletableFuture;

import lombok.extern.slf4j.Slf4j;

/**
 * @author zhangbin
 * @date 2019-05-30 16:34
 * @description: TODO
 */
@Slf4j
public class HttpUtil {

    private HttpClient httpClient;
    private HttpClient httpPureClient;
    private BasicCookieStore basicCookieStore;
    private Config config;

    public HttpUtil() {
        this.init();
    }

    public HttpUtil(BasicCookieStore basicCookieStore) {
        init(basicCookieStore);
    }

    public void init() {
        BasicCookieStore basicCookieStore = new BasicCookieStore();
        this.init(basicCookieStore);
    }

    public void init(BasicCookieStore basicCookieStore) {
        this.basicCookieStore = basicCookieStore;
        this.config = ApplicationContextHelper.getBean(Config.class);
        httpClient = HttpClients.custom().setDefaultCookieStore(basicCookieStore).build();
        if (config != null && config.getEnableProxy()) {
            HttpHost proxy = new HttpHost(config.getProxyIp().getIp(), config.getProxyIp().getPort());
            DefaultProxyRoutePlanner routePlanner = new DefaultProxyRoutePlanner(proxy);
            httpClient = HttpClients.custom().setRoutePlanner(routePlanner)
                    .setDefaultCookieStore(basicCookieStore).build();
        }
        httpPureClient = HttpClients.createDefault();
    }

    public BasicCookieStore getBasicCookieStore() {
        return basicCookieStore;
    }

    public String get(HttpGet httpGet) {
        return this.doAction(httpGet,httpClient);
    }


    public String post(HttpPost httpPost) {
        return this.doAction(httpPost,httpClient);
    }

    private String doAction(HttpRequestBase httpRequestBase,HttpClient client) {
        httpRequestBase.addHeader(new BasicHeader("Origin",config.getBaseUrl()));
        httpRequestBase.addHeader(new BasicHeader(HttpHeaders.REFERER,config.getBaseUrl()));
        httpRequestBase.addHeader(new BasicHeader(HttpHeaders.HOST, config.getHost()));
        return sendRequest(httpRequestBase,client);
    }

    /**
     * 发送 http 请求
     * @param httpRequestBase request
     * @param client http client
     * @return response
     */
    private String sendRequest(HttpRequestBase httpRequestBase,HttpClient client){
        String result = null;
        try {
            HttpResponse response = client.execute(httpRequestBase);
            if (response.getStatusLine().getStatusCode() == HttpStatus.SC_OK) {
                HttpEntity httpEntity = response.getEntity();
                result = EntityUtils.toString(httpEntity, "UTF-8");
                EntityUtils.consume(httpEntity);
            } else {
                log.error("请求异常：{},{}", httpRequestBase.getURI(), response.getStatusLine());
            }
            return result;
        } catch (IOException e) {
            log.error("请求异常：{},{}", httpRequestBase.getURI(), e.getMessage());
        }
        return "";
    }

    /**
     * 异步http Get请求 无12306cookies植入等 纯净版
     * @param httpGet 请求参数
     * @return future response
     */
    public CompletableFuture<String> asyncGet(HttpGet httpGet){
        return CompletableFuture.supplyAsync(()->sendRequest(httpGet,httpPureClient));
    }

    /**
     * 异步http Post请求 无12306cookies植入等 纯净版
     * @param httpPost 请求参数
     * @return future response
     */
    public CompletableFuture<String> asyncPost(HttpPost httpPost){
        return CompletableFuture.supplyAsync(()->sendRequest(httpPost,httpPureClient));
    }
}