build-system/gradle-core/src/main/groovy/com/android/build/gradle/tasks/ProcessTestManifest.groovy

/*
 * Copyright (C) 2014 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.DependencyManager
import com.android.build.gradle.internal.dependency.ManifestDependencyImpl
import com.android.build.gradle.internal.scope.ConventionMappingHelper
import com.android.build.gradle.internal.scope.TaskConfigAction
import com.android.build.gradle.internal.scope.VariantScope
import com.android.build.gradle.internal.variant.BaseVariantOutputData
import com.android.builder.core.VariantConfiguration
import com.google.common.collect.Lists
import org.gradle.api.tasks.Input
import org.gradle.api.tasks.InputFile
import org.gradle.api.tasks.InputFiles
import org.gradle.api.tasks.Optional
import org.gradle.api.tasks.ParallelizableTask
/**
 * A task that processes the manifest
 */
@ParallelizableTask
public class ProcessTestManifest extends ManifestProcessorTask {

    @InputFile
    @Optional
    File testManifestFile

    File tmpDir

    // ----- PRIVATE TASK API -----

    @Input
    String testApplicationId

    @Input @Optional
    String minSdkVersion

    @Input @Optional
    String targetSdkVersion

    @Input
    String testedApplicationId

    @Input
    String instrumentationRunner

    @Input
    Boolean handleProfiling;

    @Input
    Boolean functionalTest;

    @Input
    Map<String, Object> placeholdersValues;

    List<ManifestDependencyImpl> libraries

    /*
     * since libraries above can't return it's input files (@Nested doesn't
     * work on lists), so do a method that will gather them and return them.
     */
    @InputFiles
    List<File> getLibraryManifests() {
        List<ManifestDependencyImpl> libs = getLibraries()
        if (libs == null || libs.isEmpty()) {
            return Collections.emptyList();
        }

        List<File> files = Lists.newArrayListWithCapacity(libs.size() * 2)
        for (ManifestDependencyImpl mdi : libs) {
            files.addAll(mdi.getAllManifests())
        }

        return files;
    }

    @Override
    protected void doFullTaskAction() {
        getBuilder().processTestManifest(
                getTestApplicationId(),
                getMinSdkVersion(),
                getTargetSdkVersion(),
                getTestedApplicationId(),
                getInstrumentationRunner(),
                getHandleProfiling(),
                getFunctionalTest(),
                getTestManifestFile(),
                getLibraries(),
                getPlaceholdersValues(),
                getManifestOutputFile(),
                getTmpDir())
    }

    // ----- ConfigAction -----


    public static class ConfigAction implements TaskConfigAction<ProcessTestManifest> {

        VariantScope scope

        ConfigAction(VariantScope scope) {
            this.scope = scope
        }

        @Override
        String getName() {
            return scope.getTaskName("process", "Manifest")
        }

        @Override
        Class<ProcessTestManifest> getType() {
            return ProcessTestManifest
        }

        @Override
        void execute(ProcessTestManifest processTestManifestTask) {

            VariantConfiguration config = scope.variantConfiguration
            ConventionMappingHelper.map(processTestManifestTask, "testManifestFile") {
                config.getMainManifest()
            }
            ConventionMappingHelper.map(processTestManifestTask, "tmpDir") {
                new File(scope.globalScope.getIntermediatesDir(), "manifest/tmp")
            }

            // get single output for now.
            BaseVariantOutputData variantOutputData = scope.variantData.outputs.get(0)

            variantOutputData.manifestProcessorTask = processTestManifestTask

            processTestManifestTask.androidBuilder = scope.globalScope.androidBuilder
            processTestManifestTask.setVariantName(config.getFullName())

            ConventionMappingHelper.map(processTestManifestTask, "testApplicationId") {
                config.applicationId
            }
            ConventionMappingHelper.map(processTestManifestTask, "minSdkVersion") {
                if (scope.globalScope.androidBuilder.isPreviewTarget()) {
                    return scope.globalScope.androidBuilder.getTargetCodename()
                }

                config.minSdkVersion?.apiString
            }
            ConventionMappingHelper.map(processTestManifestTask, "targetSdkVersion") {
                if (scope.globalScope.androidBuilder.isPreviewTarget()) {
                    return scope.globalScope.androidBuilder.getTargetCodename()
                }

                return config.targetSdkVersion?.apiString
            }
            ConventionMappingHelper.map(processTestManifestTask, "testedApplicationId") {
                config.testedApplicationId
            }
            ConventionMappingHelper.map(processTestManifestTask, "instrumentationRunner") {
                config.instrumentationRunner
            }
            ConventionMappingHelper.map(processTestManifestTask, "handleProfiling") {
                config.handleProfiling
            }
            ConventionMappingHelper.map(processTestManifestTask, "functionalTest") {
                config.functionalTest
            }
            ConventionMappingHelper.map(processTestManifestTask, "libraries") {
                DependencyManager.getManifestDependencies(config.directLibraries)
            }
            ConventionMappingHelper.map(processTestManifestTask, "manifestOutputFile") {
                variantOutputData.getScope().getManifestOutputFile()
            }
            ConventionMappingHelper.map(processTestManifestTask, "placeholdersValues") {
                config.getManifestPlaceholders()
            }
        }
    }
}