package com.coloryr.allmusic.server.netapi;

import com.coloryr.allmusic.server.core.AllMusic;
import com.coloryr.allmusic.server.core.music.MusicHttpClient;
import com.coloryr.allmusic.server.core.objs.HttpResObj;
import com.coloryr.allmusic.server.netapi.obj.EncResObj;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import org.apache.hc.client5.http.classic.methods.HttpGet;
import org.apache.hc.client5.http.classic.methods.HttpPost;
import org.apache.hc.client5.http.cookie.Cookie;
import org.apache.hc.client5.http.cookie.CookieStore;
import org.apache.hc.client5.http.entity.UrlEncodedFormEntity;
import org.apache.hc.client5.http.impl.classic.CloseableHttpResponse;
import org.apache.hc.client5.http.protocol.HttpClientContext;
import org.apache.hc.core5.http.HttpEntity;
import org.apache.hc.core5.http.NameValuePair;
import org.apache.hc.core5.http.io.entity.EntityUtils;
import org.apache.hc.core5.http.message.BasicNameValuePair;

import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.net.URI;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;

public class NetApiHttpClient {

    public static HttpResObj get(String path, String data) {
        try {
            data = URLEncoder.encode(data, StandardCharsets.UTF_8);
            HttpGet request = new HttpGet(path + data);
            request.setHeader("referer", "https://music.163.com");
            request.setHeader("content-type", "application/json;charset=UTF-8");
            request.setHeader("user-agent", MusicHttpClient.UserAgent);
            HttpClientContext context = HttpClientContext.create();
            CookieStore cookieStore = MusicHttpClient.createCookieStore();
            context.setCookieStore(cookieStore);
            try (CloseableHttpResponse response = MusicHttpClient.client.execute(request, context)) {
                int httpCode = response.getCode();
                HttpEntity entity = response.getEntity();
                if (entity == null) {
                    AllMusic.log.warning("<light_purple>[AllMusic3]<red>获取网页错误");
                    return null;
                }
                InputStream inputStream = entity.getContent();
                boolean ok = httpCode == 200;
                ByteArrayOutputStream result = new ByteArrayOutputStream();
                byte[] buffer = new byte[4096];
                int length;
                while ((length = inputStream.read(buffer)) != -1) {
                    result.write(buffer, 0, length);
                }
                inputStream.close();
                EntityUtils.consume(entity);
                String data1 = result.toString(StandardCharsets.UTF_8);
                if (!ok) {
                    AllMusic.log.warning("<light_purple>[AllMusic3]<red>服务器返回错误：" + data1);
                }
                // 保存 cookies
                MusicHttpClient.saveCookies(cookieStore);
                return new HttpResObj(data1, ok);
            }
        } catch (Exception e) {
            AllMusic.log.warning("<light_purple>[AllMusic3]<red>获取网页错误");
            e.printStackTrace();
        }
        return null;
    }

    public static HttpResObj post(String url, JsonObject data, EncryptType type, String ourl) {
        try {
            HttpPost request = new HttpPost(url);
            request.setHeader("Content-Type", "application/x-www-form-urlencoded");
            request.setHeader("Referer", "https://music.163.com");
            HttpClientContext context = HttpClientContext.create();
            CookieStore cookieStore = MusicHttpClient.createCookieStore();
            context.setCookieStore(cookieStore);
            EncResObj res;
            List<Cookie> cookies = cookieStore.getCookies();
            // 注意：这里需要根据域名过滤，但为了简化，我们使用所有 cookies
            if (type == EncryptType.WEAPI) {
                request.setHeader("User-Agent", MusicHttpClient.UserAgent);
                String csrfToken = "";
                for (Cookie cookie : cookies) {
                    if (cookie.getName().equalsIgnoreCase("__csrf")) {
                        csrfToken = cookie.getValue();
                    }
                }
                data.addProperty("csrf_token", csrfToken);
                res = CryptoUtil.weapiEncrypt(AllMusic.gson.toJson(data));
                url = url.replaceFirst("\\w*api", "weapi");
                request.setUri(new URI(url));
                List<NameValuePair> params = new ArrayList<>();
                params.add(new BasicNameValuePair("params", res.params));
                params.add(new BasicNameValuePair("encSecKey", res.encSecKey));
                request.setEntity(new UrlEncodedFormEntity(params));
            } else if (type == EncryptType.EAPI) {
                request.setHeader("User-Agent", "Mozilla/5.0 (Linux; Android 9; PCT-AL10) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.64 HuaweiBrowser/10.0.3.311 Mobile Safari/537.36");
                JsonObject header = new JsonObject();
                header.addProperty("appver", "8.10.90");
                header.addProperty("versioncode", "140");
                header.addProperty("buildver", new Date().toString().substring(0, 10));
                header.addProperty("resolution", "1920x1080");
                header.addProperty("os", "android");
                String requestId = "0000" + (new Date() + "_" + Math.floor(Math.random() * 1000));
                header.addProperty("requestId", requestId);
                for (Cookie cookie : cookies) {
                    if (cookie.getName().equalsIgnoreCase("MUSIC_U")) {
                        header.addProperty("MUSIC_U", cookie.getValue());
                    } else if (cookie.getName().equalsIgnoreCase("MUSIC_A")) {
                        header.addProperty("MUSIC_A", cookie.getValue());
                    } else if (cookie.getName().equalsIgnoreCase("channel")) {
                        header.addProperty("channel", cookie.getValue());
                    } else if (cookie.getName().equalsIgnoreCase("mobilename")) {
                        header.addProperty("mobilename", cookie.getValue());
                    } else if (cookie.getName().equalsIgnoreCase("osver")) {
                        header.addProperty("osver", cookie.getValue());
                    } else if (cookie.getName().equalsIgnoreCase("__csrf")) {
                        header.addProperty("__csrf", cookie.getValue());
                    }
                }
                data.add("header", header);
                res = CryptoUtil.eapi(ourl, data);
                url = url.replaceFirst("\\w*api", "eapi");
                request.setUri(new URI(url));
                List<NameValuePair> params = new ArrayList<>();
                params.add(new BasicNameValuePair("params", res.params));
                request.setEntity(new UrlEncodedFormEntity(params));
            } else {
                request.setUri(new URI(url));
                List<NameValuePair> params = new ArrayList<>();
                for (Map.Entry<String, JsonElement> item : data.entrySet()) {
                    params.add(new BasicNameValuePair(item.getKey(), item.getValue().getAsString()));
                }
                request.setEntity(new UrlEncodedFormEntity(params));
            }
            try (CloseableHttpResponse response = MusicHttpClient.client.execute(request, context)) {
                int httpCode = response.getCode();
                HttpEntity entity = response.getEntity();
                if (entity == null) {
                    AllMusic.log.warning("<light_purple>[AllMusic3]<red>获取网页错误");
                    return null;
                }
                InputStream inputStream = entity.getContent();
                boolean ok = httpCode == 200;
                ByteArrayOutputStream result = new ByteArrayOutputStream();
                byte[] buffer = new byte[4096];
                int length;
                while ((length = inputStream.read(buffer)) != -1) {
                    result.write(buffer, 0, length);
                }
                String data1 = result.toString(StandardCharsets.UTF_8);
                EntityUtils.consume(entity);
                if (!ok) {
                    AllMusic.log.warning("<light_purple>[AllMusic3]<red>服务器返回错误：" + data1);
                }
                // 保存 cookies
                MusicHttpClient.saveCookies(cookieStore);
                return new HttpResObj(data1, ok);
            }
        } catch (Exception e) {
            AllMusic.log.warning("<light_purple>[AllMusic3]<red>获取网页错误");
            e.printStackTrace();
        }
        return null;
    }
}