package com.qianwen.smartman.common.utils;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import org.apache.http.Header;
import org.apache.http.HttpEntity;
import org.apache.http.client.CookieStore;
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.client.methods.HttpRequestBase;
import org.apache.http.client.protocol.HttpClientContext;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.BasicCookieStore;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.message.BasicHeader;
import org.apache.http.util.EntityUtils;
import org.assertj.core.util.Lists;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.qianwen.smartman.common.constant.DncConstant;

public class SimpleHttpClient {
    private static final Logger log = LoggerFactory.getLogger(SimpleHttpClient.class);
    private static final SimpleHttpClient simpleHttpClient = new SimpleHttpClient();
    private CookieStore cookieStore = new BasicCookieStore();
    private final List<Header> DEF_HEADER = Lists.newArrayList();
    private CloseableHttpClient http;

    public SimpleHttpClient() {
        this.DEF_HEADER.add(new BasicHeader("User-Agent", "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/61.0.3163.100 Safari/537.36"));
        this.DEF_HEADER.add(new BasicHeader("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9"));
        this.DEF_HEADER.add(new BasicHeader("Accept-Language", "zh-CN,zh;q=0.9"));
        this.http = HttpClients.custom().setDefaultHeaders(this.DEF_HEADER).setDefaultCookieStore(this.cookieStore).build();
    }

    public void setCookieStore(final CookieStore cookieStore) {
        this.cookieStore = cookieStore;
    }

    public void setHttp(final CloseableHttpClient http) {
        this.http = http;
    }

    public boolean equals(final Object o) {
        if (o == this) {
            return true;
        }
        if (o instanceof SimpleHttpClient) {
            SimpleHttpClient other = (SimpleHttpClient) o;
            if (other.canEqual(this)) {
                Object this$cookieStore = getCookieStore();
                Object other$cookieStore = other.getCookieStore();
                if (this$cookieStore == null) {
                    if (other$cookieStore != null) {
                        return false;
                    }
                } else if (!this$cookieStore.equals(other$cookieStore)) {
                    return false;
                }
                Object this$DEF_HEADER = getDEF_HEADER();
                Object other$DEF_HEADER = other.getDEF_HEADER();
                if (this$DEF_HEADER == null) {
                    if (other$DEF_HEADER != null) {
                        return false;
                    }
                } else if (!this$DEF_HEADER.equals(other$DEF_HEADER)) {
                    return false;
                }
                Object this$http = getHttp();
                Object other$http = other.getHttp();
                return this$http == null ? other$http == null : this$http.equals(other$http);
            }
            return false;
        }
        return false;
    }

    protected boolean canEqual(final Object other) {
        return other instanceof SimpleHttpClient;
    }

    public int hashCode() {
        Object $cookieStore = getCookieStore();
        int result = (1 * 59) + ($cookieStore == null ? 43 : $cookieStore.hashCode());
        Object $DEF_HEADER = getDEF_HEADER();
        int result2 = (result * 59) + ($DEF_HEADER == null ? 43 : $DEF_HEADER.hashCode());
        Object $http = getHttp();
        return (result2 * 59) + ($http == null ? 43 : $http.hashCode());
    }

    public String toString() {
        return "SimpleHttpClient(cookieStore=" + getCookieStore() + ", DEF_HEADER=" + getDEF_HEADER() + ", http=" + getHttp() + ")";
    }

    public static SimpleHttpClient instance() {
        return simpleHttpClient;
    }

    public CookieStore getCookieStore() {
        return this.cookieStore;
    }

    public List<Header> getDEF_HEADER() {
        return this.DEF_HEADER;
    }

    public CloseableHttpClient getHttp() {
        return this.http;
    }

    public static void download(String url, String filepath, int cache) throws Exception {
        InputStream inputStream = null;
        FileOutputStream fileOutputStream = null;
        try {
            try {
                HttpEntity entity = simpleHttpClient.execute(new HttpGet(url)).getEntity();
                inputStream = entity.getContent();
                File file = new File(filepath);
                file.getParentFile().mkdirs();
                fileOutputStream = new FileOutputStream(file);
                byte[] buffer = new byte[cache];
                while (true) {
                    int ch = inputStream.read(buffer);
                    if (ch == -1) {
                        break;
                    }
                    fileOutputStream.write(buffer, 0, ch);
                }
                if (inputStream != null) {
                    inputStream.close();
                }
                if (fileOutputStream != null) {
                    fileOutputStream.flush();
                    fileOutputStream.close();
                }
            } catch (Exception e) {
                e.printStackTrace();
                throw e;
            }
        } catch (Throwable th) {
            if (inputStream != null) {
                inputStream.close();
            }
            if (fileOutputStream != null) {
                fileOutputStream.flush();
                fileOutputStream.close();
            }
            throw th;
        }
    }

    public String get(String url, List<Header> headers) throws Exception {
        HttpGet httpget = new HttpGet(url);
        if (headers != null) {
            headers.forEach(header -> {
                httpget.addHeader(header);
            });
        }
        CloseableHttpResponse response = execute(httpget);
        try {
            String res = EntityUtils.toString(response.getEntity(), DncConstant.CHARSET);
            httpget.abort();
            response.close();
            return res;
        } catch (Throwable th) {
            response.close();
            throw th;
        }
    }

    private void addCookies(String host, CookieStore cookieStore) {
        cookieStore.getCookies().forEach(cookie -> {
            this.cookieStore.addCookie(cookie);
        });
    }

    public void close() {
        try {
            this.http.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public String post(String url, String json) throws Exception {
        CloseableHttpResponse response = null;
        try {
            HttpPost httppost = new HttpPost(url);
            System.out.println("执行请求：" + httppost.getURI());
            StringEntity s = new StringEntity(json);
            s.setContentEncoding(DncConstant.CHARSET);
            s.setContentType("application/json");
            httppost.setEntity(s);
            response = execute(httppost);
            String entityUtils = EntityUtils.toString(response.getEntity());
            if (response != null) {
                response.close();
            }
            return entityUtils;
        } catch (Throwable th) {
            if (response != null) {
                response.close();
            }
            throw th;
        }
    }

    public CloseableHttpResponse execute(HttpRequestBase url) {
		try {
			HttpClientContext context = HttpClientContext.create();
			CloseableHttpResponse response = this.http.execute(url, context);
			addCookies(url.getURI().getHost(), context.getCookieStore());
			return response;
		} catch (IOException ex) {
			log.error("执行http请求异常",ex);
			throw new RuntimeException(ex);
		}
    }

    
    public static class RangeInfo {
        String acceptRanges;
        String etag;
        String lastModified;
        Long contentLength;

        public void setAcceptRanges(final String acceptRanges) {
            this.acceptRanges = acceptRanges;
        }

        public void setEtag(final String etag) {
            this.etag = etag;
        }

        public void setLastModified(final String lastModified) {
            this.lastModified = lastModified;
        }

        public void setContentLength(final Long contentLength) {
            this.contentLength = contentLength;
        }

        public boolean equals(final Object o) {
            if (o == this) {
                return true;
            }
            if (o instanceof RangeInfo) {
                RangeInfo other = (RangeInfo) o;
                if (other.canEqual(this)) {
                    Object this$contentLength = getContentLength();
                    Object other$contentLength = other.getContentLength();
                    if (this$contentLength == null) {
                        if (other$contentLength != null) {
                            return false;
                        }
                    } else if (!this$contentLength.equals(other$contentLength)) {
                        return false;
                    }
                    Object this$acceptRanges = getAcceptRanges();
                    Object other$acceptRanges = other.getAcceptRanges();
                    if (this$acceptRanges == null) {
                        if (other$acceptRanges != null) {
                            return false;
                        }
                    } else if (!this$acceptRanges.equals(other$acceptRanges)) {
                        return false;
                    }
                    Object this$etag = getEtag();
                    Object other$etag = other.getEtag();
                    if (this$etag == null) {
                        if (other$etag != null) {
                            return false;
                        }
                    } else if (!this$etag.equals(other$etag)) {
                        return false;
                    }
                    Object this$lastModified = getLastModified();
                    Object other$lastModified = other.getLastModified();
                    return this$lastModified == null ? other$lastModified == null : this$lastModified.equals(other$lastModified);
                }
                return false;
            }
            return false;
        }

        protected boolean canEqual(final Object other) {
            return other instanceof RangeInfo;
        }

        public int hashCode() {
            Object $contentLength = getContentLength();
            int result = (1 * 59) + ($contentLength == null ? 43 : $contentLength.hashCode());
            Object $acceptRanges = getAcceptRanges();
            int result2 = (result * 59) + ($acceptRanges == null ? 43 : $acceptRanges.hashCode());
            Object $etag = getEtag();
            int result3 = (result2 * 59) + ($etag == null ? 43 : $etag.hashCode());
            Object $lastModified = getLastModified();
            return (result3 * 59) + ($lastModified == null ? 43 : $lastModified.hashCode());
        }

        public String toString() {
            return "SimpleHttpClient.RangeInfo(acceptRanges=" + getAcceptRanges() + ", etag=" + getEtag() + ", lastModified=" + getLastModified() + ", contentLength=" + getContentLength() + ")";
        }

        public RangeInfo(final String acceptRanges, final String etag, final String lastModified, final Long contentLength) {
            this.acceptRanges = acceptRanges;
            this.etag = etag;
            this.lastModified = lastModified;
            this.contentLength = contentLength;
        }

        public String getAcceptRanges() {
            return this.acceptRanges;
        }

        public String getEtag() {
            return this.etag;
        }

        public String getLastModified() {
            return this.lastModified;
        }

        public Long getContentLength() {
            return this.contentLength;
        }
    }
}