import com.android.build.gradle.tasks.ExternalNativeBuildJsonTask
import groovy.json.JsonSlurper
import org.apache.tools.ant.taskdefs.condition.Os
import javax.inject.Inject

def safeExtGet(prop, fallback) {
    rootProject.ext.has(prop) ? rootProject.ext.get(prop) : fallback
}

def safeAppExtGet(prop, fallback) {
    def appProject = rootProject.allprojects.find { it.plugins.hasPlugin('com.android.application') }
    appProject?.ext?.has(prop) ? appProject.ext.get(prop) : fallback
}

def isNewArchitectureEnabled() {
    // In React Native 0.82+, users can no longer opt-out of the New Architecture.
    if (getReactNativeMinorVersion() >= 82) {
        return true
    }

    // In older versions, to opt-in for the New Architecture, you can either:
    // - Set `newArchEnabled` to true inside the `gradle.properties` file
    // - Invoke gradle with `-newArchEnabled=true`
    // - Set an environment variable `ORG_GRADLE_PROJECT_newArchEnabled=true`
    return project.hasProperty("newArchEnabled") && project.newArchEnabled == "true"
}

def resolveReactNativeDirectory() {
    def reactNativeLocation = safeAppExtGet("REACT_NATIVE_NODE_MODULES_DIR", null)
    if (reactNativeLocation != null) {
        return file(reactNativeLocation)
    }

    // Fallback to node resolver for custom directory structures like monorepos.
    def reactNativePackage = file(providers.exec {
        workingDir(rootDir)
        commandLine("node", "--print", "require.resolve('react-native/package.json')")
    }.standardOutput.asText.get().trim())
    if (reactNativePackage.exists()) {
        return reactNativePackage.parentFile
    }

    throw new GradleException("[Worklets] Unable to resolve react-native location in node_modules. You should set project extension property (in `app/build.gradle`) named `REACT_NATIVE_NODE_MODULES_DIR` with the path to react-native in node_modules.")
}

def getReactNativeVersion() {
    def reactNativeRootDir = resolveReactNativeDirectory()
    def reactProperties = new Properties()
    file("$reactNativeRootDir/ReactAndroid/gradle.properties").withInputStream { reactProperties.load(it) }
    return reactProperties.getProperty("VERSION_NAME")
}

def getReactNativeMinorVersion() {
    def reactNativeVersion = getReactNativeVersion()
    return reactNativeVersion.startsWith("0.0.0-") ? 1000 : reactNativeVersion.split("\\.")[1].toInteger()
}

def getHermesV1Enabled() {
    // Even though `HERMES_V1_ENABLED` is now centralized
    // in `react-native-flags.cmake` for React Native >= 0.84
    // that CMake file depends on definitions provided
    // in local `externalNativeBuild` configuration of the LIBRARY.
    // I hope this is only a temporary workaround.
    if (getReactNativeMinorVersion() >= 84) {
        return safeAppExtGet("hermesV1Enabled", true)
    } else {
        return safeAppExtGet("hermesV1Enabled", false)
    }
}

def getWorkletsVersion() {
    def inputFile = file(projectDir.path + '/../package.json')
    def json = new JsonSlurper().parseText(inputFile.text)
    return json.version
}

static def toPlatformFileString(String path) {
    if (Os.isFamily(Os.FAMILY_WINDOWS)) {
        path = path.replace(File.separatorChar, '/' as char)
    }
    return path
}

def getStaticFeatureFlags() {
    def featureFlags = new HashMap<String, String>()

    def staticFeatureFlagsFile = file(projectDir.path + "/../src/featureFlags/staticFlags.json")
    if (!staticFeatureFlagsFile.exists()) {
        throw new GradleException("[Worklets] Feature flags file not found at ${staticFeatureFlagsFile.absolutePath}.")
    }
    new JsonSlurper().parseText(staticFeatureFlagsFile.text).each { key, value -> featureFlags[key] = value.toString()
    }

    def packageJsonFile = file(rootDir.path + "/../package.json")
    if (packageJsonFile.exists()) {
        def packageJson = new JsonSlurper().parseText(packageJsonFile.text)
        packageJson.worklets?.staticFeatureFlags?.each { key, value -> featureFlags[key] = value.toString()
        }
    }

    return featureFlags;
}

def getStaticFeatureFlagsString(featureFlags) {
    return featureFlags.collect { key, value -> "[${key}:${value}]" }.join("")
}

def isFlagEnabled(featureFlags, flagName) {
    return (featureFlags.containsKey(flagName) && featureFlags[flagName] == "true");
}

if (isNewArchitectureEnabled() && project != rootProject) {
    apply plugin: "com.facebook.react"
}

def featureFlags = getStaticFeatureFlags()

def packageDir = project.projectDir.parentFile
def reactNativeRootDir = resolveReactNativeDirectory()
def REACT_NATIVE_MINOR_VERSION = getReactNativeMinorVersion()
def REACT_NATIVE_VERSION = getReactNativeVersion()
def WORKLETS_VERSION = getWorkletsVersion()
def IS_NEW_ARCHITECTURE_ENABLED = isNewArchitectureEnabled()
def IS_REANIMATED_EXAMPLE_APP = safeAppExtGet("isReanimatedExampleApp", false)
def FETCH_PREVIEW_ENABLED = isFlagEnabled(featureFlags, "FETCH_PREVIEW_ENABLED");
def WORKLETS_FEATURE_FLAGS = getStaticFeatureFlagsString(featureFlags)
def HERMES_V1_ENABLED = getHermesV1Enabled()
def WORKLETS_PROFILING = safeAppExtGet("enableWorkletsProfiling", false)

// Set version for prefab
version WORKLETS_VERSION

def workletsPrefabHeadersDir = project.file("${getLayout().getBuildDirectory().getAsFile().get().absolutePath}/prefab-headers/worklets")

def JS_RUNTIME = {
    // Override JS runtime with environment variable
    if (System.getenv("JS_RUNTIME")) {
        return System.getenv("JS_RUNTIME")
    }

    // Check if Hermes is enabled in app setup
    def appProject = rootProject.allprojects.find { it.plugins.hasPlugin('com.android.application') }
    if (appProject?.hermesEnabled?.toBoolean() || appProject?.ext?.react?.enableHermes?.toBoolean()) {
        return "hermes"
    }

    // Use JavaScriptCore (JSC) by default
    return "jsc"
}.call()

def reactNativeArchitectures() {
    def value = project.getProperties().get("reactNativeArchitectures")
    return value ? value.split(",") : ["armeabi-v7a", "x86", "x86_64", "arm64-v8a"]
}

buildscript {
    repositories {
        google()
        mavenCentral()
    }
    dependencies {
        classpath "com.android.tools.build:gradle:8.13.1"
        classpath "de.undercouch:gradle-download-task:5.6.0"
        classpath "com.diffplug.spotless:spotless-plugin-gradle:8.1.0"
    }
}

apply plugin: "com.android.library"
apply plugin: "maven-publish"
apply plugin: "de.undercouch.download"

if (project == rootProject) {
    apply from: "spotless.gradle"
} else {
    apply plugin: "org.jetbrains.kotlin.android"
}

apply from: "./fix-prefab.gradle"


android {
    compileSdkVersion safeExtGet("compileSdkVersion", 36)

    namespace "com.swmansion.worklets"

    if (rootProject.hasProperty("ndkPath")) {
        ndkPath rootProject.ext.ndkPath
    }
    if (rootProject.hasProperty("ndkVersion")) {
        ndkVersion rootProject.ext.ndkVersion
    }

    buildFeatures {
        prefab true
        prefabPublishing true
        buildConfig true
    }

    prefab {
        worklets {
            headers workletsPrefabHeadersDir.absolutePath
        }
    }

    defaultConfig {
        minSdkVersion safeExtGet("minSdkVersion", 24)
        targetSdkVersion safeExtGet("targetSdkVersion", 36)
        versionCode 1
        versionName WORKLETS_VERSION

        buildConfigField("boolean", "WORKLETS_PROFILING", WORKLETS_PROFILING.toString())
        buildConfigField("boolean", "IS_INTERNAL_BUILD", "false")
        buildConfigField("int", "EXOPACKAGE_FLAGS", "0")
        buildConfigField("int", "REACT_NATIVE_MINOR_VERSION", REACT_NATIVE_MINOR_VERSION.toString())

        externalNativeBuild {
            cmake {
                arguments "-DANDROID_STL=c++_shared",
                        "-DREACT_NATIVE_MINOR_VERSION=${REACT_NATIVE_MINOR_VERSION}",
                        "-DANDROID_TOOLCHAIN=clang",
                        "-DREACT_NATIVE_DIR=${toPlatformFileString(reactNativeRootDir.path)}",
                        "-DJS_RUNTIME=${JS_RUNTIME}",
                        "-DIS_REANIMATED_EXAMPLE_APP=${IS_REANIMATED_EXAMPLE_APP}",
                        "-DWORKLETS_FETCH_PREVIEW_ENABLED=${FETCH_PREVIEW_ENABLED}",
                        "-DWORKLETS_PROFILING=${WORKLETS_PROFILING}",
                        "-DWORKLETS_VERSION=${WORKLETS_VERSION}",
                        "-DANDROID_SUPPORT_FLEXIBLE_PAGE_SIZES=ON",
                        "-DWORKLETS_FEATURE_FLAGS=${WORKLETS_FEATURE_FLAGS}",
                        "-DHERMES_V1_ENABLED=${HERMES_V1_ENABLED}"
                abiFilters(*reactNativeArchitectures())
                targets("worklets")
            }
        }

        consumerProguardFiles 'proguard-rules.pro'
    }
    externalNativeBuild {
        cmake {
            version = System.getenv("CMAKE_VERSION") ?: "3.22.1"
            path "CMakeLists.txt"
        }
    }
    buildTypes {
        debug {
            externalNativeBuild {
                cmake {
                    if (JS_RUNTIME == "hermes") {
                        //  React Native doesn't expose these flags, but not having them
                        //  can lead to runtime errors due to ABI mismatches.
                        //  There's also
                        //    HERMESVM_PROFILER_OPCODE
                        //    HERMESVM_PROFILER_BB
                        //  which shouldn't be defined in standard setups.
                        arguments "-DHERMES_ENABLE_DEBUGGER=1"
                    }
                }
            }
            packagingOptions {
                doNotStrip "**/**/*.so"
            }
        }
    }
    lintOptions {
        abortOnError false
    }
    packagingOptions {
        // For some reason gradle only complains about the duplicated version of librrc_root and libreact_render libraries
        // while there are more libraries copied in intermediates folder of the lib build directory, we exclude
        // only the ones that make the build fail (ideally we should only include libworklets but we
        // are only allowed to specify exclude patterns)
        excludes = ["META-INF",
                    "META-INF/**",
                    "**/libc++_shared.so",
                    "**/libfbjni.so",
                    "**/libjsi.so",
                    "**/libhermes.so",
                    "**/libhermesvm.so",
                    "**/libhermestooling.so",
                    "**/libreactnative.so",
                    "**/libjscexecutor.so",]
    }
    compileOptions {
        sourceCompatibility JavaVersion.VERSION_17
        targetCompatibility JavaVersion.VERSION_17
    }
    sourceSets {
        main {
            java {
                if (FETCH_PREVIEW_ENABLED) {
                    srcDirs += "src/networking"
                } else {
                    srcDirs += "src/no-networking"
                }
            }
        }
    }
    if (project != rootProject) {
        kotlinOptions {
            jvmTarget = '17'
        }
    }
    tasks.withType(ExternalNativeBuildJsonTask).tap {
        configureEach { compileTask ->
            compileTask.doLast {
                if (!IS_REANIMATED_EXAMPLE_APP) {
                    return
                }

                def generated = new File("${compileTask.abi.getCxxBuildFolder()}/compile_commands.json")
                def output = new File("${packageDir}/compile_commands.json")

                output.text = generated.text

                println("Generated clangd metadata.")
            }
        }
    }
}

def validateReactNativeVersionResult = providers.exec {
    workingDir(projectDir.path)
    commandLine("node", "./../scripts/validate-react-native-version.js", REACT_NATIVE_VERSION.toString())
    ignoreExitValue = true
}

tasks.register('assertMinimalReactNativeVersionTask') {
    doFirst {
        if (validateReactNativeVersionResult.getResult().get().exitValue != 0) {
            throw new GradleException(validateReactNativeVersionResult.getStandardError().getAsText().get().trim())
        }
    }
}

preBuild.dependsOn(assertMinimalReactNativeVersionTask)

tasks.register('assertNewArchitectureEnabledTask') {
    onlyIf { !IS_NEW_ARCHITECTURE_ENABLED }
    doFirst {
        throw new GradleException("[Worklets] Worklets require new architecture to be enabled. Please enable it by setting `newArchEnabled` to `true` in `gradle.properties`.")
    }
}

preBuild.dependsOn(assertNewArchitectureEnabledTask)

tasks.register('prepareWorkletsHeadersForPrefabs', Copy) {
    from("$projectDir/src/main/cpp")
    from("$projectDir/../Common/cpp")
    include("worklets/**/*.h")
    into(workletsPrefabHeadersDir.absolutePath)
}

interface FSService {
    @Inject
    FileSystemOperations getFs()
}

tasks.register('cleanCMakeCache') {
    def fsProvider = project.objects.newInstance(FSService)
    def cxxDir = file("${projectDir}/.cxx")
    doFirst {
        fsProvider.fs.delete {
            delete cxxDir
        }
    }
}

repositories {
    mavenCentral()
    google()
}

dependencies {
    implementation "com.facebook.yoga:proguard-annotations:1.19.0"
    implementation "androidx.transition:transition:1.6.0"
    implementation "androidx.core:core:1.15.0"

    implementation "com.facebook.react:react-android" // version substituted by RNGP
    implementation "androidx.core:core-ktx:1.17.0"
    if (JS_RUNTIME == "hermes") {
        implementation "com.facebook.react:hermes-android" // version substituted by RNGP
    }
}

preBuild.dependsOn(prepareWorkletsHeadersForPrefabs)

afterEvaluate {
    tasks.named("clean") {
        it.finalizedBy(cleanCMakeCache)
    }
}