import java.util.regex.Matcher
import java.util.regex.Pattern
import org.apache.commons.exec.CommandLine
import org.apache.commons.exec.DefaultExecutor
import org.apache.commons.exec.PumpStreamHandler

val GROUP_ANDROID = "android"
val bHackingMode = true
if (parseGradleVersion(gradle.gradleVersion) < 6) {
    logger.error("ERROR: Gradle Version MUST >= 6.0, current is {}", gradle.gradleVersion)
    throw RuntimeException("ERROR: Gradle Version")
} else {
    logger.info("Gradle Version {}", gradle.gradleVersion)
}

buildscript {
    repositories {
        mavenCentral()
    }
    dependencies {
        classpath("org.apache.commons:commons-exec:1.3")
    }
}

tasks {
    register("rr") {
        this.group = GROUP_ANDROID
        this.description = "reboot to recovery"
        this.doLast {
            logger.warn("Rebooting to recovery ...")
            Runtime.getRuntime().exec("adb root")
            Runtime.getRuntime().exec("adb reboot recovery")
        }
    }
    val unpackTask by register<JavaExec>("unpack") {
        group = GROUP_ANDROID
        main = "cfig.packable.PackableLauncherKt"
        classpath = files("bbootimg/build/libs/bbootimg.jar")
        this.maxHeapSize = "512m"
        enableAssertions = true
        args("unpack")
    }
    unpackTask.dependsOn("bbootimg:jar")

    val packTask by register<JavaExec>("pack") {
        group = GROUP_ANDROID
        main = "cfig.packable.PackableLauncherKt"
        classpath = files("bbootimg/build/libs/bbootimg.jar")
        this.maxHeapSize = "512m"
        enableAssertions = true
        args("pack")
    }
    packTask.dependsOn("bbootimg:jar", "aosp:boot_signer:build")

    val flashTask by register("flash", JavaExec::class) {
        group = GROUP_ANDROID
        main = "cfig.packable.PackableLauncherKt"
        classpath = files("bbootimg/build/libs/bbootimg.jar")
        this.maxHeapSize = "512m"
        enableAssertions = true
        args("flash")
    }
    flashTask.dependsOn("bbootimg:jar")


    val pullTask by register("pull", JavaExec::class) {
        group = GROUP_ANDROID
        main = "cfig.packable.PackableLauncherKt"
        classpath = files("bbootimg/build/libs/bbootimg.jar")
        this.maxHeapSize = "512m"
        enableAssertions = true
        args("pull")
    }
    pullTask.dependsOn("bbootimg:jar")

    val cleanTask by register("clean", JavaExec::class) {
        group = GROUP_ANDROID
        main = "cfig.packable.PackableLauncherKt"
        classpath = files("bbootimg/build/libs/bbootimg.jar")
        this.maxHeapSize = "512m"
        enableAssertions = true
        args("clean")
    }
    cleanTask.dependsOn("bbootimg:jar")

    //sparse image dependencies
    if (bHackingMode) {
        logger.info("Hacking mode!")
        //C++ mkbootfs
        if (System.getProperty("os.name").contains("Mac")) {
            //mac
            unpackTask.dependsOn("aosp:libsparse:simg2img:installReleaseMacos")
            packTask.dependsOn("aosp:libsparse:img2simg:installReleaseMacos")
            // common
            packTask.dependsOn("aosp:mkbootfs.10:mkbootfsExecutable")
            packTask.dependsOn("aosp:mkbootfs.11:mkbootfsExecutable")
            unpackTask.dependsOn("aosp:mkbootfs.10:mkbootfsExecutable")
            unpackTask.dependsOn("aosp:mkbootfs.11:mkbootfsExecutable")
        } else if (System.getProperty("os.name").contains("Linux")) {
            //linux
            unpackTask.dependsOn("aosp:libsparse:simg2img:installReleaseLinux")
            packTask.dependsOn("aosp:libsparse:img2simg:installReleaseLinux")
            // common
            packTask.dependsOn("aosp:mkbootfs.10:mkbootfsExecutable")
            packTask.dependsOn("aosp:mkbootfs.11:mkbootfsExecutable")
            unpackTask.dependsOn("aosp:mkbootfs.10:mkbootfsExecutable")
            unpackTask.dependsOn("aosp:mkbootfs.11:mkbootfsExecutable")
        } else {
            logger.info("Disable C++ modules on Window$")
        }
    } else {
        logger.info("Release mode")
    }
}

fun parseGradleVersion(version: String): Int {
    val VERSION_PATTERN = Pattern.compile("((\\d+)(\\.\\d+)+)(-(\\p{Alpha}+)-(\\w+))?(-(SNAPSHOT|\\d{14}([-+]\\d{4})?))?")
    val matcher = VERSION_PATTERN.matcher(version)
    if (!matcher.matches()) {
        throw IllegalArgumentException(String.format("'%s' is not a valid Gradle version string (examples: '1.0', '1.0-rc-1')", version))
    }
    val versionPart: String = matcher.group(1)
    val majorPart = Integer.parseInt(matcher.group(2), 10)
    logger.info("Gradle: versionPart {}, majorPart {}", versionPart, majorPart)
    return majorPart
}