com.google.idea.blaze.android.run.binary.instantrun.BlazeInstantRunGradleIntegration.java Source code

Introduction

Here is the source code for com.google.idea.blaze.android.run.binary.instantrun.BlazeInstantRunGradleIntegration.java

Source

/*
 * Copyright 2016 The Bazel Authors. All rights reserved.
 *
 * 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.google.idea.blaze.android.run.binary.instantrun;

import com.android.SdkConstants;
import com.google.common.base.Joiner;
import com.google.common.collect.ImmutableList;
import com.google.common.hash.Hashing;
import com.google.idea.blaze.base.async.process.ExternalTask;
import com.google.idea.blaze.base.async.process.LineProcessingOutputStream;
import com.google.idea.blaze.base.async.process.PrintOutputLineProcessor;
import com.google.idea.blaze.base.experiments.DeveloperFlag;
import com.google.idea.blaze.base.experiments.StringExperiment;
import com.google.idea.blaze.base.model.primitives.Label;
import com.google.idea.blaze.base.scope.BlazeContext;
import com.google.idea.blaze.base.scope.output.IssueOutput;
import com.google.idea.blaze.base.scope.output.PrintOutput;
import com.google.idea.blaze.base.settings.BlazeImportSettings;
import com.google.idea.blaze.base.settings.BlazeImportSettingsManager;
import com.google.idea.blaze.base.sync.projectstructure.ModuleDataStorage;
import com.intellij.openapi.application.PathManager;
import com.intellij.openapi.project.Project;

import javax.annotation.Nullable;
import java.io.File;

/**
 * Defines where instant run storage and artifacts go.
 */
class BlazeInstantRunGradleIntegration {
    private static final String INSTANT_RUN_SUBDIRECTORY = "instantrun";

    private static StringExperiment LOCAL_GRADLE_VERSION = new StringExperiment("use.local.gradle.version");
    private static DeveloperFlag REBUILD_LOCAL_GRADLE = new DeveloperFlag("rebuild.local.gradle");

    /**
     * Gets a unique directory for a given target that can be used for the build process.
     */
    static File getInstantRunArtifactDirectory(Project project, Label target) {
        BlazeImportSettings importSettings = BlazeImportSettingsManager.getInstance(project).getImportSettings();
        assert importSettings != null;
        File dataSubDirectory = new File(importSettings.getProjectDataDirectory(),
                ModuleDataStorage.DATA_SUBDIRECTORY);
        File instantRunDirectory = new File(dataSubDirectory, INSTANT_RUN_SUBDIRECTORY);
        String targetHash = Hashing.md5().hashUnencodedChars(target.toString()).toString();
        return new File(instantRunDirectory, targetHash);
    }

    @Nullable
    static String getGradleUrl(BlazeContext context) {
        String localGradleVersion = LOCAL_GRADLE_VERSION.getValue();
        boolean isDevMode = localGradleVersion != null;

        if (isDevMode) {
            String toolsIdeaPath = PathManager.getHomePath();
            File toolsDir = new File(toolsIdeaPath).getParentFile();
            File repoDir = toolsDir.getParentFile();
            File localGradleDirectory = new File(new File(repoDir, "out/repo/com/android/tools/build/builder"),
                    localGradleVersion);
            if (REBUILD_LOCAL_GRADLE.getValue() || !localGradleDirectory.exists()) {
                // Build gradle
                context.output(PrintOutput.output("Building local Gradle..."));
                int retVal = ExternalTask.builder(toolsDir, ImmutableList.of("./gradlew", ":init", ":publishLocal"))
                        .stdout(LineProcessingOutputStream.of(new PrintOutputLineProcessor(context))).build().run();

                if (retVal != 0) {
                    IssueOutput.error("Gradle build failed.").submit(context);
                    return null;
                }
            }
            return new File(repoDir, "out/repo").getPath();
        }

        // Not supported yet
        IssueOutput.error(
                "You must specify 'use.local.gradle.version' experiment, non-local gradle not supported yet.")
                .submit(context);
        return null;
    }

    static String getGradlePropertiesString() {
        return Joiner.on('\n').join("org.gradle.daemon=true", "org.gradle.jvmargs=-XX:MaxPermSize=1024m -Xmx4096m");
    }

    static String getGradleBuildInfoString(String gradleUrl, File executionRoot, File apkManifestFile) {
        String template = Joiner.on('\n').join("buildscript {", "  repositories {", "    jcenter()",
                "    maven { url '%s' }", "  }", "  dependencies {",
                "    classpath 'com.android.tools.build:gradle:%s'", "  }", "}",
                "apply plugin: 'com.android.external.build'", "externalBuild {", "  executionRoot = '%s'",
                "  buildManifestPath = '%s'", "}");
        String gradleVersion = LOCAL_GRADLE_VERSION.getValue();
        gradleVersion = gradleVersion != null ? gradleVersion : SdkConstants.GRADLE_LATEST_VERSION;

        return String.format(template, gradleUrl, gradleVersion, executionRoot.getPath(),
                apkManifestFile.getPath());
    }
}