buildscript {
    repositories {
        mavenCentral()
        maven(url = "https://plugins.gradle.org/m2/")
    }

    dependencies {
        classpath("com.google.code.gson:gson:2.8.1")
    }
}

plugins {
    id("checkstyle")
}

group = "org.jackhuang"
version = "3.0"

subprojects {
    apply {
        plugin("java")
        plugin("idea")
        plugin("maven-publish")
        plugin("checkstyle")
    }

    repositories {
        flatDir {
            name = "libs"
            dirs = setOf(rootProject.file("lib"))
        }
        mavenCentral()
        maven(url = "https://jitpack.io")
    }

    tasks.withType<JavaCompile> {
        sourceCompatibility = "1.8"
        targetCompatibility = "1.8"

        options.encoding = "UTF-8"
    }

    configure<CheckstyleExtension> {
        sourceSets = setOf()
    }

    tasks.withType<Checkstyle> {
        exclude("de/javawi/jstun")
    }

    dependencies {
        "testImplementation"("junit:junit:4.12")
    }

    configure<PublishingExtension> {
        publications {
            create<MavenPublication>("maven") {
                from(components["java"])
            }
        }
        repositories {
            mavenLocal()
        }
    }
}

tasks.create("checkTranslations") {
    doLast {
        val hmclLangDir = file("HMCL/src/main/resources/assets/lang")

        val en = java.util.Properties().apply {
            hmclLangDir.resolve("I18N.properties").bufferedReader().use { load(it) }
        }

        val zh = java.util.Properties().apply {
            hmclLangDir.resolve("I18N_zh.properties").bufferedReader().use { load(it) }
        }

        val zh_CN = java.util.Properties().apply {
            hmclLangDir.resolve("I18N_zh_CN.properties").bufferedReader().use { load(it) }
        }

        var success = true

        zh_CN.forEach {
            if (!en.containsKey(it.key)) {
                project.logger.warn("I18N.properties missing key '${it.key}'")
                success = false
            }
        }

        zh_CN.forEach {
            if (!zh.containsKey(it.key)) {
                project.logger.warn("I18N_zh.properties missing key '${it.key}'")
                success = false
            }
        }

        if (!success) {
            throw GradleException("Part of the translation is missing")
        }
    }
}

defaultTasks("clean", "build")

val jfxModules = listOf("base", "graphics", "controls", "fxml", "media", "web")
val jfxClassifier = listOf(
    "linux", "linux-arm32-monocle", "linux-aarch64",
    "mac", "mac-aarch64",
    "win", "win-x86"
)
val jfxVersion = "17"
val jfxMirrorRepos = listOf("https://maven.aliyun.com/repository/central")
val jfxDependenciesFile = rootProject.file("HMCL/src/main/resources/assets/openjfx-dependencies.json")
val jfxUnsupported = mapOf(
    "linux-arm32-monocle" to listOf("media", "web")
)

fun isSupported(module: String, classifier: String) = when (classifier) {
    "linux-arm32-monocle" -> module != "media" && module != "web"
    else -> true
}

val jfxInClasspath =
    try {
        Class.forName("javafx.application.Application", false, this.javaClass.classLoader)
        true
    } catch (ignored: Throwable) {
        false
    }

if (!jfxInClasspath && JavaVersion.current() >= JavaVersion.VERSION_11) {
    val os = System.getProperty("os.name").toLowerCase().let { osName ->
        when {
            osName.contains("win") -> "win"
            osName.contains("mac") -> "mac"
            osName.contains("linux") || osName.contains("unix") -> "linux"
            else -> null
        }
    }

    val classifier = if (os == null) null else when (System.getProperty("os.arch").toLowerCase()) {
        "x86_64", "x86-64", "amd64", "ia32e", "em64t", "x64" -> os
        "x86", "x86_32", "x86-32", "i386", "i486", "i586", "i686", "i86pc", "ia32", "x32" -> "$os-x86"
        "arm64", "aarch64", "armv8", "armv9" -> "$os-aarch64"
        else -> null
    }

    if (classifier != null && classifier in jfxClassifier) {
        rootProject.subprojects {
            for (module in jfxModules) {
                dependencies.add("compileOnly", "org.openjfx:javafx-$module:$jfxVersion:$classifier")
            }
        }
    }
}

rootProject.tasks.create("generateOpenJFXDependencies") {
    doLast {
        val jfxDependencies = jfxModules.map { module ->
            linkedMapOf(
                "module" to "javafx.$module",
                "groupId" to "org.openjfx",
                "artifactId" to "javafx-$module",
                "version" to jfxVersion,
                "sha1" to jfxClassifier.filter { classifier -> isSupported(module, classifier) }
                    .associateWith { classifier ->
                        java.net.URL("https://repo1.maven.org/maven2/org/openjfx/javafx-$module/$jfxVersion/javafx-$module-$jfxVersion-$classifier.jar.sha1")
                            .readText()
                    }
            )
        }

        jfxDependenciesFile.writeText(
            com.google.gson.GsonBuilder().setPrettyPrinting().create().toJson(jfxDependencies)
        )
    }
}

// Ensure that the mirror repository caches files
rootProject.tasks.create("preTouchOpenJFXDependencies") {
    doLast {
        for (repo in jfxMirrorRepos) {
            for (module in jfxModules) {
                for (classifier in jfxClassifier) {
                    if (isSupported(module, classifier)) {
                        val jarUrl =
                            java.net.URL("$repo/org/openjfx/javafx-$module/$jfxVersion/javafx-$module-$jfxVersion-$classifier.jar")
                        try {
                            jarUrl.readBytes()
                        } catch (e: Throwable) {
                            logger.warn("An exception occurred while pre touching $jarUrl", e)
                        }
                    }
                }
            }
        }
    }
}