package qz.build.jlink;

import com.github.zafarkhaja.semver.Version;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import qz.build.provision.params.Arch;

import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.HashMap;

import static qz.build.jlink.Vendor.*;

/**
 * Each JDK provider uses their own url format
 */
public class Url {
    static HashMap<Vendor, String> VENDOR_URL_MAP = new HashMap<>();
    static {
        VENDOR_URL_MAP.put(BELLSOFT, "https://download.bell-sw.com/java/%s/bellsoft-jdk%s-%s-%s.%s");
        VENDOR_URL_MAP.put(ECLIPSE, "https://github.com/adoptium/temurin%s-binaries/releases/download/jdk-%s/OpenJDK%sU-jdk_%s_%s_%s_%s.%s");
        VENDOR_URL_MAP.put(IBM, "https://github.com/ibmruntimes/semeru%s-binaries/releases/download/jdk-%s_%s-%s/ibm-semeru-open-jdk_%s_%s_%s_%s-%s.%s");
        VENDOR_URL_MAP.put(MICROSOFT, "https://aka.ms/download-jdk/microsoft-jdk-%s-%s-%s.%s");
        VENDOR_URL_MAP.put(AMAZON, "https://corretto.aws/downloads/resources/%s/amazon-corretto-%s-%s-%s.%s");
        VENDOR_URL_MAP.put(AZUL, "https://cdn.azul.com/zulu%s/bin/zulu%s-ca-jdk%s-%s_%s.%s");
    }

    private static final Logger log = LogManager.getLogger(Url.class);

    Vendor vendor;
    String pattern;
    public Url(Vendor vendor) {
        this.vendor = vendor;
        if(!VENDOR_URL_MAP.containsKey(vendor)) {
            throw new UnsupportedOperationException(String.format("Vendor provided '%s' couldn't be matched to a URL pattern, aborting.", vendor));
        }
        pattern = VENDOR_URL_MAP.get(vendor);
    }

    public String format(Arch arch, Platform platform, String gcEngine, Version javaSemver, String javaVersion, String gcVer) throws UnsupportedEncodingException {
        Url pattern = new Url(vendor);
        String urlArch = vendor.getUrlArch(arch);
        String fileExt = vendor.getUrlExtension(platform);
        String urlPlatform = vendor.getUrlPlatform(platform);
        String urlJavaVersion = vendor.getUrlJavaVersion(javaSemver);

        // Convert "+" to "%2B"
        String urlJavaVersionEncode = URLEncoder.encode(javaVersion, "UTF-8");

        int javaMajor = javaSemver.getMajorVersion();

        switch(vendor) {
            case BELLSOFT:
                return String.format(pattern.pattern, urlJavaVersionEncode, urlJavaVersionEncode, urlPlatform, urlArch, fileExt);
            case ECLIPSE:
                return String.format(pattern.pattern, javaMajor, urlJavaVersionEncode, javaMajor, urlArch, urlPlatform, gcEngine, urlJavaVersion, fileExt);
            case IBM:
                return String.format(pattern.pattern, javaMajor, urlJavaVersionEncode, gcEngine, gcVer, urlArch, urlPlatform, urlJavaVersion, gcEngine, gcVer, fileExt);
            case MICROSOFT:
                return String.format(pattern.pattern, urlJavaVersion, urlPlatform, urlArch, fileExt);
            case AMAZON:
                return String.format(pattern.pattern, urlJavaVersion, urlJavaVersion, urlPlatform, urlArch, fileExt);
            case AZUL:
                // Special handling of Linux aarch64
                String embedded = platform == Platform.LINUX ? "-embedded" : "";
                return String.format(pattern.pattern, embedded, gcVer, urlJavaVersion, urlPlatform, urlArch, fileExt);
            default:
                throw new UnsupportedOperationException(String.format("URL pattern for '%s' (%s) is missing a format implementation.", vendor, pattern));
        }
    }

}