com.github.triplet.gradle.play.tasks.internal.Artifacts.kt Maven / Gradle / Ivy

package com.github.triplet.gradle.play.tasks.internal

import com.android.build.VariantOutput.OutputType
import com.android.build.gradle.api.ApkVariantOutput
import com.android.build.gradle.internal.api.InstallableVariantImpl
import com.android.build.gradle.internal.scope.InternalArtifactType
import com.github.triplet.gradle.play.internal.config
import java.io.File

internal fun PublishTaskBase.findBundleFile(): File? {
    val customDir = extension.config.artifactDir

    return if (customDir == null) {
        val installable = variant as InstallableVariantImpl
        installable.getFinalArtifact(InternalArtifactType.BUNDLE).get().files.singleOrNull()
    } else if (customDir.isFile && customDir.extension == "aab") {
        customDir
    } else {
        val bundles = customDir.listFiles().orEmpty().filter { it.extension == "aab" }
        if (bundles.isEmpty()) {
            logger.warn("Warning: '$customDir' does not yet contain an App Bundle.")
        } else if (bundles.size > 1) {
            logger.warn("Warning: '$customDir' contains multiple App Bundles. Only one is allowed.")
        }
        bundles.singleOrNull()
    }
}

internal fun PublishTaskBase.findApkFiles(allowSplits: Boolean): List? {
    val customDir = extension.config.artifactDir

    return if (customDir == null) {
        var result = variant.outputs.filterIsInstance()
        if (!allowSplits) {
            result = result.filter {
                OutputType.valueOf(it.outputType) == OutputType.MAIN || it.filters.isEmpty()
            }
        }
        result.map { it.outputFile }
    } else if (customDir.isFile && customDir.extension == "apk") {
        listOf(customDir)
    } else {
        val apks = customDir.listFiles().orEmpty().filter { it.extension == "apk" }
        if (apks.isEmpty()) {
            logger.warn("Warning: '$customDir' does not yet contain any APKs.")
        }
        apks
    }.ifEmpty { null }
}