import de.undercouch.gradle.tasks.download.Download
import org.gradle.api.DefaultTask
import org.gradle.api.Project
import org.gradle.api.provider.Provider
import org.gradle.api.tasks.Copy
import org.gradle.api.tasks.TaskProvider
import org.gradle.api.tasks.bundling.Jar
import org.gradle.kotlin.dsl.register
import org.gradle.kotlin.dsl.withType
import org.jetbrains.kotlin.gradle.dsl.KotlinMultiplatformExtension
import java.io.File

class SkikoProjectContext(
    val project: Project,
    val skiko: SkikoProperties,
    val kotlin: KotlinMultiplatformExtension,
    val windowsSdkPathProvider: () -> WindowsSdkPaths,
    val createChecksumsTask: (OS, Arch, Provider<File>) -> TaskProvider<*>,
    val additionalRuntimeLibraries: List<AdditionalRuntimeLibrary>,
) {

    val buildType = skiko.buildType

    val windowsSdkPaths: WindowsSdkPaths by lazy {
        windowsSdkPathProvider()
    }

    val allJvmRuntimeJars = mutableMapOf<Pair<OS, Arch>, TaskProvider<Jar>>()
}

fun SkikoProjectContext.declareSkiaTasks() {
    mapOf(
        "android" to listOf("arm64", "x64"),
        "ios" to listOf("arm64", "x64"),
        "iosSim" to listOf("arm64", "x64"),
        "linux" to listOf("arm64", "x64"),
        "macos" to listOf("arm64", "x64"),
        "tvos" to listOf("arm64"),
        "tvosSim" to listOf("arm64", "x64"),
        "wasm" to listOf("wasm"),
        "windows" to listOf("arm64", "x64"),
    ).forEach { (config, architectures) ->
        architectures.forEach { arch ->
            val taskNameSuffix = joinToTitleCamelCase(config, arch)
            val target = "$config-$arch"

            val skiaReleaseTag = project.skiaVersion(target)

            val skiaBaseUrl = "https://github.com/JetBrains/skia/releases/download/$skiaReleaseTag"

            val artifactId = "Skia-${skiaReleaseTag}-${config}-$buildType-${arch}"

            val downloadSkiaTask = project.tasks.register<Download>("downloadSkia$buildType$taskNameSuffix") {
                group = "Skia Binaries"

                val skiaUrl = "$skiaBaseUrl/$artifactId.zip"
                description = "downloads $skiaUrl"

                onlyIfModified(true)
                src(skiaUrl)
                dest(skiko.dependenciesDir.resolve(
                    "skia/$skiaReleaseTag/Skia-$skiaReleaseTag-$config-$buildType-${arch}.zip")
                )
            }

            project.tasks.register<Copy>("unzipSkia$buildType$taskNameSuffix") {
                group = "Skia Binaries"

                val outputDir = skiko.dependenciesDir.resolve("skia/$skiaReleaseTag/$artifactId")
                description = "unzips to $outputDir"

                dependsOn(downloadSkiaTask)
                from(project.zipTree(downloadSkiaTask.get().dest))

                into(outputDir)
            }
        }
    }
}


/**
 * Do not call inside tasks.register or tasks.call callback
 * (tasks' registration during other task's registration is prohibited)
 */
fun SkikoProjectContext.registerOrGetSkiaDirProvider(
    os: OS, arch: Arch, isUikitSim: Boolean = false
): Provider<File> {
    val taskNameSuffix = joinToTitleCamelCase(buildType.id, os.idWithSuffix(isUikitSim = isUikitSim), arch.id)

    val skiaDir = skiko.skiaDir
    return if (skiaDir != null) {
        project.tasks.registerOrGetTask<DefaultTask>("skiaDir$taskNameSuffix") {
            // dummy task to simplify usage of the resulting provider (see `else` branch)
            // if a file provider is not created from a task provider,
            // then it cannot be used instead of a task in `dependsOn` clauses of other tasks.
            // e.g. the resulting `skiaDir` could not be used in `dependsOn` of CppCompile configuration
            enabled = false
        }.map {
            skiaDir.absoluteFile
        }
    } else {
        project.tasks.withType<Copy>().named("unzipSkia$taskNameSuffix").map { it.destinationDir.absoluteFile }
    }
}

internal val Project.isInIdea: Boolean
    get() {
        return System.getProperty("idea.active")?.toBoolean() == true
    }

val Project.supportAndroid: Boolean
    get() = findProperty(SkikoGradleProperties.ANDROID_ENABLED) == "true" // || isInIdea

val Project.supportAwt: Boolean
    get() = findProperty(SkikoGradleProperties.AWT_ENABLED) == "true" || isInIdea

val Project.supportAllNative: Boolean
    get() = findProperty(SkikoGradleProperties.NATIVE_ENABLED) == "true" || isInIdea

val Project.supportAllNativeIos: Boolean
    get() = supportAllNative || findProperty("${SkikoGradleProperties.NATIVE_IOS}.enabled") == "true" || isInIdea

val Project.supportNativeIosArm64: Boolean
    get() = supportAllNativeIos || findProperty(SkikoGradleProperties.NATIVE_IOS_ARM64) == "true" || isInIdea

val Project.supportNativeIosSimulatorArm64: Boolean
    get() = supportAllNativeIos || findProperty(SkikoGradleProperties.NATIVE_IOS_SIMULATOR_ARM64) == "true" || isInIdea

val Project.supportNativeIosX64: Boolean
    get() = supportAllNativeIos || findProperty(SkikoGradleProperties.NATIVE_IOS_X64) == "true" || isInIdea

val Project.supportAnyNativeIos: Boolean
    get() = supportAllNativeIos || supportNativeIosArm64 || supportNativeIosSimulatorArm64 || supportNativeIosX64

val Project.supportAllNativeTvos: Boolean
    get() = supportAllNative || findProperty("${SkikoGradleProperties.NATIVE_TVOS}.enabled") == "true" || isInIdea

val Project.supportNativeTvosArm64: Boolean
    get() = supportAllNativeTvos || findProperty(SkikoGradleProperties.NATIVE_TVOS_ARM64) == "true" || isInIdea

val Project.supportNativeTvosSimulatorArm64: Boolean
    get() = supportAllNativeTvos || findProperty(SkikoGradleProperties.NATIVE_TVOS_SIMULATOR_ARM64) == "true" || isInIdea

val Project.supportNativeTvosX64: Boolean
    get() = supportAllNativeTvos || findProperty(SkikoGradleProperties.NATIVE_TVOS_X64) == "true" || isInIdea

val Project.supportAnyNativeTvos: Boolean
    get() = supportAllNativeTvos || supportNativeTvosArm64 || supportNativeTvosSimulatorArm64 || supportNativeTvosX64

val Project.supportNativeMac: Boolean
    get() = supportAllNative || findProperty(SkikoGradleProperties.NATIVE_MAC) == "true" || isInIdea

val Project.supportNativeLinux: Boolean
    get() = supportAllNative || findProperty(SkikoGradleProperties.NATIVE_LINUX) == "true" || isInIdea

val Project.supportAnyNative: Boolean
    get() = supportAllNative || supportAnyNativeIos || supportNativeMac || supportNativeLinux

val Project.supportWeb: Boolean
    get() = findProperty(SkikoGradleProperties.WASM_ENABLED) == "true" || isInIdea

fun Project.skiaVersion(target: String): String {
    val platformSpecificVersion = "dependencies.skia.$target"

    return if (hasProperty(platformSpecificVersion)) {
        property(platformSpecificVersion) as String
    } else {
        property("dependencies.skia") as String
    }
}
