blob: 1dc8ead0813dd1ce65e80573f81a7b8815996da1 [file] [log] [blame]
// Copyright 2014 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.devtools.build.lib.analysis;
import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Streams;
import com.google.devtools.build.lib.analysis.MakeVariableSupplier.MapBackedMakeVariableSupplier;
import com.google.devtools.build.lib.analysis.MakeVariableSupplier.TemplateVariableInfoBackedMakeVariableSupplier;
import com.google.devtools.build.lib.analysis.config.BuildConfigurationValue;
import com.google.devtools.build.lib.analysis.stringtemplate.ExpansionException;
import com.google.devtools.build.lib.analysis.stringtemplate.TemplateContext;
import com.google.devtools.build.lib.packages.Package;
import java.util.List;
import java.util.stream.Collectors;
import net.starlark.java.eval.Dict;
/**
* Implements make variable expansion for make variables that depend on the configuration and the
* target (not on behavior of the {@link ConfiguredTarget} implementation). Retrieved Make variable
* value can be modified using {@link MakeVariableSupplier}
*/
public class ConfigurationMakeVariableContext implements TemplateContext {
private static ImmutableList<TemplateVariableInfo> getRuleTemplateVariableProviders(
RuleContext ruleContext, Iterable<String> attributeNames) {
ImmutableList.Builder<TemplateVariableInfo> providers = new ImmutableList.Builder<>();
// Get template variable providers from the attributes.
List<TemplateVariableInfo> fromAttributes =
Streams.stream(attributeNames)
// Only process this attribute it if is present in the rule.
.filter(attrName -> ruleContext.attributes().has(attrName))
// Get the TemplateVariableInfo providers from this attribute.
.flatMap(
attrName ->
ruleContext.getPrerequisites(attrName, TemplateVariableInfo.PROVIDER).stream())
.collect(Collectors.toList());
providers.addAll(fromAttributes);
// Also collect template variable providers from any resolved toolchains.
if (ruleContext.getToolchainContexts() != null) {
ruleContext
.getToolchainContexts()
.getContextMap()
.values()
.forEach(context -> providers.addAll(context.templateVariableProviders()));
}
return providers.build();
}
private final ImmutableList<? extends MakeVariableSupplier> allMakeVariableSuppliers;
// TODO(b/37567440): Remove when Starlark callers can be updated to get this from
// CcToolchainProvider. We should use CcCommon.CC_TOOLCHAIN_ATTRIBUTE_NAME, but we didn't want to
// pollute core with C++ specific constant.
protected static final ImmutableList<String> DEFAULT_MAKE_VARIABLE_ATTRIBUTES =
ImmutableList.of("toolchains", ":cc_toolchain", "$toolchains", "$cc_toolchain");
public ConfigurationMakeVariableContext(
RuleContext ruleContext, Package pkg, BuildConfigurationValue configuration) {
this(ruleContext, pkg, configuration, ImmutableList.<MakeVariableSupplier>of());
}
public ConfigurationMakeVariableContext(
RuleContext ruleContext,
Package pkg,
BuildConfigurationValue configuration,
Iterable<? extends MakeVariableSupplier> makeVariableSuppliers) {
this(
getRuleTemplateVariableProviders(ruleContext, DEFAULT_MAKE_VARIABLE_ATTRIBUTES),
pkg,
configuration,
makeVariableSuppliers);
}
private ConfigurationMakeVariableContext(
ImmutableList<TemplateVariableInfo> ruleTemplateVariableProviders,
Package pkg,
BuildConfigurationValue configuration,
Iterable<? extends MakeVariableSupplier> extraMakeVariableSuppliers) {
this.allMakeVariableSuppliers =
ImmutableList.<MakeVariableSupplier>builder()
// These should be in priority order:
// 1) extra suppliers passed in (assume the caller knows what they are doing)
// 2) variables from the command-line
// 3) package-level overrides (ie, vardef)
// 4) variables from the rule (including from resolved toolchains)
// 5) variables from the global configuration
.addAll(Preconditions.checkNotNull(extraMakeVariableSuppliers))
.add(new MapBackedMakeVariableSupplier(configuration.getCommandLineBuildVariables()))
.add(new MapBackedMakeVariableSupplier(pkg.getMakeEnvironment()))
.add(new TemplateVariableInfoBackedMakeVariableSupplier(ruleTemplateVariableProviders))
.add(new MapBackedMakeVariableSupplier(configuration.getGlobalMakeEnvironment()))
.build();
}
@Override
public String lookupVariable(String name) throws ExpansionException, InterruptedException {
for (MakeVariableSupplier supplier : allMakeVariableSuppliers) {
String variableValue = supplier.getMakeVariable(name);
if (variableValue != null) {
return variableValue;
}
}
throw new ExpansionException(String.format("$(%s) not defined", name));
}
public Dict.Builder<String, String> collectMakeVariables()
throws ExpansionException, InterruptedException {
Dict.Builder<String, String> map = Dict.builder();
// Collect variables in the reverse order as in lookupMakeVariable
// because each update is overwriting.
for (MakeVariableSupplier supplier : allMakeVariableSuppliers.reverse()) {
map.putAll(supplier.getAllMakeVariables());
}
return map;
}
@Override
public String lookupFunction(String name, String param) throws ExpansionException {
throw new ExpansionException(String.format("$(%s) not defined", name));
}
}