build-system/gradle-core/src/main/java/com/android/build/gradle/tasks/GeneratePrefabPackages.kt

/*
 * Copyright (C) 2019 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.android.build.gradle.tasks

import com.android.build.gradle.internal.cxx.logging.errorln
import com.android.build.gradle.internal.cxx.model.CxxAbiModel
import com.android.build.gradle.internal.cxx.model.CxxModuleModel
import com.android.build.gradle.internal.cxx.model.DetermineUsedStlResult
import com.android.build.gradle.internal.cxx.model.determineUsedStl
import com.android.build.gradle.internal.cxx.model.soFolder
import com.android.build.gradle.internal.cxx.services.createProcessOutputJunction
import com.android.ide.common.process.ProcessInfoBuilder
import org.gradle.api.Action
import org.gradle.process.ExecResult
import org.gradle.process.JavaExecSpec
import java.io.File

fun generatePrefabPackages(
    moduleModel: CxxModuleModel,
    abiModel: CxxAbiModel,
    packages: List<File>,
    execOperation: (Action<in JavaExecSpec?>) -> ExecResult
) {
    val packagePaths = packages.map { it.path }

    val buildSystem = when (abiModel.variant.module.buildSystem) {
        NativeBuildSystem.NDK_BUILD -> "ndk-build"
        NativeBuildSystem.CMAKE -> "cmake"
    }

    val osVersion = abiModel.abiPlatformVersion

    val prefabClassPath: File = abiModel.variant.prefabClassPath
        ?: throw RuntimeException(
                "CxxAbiModule.prefabClassPath cannot be null when Prefab is used"
        )

    val selectedStl = when (val result = abiModel.variant.determineUsedStl()) {
        is DetermineUsedStlResult.Success -> result.stl
        is DetermineUsedStlResult.Failure -> {
            errorln(result.error)
            return
        }
    }

    // TODO: Get main class from manifest.
    val builder = ProcessInfoBuilder().setClasspath(prefabClassPath.toString())
        .setMain("com.google.prefab.cli.AppKt")
        .addArgs("--build-system", buildSystem)
        .addArgs("--platform", "android")
        .addArgs("--abi", abiModel.abi.tag)
        .addArgs("--os-version", osVersion.toString())
        .addArgs("--stl", selectedStl.argumentName)
        .addArgs("--ndk-version", abiModel.variant.module.ndkVersion.major.toString())
        .addArgs("--output", abiModel.prefabFolder.resolve("prefab").toString())
        .addArgs(packagePaths)

    moduleModel.createProcessOutputJunction(
        abiModel.soFolder,
        "prefab_${buildSystem}_${abiModel.abi.tag}",
        builder, "prefab"
    ).javaProcess().logStderrToInfo().execute(execOperation)
}