package cc.mrbird.febs.mall.chain.trcCoin;

import okhttp3.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.Map;
import java.util.Objects;

public class OkHttpUtil2 {
    protected final static Logger logger = LoggerFactory.getLogger(OkHttpUtil2.class);
    private static Dispatcher dispatcher = new Dispatcher();

    static {
        dispatcher.setMaxRequests(200);
        dispatcher.setMaxRequestsPerHost(100);
    }

    private static OkHttpClient httpClient = new OkHttpClient.Builder().dispatcher(dispatcher).build();

    public static byte[] doGet(String url, Map<String, String[]> header, Map<String, String[]> params, String expectContentType) {
        Request.Builder builder = new Request.Builder();
        addHeader(builder, header);
        addUrlParam(builder, url, params);
        return requestExec(builder.build(), expectContentType);
    }

    public static byte[] doGetSingle(String url, Map<String, String> header, Map<String, String> params, String expectContentType) {
        Request.Builder builder = new Request.Builder();
        addHeaderSingle(builder, header);
        addUrlParamSingle(builder, url, params);
        return requestExec(builder.build(), expectContentType);
    }

    public static byte[] doPost(String url, Map<String, String[]> header, Map<String, String[]> body, String expectContentType) {
        Request.Builder builder = new Request.Builder().url(url);
        addHeader(builder, header);
        addBodyParam(builder, body, "POST");
        return requestExec(builder.build(), expectContentType);
    }



    private static void addHeaderSingle(Request.Builder builder, Map<String, String> header) {
        if (header == null) {
            return;
        }
        for (String key : header.keySet()) {
            String value = header.get(key);
            if (value != null) {
                builder.addHeader(key, value);
            }
        }
    }

    private static void addHeader(Request.Builder builder, Map<String, String[]> header) {
        if (header == null) {
            return;
        }
        for (String key : header.keySet()) {
            String[] values = header.get(key);
            if (values != null) {
                for (String value : values) {
                    builder.addHeader(key, value);
                }
            }
        }
    }

    private static void addUrlParam(Request.Builder builder, String url, Map<String, String[]> params) {
        if (params == null) {
            return;
        }
        HttpUrl.Builder urlBuilder = HttpUrl.parse(url).newBuilder();
        for (String key : params.keySet()) {
            String[] values = params.get(key);
            if (values != null) {
                for (String value : values) {
                    urlBuilder.addQueryParameter(key, value);
                }
            }
        }
        builder.url(urlBuilder.build());
    }

    private static void addUrlParamSingle(Request.Builder builder, String url, Map<String, String> params) {
        if (params == null) {
            return;
        }
        HttpUrl.Builder urlBuilder = HttpUrl.parse(url).newBuilder();
        for (String key : params.keySet()) {
            String value = params.get(key);
            if (value != null) {
                urlBuilder.addQueryParameter(key, value);
            }
        }
        builder.url(urlBuilder.build());
    }

    private static void addBodyParam(Request.Builder builder, Map<String, String[]> body, String method) {
        if (body == null) {
            return;
        }
        FormBody.Builder formBodyBuilder = new FormBody.Builder(StandardCharsets.UTF_8);
        for (String key : body.keySet()) {
            String[] values = body.get(key);
            if (values != null) {
                for (String value : values) {
                    formBodyBuilder.add(key, value);
                }
            }
        }
        builder.method(method, formBodyBuilder.build());
    }

    private static byte[] requestExec(Request request, String expectContentType) {
        Objects.requireNonNull(request, "okHttp request is null");

        try (Response response = httpClient.newCall(request).execute()) {
            if (response.code() == 200) {
                ResponseBody body = response.body();
                if (body != null) {
                    byte[] bytes = body.bytes();
                    String contentType = response.header("Content-Type");
                    if (contentType != null && !contentType.contains(expectContentType)) {
                        String res = new String(bytes, StandardCharsets.UTF_8);
                        System.out.println(res);
                        return bytes;
                    }
                    return bytes;
                }
                logger.error("response body is null");
                System.out.println("response body is null");
            } else {
                ResponseBody body = response.body();
                String res = "";
                byte[] bytes = null;
                if (body != null) {
                    bytes = body.bytes();
                    String contentType = response.header("Content-Type");
                    if (contentType != null && !contentType.contains(expectContentType)) {
                        //res = new String(body.bytes(), StandardCharsets.UTF_8);
                        //return body.bytes();
                    }
                    res = new String(bytes, StandardCharsets.UTF_8);

                }
                logger.error("request failed, http code:{}，responseBody:{} ", response.code(), res);
                System.out.println("request failed, http code: " + response.code());

                return bytes;
            }
        } catch (IOException ioException) {
            logger.error("request exec error:", ioException);
            System.out.println("request exec error: " + ioException.getMessage());
        }
        return null;
    }
}