diff options
author | 2017-04-26 21:30:50 +0200 | |
---|---|---|
committer | 2017-04-27 11:17:55 +0200 | |
commit | ce33ab7a49126a513d7d5a6bc16f86154d9a85b6 (patch) | |
tree | fac0eff07298229cc7547122e92e80fd3715427e /src/main/java/com/google/devtools/build/lib/rules | |
parent | 46da1fca1b3b0b4cd4eb28c4ec4f3cb1b5a22dd4 (diff) |
Give RuleContext the ability to add Make Variables
This CL also makes CcToolchain responsible for adding the sysroot to CC_FLAGS.
PiperOrigin-RevId: 154329746
Diffstat (limited to 'src/main/java/com/google/devtools/build/lib/rules')
6 files changed, 71 insertions, 31 deletions
diff --git a/src/main/java/com/google/devtools/build/lib/rules/MakeVariableProvider.java b/src/main/java/com/google/devtools/build/lib/rules/MakeVariableProvider.java index 1c56c20bfb..15d4998c7f 100644 --- a/src/main/java/com/google/devtools/build/lib/rules/MakeVariableProvider.java +++ b/src/main/java/com/google/devtools/build/lib/rules/MakeVariableProvider.java @@ -15,34 +15,37 @@ package com.google.devtools.build.lib.rules; import com.google.common.collect.ImmutableMap; -import com.google.devtools.build.lib.analysis.RuleConfiguredTarget.Mode; -import com.google.devtools.build.lib.analysis.RuleContext; import com.google.devtools.build.lib.analysis.TransitiveInfoProvider; import com.google.devtools.build.lib.concurrent.ThreadSafety.Immutable; -import java.util.TreeMap; +import com.google.devtools.build.lib.packages.ClassObjectConstructor; +import com.google.devtools.build.lib.packages.NativeClassObjectConstructor; +import com.google.devtools.build.lib.packages.SkylarkClassObject; +import com.google.devtools.build.lib.skylarkinterface.SkylarkCallable; +import com.google.devtools.build.lib.skylarkinterface.SkylarkModule; /** Provides access to make variables from the current fragments. */ +@SkylarkModule(name = "MakeVariables", doc = "Make variables exposed by the current target.") @Immutable -public final class MakeVariableProvider implements TransitiveInfoProvider { +public final class MakeVariableProvider extends SkylarkClassObject + implements TransitiveInfoProvider { + public static final String SKYLARK_NAME = "MakeVariableInfo"; + + public static final ClassObjectConstructor SKYLARK_CONSTRUCTOR = + new NativeClassObjectConstructor(SKYLARK_NAME) {}; + private final ImmutableMap<String, String> makeVariables; public MakeVariableProvider(ImmutableMap<String, String> makeVariables) { + super(SKYLARK_CONSTRUCTOR, ImmutableMap.<String, Object>of()); this.makeVariables = makeVariables; } + @SkylarkCallable( + name = "make_variables", + doc = "Returns the make variables defined by this target.", + structField = true + ) public ImmutableMap<String, String> getMakeVariables() { return makeVariables; } - - public static ImmutableMap<String, String> getToolchainMakeVariables( - RuleContext ruleContext, String attributeName) { - // Cannot be an ImmutableMap.Builder because we want to support duplicate keys - TreeMap<String, String> result = new TreeMap<>(); - for (MakeVariableProvider provider : - ruleContext.getPrerequisites(attributeName, Mode.TARGET, MakeVariableProvider.class)) { - result.putAll(provider.getMakeVariables()); - } - - return ImmutableMap.copyOf(result); - } } diff --git a/src/main/java/com/google/devtools/build/lib/rules/SkylarkRuleContext.java b/src/main/java/com/google/devtools/build/lib/rules/SkylarkRuleContext.java index 897d77ac01..7ab8097ed9 100644 --- a/src/main/java/com/google/devtools/build/lib/rules/SkylarkRuleContext.java +++ b/src/main/java/com/google/devtools/build/lib/rules/SkylarkRuleContext.java @@ -986,9 +986,11 @@ public final class SkylarkRuleContext implements SkylarkValue { public String expandMakeVariables(String attributeName, String command, final Map<String, String> additionalSubstitutions) throws EvalException { checkMutable("expand_make_variables"); - return ruleContext.expandMakeVariables(attributeName, - command, new ConfigurationMakeVariableContext(ruleContext.getRule().getPackage(), - ruleContext.getConfiguration()) { + return ruleContext.expandMakeVariables( + attributeName, + command, + new ConfigurationMakeVariableContext( + ruleContext, ruleContext.getRule().getPackage(), ruleContext.getConfiguration()) { @Override public String lookupMakeVariable(String name) throws ExpansionException { if (additionalSubstitutions.containsKey(name)) { diff --git a/src/main/java/com/google/devtools/build/lib/rules/cpp/CcToolchain.java b/src/main/java/com/google/devtools/build/lib/rules/cpp/CcToolchain.java index c4d3abc469..cc01e6c7f4 100644 --- a/src/main/java/com/google/devtools/build/lib/rules/cpp/CcToolchain.java +++ b/src/main/java/com/google/devtools/build/lib/rules/cpp/CcToolchain.java @@ -39,6 +39,7 @@ import com.google.devtools.build.lib.collect.nestedset.NestedSet; import com.google.devtools.build.lib.collect.nestedset.NestedSetBuilder; import com.google.devtools.build.lib.collect.nestedset.Order; import com.google.devtools.build.lib.packages.License; +import com.google.devtools.build.lib.rules.MakeVariableProvider; import com.google.devtools.build.lib.rules.RuleConfiguredTargetFactory; import com.google.devtools.build.lib.rules.cpp.FdoSupport.FdoException; import com.google.devtools.build.lib.util.Pair; @@ -49,6 +50,7 @@ import com.google.devtools.build.skyframe.SkyFunction; import com.google.devtools.build.skyframe.SkyKey; import java.io.IOException; import java.util.ArrayList; +import java.util.HashMap; import java.util.List; import java.util.Map; @@ -204,7 +206,7 @@ public class CcToolchain implements RuleConfiguredTargetFactory { coverage = crosstool; } - CcToolchainProvider provider = + CcToolchainProvider ccProvider = new CcToolchainProvider( cppConfiguration, crosstool, @@ -231,10 +233,19 @@ public class CcToolchain implements RuleConfiguredTargetFactory { ruleContext.getPrerequisiteArtifact("$link_dynamic_library_tool", Mode.HOST), getEnvironment(ruleContext), cppConfiguration.getBuiltInIncludeDirectories()); + + // TODO(kmensah): Remove sysroot from cppConfiguration and calculate it here. + PathFragment sysroot = cppConfiguration.getSysroot(); + + MakeVariableProvider makeVariableProvider = + createMakeVariableProvider(cppConfiguration, sysroot); + RuleConfiguredTargetBuilder builder = new RuleConfiguredTargetBuilder(ruleContext) - .addProvider(provider) - .addNativeDeclaredProvider(provider) + .addProvider(ccProvider) + .addNativeDeclaredProvider(ccProvider) + .addProvider(makeVariableProvider) + .addNativeDeclaredProvider(makeVariableProvider) .addProvider(fdoSupport.getFdoSupport().createFdoSupportProvider(ruleContext)) .setFilesToBuild(new NestedSetBuilder<Artifact>(Order.STABLE_ORDER).build()) .addProvider(RunfilesProvider.simple(Runfiles.EMPTY)); @@ -354,6 +365,22 @@ public class CcToolchain implements RuleConfiguredTargetFactory { : NestedSetBuilder.<Artifact>emptySet(Order.STABLE_ORDER); } + private MakeVariableProvider createMakeVariableProvider( + CppConfiguration cppConfiguration, PathFragment sysroot) { + + HashMap<String, String> makeVariables = + new HashMap<>(cppConfiguration.getAdditionalMakeVariables()); + + // Overwrite the CC_FLAGS variable to include sysroot, if it's available. + if (sysroot != null) { + String sysrootFlag = "--sysroot=" + sysroot; + String ccFlags = makeVariables.get("CC_FLAGS"); + ccFlags = ccFlags.isEmpty() ? sysrootFlag : ccFlags + " " + sysrootFlag; + makeVariables.put("CC_FLAGS", ccFlags); + } + return new MakeVariableProvider(ImmutableMap.copyOf(makeVariables)); + } + /** * Returns a map that should be templated into the crosstool as build variables. Is meant to * be overridden by subclasses of CcToolchain. diff --git a/src/main/java/com/google/devtools/build/lib/rules/cpp/CppConfiguration.java b/src/main/java/com/google/devtools/build/lib/rules/cpp/CppConfiguration.java index 95efb13dd9..d07ec2aaf8 100644 --- a/src/main/java/com/google/devtools/build/lib/rules/cpp/CppConfiguration.java +++ b/src/main/java/com/google/devtools/build/lib/rules/cpp/CppConfiguration.java @@ -718,6 +718,7 @@ public class CppConfiguration extends BuildConfiguration.Fragment { for (CrosstoolConfig.MakeVariable variable : toolchain.getMakeVariableList()) { makeVariablesBuilder.put(variable.getName(), variable.getValue()); } + // TODO(kmensah): Remove once targets can depend on the cc_toolchain in skylark. if (sysrootFlag != null) { String ccFlags = makeVariablesBuilder.get("CC_FLAGS"); ccFlags = ccFlags.isEmpty() ? sysrootFlag : ccFlags + " " + sysrootFlag; @@ -1655,7 +1656,6 @@ public class CppConfiguration extends BuildConfiguration.Fragment { * <p>The returned map must contain an entry for {@code STACK_FRAME_UNLIMITED}, * though the entry may be an empty string. */ - @VisibleForTesting public ImmutableMap<String, String> getAdditionalMakeVariables() { return additionalMakeVariables; } diff --git a/src/main/java/com/google/devtools/build/lib/rules/extra/ExtraActionFactory.java b/src/main/java/com/google/devtools/build/lib/rules/extra/ExtraActionFactory.java index ef7062303b..f0bc1c7679 100644 --- a/src/main/java/com/google/devtools/build/lib/rules/extra/ExtraActionFactory.java +++ b/src/main/java/com/google/devtools/build/lib/rules/extra/ExtraActionFactory.java @@ -64,9 +64,11 @@ public final class ExtraActionFactory implements RuleConfiguredTargetFactory { command = command.replace("$(OWNER_LABEL_DIGEST)", "$$(OWNER_LABEL_DIGEST)"); command = command.replace("$(output ", "$$(output "); try { - command = MakeVariableExpander.expand( - command, new ConfigurationMakeVariableContext( - context.getTarget().getPackage(), context.getConfiguration())); + command = + MakeVariableExpander.expand( + command, + new ConfigurationMakeVariableContext( + context, context.getTarget().getPackage(), context.getConfiguration())); } catch (MakeVariableExpander.ExpansionException e) { context.ruleError(String.format("Unable to expand make variables: %s", e.getMessage())); diff --git a/src/main/java/com/google/devtools/build/lib/rules/genrule/GenRuleBase.java b/src/main/java/com/google/devtools/build/lib/rules/genrule/GenRuleBase.java index ca50836a59..49372a9c3a 100644 --- a/src/main/java/com/google/devtools/build/lib/rules/genrule/GenRuleBase.java +++ b/src/main/java/com/google/devtools/build/lib/rules/genrule/GenRuleBase.java @@ -39,7 +39,6 @@ import com.google.devtools.build.lib.collect.nestedset.NestedSetBuilder; import com.google.devtools.build.lib.collect.nestedset.Order; import com.google.devtools.build.lib.packages.TargetUtils; import com.google.devtools.build.lib.rules.AliasProvider; -import com.google.devtools.build.lib.rules.MakeVariableProvider; import com.google.devtools.build.lib.rules.RuleConfiguredTargetFactory; import com.google.devtools.build.lib.rules.cpp.CppHelper; import com.google.devtools.build.lib.rules.java.JavaHelper; @@ -290,12 +289,17 @@ public abstract class GenRuleBase implements RuleConfiguredTargetFactory { private final NestedSet<Artifact> resolvedSrcs; private final NestedSet<Artifact> filesToBuild; - public CommandResolverContext(RuleContext ruleContext, NestedSet<Artifact> resolvedSrcs, + private static final ImmutableList<String> makeVariableAttributes = + ImmutableList.of(":cc_toolchain", "toolchains"); + + public CommandResolverContext( + RuleContext ruleContext, + NestedSet<Artifact> resolvedSrcs, NestedSet<Artifact> filesToBuild) { super( + ruleContext.getMakeVariables(makeVariableAttributes), ruleContext.getRule().getPackage(), - ruleContext.getConfiguration(), - MakeVariableProvider.getToolchainMakeVariables(ruleContext, "toolchains")); + ruleContext.getConfiguration()); this.ruleContext = ruleContext; this.resolvedSrcs = resolvedSrcs; this.filesToBuild = filesToBuild; @@ -344,7 +348,9 @@ public abstract class GenRuleBase implements RuleConfiguredTargetFactory { } } else if (JDK_MAKE_VARIABLE.matcher("$(" + name + ")").find()) { return new ConfigurationMakeVariableContext( - ruleContext.getTarget().getPackage(), ruleContext.getHostConfiguration()) + ruleContext.getMakeVariables(makeVariableAttributes), + ruleContext.getTarget().getPackage(), + ruleContext.getHostConfiguration()) .lookupMakeVariable(name); } else { return super.lookupMakeVariable(name); |