package cc.polyfrost.oneconfig.utils;

import com.google.gson.JsonElement;
import gg.essential.universal.UDesktop;
import org.apache.commons.io.IOUtils;

import java.io.*;
import java.net.HttpURLConnection;
import java.net.URI;
import java.net.URL;
import java.nio.charset.StandardCharsets;
import java.security.MessageDigest;

/**
 * Utility class for accessing the internet.
 */
public final class NetworkUtils {

    /**
     * Gets the contents of a URL as a String.
     *
     * @param url       The URL to read.
     * @param userAgent The user agent to use.
     * @param timeout   The timeout in milliseconds.
     * @param useCaches Whether to use caches.
     * @return The contents of the URL.
     */
    public static String getString(String url, String userAgent, int timeout, boolean useCaches) {
        try (InputStreamReader input = new InputStreamReader(setupConnection(url, userAgent, timeout, useCaches), StandardCharsets.UTF_8)) {
            return IOUtils.toString(input);
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    /**
     * Gets the contents of a URL as a String.
     *
     * @param url The URL to read.
     * @return The contents of the URL.
     * @see NetworkUtils#getString(String, String, int, boolean)
     */
    public static String getString(String url) {
        return getString(url, "OneConfig/1.0.0", 5000, false);
    }

    /**
     * Gets the contents of a URL as a JsonElement.
     *
     * @param url       The URL to read.
     * @param userAgent The user agent to use.
     * @param timeout   The timeout in milliseconds.
     * @param useCaches Whether to use caches.
     * @return The contents of the URL.
     * @see NetworkUtils#getString(String, String, int, boolean)
     * @see JsonUtils#parseString(String)
     */
    public static JsonElement getJsonElement(String url, String userAgent, int timeout, boolean useCaches) {
        return JsonUtils.parseString(getString(url, userAgent, timeout, useCaches));
    }

    /**
     * Gets the contents of a URL as a JsonElement.
     *
     * @param url The URL to read.
     * @return The contents of the URL.
     * @see NetworkUtils#getJsonElement(String, String, int, boolean)
     */
    public static JsonElement getJsonElement(String url) {
        return getJsonElement(url, "OneConfig/1.0.0", 5000, false);
    }

    /**
     * Downloads a file from a URL.
     *
     * @param url       The URL to download from.
     * @param file      The file to download to.
     * @param userAgent The user agent to use.
     * @param timeout   The timeout in milliseconds.
     * @param useCaches Whether to use caches.
     * @return Whether the download was successful.
     */
    public static boolean downloadFile(String url, File file, String userAgent, int timeout, boolean useCaches) {
        url = url.replace(" ", "%20");
        try (FileOutputStream fileOut = new FileOutputStream(file); BufferedInputStream in = new BufferedInputStream(setupConnection(url, userAgent, timeout, useCaches))) {
            IOUtils.copy(in, fileOut);
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
        return true;
    }

    /**
     * Downloads a file from a URL.
     *
     * @param url  The URL to download from.
     * @param file The file to download to.
     * @return Whether the download was successful.
     * @see NetworkUtils#downloadFile(String, File, String, int, boolean)
     */
    public static boolean downloadFile(String url, File file) {
        return downloadFile(url, file, "OneConfig/1.0.0", 5000, false);
    }

    /**
     * Gets the SHA-256 hash of a file.
     *
     * @param file The file to hash.
     * @return The SHA-256 hash of the file.
     */
    public static String getFileChecksum(File file) {
        try (FileInputStream inputStream = new FileInputStream(file)) {
            MessageDigest digest = MessageDigest.getInstance("SHA-256");
            byte[] bytesBuffer = new byte[1024];
            int bytesRead;

            while ((bytesRead = inputStream.read(bytesBuffer)) != -1) {
                digest.update(bytesBuffer, 0, bytesRead);
            }

            return convertByteArrayToHexString(digest.digest());
        } catch (Exception e) {
            e.printStackTrace();
        }
        return "";
    }

    /**
     * Launches a URL in the default browser.
     *
     * @param uri The URI to launch.
     * @see UDesktop#browse(URI)
     * @see java.awt.Desktop#browse(URI)
     */
    public static void browseLink(String uri) {
        UDesktop.browse(URI.create(uri));
    }

    private static InputStream setupConnection(String url, String userAgent, int timeout, boolean useCaches) throws IOException {
        HttpURLConnection connection = ((HttpURLConnection) new URL(url).openConnection());
        connection.setRequestMethod("GET");
        connection.setUseCaches(useCaches);
        connection.addRequestProperty("User-Agent", userAgent);
        connection.setReadTimeout(timeout);
        connection.setConnectTimeout(timeout);
        connection.setDoOutput(true);
        return connection.getInputStream();
    }

    private static String convertByteArrayToHexString(byte[] arrayBytes) {
        StringBuilder stringBuffer = new StringBuilder();
        for (byte arrayByte : arrayBytes) {
            stringBuffer.append(Integer.toString((arrayByte & 0xff) + 0x100, 16)
                    .substring(1));
        }
        return stringBuffer.toString();
    }
}