All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.android.build.gradle.internal.TaskManager Maven / Gradle / Ivy

/*
 * 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.internal;

import static com.android.build.OutputFile.DENSITY;
import static com.android.builder.core.BuilderConstants.CONNECTED;
import static com.android.builder.core.BuilderConstants.DEVICE;
import static com.android.builder.core.BuilderConstants.FD_ANDROID_RESULTS;
import static com.android.builder.core.BuilderConstants.FD_ANDROID_TESTS;
import static com.android.builder.core.BuilderConstants.FD_FLAVORS_ALL;
import static com.android.builder.core.VariantType.ANDROID_TEST;
import static com.android.builder.core.VariantType.DEFAULT;
import static com.android.builder.core.VariantType.UNIT_TEST;
import static com.android.sdklib.BuildToolInfo.PathId.ZIP_ALIGN;
import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Preconditions.checkState;

import com.android.SdkConstants;
import com.android.annotations.NonNull;
import com.android.annotations.Nullable;
import com.android.build.OutputFile;
import com.android.build.gradle.AndroidConfig;
import com.android.build.gradle.internal.core.Abi;
import com.android.build.gradle.internal.core.GradleVariantConfiguration;
import com.android.build.gradle.internal.coverage.JacocoInstrumentTask;
import com.android.build.gradle.internal.coverage.JacocoPlugin;
import com.android.build.gradle.internal.coverage.JacocoReportTask;
import com.android.build.gradle.internal.dependency.LibraryDependencyImpl;
import com.android.build.gradle.internal.dependency.ManifestDependencyImpl;
import com.android.build.gradle.internal.dependency.VariantDependencies;
import com.android.build.gradle.internal.dsl.AaptOptions;
import com.android.build.gradle.internal.dsl.AbiSplitOptions;
import com.android.build.gradle.internal.dsl.CoreNdkOptions;
import com.android.build.gradle.internal.dsl.PackagingOptions;
import com.android.build.gradle.internal.publishing.ApkPublishArtifact;
import com.android.build.gradle.internal.publishing.MappingPublishArtifact;
import com.android.build.gradle.internal.publishing.MetadataPublishArtifact;
import com.android.build.gradle.internal.scope.AndroidTask;
import com.android.build.gradle.internal.scope.AndroidTaskRegistry;
import com.android.build.gradle.internal.scope.ConventionMappingHelper;
import com.android.build.gradle.internal.scope.GlobalScope;
import com.android.build.gradle.internal.scope.TaskConfigAction;
import com.android.build.gradle.internal.scope.VariantOutputScope;
import com.android.build.gradle.internal.scope.VariantScope;
import com.android.build.gradle.internal.tasks.AndroidReportTask;
import com.android.build.gradle.internal.tasks.CheckManifest;
import com.android.build.gradle.internal.tasks.DependencyReportTask;
import com.android.build.gradle.internal.tasks.DeviceProviderInstrumentTestTask;
import com.android.build.gradle.internal.tasks.FileSupplier;
import com.android.build.gradle.internal.tasks.GenerateApkDataTask;
import com.android.build.gradle.internal.tasks.InstallVariantTask;
import com.android.build.gradle.internal.tasks.MockableAndroidJarTask;
import com.android.build.gradle.internal.tasks.PrepareDependenciesTask;
import com.android.build.gradle.internal.tasks.SigningReportTask;
import com.android.build.gradle.internal.tasks.SourceSetsTask;
import com.android.build.gradle.internal.tasks.TestServerTask;
import com.android.build.gradle.internal.tasks.UninstallTask;
import com.android.build.gradle.internal.tasks.multidex.CreateMainDexList;
import com.android.build.gradle.internal.tasks.multidex.CreateManifestKeepList;
import com.android.build.gradle.internal.tasks.multidex.JarMergingTask;
import com.android.build.gradle.internal.tasks.multidex.RetraceMainDexList;
import com.android.build.gradle.internal.test.TestDataImpl;
import com.android.build.gradle.internal.test.report.ReportType;
import com.android.build.gradle.internal.variant.ApkVariantData;
import com.android.build.gradle.internal.variant.ApkVariantOutputData;
import com.android.build.gradle.internal.variant.ApplicationVariantData;
import com.android.build.gradle.internal.variant.BaseVariantData;
import com.android.build.gradle.internal.variant.BaseVariantOutputData;
import com.android.build.gradle.internal.variant.LibraryVariantData;
import com.android.build.gradle.internal.variant.TestVariantData;
import com.android.build.gradle.tasks.AidlCompile;
import com.android.build.gradle.tasks.AndroidJarTask;
import com.android.build.gradle.tasks.AndroidProGuardTask;
import com.android.build.gradle.tasks.CompatibleScreensManifest;
import com.android.build.gradle.tasks.Dex;
import com.android.build.gradle.tasks.GenerateBuildConfig;
import com.android.build.gradle.tasks.GenerateResValues;
import com.android.build.gradle.tasks.GenerateSplitAbiRes;
import com.android.build.gradle.tasks.JackTask;
import com.android.build.gradle.tasks.JillTask;
import com.android.build.gradle.tasks.Lint;
import com.android.build.gradle.tasks.MergeAssets;
import com.android.build.gradle.tasks.MergeManifests;
import com.android.build.gradle.tasks.MergeResources;
import com.android.build.gradle.tasks.NdkCompile;
import com.android.build.gradle.tasks.PackageApplication;
import com.android.build.gradle.tasks.PackageSplitAbi;
import com.android.build.gradle.tasks.PackageSplitRes;
import com.android.build.gradle.tasks.PreDex;
import com.android.build.gradle.tasks.PreprocessResourcesTask;
import com.android.build.gradle.tasks.ProcessAndroidResources;
import com.android.build.gradle.tasks.ProcessManifest;
import com.android.build.gradle.tasks.ProcessTestManifest;
import com.android.build.gradle.tasks.RenderscriptCompile;
import com.android.build.gradle.tasks.ShrinkResources;
import com.android.build.gradle.tasks.SplitZipAlign;
import com.android.build.gradle.tasks.ZipAlign;
import com.android.build.gradle.tasks.factory.JavaCompileConfigAction;
import com.android.build.gradle.tasks.factory.ProGuardTaskConfigAction;
import com.android.build.gradle.tasks.factory.ProcessJavaResConfigAction;
import com.android.builder.core.AndroidBuilder;
import com.android.builder.core.VariantConfiguration;
import com.android.builder.core.VariantType;
import com.android.builder.dependency.LibraryDependency;
import com.android.builder.internal.testing.SimpleTestCallable;
import com.android.builder.sdk.TargetInfo;
import com.android.builder.testing.ConnectedDeviceProvider;
import com.android.builder.testing.api.DeviceProvider;
import com.android.builder.testing.api.TestServer;
import com.android.sdklib.IAndroidTarget;
import com.android.utils.StringHelper;
import com.google.common.base.CharMatcher;
import com.google.common.base.Function;
import com.google.common.base.Objects;
import com.google.common.base.Predicate;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Iterables;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.collect.Sets;

import org.gradle.api.Action;
import org.gradle.api.GradleException;
import org.gradle.api.Project;
import org.gradle.api.Task;
import org.gradle.api.artifacts.Configuration;
import org.gradle.api.execution.TaskExecutionGraph;
import org.gradle.api.file.ConfigurableFileCollection;
import org.gradle.api.file.FileCollection;
import org.gradle.api.file.FileTree;
import org.gradle.api.internal.file.collections.DefaultConfigurableFileCollection;
import org.gradle.api.logging.LogLevel;
import org.gradle.api.logging.Logger;
import org.gradle.api.logging.Logging;
import org.gradle.api.plugins.BasePlugin;
import org.gradle.api.plugins.JavaBasePlugin;
import org.gradle.api.plugins.JavaPlugin;
import org.gradle.api.reporting.ConfigurableReport;
import org.gradle.api.tasks.Copy;
import org.gradle.api.tasks.bundling.Jar;
import org.gradle.api.tasks.compile.AbstractCompile;
import org.gradle.api.tasks.compile.JavaCompile;
import org.gradle.api.tasks.testing.Test;
import org.gradle.api.tasks.testing.TestTaskReports;
import org.gradle.tooling.provider.model.ToolingModelBuilderRegistry;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.Callable;

import groovy.lang.Closure;
import proguard.gradle.ProGuardTask;

/**
 * Manages tasks creation.
 */
public abstract class TaskManager {

    public static final String FILE_JACOCO_AGENT = "jacocoagent.jar";

    public static final String DEFAULT_PROGUARD_CONFIG_FILE = "proguard-android.txt";

    public static final String DIR_BUNDLES = "bundles";

    public static final String INSTALL_GROUP = "Install";

    public static final String BUILD_GROUP = BasePlugin.BUILD_GROUP;

    public static final String ANDROID_GROUP = "Android";

    /** Property used to define extra instrumentation test runner arguments. */
    public static final String TEST_RUNNER_ARGS_PROP =
            "android.testInstrumentationRunnerArguments.";

    protected Project project;

    protected AndroidBuilder androidBuilder;

    private DependencyManager dependencyManager;

    protected SdkHandler sdkHandler;

    protected AndroidConfig extension;

    protected ToolingModelBuilderRegistry toolingRegistry;

    private final GlobalScope globalScope;

    private AndroidTaskRegistry androidTasks = new AndroidTaskRegistry();

    private Logger logger;

    protected boolean isNdkTaskNeeded = true;

    // Task names
    // TODO: Convert to AndroidTask.
    private static final String MAIN_PREBUILD = "preBuild";

    private static final String UNINSTALL_ALL = "uninstallAll";

    private static final String DEVICE_CHECK = "deviceCheck";

    protected static final String CONNECTED_CHECK = "connectedCheck";

    private static final String ASSEMBLE_ANDROID_TEST = "assembleAndroidTest";

    private static final String SOURCE_SETS = "sourceSets";

    private static final String LINT = "lint";

    protected static final String LINT_COMPILE = "compileLint";

    // Tasks
    private Copy jacocoAgentTask;

    public MockableAndroidJarTask createMockableJar;

    public TaskManager(
            Project project,
            AndroidBuilder androidBuilder,
            AndroidConfig extension,
            SdkHandler sdkHandler,
            DependencyManager dependencyManager,
            ToolingModelBuilderRegistry toolingRegistry) {
        this.project = project;
        this.androidBuilder = androidBuilder;
        this.sdkHandler = sdkHandler;
        this.extension = extension;
        this.toolingRegistry = toolingRegistry;
        this.dependencyManager = dependencyManager;
        logger = Logging.getLogger(this.getClass());

        globalScope = new GlobalScope(
                project,
                androidBuilder,
                extension,
                sdkHandler,
                toolingRegistry);
    }

    private boolean isVerbose() {
        return project.getLogger().isEnabled(LogLevel.INFO);
    }

    private boolean isDebugLog() {
        return project.getLogger().isEnabled(LogLevel.DEBUG);
    }

    /**
     * Creates the tasks for a given BaseVariantData.
     */
    public abstract void createTasksForVariantData(@NonNull TaskFactory tasks,
            @NonNull BaseVariantData variantData);

    public GlobalScope getGlobalScope() {
        return globalScope;
    }

    /**
     * Returns a collection of buildables that creates native object.
     *
     * A buildable is considered to be any object that can be used as the argument to
     * Task.dependsOn.  This could be a Task or a BuildableModelElement (e.g. BinarySpec).
     */
    protected Collection getNdkBuildable(BaseVariantData variantData) {
        return Collections.singleton(variantData.ndkCompileTask);
    }

    /**
     * Override to configure NDK data in the scope.
     */
    public void configureScopeForNdk(@NonNull VariantScope scope) {
        final BaseVariantData variantData = scope.getVariantData();
        scope.setNdkSoFolder(Collections.singleton(new File(
                scope.getGlobalScope().getIntermediatesDir(),
                "ndk/" + variantData.getVariantConfiguration().getDirName() + "/lib")));
        File objFolder = new File(scope.getGlobalScope().getIntermediatesDir(),
                "ndk/" + variantData.getVariantConfiguration().getDirName() + "/obj");
        scope.setNdkObjFolder(objFolder);
        for (Abi abi : NdkHandler.getAbiList()) {
            scope.addNdkDebuggableLibraryFolders(abi,
                    new File(objFolder, "local/" + abi.getName()));
        }

    }

    protected AndroidConfig getExtension() {
        return extension;
    }

    public void resolveDependencies(
            @NonNull VariantDependencies variantDeps,
            @Nullable VariantDependencies testedVariantDeps,
            @Nullable String testedProjectPath) {
        dependencyManager.resolveDependencies(variantDeps, testedVariantDeps, testedProjectPath);
    }

    /**
     * Create tasks before the evaluation (on plugin apply). This is useful for tasks that
     * could be referenced by custom build logic.
     */
    public void createTasksBeforeEvaluate(@NonNull TaskFactory tasks) {
        tasks.create(UNINSTALL_ALL, new Action() {
            @Override
            public void execute(Task uninstallAllTask) {
                uninstallAllTask.setDescription("Uninstall all applications.");
                uninstallAllTask.setGroup(INSTALL_GROUP);
            }
        });

        tasks.create(DEVICE_CHECK, new Action() {
            @Override
            public void execute(Task deviceCheckTask) {
                deviceCheckTask.setDescription(
                        "Runs all device checks using Device Providers and Test Servers.");
                deviceCheckTask.setGroup(JavaBasePlugin.VERIFICATION_GROUP);
            }
        });

        tasks.create(CONNECTED_CHECK, new Action() {
            @Override
            public void execute(Task connectedCheckTask) {
                connectedCheckTask.setDescription(
                        "Runs all device checks on currently connected devices.");
                connectedCheckTask.setGroup(JavaBasePlugin.VERIFICATION_GROUP);
            }
        });

        tasks.create(MAIN_PREBUILD);

        tasks.create(SOURCE_SETS, SourceSetsTask.class, new Action() {
            @Override
            public void execute(SourceSetsTask sourceSetsTask) {
                sourceSetsTask.setConfig(extension);
                sourceSetsTask.setDescription(
                        "Prints out all the source sets defined in this project.");
                sourceSetsTask.setGroup(ANDROID_GROUP);
            }
        });

        tasks.create(ASSEMBLE_ANDROID_TEST, new Action() {
            @Override
            public void execute(Task assembleAndroidTestTask) {
                assembleAndroidTestTask.setGroup(BasePlugin.BUILD_GROUP);
                assembleAndroidTestTask.setDescription("Assembles all the Test applications.");
            }
        });

        tasks.create(LINT, Lint.class, new Action() {
            @Override
            public void execute(Lint lintTask) {
                lintTask.setDescription("Runs lint on all variants.");
                lintTask.setGroup(JavaBasePlugin.VERIFICATION_GROUP);
                lintTask.setLintOptions(getExtension().getLintOptions());
                lintTask.setSdkHome(sdkHandler.getSdkFolder());
                lintTask.setToolingRegistry(toolingRegistry);
            }
        });
        tasks.named(JavaBasePlugin.CHECK_TASK_NAME, new Action() {
            @Override
            public void execute(Task it) {
                it.dependsOn(LINT);
            }
        });
        createLintCompileTask(tasks);
    }

    public void createMockableJarTask() {
        createMockableJar =
                project.getTasks().create("mockableAndroidJar", MockableAndroidJarTask.class);
        createMockableJar.setGroup(BUILD_GROUP);
        createMockableJar.setDescription(
                "Creates a version of android.jar that's suitable for unit tests.");

        ConventionMappingHelper.map(createMockableJar, "androidJar", new Callable() {
            @Override
            public File call() throws Exception {
                return new File(androidBuilder.getTarget().getPath(IAndroidTarget.ANDROID_JAR));
            }
        });

        CharMatcher safeCharacters = CharMatcher.JAVA_LETTER_OR_DIGIT.or(CharMatcher.anyOf("-."));
        String sdkName = safeCharacters.negate().replaceFrom(extension.getCompileSdkVersion(), '-');

        createMockableJar.setOutputFile(
                new File(globalScope.getIntermediatesDir(), "mockable-" + sdkName + ".jar"));

        ConventionMappingHelper.map(createMockableJar, "returnDefaultValues", new Callable() {
            @Override
            public Boolean call() {
                return extension.getTestOptions().getUnitTests().isReturnDefaultValues();
            }
        });
    }

    public void createMergeAppManifestsTask(
            @NonNull TaskFactory tasks,
            @NonNull VariantScope variantScope) {

        ApplicationVariantData appVariantData =
                (ApplicationVariantData) variantScope.getVariantData();
        Set screenSizes = appVariantData.getCompatibleScreens();

        // loop on all outputs. The only difference will be the name of the task, and location
        // of the generated manifest
        for (final BaseVariantOutputData vod : appVariantData.getOutputs()) {
            VariantOutputScope scope = vod.getScope();

            AndroidTask csmTask = null;
            if (vod.getMainOutputFile().getFilter(DENSITY) != null) {
                csmTask = androidTasks.create(tasks,
                        new CompatibleScreensManifest.ConfigAction(scope, screenSizes));
                scope.setCompatibleScreensManifestTask(csmTask);
            }

            scope.setManifestProcessorTask(androidTasks.create(tasks,
                    new MergeManifests.ConfigAction(scope)));

            if (csmTask != null) {
                scope.getManifestProcessorTask().dependsOn(tasks, csmTask);
            }
        }
    }

    public void createMergeLibManifestsTask(
            @NonNull TaskFactory tasks,
            @NonNull VariantScope scope) {

        AndroidTask processManifest = androidTasks.create(tasks,
                new ProcessManifest.ConfigAction(scope));

        processManifest.dependsOn(tasks, scope.getVariantData().prepareDependenciesTask);

        BaseVariantOutputData variantOutputData = scope.getVariantData().getOutputs().get(0);
        variantOutputData.getScope().setManifestProcessorTask(processManifest);
    }

    protected void createProcessTestManifestTask(
            @NonNull TaskFactory tasks,
            @NonNull VariantScope scope) {

        AndroidTask processTestManifestTask = androidTasks.create(tasks,
                new ProcessTestManifest.ConfigAction(scope));

        processTestManifestTask.dependsOn(tasks, scope.getVariantData().prepareDependenciesTask);

        BaseVariantOutputData variantOutputData = scope.getVariantData().getOutputs().get(0);
        variantOutputData.getScope().setManifestProcessorTask(processTestManifestTask);
    }

    public void createRenderscriptTask(
            @NonNull TaskFactory tasks,
            @NonNull VariantScope scope) {
        scope.setRenderscriptCompileTask(
                androidTasks.create(tasks, new RenderscriptCompile.ConfigAction(scope)));

        BaseVariantData variantData = scope.getVariantData();
        GradleVariantConfiguration config = variantData.getVariantConfiguration();
        // get single output for now.
        BaseVariantOutputData variantOutputData = variantData.getOutputs().get(0);

        scope.getRenderscriptCompileTask().dependsOn(tasks, variantData.prepareDependenciesTask);
        if (config.getType().isForTesting()) {
            scope.getRenderscriptCompileTask().dependsOn(tasks,
                    variantOutputData.getScope().getManifestProcessorTask());
        } else {
            scope.getRenderscriptCompileTask().dependsOn(tasks, scope.getCheckManifestTask());
        }

        scope.getResourceGenTask().dependsOn(tasks, scope.getRenderscriptCompileTask());
        // only put this dependency if rs will generate Java code
        if (!config.getRenderscriptNdkModeEnabled()) {
            scope.getSourceGenTask().dependsOn(tasks, scope.getRenderscriptCompileTask());
        }

    }

    public AndroidTask createMergeResourcesTask(
            @NonNull TaskFactory tasks,
            @NonNull VariantScope scope) {
        return basicCreateMergeResourcesTask(
                tasks,
                scope,
                "merge",
                null /*outputLocation*/,
                true /*includeDependencies*/,
                true /*process9patch*/);
    }

    public AndroidTask basicCreateMergeResourcesTask(
            @NonNull TaskFactory tasks,
            @NonNull VariantScope scope,
            @NonNull String taskNamePrefix,
            @Nullable File outputLocation,
            final boolean includeDependencies,
            final boolean process9Patch) {
        AndroidTask mergeResourcesTask = androidTasks.create(tasks,
                new MergeResources.ConfigAction(
                        scope,
                        taskNamePrefix,
                        outputLocation,
                        includeDependencies,
                        process9Patch));
        mergeResourcesTask.dependsOn(tasks,
                scope.getVariantData().prepareDependenciesTask,
                scope.getResourceGenTask());
        scope.setMergeResourcesTask(mergeResourcesTask);
        scope.setResourceOutputDir(
                Objects.firstNonNull(outputLocation, scope.getDefaultMergeResourcesOutputDir()));
        return scope.getMergeResourcesTask();
    }


    public void createMergeAssetsTask(TaskFactory tasks, VariantScope scope) {
        AndroidTask mergeAssetsTask = androidTasks.create(tasks, new MergeAssets.ConfigAction(scope));
        mergeAssetsTask.dependsOn(tasks,
                scope.getVariantData().prepareDependenciesTask,
                scope.getAssetGenTask());
        scope.setMergeAssetsTask(mergeAssetsTask);
    }

    public void createBuildConfigTask(@NonNull TaskFactory tasks, @NonNull VariantScope scope) {
        AndroidTask generateBuildConfigTask =
                androidTasks.create(tasks, new GenerateBuildConfig.ConfigAction(scope));
        scope.setGenerateBuildConfigTask(generateBuildConfigTask);
        scope.getSourceGenTask().dependsOn(tasks, generateBuildConfigTask.getName());
        if (scope.getVariantConfiguration().getType().isForTesting()) {
            // in case of a test project, the manifest is generated so we need to depend
            // on its creation.

            // For test apps there should be a single output, so we get it.
            BaseVariantOutputData variantOutputData = scope.getVariantData().getOutputs().get(0);

            generateBuildConfigTask.dependsOn(
                    tasks, variantOutputData.getScope().getManifestProcessorTask());
        } else {
            generateBuildConfigTask.dependsOn(tasks, scope.getCheckManifestTask());
        }
    }

    public void createGenerateResValuesTask(
            @NonNull TaskFactory tasks,
            @NonNull VariantScope scope) {
        AndroidTask generateResValuesTask = androidTasks.create(
                tasks, new GenerateResValues.ConfigAction(scope));
        scope.getResourceGenTask().dependsOn(tasks, generateResValuesTask);
    }

    public void createPreprocessResourcesTask(
            @NonNull TaskFactory tasks,
            @NonNull final VariantScope scope) {
        if (!"true".equals(
                project.getProperties().get(
                        "com.android.build.gradle.experimentalPreprocessResources"))) {
            return;
        }

        final BaseVariantData variantData = scope.getVariantData();
        int minSdk = variantData.getVariantConfiguration().getMinSdkVersion().getApiLevel();

        if (extension.getPreprocessingOptions().getPreprocessResources()
                && minSdk < PreprocessResourcesTask.MIN_SDK) {
            // Otherwise mergeResources will rename files when merging and it's hard to keep track
            // of PNGs that the user wanted to use instead of the generated ones.
            checkArgument(extension.getBuildToolsRevision().compareTo(
                            MergeResources.NORMALIZE_RESOURCES_BUILD_TOOLS) >= 0,
                    "To preprocess resources, you have to use build tools >= %1$s",
                    MergeResources.NORMALIZE_RESOURCES_BUILD_TOOLS);

            scope.setPreprocessResourcesTask(androidTasks.create(
                    tasks,
                    scope.getTaskName("preprocess", "Resources"),
                    PreprocessResourcesTask.class,
                    new Action() {
                        @Override
                        public void execute(PreprocessResourcesTask preprocessResourcesTask) {
                            variantData.preprocessResourcesTask = preprocessResourcesTask;

                            preprocessResourcesTask.dependsOn(variantData.mergeResourcesTask);

                            preprocessResourcesTask.setVariantName(variantData.getName());

                            String variantDirName =
                                    variantData.getVariantConfiguration().getDirName();
                            preprocessResourcesTask.setMergedResDirectory(
                                    scope.getMergeResourcesOutputDir());
                            preprocessResourcesTask.setGeneratedResDirectory(new File(
                                    scope.getGlobalScope().getGeneratedDir(),
                                    "res/pngs/" + variantDirName));
                            preprocessResourcesTask.setOutputResDirectory(
                                    scope.getPreprocessResourceOutputDir());
                            preprocessResourcesTask.setIncrementalFolder(new File(
                                    scope.getGlobalScope().getIntermediatesDir(),
                                    "incremental/preprocessResourcesTask/" + variantDirName));

                            preprocessResourcesTask.setDensitiesToGenerate(
                                    extension.getPreprocessingOptions().getTypedDensities());
                        }
                    }));
        }

    }

    public void createProcessResTask(
            @NonNull TaskFactory tasks,
            @NonNull VariantScope scope,
            boolean generateResourcePackage) {
        createProcessResTask(
                tasks,
                scope,
                new File(globalScope.getIntermediatesDir(),
                        "symbols/" + scope.getVariantData().getVariantConfiguration().getDirName()),
                generateResourcePackage);
    }

    public void createProcessResTask(
            @NonNull TaskFactory tasks,
            @NonNull VariantScope scope,
            @Nullable File symbolLocation,
            boolean generateResourcePackage) {
        BaseVariantData variantData = scope.getVariantData();

        variantData.calculateFilters(scope.getGlobalScope().getExtension().getSplits());

        // loop on all outputs. The only difference will be the name of the task, and location
        // of the generated data.
        for (BaseVariantOutputData vod : variantData.getOutputs()) {
            final VariantOutputScope variantOutputScope = vod.getScope();

            variantOutputScope.setProcessResourcesTask(androidTasks.create(tasks,
                    new ProcessAndroidResources.ConfigAction(variantOutputScope, symbolLocation,
                            generateResourcePackage)));
            variantOutputScope.getProcessResourcesTask().dependsOn(tasks,
                    variantOutputScope.getManifestProcessorTask(),
                    scope.getMergeResourcesTask(),
                    scope.getMergeAssetsTask());

            // TODO: Make it non-optional once this is not behind a flag.
            variantOutputScope.getProcessResourcesTask().optionalDependsOn(tasks,
                    scope.getPreprocessResourcesTask());

            if (vod.getMainOutputFile().getFilter(DENSITY) == null) {
                scope.setGenerateRClassTask(variantOutputScope.getProcessResourcesTask());
                scope.getSourceGenTask().optionalDependsOn(tasks,
                        variantOutputScope.getProcessResourcesTask());
            }

        }

    }

    /**
     * Creates the split resources packages task if necessary. AAPT will produce split packages for
     * all --split provided parameters. These split packages should be signed and moved unchanged to
     * the APK build output directory.
     */
    public void createSplitResourcesTasks(@NonNull VariantScope scope) {
        BaseVariantData variantData = scope.getVariantData();

        checkState(variantData.getSplitHandlingPolicy().equals(
                        BaseVariantData.SplitHandlingPolicy.RELEASE_21_AND_AFTER_POLICY),
                "Can only create split resources tasks for pure splits.");

        final VariantConfiguration config = variantData.getVariantConfiguration();
        Set densityFilters = variantData.getFilters(OutputFile.FilterType.DENSITY);
        Set abiFilters = variantData.getFilters(OutputFile.FilterType.ABI);
        Set languageFilters = variantData.getFilters(OutputFile.FilterType.LANGUAGE);

        List outputs = variantData.getOutputs();
        if (outputs.size() != 1) {
            throw new RuntimeException(
                    "In release 21 and later, there can be only one main APK, " +
                            "found " + outputs.size());
        }

        final BaseVariantOutputData variantOutputData = outputs.get(0);
        VariantOutputScope variantOutputScope = variantOutputData.getScope();
        variantOutputData.packageSplitResourcesTask = project.getTasks().create(
                scope.getTaskName("package", "SplitResources"),
                PackageSplitRes.class);
        variantOutputData.packageSplitResourcesTask.setInputDirectory(
                variantOutputScope.getProcessResourcePackageOutputFile().getParentFile());
        variantOutputData.packageSplitResourcesTask.setDensitySplits(densityFilters);
        variantOutputData.packageSplitResourcesTask.setLanguageSplits(languageFilters);
        variantOutputData.packageSplitResourcesTask.setOutputBaseName(config.getBaseName());
        variantOutputData.packageSplitResourcesTask.setSigningConfig(config.getSigningConfig());
        variantOutputData.packageSplitResourcesTask.setOutputDirectory(new File(
                scope.getGlobalScope().getIntermediatesDir(), "splits/" + config.getDirName()));
        variantOutputData.packageSplitResourcesTask.setAndroidBuilder(androidBuilder);
        variantOutputData.packageSplitResourcesTask.dependsOn(
                variantOutputScope.getProcessResourcesTask().getName());

        SplitZipAlign zipAlign = project.getTasks().create(
                scope.getTaskName("zipAlign", "SplitPackages"),
                SplitZipAlign.class);
        ConventionMappingHelper.map(zipAlign, "zipAlignExe", new Callable() {
            @Override
            public File call() throws Exception {
                final TargetInfo info = androidBuilder.getTargetInfo();
                if (info == null) {
                    return null;
                }
                String path = info.getBuildTools().getPath(ZIP_ALIGN);
                if (path == null) {
                    return null;
                }
                return new File(path);
            }
        });

        zipAlign.setOutputDirectory(new File(scope.getGlobalScope().getBuildDir(), "outputs/apk"));
        ConventionMappingHelper.map(zipAlign, "densityOrLanguageInputFiles",
                new Callable>() {
                    @Override
                    public List call() {
                        return variantOutputData.packageSplitResourcesTask.getOutputFiles();
                    }
                });
        zipAlign.setOutputBaseName(config.getBaseName());
        zipAlign.setAbiFilters(abiFilters);
        zipAlign.setLanguageFilters(languageFilters);
        zipAlign.setDensityFilters(densityFilters);
        File metadataDirectory = new File(zipAlign.getOutputDirectory().getParentFile(),
                "metadata");
        zipAlign.setApkMetadataFile(new File(metadataDirectory, config.getFullName() + ".mtd"));
        ((ApkVariantOutputData) variantOutputData).splitZipAlign = zipAlign;
        zipAlign.dependsOn(variantOutputData.packageSplitResourcesTask);
    }

    public void createSplitAbiTasks(@NonNull final VariantScope scope) {
        ApplicationVariantData variantData = (ApplicationVariantData) scope.getVariantData();

        checkState(variantData.getSplitHandlingPolicy().equals(
                BaseVariantData.SplitHandlingPolicy.RELEASE_21_AND_AFTER_POLICY),
                "split ABI tasks are only compatible with pure splits.");

        final VariantConfiguration config = variantData.getVariantConfiguration();
        Set filters = AbiSplitOptions.getAbiFilters(
                getExtension().getSplits().getAbiFilters());
        if (filters.isEmpty()) {
            return;
        }

        List outputs = variantData.getOutputs();
        if (outputs.size() != 1) {
            throw new RuntimeException(
                    "In release 21 and later, there can be only one main APK, " +
                            "found " + outputs.size());
        }

        BaseVariantOutputData variantOutputData = outputs.get(0);
        // first create the split APK resources.
        GenerateSplitAbiRes generateSplitAbiRes = project.getTasks().create(
                scope.getTaskName("generate", "SplitAbiRes"),
                GenerateSplitAbiRes.class);
        generateSplitAbiRes.setAndroidBuilder(androidBuilder);

        generateSplitAbiRes.setOutputDirectory(new File(
                scope.getGlobalScope().getIntermediatesDir(), "abi/" + config.getDirName()));
        generateSplitAbiRes.setSplits(filters);
        generateSplitAbiRes.setOutputBaseName(config.getBaseName());
        generateSplitAbiRes.setApplicationId(config.getApplicationId());
        generateSplitAbiRes.setVersionCode(config.getVersionCode());
        generateSplitAbiRes.setVersionName(config.getVersionName());
        ConventionMappingHelper.map(generateSplitAbiRes, "debuggable", new Callable() {
            @Override
            public Boolean call() throws Exception {
                return config.getBuildType().isDebuggable();
            }
        });
        ConventionMappingHelper.map(generateSplitAbiRes, "aaptOptions", new Callable() {
            @Override
            public AaptOptions call() throws Exception {
                return getExtension().getAaptOptions();
            }
        });
        generateSplitAbiRes.dependsOn(
                variantOutputData.getScope().getProcessResourcesTask().getName());

        // then package those resources with the appropriate JNI libraries.
        variantOutputData.packageSplitAbiTask = project.getTasks().create(
                scope.getTaskName("package", "SplitAbi"), PackageSplitAbi.class);
        variantOutputData.packageSplitAbiTask.setInputFiles(generateSplitAbiRes.getOutputFiles());
        variantOutputData.packageSplitAbiTask.setSplits(filters);
        variantOutputData.packageSplitAbiTask.setOutputBaseName(config.getBaseName());
        variantOutputData.packageSplitAbiTask.setSigningConfig(config.getSigningConfig());
        variantOutputData.packageSplitAbiTask.setOutputDirectory(new File(
                scope.getGlobalScope().getIntermediatesDir(), "splits/" + config.getDirName()));
        variantOutputData.packageSplitAbiTask.setMergingFolder(
                new File(scope.getGlobalScope().getIntermediatesDir(),
                        "package-merge/" + variantOutputData.getDirName()));
        variantOutputData.packageSplitAbiTask.setAndroidBuilder(androidBuilder);
        variantOutputData.packageSplitAbiTask.dependsOn(generateSplitAbiRes);
        variantOutputData.packageSplitAbiTask.dependsOn(scope.getNdkBuildable());

        ConventionMappingHelper.map(variantOutputData.packageSplitAbiTask, "jniFolders",
                new Callable>() {
                    @Override
                    public Set call() throws Exception {
                        return getJniFolders(scope);
                    }

                });
        ConventionMappingHelper.map(variantOutputData.packageSplitAbiTask, "jniDebuggable",
                new Callable() {
                    @Override
                    public Boolean call() throws Exception {
                        return config.getBuildType().isJniDebuggable();
                    }
                });
        ConventionMappingHelper.map(variantOutputData.packageSplitAbiTask, "packagingOptions",
                new Callable() {
                    @Override
                    public PackagingOptions call() throws Exception {
                        return getExtension().getPackagingOptions();
                    }
                });

        ((ApkVariantOutputData) variantOutputData).splitZipAlign.getAbiInputFiles().addAll(
                variantOutputData.packageSplitAbiTask.getOutputFiles());

        ((ApkVariantOutputData) variantOutputData).splitZipAlign.dependsOn(
                variantOutputData.packageSplitAbiTask);
    }

    /**
     * Calculate the list of folders that can contain jni artifacts for this variant.
     *
     * @return a potentially empty list of directories that exist or not and that may contains
     * native resources.
     */
    @NonNull
    public Set getJniFolders(@NonNull VariantScope scope) {

        BaseVariantData variantData = scope.getVariantData();
        VariantConfiguration config = variantData.getVariantConfiguration();
        // for now only the project's compilation output.
        Set set = Sets.newHashSet();
        addAllIfNotNull(set, scope.getNdkSoFolder());
        set.add(variantData.renderscriptCompileTask.getLibOutputDir());
        //noinspection unchecked
        addAllIfNotNull(set, config.getLibraryJniFolders());
        //noinspection unchecked
        addAllIfNotNull(set, config.getJniLibsList());

        if (Boolean.TRUE.equals(config.getMergedFlavor().getRenderscriptSupportModeEnabled())) {
            File rsLibs = androidBuilder.getSupportNativeLibFolder();
            if (rsLibs != null && rsLibs.isDirectory()) {
                set.add(rsLibs);
            }

        }

        return set;
    }

    private static  void addAllIfNotNull(@NonNull Collection main, @Nullable Collection toAdd) {
        if (toAdd != null) {
            main.addAll(toAdd);
        }
    }

    public void createProcessJavaResTask(@NonNull TaskFactory tasks, @NonNull VariantScope scope) {
        scope.setProcessJavaResourcesTask(
                androidTasks.create(tasks, new ProcessJavaResConfigAction(scope)));
    }

    public void createAidlTask(@NonNull TaskFactory tasks, @NonNull VariantScope scope) {
        scope.setAidlCompileTask(androidTasks.create(tasks, new AidlCompile.ConfigAction(scope)));
        scope.getSourceGenTask().dependsOn(tasks, scope.getAidlCompileTask());
        scope.getAidlCompileTask().dependsOn(tasks, scope.getVariantData().prepareDependenciesTask);
    }

    /**
     * Creates the task for creating *.class files using javac. These tasks are created regardless
     * of whether Jack is used or not, but assemble will not depend on them if it is. They are
     * always used when running unit tests.
     */
    public AndroidTask createJavacTask(
            @NonNull final TaskFactory tasks,
            @NonNull final VariantScope scope) {
        final BaseVariantData variantData = scope.getVariantData();
        final AndroidTask javacTask = androidTasks.create(tasks,
                new JavaCompileConfigAction(scope));
        scope.setJavacTask(javacTask);

        javacTask.optionalDependsOn(tasks, scope.getSourceGenTask());
        javacTask.dependsOn(tasks,
                scope.getVariantData().prepareDependenciesTask,
                scope.getProcessJavaResourcesTask());

        // TODO - dependency information for the compile classpath is being lost.
        // Add a temporary approximation
        javacTask.dependsOn(tasks,
                scope.getVariantData().getVariantDependency().getCompileConfiguration()
                        .getBuildDependencies());

        if (variantData.getType().isForTesting()) {
            BaseVariantData testedVariantData =
                    (BaseVariantData) ((TestVariantData) variantData).getTestedVariantData();
            final JavaCompile testedJavacTask = testedVariantData.javacTask;
            javacTask.dependsOn(tasks,
                    testedJavacTask != null ? testedJavacTask :
                            testedVariantData.getScope().getJavacTask());
        }

        // Create jar task for uses by external modules.
        if (variantData.getVariantDependency().getClassesConfiguration() != null) {
            tasks.create(scope.getTaskName("package", "JarArtifact"), Jar.class, new Action() {
                @Override
                public void execute(Jar jar) {
                    variantData.classesJarTask = jar;
                    jar.dependsOn(javacTask.getName());

                    // add the class files (whether they are instrumented or not.
                    jar.from(scope.getJavaOutputDir());

                    jar.setDestinationDir(new File(
                            scope.getGlobalScope().getIntermediatesDir(),
                            "classes-jar/" +
                                    variantData.getVariantConfiguration().getDirName()));
                    jar.setArchiveName("classes.jar");
                }
            });
        }

        return javacTask;
    }

    /**
     * Makes the given task the one used by top-level "compile" task.
     */
    public static void setJavaCompilerTask(
            @NonNull AndroidTask javaCompilerTask,
            @NonNull TaskFactory tasks,
            @NonNull VariantScope scope) {
        scope.getCompileTask().dependsOn(tasks, javaCompilerTask);
        scope.setJavaCompilerTask(javaCompilerTask);

        // TODO: Get rid of it once we stop keeping tasks in variant data.
        //noinspection VariableNotUsedInsideIf
        if (scope.getVariantData().javacTask != null) {
            // This is not the experimental plugin, let's update variant data, so Variants API
            // keeps working.
            scope.getVariantData().javaCompilerTask =  (AbstractCompile) tasks.named(javaCompilerTask.getName());
        }

    }

    public void createGenerateMicroApkDataTask(
            @NonNull TaskFactory tasks,
            @NonNull VariantScope scope,
            @NonNull Configuration config) {
        AndroidTask generateMicroApkTask = androidTasks.create(tasks,
                new GenerateApkDataTask.ConfigAction(scope, config));
        generateMicroApkTask.dependsOn(tasks, config);

        // the merge res task will need to run after this one.
        scope.getResourceGenTask().dependsOn(tasks, generateMicroApkTask);
    }

    public void createNdkTasks(@NonNull VariantScope scope) {
        final BaseVariantData variantData = scope.getVariantData();
        NdkCompile ndkCompile = project.getTasks().create(
                scope.getTaskName("compile", "Ndk"),
                NdkCompile.class);

        ndkCompile.dependsOn(variantData.preBuildTask);

        ndkCompile.setAndroidBuilder(androidBuilder);
        ndkCompile.setNdkDirectory(sdkHandler.getNdkFolder());
        ndkCompile.setIsForTesting(variantData.getType().isForTesting());
        variantData.ndkCompileTask = ndkCompile;
        variantData.compileTask.dependsOn(variantData.ndkCompileTask);

        final GradleVariantConfiguration variantConfig = variantData.getVariantConfiguration();

        if (Boolean.TRUE.equals(variantConfig.getMergedFlavor().getRenderscriptNdkModeEnabled())) {
            ndkCompile.setNdkRenderScriptMode(true);
            ndkCompile.dependsOn(variantData.renderscriptCompileTask);
        } else {
            ndkCompile.setNdkRenderScriptMode(false);
        }

        ConventionMappingHelper.map(ndkCompile, "sourceFolders", new Callable>() {
            @Override
            public List call() {
                List sourceList = variantConfig.getJniSourceList();
                if (Boolean.TRUE.equals(
                        variantConfig.getMergedFlavor().getRenderscriptNdkModeEnabled())) {
                    sourceList.add(variantData.renderscriptCompileTask.getSourceOutputDir());
                }

                return sourceList;
            }
        });

        ndkCompile.setGeneratedMakefile(new File(scope.getGlobalScope().getIntermediatesDir(),
                "ndk/" + variantData.getVariantConfiguration().getDirName() + "/Android.mk"));

        ConventionMappingHelper.map(ndkCompile, "ndkConfig", new Callable() {
            @Override
            public CoreNdkOptions call() {
                return variantConfig.getNdkConfig();
            }
        });

        ConventionMappingHelper.map(ndkCompile, "debuggable", new Callable() {
            @Override
            public Boolean call() {
                return variantConfig.getBuildType().isJniDebuggable();
            }
        });

        ndkCompile.setObjFolder(new File(scope.getGlobalScope().getIntermediatesDir(),
                "ndk/" + variantData.getVariantConfiguration().getDirName() + "/obj"));

        Collection ndkSoFolder = scope.getNdkSoFolder();
        if (ndkSoFolder != null && !ndkSoFolder.isEmpty()) {
            ndkCompile.setSoFolder(ndkSoFolder.iterator().next());
        }
    }

    /**
     * Creates the tasks to build unit tests.
     */
    public void createUnitTestVariantTasks(
            @NonNull TaskFactory tasks,
            @NonNull TestVariantData variantData) {
        variantData.assembleVariantTask.dependsOn(createMockableJar);
        VariantScope variantScope = variantData.getScope();

        createPreBuildTasks(tasks, variantScope);
        createProcessJavaResTask(tasks, variantScope);
        createCompileAnchorTask(tasks, variantScope);
        AndroidTask javacTask = createJavacTask(tasks, variantScope);
        setJavaCompilerTask(javacTask, tasks, variantScope);
        createUnitTestTask(tasks, variantData);

        // This hides the assemble unit test task from the task list.
        variantData.assembleVariantTask.setGroup(null);
    }

    /**
     * Creates the tasks to build android tests.
     */
    public void createAndroidTestVariantTasks(@NonNull TaskFactory tasks,
            @NonNull TestVariantData variantData) {
        VariantScope variantScope = variantData.getScope();

        // get single output for now (though this may always be the case for tests).
        final BaseVariantOutputData variantOutputData = variantData.getOutputs().get(0);

        final BaseVariantData baseTestedVariantData =
                (BaseVariantData) variantData.getTestedVariantData();
        final BaseVariantOutputData testedVariantOutputData =
                baseTestedVariantData.getOutputs().get(0);

        createAnchorTasks(tasks, variantScope);

        // Add a task to process the manifest
        createProcessTestManifestTask(tasks, variantScope);

        // Add a task to create the res values
        createGenerateResValuesTask(tasks, variantScope);

        // Add a task to compile renderscript files.
        createRenderscriptTask(tasks, variantScope);

        // Add a task to merge the resource folders
        createMergeResourcesTask(tasks, variantScope);

        // Add a task to merge the assets folders
        createMergeAssetsTask(tasks, variantScope);

        if (variantData.getTestedVariantData().getVariantConfiguration().getType().equals(
                VariantType.LIBRARY)) {
            // in this case the tested library must be fully built before test can be built!
            if (testedVariantOutputData.assembleTask != null) {
                variantOutputData.getScope().getManifestProcessorTask().dependsOn(
                        tasks, testedVariantOutputData.assembleTask);
                variantScope.getMergeResourcesTask().dependsOn(
                        tasks, testedVariantOutputData.assembleTask);
            }

        }

        // Add a task to create the BuildConfig class
        createBuildConfigTask(tasks, variantScope);

        createPreprocessResourcesTask(tasks, variantScope);

        // Add a task to generate resource source files
        createProcessResTask(tasks, variantScope, true /*generateResourcePackage*/);

        // process java resources
        createProcessJavaResTask(tasks, variantScope);

        createAidlTask(tasks, variantScope);

        // Add NDK tasks
        if (isNdkTaskNeeded) {
            createNdkTasks(variantScope);
        }

        variantScope.setNdkBuildable(getNdkBuildable(variantData));

        // Add a task to compile the test application
        if (variantData.getVariantConfiguration().getUseJack()) {
            createJackTask(tasks, variantScope);
        } else {
            AndroidTask javacTask = createJavacTask(tasks, variantScope);
            setJavaCompilerTask(javacTask, tasks, variantScope);
            createPostCompilationTasks(tasks, variantScope);
        }

        createPackagingTask(tasks, variantScope, false /*publishApk*/);

        tasks.named(ASSEMBLE_ANDROID_TEST, new Action() {
            @Override
            public void execute(Task it) {
                it.dependsOn(variantOutputData.assembleTask);
            }
        });

        createConnectedTestForVariant(tasks, variantScope);
    }

    // TODO - should compile src/lint/java from src/lint/java and jar it into build/lint/lint.jar
    private void createLintCompileTask(TaskFactory tasks) {

        // TODO: move doFirst into dedicated task class.
        tasks.create(LINT_COMPILE, Task.class,
                new Action() {
                    @Override
                    public void execute(Task lintCompile) {
                        final File outputDir =
                                new File(getGlobalScope().getIntermediatesDir(), "lint");

                        lintCompile.doFirst(new Action() {
                            @Override
                            public void execute(Task task) {
                                // create the directory for lint output if it does not exist.
                                if (!outputDir.exists()) {
                                    boolean mkdirs = outputDir.mkdirs();
                                    if (!mkdirs) {
                                        throw new GradleException(
                                                "Unable to create lint output directory.");
                                    }
                                }
                            }
                        });
                    }
                });
    }

    /**
     * Is the given variant relevant for lint?
     */
    private static boolean isLintVariant(
            @NonNull BaseVariantData baseVariantData) {
        // Only create lint targets for variants like debug and release, not debugTest
        VariantConfiguration config = baseVariantData.getVariantConfiguration();
        return !config.getType().isForTesting();
    }

    /**
     * Add tasks for running lint on individual variants. We've already added a
     * lint task earlier which runs on all variants.
     */
    public void createLintTasks(TaskFactory tasks, final VariantScope scope) {
        final BaseVariantData baseVariantData =
                scope.getVariantData();
        if (!isLintVariant(baseVariantData)) {
            return;
        }

        // wire the main lint task dependency.
        tasks.named(LINT, new Action() {
            @Override
            public void execute(Task it) {
                it.dependsOn(LINT_COMPILE);
                it.dependsOn(scope.getJavacTask().getName());
            }
        });

        AndroidTask variantLintCheck = androidTasks.create(
                tasks, new Lint.ConfigAction(scope));
        variantLintCheck.dependsOn(tasks, LINT_COMPILE, scope.getJavacTask());
    }

    private void createLintVitalTask(@NonNull ApkVariantData variantData) {
        checkState(getExtension().getLintOptions().isCheckReleaseBuilds());
        // TODO: re-enable with Jack when possible
        if (!variantData.getVariantConfiguration().getBuildType().isDebuggable() &&
                !variantData.getVariantConfiguration().getUseJack()) {
            String variantName = variantData.getVariantConfiguration().getFullName();
            String capitalizedVariantName = StringHelper.capitalize(variantName);
            String taskName = "lintVital" + capitalizedVariantName;
            final Lint lintReleaseCheck = project.getTasks().create(taskName, Lint.class);
            // TODO: Make this task depend on lintCompile too (resolve initialization order first)
            optionalDependsOn(lintReleaseCheck, variantData.javacTask);
            lintReleaseCheck.setLintOptions(getExtension().getLintOptions());
            lintReleaseCheck.setSdkHome(
                    checkNotNull(sdkHandler.getSdkFolder(), "SDK not set up."));
            lintReleaseCheck.setVariantName(variantName);
            lintReleaseCheck.setToolingRegistry(toolingRegistry);
            lintReleaseCheck.setFatalOnly(true);
            lintReleaseCheck.setDescription(
                    "Runs lint on just the fatal issues in the " + capitalizedVariantName
                            + " build.");

            variantData.assembleVariantTask.dependsOn(lintReleaseCheck);

            // If lint is being run, we do not need to run lint vital.
            // TODO: Find a better way to do this.
            project.getGradle().getTaskGraph().whenReady(new Closure(this, this) {
                public void doCall(TaskExecutionGraph taskGraph) {
                    if (taskGraph.hasTask(LINT)) {
                        lintReleaseCheck.setEnabled(false);
                    }
                }
            });
        }
    }

    private void createUnitTestTask(@NonNull TaskFactory tasks,
            @NonNull final TestVariantData variantData) {
        final BaseVariantData testedVariantData =
                (BaseVariantData) variantData.getTestedVariantData();

        final Test runTestsTask = project.getTasks().create(
                variantData.getScope().getTaskName(UNIT_TEST.getPrefix()),
                Test.class);
        runTestsTask.setGroup(JavaBasePlugin.VERIFICATION_GROUP);
        runTestsTask.setDescription(
                "Run unit tests for the " +
                        testedVariantData.getVariantConfiguration().getFullName() + " build.");

        fixTestTaskSources(runTestsTask);

        runTestsTask.dependsOn(variantData.assembleVariantTask);

        final AbstractCompile testCompileTask = variantData.javacTask;
        runTestsTask.setTestClassesDir(testCompileTask.getDestinationDir());

        ConventionMappingHelper.map(runTestsTask, "classpath",
                new Callable() {
                    @Override
                    public ConfigurableFileCollection call() throws Exception {
                        Iterable filteredBootClasspath = Iterables.filter(
                                androidBuilder.getBootClasspath(),
                                new Predicate() {
                                    @Override
                                    public boolean apply(@Nullable File file) {
                                        return file != null &&
                                                !SdkConstants.FN_FRAMEWORK_LIBRARY
                                                        .equals(file.getName());
                                    }
                                });

                        return project.files(
                                testCompileTask.getClasspath(),
                                testCompileTask.getOutputs().getFiles(),
                                variantData.processJavaResourcesTask.getOutputs(),
                                testedVariantData.processJavaResourcesTask.getOutputs(),
                                filteredBootClasspath,
                                // Mockable JAR is last, to make sure you can shadow the classes
                                // withdependencies.
                                createMockableJar.getOutputFile());
                    }
                });

        // Put the variant name in the report path, so that different testing tasks don't
        // overwrite each other's reports.
        TestTaskReports testTaskReports = runTestsTask.getReports();

        for (ConfigurableReport report : new ConfigurableReport[] {
                testTaskReports.getJunitXml(), testTaskReports.getHtml()}) {
            report.setDestination(new File(report.getDestination(), testedVariantData.getName()));
        }

        tasks.named(JavaPlugin.TEST_TASK_NAME, new Action() {
            @Override
            public void execute(Task test) {
                test.dependsOn(runTestsTask);
            }

        });

        extension.getTestOptions().getUnitTests().applyConfiguration(runTestsTask);
    }

    private static void fixTestTaskSources(@NonNull Test testTask) {
        // We are running in afterEvaluate, so the JavaBasePlugin has already added a
        // callback to add test classes to the list of source files of the newly created task.
        // The problem is that we haven't configured the test classes yet (JavaBasePlugin
        // assumes all Test tasks are fully configured at this point), so we have to remove the
        // "directory null" entry from source files and add the right value.
        //
        // This is an ugly hack, since we assume sourceFiles is an instance of
        // DefaultConfigurableFileCollection.
        ((DefaultConfigurableFileCollection) testTask.getInputs().getSourceFiles()).getFrom().clear();
    }

    public void createTopLevelTestTasks(final TaskFactory tasks, boolean hasFlavors) {
        final List reportTasks = Lists.newArrayListWithExpectedSize(2);

        List providers = getExtension().getDeviceProviders();

        final String connectedRootName = CONNECTED + ANDROID_TEST.getSuffix();
        final String defaultReportsDir = getGlobalScope().getReportsDir().getAbsolutePath()
                + "/" + FD_ANDROID_TESTS;
        final String defaultResultsDir = getGlobalScope().getOutputsDir().getAbsolutePath()
                + "/" + FD_ANDROID_RESULTS;

        // If more than one flavor, create a report aggregator task and make this the parent
        // task for all new connected tasks.  Otherwise, create a top level connectedAndroidTest
        // DefaultTask.
        if (hasFlavors) {
            tasks.create(connectedRootName, AndroidReportTask.class,
                    new Action() {
                        @Override
                        public void execute(AndroidReportTask mainConnectedTask) {
                            mainConnectedTask.setGroup(JavaBasePlugin.VERIFICATION_GROUP);
                            mainConnectedTask.setDescription("Installs and runs instrumentation "
                                    + "tests for all flavors on connected devices.");
                            mainConnectedTask.setReportType(ReportType.MULTI_FLAVOR);
                            ConventionMappingHelper.map(mainConnectedTask, "resultsDir",
                                    new Callable() {
                                        @Override
                                        public File call() {
                                            final String dir =
                                                    extension.getTestOptions().getResultsDir();
                                            String rootLocation = dir != null && !dir.isEmpty()
                                                    ? dir : defaultResultsDir;
                                            return project.file(rootLocation + "/connected/"
                                                    + FD_FLAVORS_ALL);
                                        }
                                    });
                            ConventionMappingHelper.map(mainConnectedTask, "reportsDir",
                                    new Callable() {
                                        @Override
                                        public File call() {
                                            final String dir =
                                                    extension.getTestOptions().getReportDir();
                                            String rootLocation = dir != null && !dir.isEmpty()
                                                    ? dir : defaultReportsDir;
                                            return project.file(rootLocation + "/connected/"
                                                    + FD_FLAVORS_ALL);
                                        }
                                    });

                        }

                    });
            reportTasks.add(connectedRootName);
        } else {
            tasks.create(connectedRootName, new Action() {
                @Override
                public void execute(Task connectedTask) {
                    connectedTask.setGroup(JavaBasePlugin.VERIFICATION_GROUP);
                    connectedTask.setDescription(
                            "Installs and runs instrumentation tests for all flavors on connected devices.");
                }

            });
        }

        tasks.named(CONNECTED_CHECK, new Action() {
            @Override
            public void execute(Task it) {
                it.dependsOn(connectedRootName);
            }

        });

        final String mainProviderTaskName = DEVICE + ANDROID_TEST.getSuffix();
        // if more than one provider tasks, either because of several flavors, or because of
        // more than one providers, then create an aggregate report tasks for all of them.
        if (providers.size() > 1 || hasFlavors) {
            tasks.create(mainProviderTaskName, AndroidReportTask.class,
                    new Action() {
                        @Override
                        public void execute(AndroidReportTask mainProviderTask) {
                            mainProviderTask.setGroup(JavaBasePlugin.VERIFICATION_GROUP);
                            mainProviderTask.setDescription(
                                    "Installs and runs instrumentation tests using all Device Providers.");
                            mainProviderTask.setReportType(ReportType.MULTI_FLAVOR);

                            ConventionMappingHelper.map(mainProviderTask, "resultsDir",
                                    new Callable() {
                                        @Override
                                        public File call() throws Exception {
                                            final String dir =
                                                    extension.getTestOptions().getResultsDir();
                                            String rootLocation =  dir != null && !dir.isEmpty()
                                                    ? dir : defaultResultsDir;

                                            return project.file(rootLocation + "/devices/"
                                                    + FD_FLAVORS_ALL);
                                        }
                                    });
                            ConventionMappingHelper.map(mainProviderTask, "reportsDir",
                                    new Callable() {
                                        @Override
                                        public File call() throws Exception {
                                            final String dir =
                                                    extension.getTestOptions().getReportDir();
                                            String rootLocation =  dir != null && !dir.isEmpty()
                                                    ? dir : defaultReportsDir;
                                            return project.file(rootLocation + "/devices/"
                                                    + FD_FLAVORS_ALL);
                                        }
                                    });
                        }

                    });
            reportTasks.add(mainProviderTaskName);
        } else {
            tasks.create(mainProviderTaskName, new Action() {
                @Override
                public void execute(Task providerTask) {
                    providerTask.setGroup(JavaBasePlugin.VERIFICATION_GROUP);
                    providerTask.setDescription(
                            "Installs and runs instrumentation tests using all Device Providers.");
                }
            });
        }

        tasks.named(DEVICE_CHECK, new Action() {
            @Override
            public void execute(Task it) {
                it.dependsOn(mainProviderTaskName);
            }
        });

        // Create top level unit test tasks.
        tasks.create(JavaPlugin.TEST_TASK_NAME, new Action() {
            @Override
            public void execute(Task unitTestTask) {
                unitTestTask.setGroup(JavaBasePlugin.VERIFICATION_GROUP);
                unitTestTask.setDescription("Run unit tests for all variants.");
            }

        });
        tasks.named(JavaBasePlugin.CHECK_TASK_NAME,
                new Action() {
                    @Override
                    public void execute(Task check) {
                        check.dependsOn(JavaPlugin.TEST_TASK_NAME);
                    }
                });

        // If gradle is launched with --continue, we want to run all tests and generate an
        // aggregate report (to help with the fact that we may have several build variants, or
        // or several device providers).
        // To do that, the report tasks must run even if one of their dependent tasks (flavor
        // or specific provider tasks) fails, when --continue is used, and the report task is
        // meant to run (== is in the task graph).
        // To do this, we make the children tasks ignore their errors (ie they won't fail and
        // stop the build).
        if (!reportTasks.isEmpty() && project.getGradle().getStartParameter()
                .isContinueOnFailure()) {
            project.getGradle().getTaskGraph().whenReady(new Closure(this, this) {
                public void doCall(TaskExecutionGraph taskGraph) {
                    for (String reportTask : reportTasks) {
                        if (taskGraph.hasTask(reportTask)) {
                            tasks.named(reportTask, new Action() {
                                @Override
                                public void execute(Task task) {
                                    ((AndroidReportTask) task).setWillRun();
                                }
                            });
                        }
                    }
                }
            });
        }
    }

    protected void createConnectedTestForVariant(
            @NonNull TaskFactory tasks,
            @NonNull final VariantScope variantScope) {
        final BaseVariantData baseVariantData =
                variantScope.getTestedVariantData();
        final TestVariantData testVariantData = (TestVariantData) variantScope.getVariantData();

        // get single output for now
        final BaseVariantOutputData variantOutputData = baseVariantData.getOutputs().get(0);
        final BaseVariantOutputData testVariantOutputData = testVariantData.getOutputs().get(0);

        String connectedRootName = CONNECTED + ANDROID_TEST.getSuffix();

        TestDataImpl testData = new TestDataImpl(testVariantData);
        testData.setExtraInstrumentationTestRunnerArgs(getExtraInstrumentationTestRunnerArgsMap());

        // create the check tasks for this test
        // first the connected one.
        ImmutableList artifactsTasks = ImmutableList.of(
                testVariantData.getOutputs().get(0).assembleTask,
                baseVariantData.assembleVariantTask);

        final AndroidTask connectedTask = androidTasks.create(
                tasks,
                new DeviceProviderInstrumentTestTask.ConfigAction(
                        testVariantData.getScope(),
                        new ConnectedDeviceProvider(sdkHandler.getSdkInfo().getAdb(),
                                new LoggerWrapper(logger)), testData));

        connectedTask.dependsOn(tasks, artifactsTasks);

        tasks.named(connectedRootName, new Action() {
            @Override
            public void execute(Task it) {
                it.dependsOn(connectedTask.getName());
            }
        });

        if (baseVariantData.getVariantConfiguration().getBuildType().isTestCoverageEnabled()
                && !baseVariantData.getVariantConfiguration().getUseJack()) {
            final AndroidTask reportTask = androidTasks.create(
                    tasks,
                    variantScope.getTaskName("create", "CoverageReport"),
                    JacocoReportTask.class,
                    new Action() {
                        @Override
                        public void execute(JacocoReportTask reportTask) {
                            reportTask.setDescription("Creates JaCoCo test coverage report from "
                                    + "data gathered on the device.");

                            reportTask.setReportName(
                                    baseVariantData.getVariantConfiguration().getFullName());

                            ConventionMappingHelper.map(reportTask, "jacocoClasspath",
                                    new Callable() {
                                        @Override
                                        public FileCollection call() throws Exception {
                                            return project.getConfigurations().getAt(
                                                    JacocoPlugin.ANT_CONFIGURATION_NAME);
                                        }
                                    });
                            ConventionMappingHelper
                                    .map(reportTask, "coverageFile", new Callable() {
                                        @Override
                                        public File call() {
                                            return new File(
                                                    ((TestVariantData) testVariantData.getScope()
                                                            .getVariantData()).connectedTestTask
                                                            .getCoverageDir(),
                                                    SimpleTestCallable.FILE_COVERAGE_EC);
                                        }
                                    });
                            ConventionMappingHelper
                                    .map(reportTask, "classDir", new Callable() {
                                        @Override
                                        public File call() {
                                            return baseVariantData.javacTask.getDestinationDir();
                                        }
                                    });
                            ConventionMappingHelper
                                    .map(reportTask, "sourceDir", new Callable>() {
                                        @Override
                                        public List call() {
                                            return baseVariantData
                                                    .getJavaSourceFoldersForCoverage();
                                        }
                                    });

                            ConventionMappingHelper
                                    .map(reportTask, "reportDir", new Callable() {
                                        @Override
                                        public File call() {
                                            return new File(
                                                    variantScope.getGlobalScope().getReportsDir(),
                                                    "/coverage/" + baseVariantData
                                                            .getVariantConfiguration()
                                                            .getDirName());
                                        }
                                    });

                            reportTask.dependsOn(connectedTask.getName());
                        }
                    });

            variantScope.setCoverageReportTask(reportTask);
            baseVariantData.getScope().getCoverageReportTask().dependsOn(tasks, reportTask);

            tasks.named(connectedRootName, new Action() {
                @Override
                public void execute(Task it) {
                    it.dependsOn(reportTask.getName());
                }
            });
        }

        String mainProviderTaskName = DEVICE + ANDROID_TEST.getSuffix();

        List providers = getExtension().getDeviceProviders();

        boolean hasFlavors = baseVariantData.getVariantConfiguration().hasFlavors();

        // now the providers.
        for (DeviceProvider deviceProvider : providers) {

            final AndroidTask providerTask = androidTasks
                    .create(tasks, new DeviceProviderInstrumentTestTask.ConfigAction(
                            testVariantData.getScope(), deviceProvider, testData));

            tasks.named(mainProviderTaskName, new Action() {
                @Override
                public void execute(Task it) {
                    it.dependsOn(providerTask.getName());
                }
            });
            providerTask.dependsOn(tasks, artifactsTasks);
        }

        // now the test servers
        List servers = getExtension().getTestServers();
        for (TestServer testServer : servers) {
            final TestServerTask serverTask = project.getTasks().create(
                    hasFlavors ?
                            baseVariantData.getScope().getTaskName(testServer.getName() + "Upload")
                            :
                                    testServer.getName() + ("Upload"),
                    TestServerTask.class);

            serverTask.setDescription(
                    "Uploads APKs for Build \'" + baseVariantData.getVariantConfiguration()
                            .getFullName() + "\' to Test Server \'" +
                            StringHelper.capitalize(testServer.getName()) + "\'.");
            serverTask.setGroup(JavaBasePlugin.VERIFICATION_GROUP);
            serverTask.dependsOn(testVariantOutputData.assembleTask,
                    variantOutputData.assembleTask);

            serverTask.setTestServer(testServer);

            ConventionMappingHelper.map(serverTask, "testApk", new Callable() {
                @Override
                public File call() throws Exception {
                    return testVariantOutputData.getOutputFile();
                }
            });
            if (!(baseVariantData instanceof LibraryVariantData)) {
                ConventionMappingHelper.map(serverTask, "testedApk", new Callable() {
                    @Override
                    public File call() throws Exception {
                        return variantOutputData.getOutputFile();
                    }
                });
            }

            ConventionMappingHelper.map(serverTask, "variantName", new Callable() {
                @Override
                public String call() throws Exception {
                    return baseVariantData.getVariantConfiguration().getFullName();
                }
            });

            tasks.named(DEVICE_CHECK, new Action() {
                @Override
                public void execute(Task it) {
                    it.dependsOn(serverTask);
                }
            });

            if (!testServer.isConfigured()) {
                serverTask.setEnabled(false);
            }
        }
    }

    private Map getExtraInstrumentationTestRunnerArgsMap() {
        Map argsMap = Maps.newHashMap();
        for (Map.Entry entry : project.getProperties().entrySet()) {
            if (entry.getKey().startsWith(TEST_RUNNER_ARGS_PROP)) {
                String argName = entry.getKey().substring(TEST_RUNNER_ARGS_PROP.length());
                String argValue = entry.getValue().toString();

                argsMap.put(argName, argValue);
            }
        }

        return argsMap;
    }

    public static void createJarTask(@NonNull TaskFactory tasks, @NonNull final VariantScope scope) {
        final BaseVariantData variantData = scope.getVariantData();

        final GradleVariantConfiguration config = variantData.getVariantConfiguration();
        tasks.create(
                scope.getTaskName("jar", "Classes"),
                AndroidJarTask.class,
                new Action() {
                    @Override
                    public void execute(AndroidJarTask jarTask) {
                        //        AndroidJarTask jarTask = project.tasks.create(
                        //                "jar${config.fullName.capitalize()}Classes",
                        //                AndroidJarTask)

                        jarTask.setArchiveName("classes.jar");
                        jarTask.setDestinationDir(new File(
                                scope.getGlobalScope().getIntermediatesDir(),
                                "packaged/" + config.getDirName() + "/"));
                        jarTask.from(scope.getJavaOutputDir());
                        jarTask.dependsOn(scope.getJavacTask().getName());
                        variantData.binayFileProviderTask = jarTask;
                    }

                });
    }

    /**
     * Creates the post-compilation tasks for the given Variant.
     *
     * These tasks create the dex file from the .class files, plus optional intermediary steps like
     * proguard and jacoco
     *
     */
    public void createPostCompilationTasks(TaskFactory tasks, @NonNull final VariantScope scope) {
        checkNotNull(scope.getJavacTask());

        final ApkVariantData variantData = (ApkVariantData) scope.getVariantData();
        final GradleVariantConfiguration config = variantData.getVariantConfiguration();

        // data holding dependencies and input for the dex. This gets updated as new
        // post-compilation steps are inserted between the compilation and dx.
        PostCompilationData pcData = new PostCompilationData();
        pcData.setClassGeneratingTasks(Collections.singletonList(scope.getJavacTask().getName()));
        pcData.setLibraryGeneratingTasks(ImmutableList.of(variantData.prepareDependenciesTask,
                variantData.getVariantDependency().getPackageConfiguration()
                        .getBuildDependencies()));
        pcData.setInputFilesCallable(new Callable>() {
            @Override
            public List call() {
                return new ArrayList(
                        variantData.javacTask.getOutputs().getFiles().getFiles());
            }

        });
        pcData.setInputDir(scope.getJavaOutputDir());

        pcData.setJavaResourcesInputDir(scope.getJavaResourcesDestinationDir());

        pcData.setInputLibrariesCallable(new Callable>() {
            @Override
            public List call() {
                return new ArrayList(
                        scope.getGlobalScope().getAndroidBuilder().getPackagedJars(config));
            }

        });

        // ---- Code Coverage first -----
        boolean isTestCoverageEnabled = config.getBuildType().isTestCoverageEnabled() && !config
                .getType().isForTesting();
        if (isTestCoverageEnabled) {
            pcData = createJacocoTask(tasks, scope, pcData);
        }

        boolean isTestForApp = config.getType().isForTesting() &&
                ((TestVariantData) variantData).getTestedVariantData().getVariantConfiguration()
                        .getType().equals(DEFAULT);
        boolean isMinifyEnabled = config.isMinifyEnabled();
        boolean isMultiDexEnabled = config.isMultiDexEnabled() && !isTestForApp;
        boolean isLegacyMultiDexMode = config.isLegacyMultiDexMode();

        // ----- Minify next ----
        File outFile = maybeCreateProguardTasks(tasks, scope, pcData);
        if (outFile != null) {
            pcData.setInputFiles(Collections.singletonList(outFile));
            pcData.setInputLibraries(Collections.emptyList());
        } else if ((getExtension().getDexOptions().getPreDexLibraries() && !isMultiDexEnabled) || (
                isMultiDexEnabled && !isLegacyMultiDexMode)) {

            AndroidTask preDexTask = androidTasks
                    .create(tasks, new PreDex.ConfigAction(scope, pcData));

            // update dependency.
            preDexTask.dependsOn(tasks, pcData.getLibraryGeneratingTasks());
            pcData.setLibraryGeneratingTasks(Collections.singletonList(preDexTask.getName()));

            // update inputs
            if (isMultiDexEnabled) {
                pcData.setInputLibraries(Collections.emptyList());

            } else {
                pcData.setInputLibrariesCallable(new Callable>() {
                    @Override
                    public List call() {
                        return new ArrayList(
                                project.fileTree(scope.getPreDexOutputDir()).getFiles());
                    }

                });
            }
        }

        AndroidTask createMainDexListTask = null;
        AndroidTask retraceTask = null;

        // ----- Multi-Dex support
        if (isMultiDexEnabled && isLegacyMultiDexMode) {
            if (!isMinifyEnabled) {
                // create a task that will convert the output of the compilation
                // into a jar. This is needed by the multi-dex input.
                AndroidTask jarMergingTask = androidTasks.create(tasks,
                        new JarMergingTask.ConfigAction(scope, pcData));

                // update dependencies
                jarMergingTask.optionalDependsOn(tasks,
                        pcData.getClassGeneratingTasks(),
                        pcData.getLibraryGeneratingTasks());
                pcData.setLibraryGeneratingTasks(
                        Collections.singletonList(jarMergingTask.getName()));
                pcData.setClassGeneratingTasks(Collections.singletonList(jarMergingTask.getName()));

                // Update the inputs
                pcData.setInputFiles(Collections.singletonList(scope.getJarMergingOutputFile()));
                pcData.setInputDirCallable(null);
                pcData.setInputLibraries(Collections.emptyList());
            }

            // ----------
            // Create a task to collect the list of manifest entry points which are
            // needed in the primary dex
            AndroidTask manifestKeepListTask = androidTasks.create(tasks,
                    new CreateManifestKeepList.ConfigAction(scope, pcData));
            manifestKeepListTask.dependsOn(tasks,
                    variantData.getOutputs().get(0).getScope().getManifestProcessorTask());

            // ----------
            // Create a proguard task to shrink the classes to manifest components
            AndroidTask proguardComponentsTask =
                    androidTasks.create(tasks, new ProGuardTaskConfigAction(scope, pcData));

            // update dependencies
            proguardComponentsTask.dependsOn(tasks, manifestKeepListTask);
            proguardComponentsTask.optionalDependsOn(tasks,
                    pcData.getClassGeneratingTasks(),
                    pcData.getLibraryGeneratingTasks());

            // ----------
            // Compute the full list of classes for the main dex file
            createMainDexListTask =
                    androidTasks.create(tasks, new CreateMainDexList.ConfigAction(scope, pcData));
            createMainDexListTask.dependsOn(tasks, proguardComponentsTask);
            //createMainDexListTask.dependsOn { proguardMainDexTask }

            // ----------
            // If proguard is enabled, create a de-obfuscated list to aid debugging.
            if (isMinifyEnabled) {
                retraceTask = androidTasks.create(tasks,
                        new RetraceMainDexList.ConfigAction(scope, pcData));
                retraceTask.dependsOn(tasks, scope.getObfuscationTask(), createMainDexListTask);
            }

        }

        AndroidTask dexTask = androidTasks.create(tasks, new Dex.ConfigAction(scope, pcData));
        scope.setDexTask(dexTask);

        // dependencies, some of these could be null
        dexTask.optionalDependsOn(tasks,
                pcData.getClassGeneratingTasks(),
                pcData.getLibraryGeneratingTasks(),
                createMainDexListTask,
                retraceTask);
    }

    public PostCompilationData createJacocoTask(
            @NonNull TaskFactory tasks,
            @NonNull final VariantScope scope,
            @NonNull final PostCompilationData pcData) {
        AndroidTask jacocoTask =
                androidTasks.create(tasks, new JacocoInstrumentTask.ConfigAction(scope, pcData));

        jacocoTask.optionalDependsOn(tasks, pcData.getClassGeneratingTasks());

        final Copy agentTask = getJacocoAgentTask();
        jacocoTask.dependsOn(tasks, agentTask);

        // update dependency.
        PostCompilationData pcData2 = new PostCompilationData();
        pcData2.setClassGeneratingTasks(Collections.singletonList(jacocoTask.getName()));
        pcData2.setLibraryGeneratingTasks(
                Arrays.asList(pcData.getLibraryGeneratingTasks(), agentTask));

        // update inputs
        pcData2.setInputFilesCallable(new Callable>() {
            @Override
            public List call() {
                return new ArrayList(
                        project.files(scope.getVariantData().jacocoInstrumentTask.getOutputDir())
                                .getFiles());
            }

        });
        pcData2.setInputDirCallable(new Callable() {
            @Override
            public File call() {
                return scope.getVariantData().jacocoInstrumentTask.getOutputDir();
            }

        });
        pcData2.setInputLibrariesCallable(new Callable>() {
            @Override
            public List call() throws Exception {
                List files = null;
                files = new ArrayList(pcData.getInputLibrariesCallable().call());
                files.add(new File(agentTask.getDestinationDir(), FILE_JACOCO_AGENT));
                return files;
            }

        });

        return pcData2;
    }

    public void createJackTask(
            @NonNull TaskFactory tasks,
            @NonNull VariantScope scope) {

        final GradleVariantConfiguration config = scope.getVariantData().getVariantConfiguration();

        // ----- Create Jill tasks -----
        final AndroidTask jillRuntimeTask = androidTasks.create(tasks,
                new JillTask.RuntimeTaskConfigAction(scope));

        final AndroidTask jillPackagedTask = androidTasks.create(tasks,
                new JillTask.PackagedConfigAction(scope));

        jillPackagedTask.dependsOn(tasks,
                scope.getVariantData().getVariantDependency().getPackageConfiguration()
                        .getBuildDependencies());

        // ----- Create Jack Task -----
        AndroidTask jackTask = androidTasks.create(tasks,
                new JackTask.ConfigAction(scope, isVerbose(), isDebugLog()));


        // Jack is compiling and also providing the binary and mapping files.
        setJavaCompilerTask(jackTask, tasks, scope);
        jackTask.dependsOn(tasks,
                scope.getVariantData().sourceGenTask,
                jillRuntimeTask,
                jillPackagedTask,
                // TODO - dependency information for the compile classpath is being lost.
                // Add a temporary approximation
                scope.getVariantData().getVariantDependency().getCompileConfiguration()
                        .getBuildDependencies());

    }

    /**
     * Creates the final packaging task, and optionally the zipalign task (if the variant is signed)
     *
     * @param publishApk if true the generated APK gets published.
     */
    public void createPackagingTask(@NonNull TaskFactory tasks, @NonNull VariantScope variantScope,
            boolean publishApk) {
        final ApkVariantData variantData = (ApkVariantData) variantScope.getVariantData();

        GradleVariantConfiguration config = variantData.getVariantConfiguration();
        boolean signedApk = variantData.isSigned();
        File apkLocation = new File(variantScope.getGlobalScope().getApkLocation());

        boolean multiOutput = variantData.getOutputs().size() > 1;

        // loop on all outputs. The only difference will be the name of the task, and location
        // of the generated data.
        for (final ApkVariantOutputData variantOutputData : variantData.getOutputs()) {
            VariantOutputScope variantOutputScope = variantOutputData.getScope();

            final String outputName = variantOutputData.getFullName();

            // When shrinking resources, rather than having the packaging task
            // directly map to the packageOutputFile of ProcessAndroidResources,
            // we insert the ShrinkResources task into the chain, such that its
            // input is the ProcessAndroidResources packageOutputFile, and its
            // output is what the PackageApplication task reads.
            AndroidTask shrinkTask = null;

            if (config.isMinifyEnabled() && config.getBuildType().isShrinkResources() &&
                    !config.getUseJack()) {
                shrinkTask = androidTasks.create(
                        tasks, new ShrinkResources.ConfigAction(variantOutputScope));
                shrinkTask.dependsOn(tasks, variantScope.getObfuscationTask(),
                        variantOutputScope.getManifestProcessorTask(),
                        variantOutputScope.getProcessResourcesTask());
            }

            AndroidTask packageApp = androidTasks.create(
                    tasks, new PackageApplication.ConfigAction(variantOutputScope));

            packageApp.dependsOn(tasks, variantOutputScope.getProcessResourcesTask(),
                    variantOutputScope.getVariantScope().getProcessJavaResourcesTask(),
                    variantOutputScope.getVariantScope().getNdkBuildable());

            packageApp.optionalDependsOn(
                    tasks,
                    shrinkTask,
                    // TODO: When Jack is converted, add activeDexTask to VariantScope.
                    variantOutputScope.getVariantScope().getDexTask(),
                    variantOutputScope.getVariantScope().getJavaCompilerTask(),
                    variantData.javaCompilerTask, // TODO: Remove when Jack is converted to AndroidTask.
                    variantOutputData.packageSplitResourcesTask,
                    variantOutputData.packageSplitAbiTask);

            AndroidTask appTask = packageApp;

            if (signedApk) {
                if (variantData.getZipAlignEnabled()) {
                    AndroidTask zipAlignTask = androidTasks.create(
                            tasks, new ZipAlign.ConfigAction(variantOutputScope));
                    zipAlignTask.dependsOn(tasks, packageApp);
                    if (variantOutputData.splitZipAlign != null) {
                        zipAlignTask.dependsOn(tasks, variantOutputData.splitZipAlign);
                    }

                    appTask = zipAlignTask;
                }

            }

            checkState(variantData.assembleVariantTask != null);

            // Add an assemble task
            if (multiOutput) {
                // create a task for this output
                variantOutputData.assembleTask = createAssembleTask(variantOutputData);

                // variant assemble task depends on each output assemble task.
                variantData.assembleVariantTask.dependsOn(variantOutputData.assembleTask);
            } else {
                // single output
                variantOutputData.assembleTask = variantData.assembleVariantTask;
            }

            if (!signedApk && variantOutputData.packageSplitResourcesTask != null) {
                // in case we are not signing the resulting APKs and we have some pure splits
                // we should manually copy them from the intermediate location to the final
                // apk location unmodified.
                Copy copyTask = project.getTasks().create(
                        variantOutputScope.getTaskName("copySplit"), Copy.class);
                copyTask.setDestinationDir(apkLocation);
                copyTask.from(variantOutputData.packageSplitResourcesTask.getOutputDirectory());
                variantOutputData.assembleTask.dependsOn(copyTask);
                copyTask.mustRunAfter(appTask.getName());
            }

            variantOutputData.assembleTask.dependsOn(appTask.getName());

            if (publishApk) {
                final String projectBaseName = globalScope.getProjectBaseName();

                // if this variant is the default publish config or we also should publish non
                // defaults, proceed with declaring our artifacts.
                if (getExtension().getDefaultPublishConfig().equals(outputName)) {
                    appTask.configure(tasks, new Action() {
                        @Override
                        public void execute(Task packageTask) {
                            project.getArtifacts().add("default",
                                    new ApkPublishArtifact(projectBaseName,
                                            null,
                                            (FileSupplier) packageTask));
                        }

                    });

                    for (FileSupplier outputFileProvider :
                            variantOutputData.getSplitOutputFileSuppliers()) {
                        project.getArtifacts().add("default",
                                new ApkPublishArtifact(projectBaseName, null, outputFileProvider));
                    }

                    try {
                        if (variantOutputData.getMetadataFile() != null) {
                            project.getArtifacts().add("default-metadata",
                                    new MetadataPublishArtifact(projectBaseName, null,
                                            variantOutputData.getMetadataFile()));
                        }
                    } catch (IOException e) {
                        throw new RuntimeException(e);
                    }

                    if (variantData.getMappingFileProvider() != null) {
                        project.getArtifacts().add("default-mapping",
                                new MappingPublishArtifact(projectBaseName, null,
                                        variantData.getMappingFileProvider()));
                    }
                }

                if (getExtension().getPublishNonDefault()) {
                    appTask.configure(tasks, new Action() {
                        @Override
                        public void execute(Task packageTask) {
                            project.getArtifacts().add(
                                    variantData.getVariantDependency().getPublishConfiguration().getName(),
                                    new ApkPublishArtifact(
                                            projectBaseName,
                                            null,
                                            (FileSupplier) packageTask));
                        }

                    });

                    for (FileSupplier outputFileProvider :
                            variantOutputData.getSplitOutputFileSuppliers()) {
                        project.getArtifacts().add(
                                variantData.getVariantDependency().getPublishConfiguration().getName(),
                                new ApkPublishArtifact(
                                        projectBaseName,
                                        null,
                                        outputFileProvider));
                    }

                    try {
                        if (variantOutputData.getMetadataFile() != null) {
                            project.getArtifacts().add(
                                    variantData.getVariantDependency().getMetadataConfiguration().getName(),
                                    new MetadataPublishArtifact(
                                            projectBaseName,
                                            null,
                                            variantOutputData.getMetadataFile()));
                        }
                    } catch (IOException e) {
                        throw new RuntimeException(e);
                    }

                    if (variantData.getMappingFileProvider() != null) {
                        project.getArtifacts().add(
                                variantData.getVariantDependency().getMappingConfiguration().getName(),
                                new MappingPublishArtifact(
                                        projectBaseName,
                                        null,
                                        variantData.getMappingFileProvider()));
                    }

                    if (variantData.classesJarTask != null) {
                        project.getArtifacts().add(
                                variantData.getVariantDependency().getClassesConfiguration().getName(),
                                variantData.classesJarTask);
                    }
                }
            }
        }

        // create install task for the variant Data. This will deal with finding the
        // right output if there are more than one.
        // Add a task to install the application package
        if (signedApk) {
            AndroidTask installTask = androidTasks.create(
                    tasks, new InstallVariantTask.ConfigAction(variantScope));
            installTask.dependsOn(tasks, variantData.assembleVariantTask);
        }

        if (getExtension().getLintOptions().isCheckReleaseBuilds()) {
            createLintVitalTask(variantData);
        }

        // add an uninstall task
        final AndroidTask uninstallTask = androidTasks.create(
                tasks, new UninstallTask.ConfigAction(variantScope));

        tasks.named(UNINSTALL_ALL, new Action() {
            @Override
            public void execute(Task it) {
                it.dependsOn(uninstallTask.getName());
            }
        });
    }

    public Task createAssembleTask(@NonNull final BaseVariantOutputData variantOutputData) {
        Task assembleTask =
                project.getTasks().create(variantOutputData.getScope().getTaskName("assemble"));
        return assembleTask;
    }

    public Task createAssembleTask(TaskFactory tasks,
            @NonNull final BaseVariantData variantData) {
        Task assembleTask =
                project.getTasks().create(variantData.getScope().getTaskName("assemble"));
        return assembleTask;
    }

    public Copy getJacocoAgentTask() {
        if (jacocoAgentTask == null) {
            jacocoAgentTask = project.getTasks().create("unzipJacocoAgent", Copy.class);
            jacocoAgentTask.from(new Callable>() {
                @Override
                public List call() throws Exception {
                    return Lists.newArrayList(Iterables.transform(
                            project.getConfigurations().getByName(
                                    JacocoPlugin.AGENT_CONFIGURATION_NAME),
                            new Function() {
                                @Override
                                public FileTree apply(@Nullable Object it) {
                                    return project.zipTree(it);
                                }
                            }));
                }
            });
            jacocoAgentTask.include(FILE_JACOCO_AGENT);
            jacocoAgentTask.into(new File(getGlobalScope().getIntermediatesDir(), "jacoco"));
        }

        return jacocoAgentTask;
    }

    /**
     * creates a zip align. This does not use convention mapping, and is meant to let other plugin
     * create zip align tasks.
     *
     * @param name       the name of the task
     * @param inputFile  the input file
     * @param outputFile the output file
     * @return the task
     */
    @NonNull
    public ZipAlign createZipAlignTask(
            @NonNull String name,
            @NonNull File inputFile,
            @NonNull File outputFile) {
        // Add a task to zip align application package
        ZipAlign zipAlignTask = project.getTasks().create(name, ZipAlign.class);

        zipAlignTask.setInputFile(inputFile);
        zipAlignTask.setOutputFile(outputFile);
        ConventionMappingHelper.map(zipAlignTask, "zipAlignExe", new Callable() {
            @Override
            public File call() throws Exception {
                final TargetInfo info = androidBuilder.getTargetInfo();
                if (info != null) {
                    String path = info.getBuildTools().getPath(ZIP_ALIGN);
                    if (path != null) {
                        return new File(path);
                    }
                }

                return null;
            }
        });

        return zipAlignTask;
    }

    /**
     * Creates the proguarding task for the given Variant if necessary.
     *
     * @return null if the proguard task was not created, otherwise the expected outputFile.
     */
    @Nullable
    public File maybeCreateProguardTasks(
            @NonNull TaskFactory tasks,
            @NonNull VariantScope scope,
            @NonNull final PostCompilationData pcData) {
        if (!scope.getVariantData().getVariantConfiguration().isMinifyEnabled()) {
            return null;
        }

        AndroidTask proguardTask = androidTasks.create(
                tasks, new AndroidProGuardTask.ConfigAction(scope, pcData));
        scope.setObfuscationTask(proguardTask);

        // update dependency.
        proguardTask.optionalDependsOn(tasks, pcData.getClassGeneratingTasks(),
                pcData.getLibraryGeneratingTasks());
        pcData.setLibraryGeneratingTasks(Collections.singletonList(proguardTask.getName()));
        pcData.setClassGeneratingTasks(Collections.singletonList(proguardTask.getName()));

        // Return output file.
        return scope.getProguardOutputFile();
    }

    public void createReportTasks(
            List> variantDataList) {
        DependencyReportTask dependencyReportTask =
                project.getTasks().create("androidDependencies", DependencyReportTask.class);
        dependencyReportTask.setDescription("Displays the Android dependencies of the project.");
        dependencyReportTask.setVariants(variantDataList);
        dependencyReportTask.setGroup(ANDROID_GROUP);

        SigningReportTask signingReportTask =
                project.getTasks().create("signingReport", SigningReportTask.class);
        signingReportTask.setDescription("Displays the signing info for each variant.");
        signingReportTask.setVariants(variantDataList);
        signingReportTask.setGroup(ANDROID_GROUP);
    }

    public void createAnchorTasks(@NonNull TaskFactory tasks, @NonNull VariantScope scope) {
        createPreBuildTasks(tasks, scope);

        // also create sourceGenTask
        final BaseVariantData variantData = scope.getVariantData();
        scope.setSourceGenTask(androidTasks.create(tasks,
                scope.getTaskName("generate", "Sources"),
                Task.class,
                new Action() {
                    @Override
                    public void execute(Task task) {
                        variantData.sourceGenTask = task;
                    }
                }));
        // and resGenTask
        scope.setResourceGenTask(androidTasks.create(tasks,
                scope.getTaskName("generate","Resources"),
                Task.class,
                new Action() {
                    @Override
                    public void execute(Task task) {
                        variantData.resourceGenTask = task;
                    }

                }));

        scope.setAssetGenTask(androidTasks.create(tasks,
                scope.getTaskName("generate", "Assets"),
                Task.class,
                new Action() {
                    @Override
                    public void execute(Task task) {
                        variantData.assetGenTask = task;
                    }
                }));

        if (!variantData.getType().isForTesting()
                && variantData.getVariantConfiguration().getBuildType().isTestCoverageEnabled()) {
            scope.setCoverageReportTask(androidTasks.create(tasks,
                    scope.getTaskName("create", "CoverageReport"),
                    Task.class,
                    new Action() {
                        @Override
                        public void execute(Task task) {
                            task.setGroup(JavaBasePlugin.VERIFICATION_GROUP);
                            task.setDescription(String.format(
                                    "Creates test coverage reports for the %s variant.",
                                    variantData.getName()));
                        }
                    }));
        }

        // and compile task
        createCompileAnchorTask(tasks, scope);
    }

    private void createPreBuildTasks(@NonNull TaskFactory tasks, @NonNull VariantScope scope) {
        final BaseVariantData variantData = scope.getVariantData();
        variantData.preBuildTask = project.getTasks().create(scope.getTaskName("pre", "Build"));
        variantData.preBuildTask.dependsOn(MAIN_PREBUILD);

        PrepareDependenciesTask prepareDependenciesTask = project.getTasks().create(
                scope.getTaskName("prepare", "Dependencies"), PrepareDependenciesTask.class);

        variantData.prepareDependenciesTask = prepareDependenciesTask;
        prepareDependenciesTask.dependsOn(variantData.preBuildTask);

        prepareDependenciesTask.setAndroidBuilder(androidBuilder);
        prepareDependenciesTask.setVariant(variantData);

        // for all libraries required by the configurations of this variant, make this task
        // depend on all the tasks preparing these libraries.
        VariantDependencies configurationDependencies = variantData.getVariantDependency();
        prepareDependenciesTask.addChecker(configurationDependencies.getChecker());

        for (LibraryDependencyImpl lib : configurationDependencies.getLibraries()) {
            dependencyManager.addDependencyToPrepareTask(variantData, prepareDependenciesTask, lib);
        }

    }

    private void createCompileAnchorTask(@NonNull TaskFactory tasks, @NonNull final VariantScope scope) {
        final BaseVariantData variantData = scope.getVariantData();
        scope.setCompileTask(androidTasks.create(tasks, new TaskConfigAction() {
            @Override
            public String getName() {
                return scope.getTaskName("compile", "Sources");
            }

            @Override
            public Class getType() {
                return Task.class;
            }

            @Override
            public void execute(Task task) {
                variantData.compileTask = task;
                variantData.compileTask.setGroup(BUILD_GROUP);
            }
        }));
        variantData.assembleVariantTask.dependsOn(scope.getCompileTask().getName());
    }

    public void createCheckManifestTask(@NonNull TaskFactory tasks, @NonNull VariantScope scope) {
        final BaseVariantData variantData = scope.getVariantData();
        final String name = variantData.getVariantConfiguration().getFullName();
        scope.setCheckManifestTask(androidTasks.create(tasks,
                scope.getTaskName("check", "Manifest"),
                CheckManifest.class,
                new Action() {
                    @Override
                    public void execute(CheckManifest checkManifestTask) {
                        variantData.checkManifestTask = checkManifestTask;
                        checkManifestTask.setVariantName(name);
                        ConventionMappingHelper.map(checkManifestTask, "manifest",
                                new Callable() {
                                    @Override
                                    public File call() throws Exception {
                                        return variantData.getVariantConfiguration()
                                                .getDefaultSourceSet().getManifestFile();
                                    }
                                });
                    }

                }));
        scope.getCheckManifestTask().dependsOn(tasks, variantData.preBuildTask);
        variantData.prepareDependenciesTask.dependsOn(scope.getCheckManifestTask().getName());
    }

    public static void optionalDependsOn(@NonNull Task main, Task... dependencies) {
        for (Task dependency : dependencies) {
            if (dependency != null) {
                main.dependsOn(dependency);
            }

        }

    }

    public static void optionalDependsOn(@NonNull Task main, @NonNull List dependencies) {
        for (Object dependency : dependencies) {
            if (dependency != null) {
                main.dependsOn(dependency);
            }

        }

    }

    @NonNull
    private static List getManifestDependencies(
            List libraries) {

        List list = Lists.newArrayListWithCapacity(libraries.size());

        for (LibraryDependency lib : libraries) {
            // get the dependencies
            List children = getManifestDependencies(lib.getDependencies());
            list.add(new ManifestDependencyImpl(lib.getName(), lib.getManifest(), children));
        }

        return list;
    }

    @NonNull
    protected Logger getLogger() {
        return logger;
    }

    @NonNull
    protected AndroidTaskRegistry getAndroidTasks() {
        return androidTasks;
    }

    private File getDefaultProguardFile(String name) {
        File sdkDir = sdkHandler.getAndCheckSdkFolder();
        return new File(sdkDir,
                SdkConstants.FD_TOOLS + File.separatorChar + SdkConstants.FD_PROGUARD
                        + File.separatorChar + name);
    }

}