1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
// 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.BuildConfiguration;
import com.google.devtools.build.lib.analysis.configuredtargets.RuleConfiguredTarget.Mode;
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 com.google.devtools.build.lib.syntax.SkylarkDict;
import java.util.LinkedHashMap;
import java.util.Map;
/**
* 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) {
return 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 ->
Streams.stream(
ruleContext.getPrerequisites(
attrName, Mode.DONT_CHECK, TemplateVariableInfo.PROVIDER)))
.collect(ImmutableList.toImmutableList());
}
private final ImmutableList<? extends MakeVariableSupplier> allMakeVariableSuppliers;
// TODO(b/37567440): Remove when Skylark 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(":cc_toolchain", "toolchains", "$toolchains");
public ConfigurationMakeVariableContext(
RuleContext ruleContext, Package pkg, BuildConfiguration configuration) {
this(ruleContext, pkg, configuration, ImmutableList.<MakeVariableSupplier>of());
}
public ConfigurationMakeVariableContext(
RuleContext ruleContext,
Package pkg,
BuildConfiguration configuration,
Iterable<? extends MakeVariableSupplier> makeVariableSuppliers) {
this(
getRuleTemplateVariableProviders(ruleContext, DEFAULT_MAKE_VARIABLE_ATTRIBUTES),
pkg,
configuration,
makeVariableSuppliers);
}
private ConfigurationMakeVariableContext(
ImmutableList<TemplateVariableInfo> ruleTemplateVariableProviders,
Package pkg,
BuildConfiguration 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 (and thus the 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 {
for (MakeVariableSupplier supplier : allMakeVariableSuppliers) {
String variableValue = supplier.getMakeVariable(name);
if (variableValue != null) {
return variableValue;
}
}
throw new ExpansionException(String.format("$(%s) not defined", name));
}
public SkylarkDict<String, String> collectMakeVariables() {
Map<String, String> map = new LinkedHashMap<>();
// Collect variables in the reverse order as in lookupMakeVariable
// because each update is overwriting.
for (MakeVariableSupplier supplier : allMakeVariableSuppliers.reverse()) {
map.putAll(supplier.getAllMakeVariables());
}
return SkylarkDict.<String, String>copyOf(null, map);
}
@Override
public String lookupFunction(String name, String param) throws ExpansionException {
throw new ExpansionException(String.format("$(%s) not defined", name));
}
}
|
