/*
 * Copyright (C) 2024 Open Autonomous Connection - All Rights Reserved
 *
 * You are unauthorized to remove this copyright.
 * You have to give Credits to the Author in your project and link this GitHub site: https://github.com/Open-Autonomous-Connection
 * See LICENSE-File if exists
 */

package org.openautonomousconnection.protocol.utils;

import dev.unlegitdqrk.unlegitlibrary.utils.DefaultMethodsOverrider;

import java.net.MalformedURLException;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URL;

public class DomainUtils extends DefaultMethodsOverrider {

    public static String getTopLevelDomain(String url) throws MalformedURLException {
        URL uri = null;
        String tldString = null;

        if (url.startsWith(SiteType.PUBLIC.name + "://")) url = url.substring((SiteType.PUBLIC.name + "://").length());
        if (url.startsWith(SiteType.CLIENT.name + "://")) url = url.substring((SiteType.CLIENT.name + "://").length());
        if (url.startsWith(SiteType.SERVER.name + "://")) url = url.substring((SiteType.SERVER.name + "://").length());
        if (url.startsWith(SiteType.PROTOCOL.name + "://")) url = url.substring((SiteType.PROTOCOL.name + "://").length());
        if (url.startsWith(SiteType.LOCAL.name + "://")) url = url.substring((SiteType.LOCAL.name + "://").length());

        if (!url.startsWith("https://") && !url.startsWith("http://")) url = "https://" + url;

        uri = new URL(url);
        String[] domainNameParts = uri.getHost().split("\\.");
        tldString = domainNameParts[domainNameParts.length - 1];

        return tldString;
    }

    public static String getDomainName(String url) throws URISyntaxException, MalformedURLException {
        if (url.startsWith(SiteType.PUBLIC.name + "://")) url = url.substring((SiteType.PUBLIC.name + "://").length());
        if (url.startsWith(SiteType.CLIENT.name + "://")) url = url.substring((SiteType.CLIENT.name + "://").length());
        if (url.startsWith(SiteType.SERVER.name + "://")) url = url.substring((SiteType.SERVER.name + "://").length());
        if (url.startsWith(SiteType.PROTOCOL.name + "://")) url = url.substring((SiteType.PROTOCOL.name + "://").length());
        if (url.startsWith(SiteType.LOCAL.name + "://")) url = url.substring((SiteType.LOCAL.name + "://").length());

        if (!url.startsWith("https://") && !url.startsWith("http://")) url = "https://" + url;

        URI uri = new URI(url);
        String domain = uri.getHost().replace("." + getTopLevelDomain(url), "");
        return domain;
    }


    public static String getPath(String url) {
        if (!url.startsWith(SiteType.PUBLIC.name + "://") && !url.startsWith(SiteType.CLIENT.name + "://") &&
                !url.startsWith(SiteType.SERVER.name + "://") && !url.startsWith(SiteType.PROTOCOL.name + "://") &&
                !url.startsWith(SiteType.LOCAL.name + "://") && !url.startsWith("http") && !url.startsWith("https")) {
            url = SiteType.PUBLIC.name + "://" + url;
        }

        String[] split = url.split("/");
        if (split.length <= 3) return "";

        StringBuilder path = new StringBuilder();

        for (int i = 3; i < split.length; i++) path.append(split[i]).append("/");

        String pathStr = path.toString();
        if (pathStr.startsWith("/")) pathStr = pathStr.substring("/".length());
        if (pathStr.endsWith("/")) pathStr = pathStr.substring(0, pathStr.length() - "/".length());

        return pathStr;
    }
}