blob: 0e6596bf7cc9482d836d8a23557deec7a98bbbed [file] [log] [blame]
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001// Copyright 2020 Google Inc. All rights reserved.
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15package bp2build
16
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux0da7ce62021-08-23 17:04:20 +000017/*
18For shareable/common functionality for conversion from soong-module to build files
19for queryview/bp2build
20*/
21
Liz Kammer2dd9ca42020-11-25 16:06:39 -080022import (
Liz Kammer2dd9ca42020-11-25 16:06:39 -080023 "fmt"
24 "reflect"
Jingwen Chen49109762021-05-25 05:16:48 +000025 "sort"
Liz Kammer2dd9ca42020-11-25 16:06:39 -080026 "strings"
27
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux0da7ce62021-08-23 17:04:20 +000028 "android/soong/android"
29 "android/soong/bazel"
Liz Kammer72beb342022-02-03 08:42:10 -050030 "android/soong/starlark_fmt"
Chris Parsons39a16972023-06-08 14:28:51 +000031 "android/soong/ui/metrics/bp2build_metrics_proto"
Liz Kammer2dd9ca42020-11-25 16:06:39 -080032 "github.com/google/blueprint"
Spandan Dasea2abba2023-06-14 21:30:38 +000033 "github.com/google/blueprint/bootstrap"
Liz Kammer2dd9ca42020-11-25 16:06:39 -080034 "github.com/google/blueprint/proptools"
35)
36
37type BazelAttributes struct {
38 Attrs map[string]string
39}
40
41type BazelTarget struct {
Jingwen Chen40067de2021-01-26 21:58:43 -050042 name string
Jingwen Chenc63677b2021-06-17 05:43:19 +000043 packageName string
Jingwen Chen40067de2021-01-26 21:58:43 -050044 content string
45 ruleClass string
46 bzlLoadLocation string
47}
48
49// IsLoadedFromStarlark determines if the BazelTarget's rule class is loaded from a .bzl file,
50// as opposed to a native rule built into Bazel.
51func (t BazelTarget) IsLoadedFromStarlark() bool {
52 return t.bzlLoadLocation != ""
53}
54
Jingwen Chenc63677b2021-06-17 05:43:19 +000055// Label is the fully qualified Bazel label constructed from the BazelTarget's
56// package name and target name.
57func (t BazelTarget) Label() string {
58 if t.packageName == "." {
59 return "//:" + t.name
60 } else {
61 return "//" + t.packageName + ":" + t.name
62 }
63}
64
Spandan Dasabedff02023-03-07 19:24:34 +000065// PackageName returns the package of the Bazel target.
66// Defaults to root of tree.
67func (t BazelTarget) PackageName() string {
68 if t.packageName == "" {
69 return "."
70 }
71 return t.packageName
72}
73
Jingwen Chen40067de2021-01-26 21:58:43 -050074// BazelTargets is a typedef for a slice of BazelTarget objects.
75type BazelTargets []BazelTarget
76
Sasha Smundak8bea2672022-08-04 13:31:14 -070077func (targets BazelTargets) packageRule() *BazelTarget {
78 for _, target := range targets {
79 if target.ruleClass == "package" {
80 return &target
81 }
82 }
83 return nil
84}
85
86// sort a list of BazelTargets in-place, by name, and by generated/handcrafted types.
Jingwen Chen49109762021-05-25 05:16:48 +000087func (targets BazelTargets) sort() {
88 sort.Slice(targets, func(i, j int) bool {
Jingwen Chen49109762021-05-25 05:16:48 +000089 return targets[i].name < targets[j].name
90 })
91}
92
Jingwen Chen40067de2021-01-26 21:58:43 -050093// String returns the string representation of BazelTargets, without load
94// statements (use LoadStatements for that), since the targets are usually not
95// adjacent to the load statements at the top of the BUILD file.
96func (targets BazelTargets) String() string {
97 var res string
98 for i, target := range targets {
Sasha Smundak8bea2672022-08-04 13:31:14 -070099 if target.ruleClass != "package" {
100 res += target.content
101 }
Jingwen Chen40067de2021-01-26 21:58:43 -0500102 if i != len(targets)-1 {
103 res += "\n\n"
104 }
105 }
106 return res
107}
108
109// LoadStatements return the string representation of the sorted and deduplicated
110// Starlark rule load statements needed by a group of BazelTargets.
111func (targets BazelTargets) LoadStatements() string {
112 bzlToLoadedSymbols := map[string][]string{}
113 for _, target := range targets {
114 if target.IsLoadedFromStarlark() {
115 bzlToLoadedSymbols[target.bzlLoadLocation] =
116 append(bzlToLoadedSymbols[target.bzlLoadLocation], target.ruleClass)
117 }
118 }
119
120 var loadStatements []string
121 for bzl, ruleClasses := range bzlToLoadedSymbols {
122 loadStatement := "load(\""
123 loadStatement += bzl
124 loadStatement += "\", "
125 ruleClasses = android.SortedUniqueStrings(ruleClasses)
126 for i, ruleClass := range ruleClasses {
127 loadStatement += "\"" + ruleClass + "\""
128 if i != len(ruleClasses)-1 {
129 loadStatement += ", "
130 }
131 }
132 loadStatement += ")"
133 loadStatements = append(loadStatements, loadStatement)
134 }
135 return strings.Join(android.SortedUniqueStrings(loadStatements), "\n")
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800136}
137
138type bpToBuildContext interface {
139 ModuleName(module blueprint.Module) string
140 ModuleDir(module blueprint.Module) string
141 ModuleSubDir(module blueprint.Module) string
142 ModuleType(module blueprint.Module) string
143
Jingwen Chendaa54bc2020-12-14 02:58:54 -0500144 VisitAllModules(visit func(blueprint.Module))
145 VisitDirectDeps(module blueprint.Module, visit func(blueprint.Module))
146}
147
148type CodegenContext struct {
Jingwen Chen16d90a82021-09-17 07:16:13 +0000149 config android.Config
Paul Duffinc6390592022-11-04 13:35:21 +0000150 context *android.Context
Jingwen Chen16d90a82021-09-17 07:16:13 +0000151 mode CodegenMode
152 additionalDeps []string
Liz Kammer6eff3232021-08-26 08:37:59 -0400153 unconvertedDepMode unconvertedDepsMode
Cole Faustb85d1a12022-11-08 18:14:01 -0800154 topDir string
Jingwen Chendaa54bc2020-12-14 02:58:54 -0500155}
156
Usta Shresthadb46a9b2022-07-11 11:29:56 -0400157func (ctx *CodegenContext) Mode() CodegenMode {
158 return ctx.mode
Jingwen Chen164e0862021-02-19 00:48:40 -0500159}
160
Jingwen Chen33832f92021-01-24 22:55:54 -0500161// CodegenMode is an enum to differentiate code-generation modes.
162type CodegenMode int
163
164const (
Usta Shresthadb46a9b2022-07-11 11:29:56 -0400165 // Bp2Build - generate BUILD files with targets buildable by Bazel directly.
Jingwen Chen33832f92021-01-24 22:55:54 -0500166 //
167 // This mode is used for the Soong->Bazel build definition conversion.
168 Bp2Build CodegenMode = iota
169
Usta Shresthadb46a9b2022-07-11 11:29:56 -0400170 // QueryView - generate BUILD files with targets representing fully mutated
Jingwen Chen33832f92021-01-24 22:55:54 -0500171 // Soong modules, representing the fully configured Soong module graph with
Usta Shresthadb46a9b2022-07-11 11:29:56 -0400172 // variants and dependency edges.
Jingwen Chen33832f92021-01-24 22:55:54 -0500173 //
174 // This mode is used for discovering and introspecting the existing Soong
175 // module graph.
176 QueryView
Spandan Das5af0bd32022-09-28 20:43:08 +0000177
178 // ApiBp2build - generate BUILD files for API contribution targets
179 ApiBp2build
Jingwen Chen33832f92021-01-24 22:55:54 -0500180)
181
Liz Kammer6eff3232021-08-26 08:37:59 -0400182type unconvertedDepsMode int
183
184const (
185 // Include a warning in conversion metrics about converted modules with unconverted direct deps
186 warnUnconvertedDeps unconvertedDepsMode = iota
187 // Error and fail conversion if encountering a module with unconverted direct deps
188 // Enabled by setting environment variable `BP2BUILD_ERROR_UNCONVERTED`
189 errorModulesUnconvertedDeps
190)
191
Jingwen Chendcc329a2021-01-26 02:49:03 -0500192func (mode CodegenMode) String() string {
193 switch mode {
194 case Bp2Build:
195 return "Bp2Build"
196 case QueryView:
197 return "QueryView"
Spandan Das5af0bd32022-09-28 20:43:08 +0000198 case ApiBp2build:
199 return "ApiBp2build"
Jingwen Chendcc329a2021-01-26 02:49:03 -0500200 default:
201 return fmt.Sprintf("%d", mode)
202 }
203}
204
Liz Kammerba3ea162021-02-17 13:22:03 -0500205// AddNinjaFileDeps adds dependencies on the specified files to be added to the ninja manifest. The
206// primary builder will be rerun whenever the specified files are modified. Allows us to fulfill the
207// PathContext interface in order to add dependencies on hand-crafted BUILD files. Note: must also
208// call AdditionalNinjaDeps and add them manually to the ninja file.
209func (ctx *CodegenContext) AddNinjaFileDeps(deps ...string) {
210 ctx.additionalDeps = append(ctx.additionalDeps, deps...)
211}
212
213// AdditionalNinjaDeps returns additional ninja deps added by CodegenContext
214func (ctx *CodegenContext) AdditionalNinjaDeps() []string {
215 return ctx.additionalDeps
216}
217
Paul Duffinc6390592022-11-04 13:35:21 +0000218func (ctx *CodegenContext) Config() android.Config { return ctx.config }
219func (ctx *CodegenContext) Context() *android.Context { return ctx.context }
Jingwen Chendaa54bc2020-12-14 02:58:54 -0500220
221// NewCodegenContext creates a wrapper context that conforms to PathContext for
222// writing BUILD files in the output directory.
Cole Faustb85d1a12022-11-08 18:14:01 -0800223func NewCodegenContext(config android.Config, context *android.Context, mode CodegenMode, topDir string) *CodegenContext {
Liz Kammer6eff3232021-08-26 08:37:59 -0400224 var unconvertedDeps unconvertedDepsMode
225 if config.IsEnvTrue("BP2BUILD_ERROR_UNCONVERTED") {
226 unconvertedDeps = errorModulesUnconvertedDeps
227 }
Liz Kammerba3ea162021-02-17 13:22:03 -0500228 return &CodegenContext{
Liz Kammer6eff3232021-08-26 08:37:59 -0400229 context: context,
230 config: config,
231 mode: mode,
232 unconvertedDepMode: unconvertedDeps,
Cole Faustb85d1a12022-11-08 18:14:01 -0800233 topDir: topDir,
Jingwen Chendaa54bc2020-12-14 02:58:54 -0500234 }
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800235}
236
237// props is an unsorted map. This function ensures that
238// the generated attributes are sorted to ensure determinism.
239func propsToAttributes(props map[string]string) string {
240 var attributes string
Cole Faust18994c72023-02-28 16:02:16 -0800241 for _, propName := range android.SortedKeys(props) {
Liz Kammer0eae52e2021-10-06 10:32:26 -0400242 attributes += fmt.Sprintf(" %s = %s,\n", propName, props[propName])
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800243 }
244 return attributes
245}
246
Liz Kammer6eff3232021-08-26 08:37:59 -0400247type conversionResults struct {
248 buildFileToTargets map[string]BazelTargets
249 metrics CodegenMetrics
Liz Kammer6eff3232021-08-26 08:37:59 -0400250}
251
252func (r conversionResults) BuildDirToTargets() map[string]BazelTargets {
253 return r.buildFileToTargets
254}
255
Spandan Dasea2abba2023-06-14 21:30:38 +0000256// struct to store state of go bazel targets
257// this implements bp2buildModule interface and is passed to generateBazelTargets
258type goBazelTarget struct {
259 targetName string
260 targetPackage string
261 bazelRuleClass string
262 bazelRuleLoadLocation string
263 bazelAttributes []interface{}
264}
265
266var _ bp2buildModule = (*goBazelTarget)(nil)
267
268func (g goBazelTarget) TargetName() string {
269 return g.targetName
270}
271
272func (g goBazelTarget) TargetPackage() string {
273 return g.targetPackage
274}
275
276func (g goBazelTarget) BazelRuleClass() string {
277 return g.bazelRuleClass
278}
279
280func (g goBazelTarget) BazelRuleLoadLocation() string {
281 return g.bazelRuleLoadLocation
282}
283
284func (g goBazelTarget) BazelAttributes() []interface{} {
285 return g.bazelAttributes
286}
287
288// Creates a target_compatible_with entry that is *not* compatible with android
289func targetNotCompatibleWithAndroid() bazel.LabelListAttribute {
290 ret := bazel.LabelListAttribute{}
291 ret.SetSelectValue(bazel.OsConfigurationAxis, bazel.OsAndroid,
292 bazel.MakeLabelList(
293 []bazel.Label{
294 bazel.Label{
295 Label: "@platforms//:incompatible",
296 },
297 },
298 ),
299 )
300 return ret
301}
302
303// helper function to return labels for srcs used in bootstrap_go_package and bootstrap_go_binary
304// this function has the following limitations which make it unsuitable for widespread use
Spandan Das0a8a2752023-06-21 01:50:33 +0000305// - wildcard patterns in srcs
306// This is ok for go since build/blueprint does not support it.
Spandan Dasea2abba2023-06-14 21:30:38 +0000307//
308// Prefer to use `BazelLabelForModuleSrc` instead
Spandan Das0a8a2752023-06-21 01:50:33 +0000309func goSrcLabels(cfg android.Config, moduleDir string, srcs []string, linuxSrcs, darwinSrcs []string) bazel.LabelListAttribute {
Spandan Dasea2abba2023-06-14 21:30:38 +0000310 labels := func(srcs []string) bazel.LabelList {
311 ret := []bazel.Label{}
312 for _, src := range srcs {
313 srcLabel := bazel.Label{
Spandan Das0a8a2752023-06-21 01:50:33 +0000314 Label: src,
Spandan Dasea2abba2023-06-14 21:30:38 +0000315 }
316 ret = append(ret, srcLabel)
317 }
Spandan Das0a8a2752023-06-21 01:50:33 +0000318 // Respect package boundaries
319 return android.TransformSubpackagePaths(
320 cfg,
321 moduleDir,
322 bazel.MakeLabelList(ret),
323 )
Spandan Dasea2abba2023-06-14 21:30:38 +0000324 }
325
326 ret := bazel.LabelListAttribute{}
327 // common
328 ret.SetSelectValue(bazel.NoConfigAxis, "", labels(srcs))
329 // linux
330 ret.SetSelectValue(bazel.OsConfigurationAxis, bazel.OsLinux, labels(linuxSrcs))
331 // darwin
332 ret.SetSelectValue(bazel.OsConfigurationAxis, bazel.OsDarwin, labels(darwinSrcs))
333 return ret
334}
335
336func goDepLabels(deps []string, goModulesMap nameToGoLibraryModule) bazel.LabelListAttribute {
337 labels := []bazel.Label{}
338 for _, dep := range deps {
339 moduleDir := goModulesMap[dep].Dir
340 if moduleDir == "." {
341 moduleDir = ""
342 }
343 label := bazel.Label{
344 Label: fmt.Sprintf("//%s:%s", moduleDir, dep),
345 }
346 labels = append(labels, label)
347 }
348 return bazel.MakeLabelListAttribute(bazel.MakeLabelList(labels))
349}
350
351// attributes common to blueprint_go_binary and bootstap_go_package
352type goAttributes struct {
353 Importpath bazel.StringAttribute
354 Srcs bazel.LabelListAttribute
355 Deps bazel.LabelListAttribute
356 Target_compatible_with bazel.LabelListAttribute
357}
358
359func generateBazelTargetsGoPackage(ctx *android.Context, g *bootstrap.GoPackage, goModulesMap nameToGoLibraryModule) ([]BazelTarget, []error) {
360 ca := android.CommonAttributes{
361 Name: g.Name(),
362 }
Spandan Dasde623292023-06-14 21:30:38 +0000363
364 // For this bootstrap_go_package dep chain,
365 // A --> B --> C ( ---> depends on)
366 // Soong provides the convenience of only listing B as deps of A even if a src file of A imports C
367 // Bazel OTOH
368 // 1. requires C to be listed in `deps` expllicity.
369 // 2. does not require C to be listed if src of A does not import C
370 //
371 // bp2build does not have sufficient info on whether C is a direct dep of A or not, so for now collect all transitive deps and add them to deps
372 transitiveDeps := transitiveGoDeps(g.Deps(), goModulesMap)
373
Spandan Dasea2abba2023-06-14 21:30:38 +0000374 ga := goAttributes{
375 Importpath: bazel.StringAttribute{
376 Value: proptools.StringPtr(g.GoPkgPath()),
377 },
Spandan Das0a8a2752023-06-21 01:50:33 +0000378 Srcs: goSrcLabels(ctx.Config(), ctx.ModuleDir(g), g.Srcs(), g.LinuxSrcs(), g.DarwinSrcs()),
379 Deps: goDepLabels(
380 android.FirstUniqueStrings(transitiveDeps),
381 goModulesMap,
382 ),
Spandan Dasea2abba2023-06-14 21:30:38 +0000383 Target_compatible_with: targetNotCompatibleWithAndroid(),
384 }
385
386 lib := goBazelTarget{
387 targetName: g.Name(),
388 targetPackage: ctx.ModuleDir(g),
389 bazelRuleClass: "go_library",
390 bazelRuleLoadLocation: "@io_bazel_rules_go//go:def.bzl",
391 bazelAttributes: []interface{}{&ca, &ga},
392 }
393 // TODO - b/284483729: Create go_test target from testSrcs
394 libTarget, err := generateBazelTarget(ctx, lib)
395 if err != nil {
396 return []BazelTarget{}, []error{err}
397 }
398 return []BazelTarget{libTarget}, nil
399}
400
401type goLibraryModule struct {
402 Dir string
403 Deps []string
404}
405
406type nameToGoLibraryModule map[string]goLibraryModule
407
408// Visit each module in the graph
409// If a module is of type `bootstrap_go_package`, return a map containing metadata like its dir and deps
410func createGoLibraryModuleMap(ctx *android.Context) nameToGoLibraryModule {
411 ret := nameToGoLibraryModule{}
412 ctx.VisitAllModules(func(m blueprint.Module) {
413 moduleType := ctx.ModuleType(m)
414 // We do not need to store information about blueprint_go_binary since it does not have any rdeps
415 if moduleType == "bootstrap_go_package" {
416 ret[m.Name()] = goLibraryModule{
417 Dir: ctx.ModuleDir(m),
418 Deps: m.(*bootstrap.GoPackage).Deps(),
419 }
420 }
421 })
422 return ret
423}
424
Spandan Dasde623292023-06-14 21:30:38 +0000425// Returns the deps in the transitive closure of a go target
426func transitiveGoDeps(directDeps []string, goModulesMap nameToGoLibraryModule) []string {
427 allDeps := directDeps
428 i := 0
429 for i < len(allDeps) {
430 curr := allDeps[i]
431 allDeps = append(allDeps, goModulesMap[curr].Deps...)
432 i += 1
433 }
434 allDeps = android.SortedUniqueStrings(allDeps)
435 return allDeps
436}
437
438func generateBazelTargetsGoBinary(ctx *android.Context, g *bootstrap.GoBinary, goModulesMap nameToGoLibraryModule) ([]BazelTarget, []error) {
439 ca := android.CommonAttributes{
440 Name: g.Name(),
441 }
442
443 // For this bootstrap_go_package dep chain,
444 // A --> B --> C ( ---> depends on)
445 // Soong provides the convenience of only listing B as deps of A even if a src file of A imports C
446 // Bazel OTOH
447 // 1. requires C to be listed in `deps` expllicity.
448 // 2. does not require C to be listed if src of A does not import C
449 //
450 // bp2build does not have sufficient info on whether C is a direct dep of A or not, so for now collect all transitive deps and add them to deps
451 transitiveDeps := transitiveGoDeps(g.Deps(), goModulesMap)
452
453 ga := goAttributes{
Spandan Das0a8a2752023-06-21 01:50:33 +0000454 Srcs: goSrcLabels(ctx.Config(), ctx.ModuleDir(g), g.Srcs(), g.LinuxSrcs(), g.DarwinSrcs()),
Spandan Dasde623292023-06-14 21:30:38 +0000455 Deps: goDepLabels(transitiveDeps, goModulesMap),
456 Target_compatible_with: targetNotCompatibleWithAndroid(),
457 }
458
459 bin := goBazelTarget{
460 targetName: g.Name(),
461 targetPackage: ctx.ModuleDir(g),
462 bazelRuleClass: "go_binary",
463 bazelRuleLoadLocation: "@io_bazel_rules_go//go:def.bzl",
464 bazelAttributes: []interface{}{&ca, &ga},
465 }
466 // TODO - b/284483729: Create go_test target from testSrcs
467 binTarget, err := generateBazelTarget(ctx, bin)
468 if err != nil {
469 return []BazelTarget{}, []error{err}
470 }
471 return []BazelTarget{binTarget}, nil
472}
473
Liz Kammer6eff3232021-08-26 08:37:59 -0400474func GenerateBazelTargets(ctx *CodegenContext, generateFilegroups bool) (conversionResults, []error) {
Jingwen Chen40067de2021-01-26 21:58:43 -0500475 buildFileToTargets := make(map[string]BazelTargets)
Jingwen Chen164e0862021-02-19 00:48:40 -0500476
477 // Simple metrics tracking for bp2build
usta4f5d2c12022-10-28 23:32:01 -0400478 metrics := CreateCodegenMetrics()
Jingwen Chen164e0862021-02-19 00:48:40 -0500479
Rupert Shuttleworth2a4fc3e2021-04-21 07:10:09 -0400480 dirs := make(map[string]bool)
481
Liz Kammer6eff3232021-08-26 08:37:59 -0400482 var errs []error
483
Spandan Dasea2abba2023-06-14 21:30:38 +0000484 // Visit go libraries in a pre-run and store its state in a map
485 // The time complexity remains O(N), and this does not add significant wall time.
486 nameToGoLibMap := createGoLibraryModuleMap(ctx.Context())
487
Jingwen Chen164e0862021-02-19 00:48:40 -0500488 bpCtx := ctx.Context()
489 bpCtx.VisitAllModules(func(m blueprint.Module) {
490 dir := bpCtx.ModuleDir(m)
Chris Parsons492bd912022-01-20 12:55:05 -0500491 moduleType := bpCtx.ModuleType(m)
Rupert Shuttleworth2a4fc3e2021-04-21 07:10:09 -0400492 dirs[dir] = true
493
Liz Kammer2ada09a2021-08-11 00:17:36 -0400494 var targets []BazelTarget
Spandan Dasea2abba2023-06-14 21:30:38 +0000495 var targetErrs []error
Jingwen Chen73850672020-12-14 08:25:34 -0500496
Jingwen Chen164e0862021-02-19 00:48:40 -0500497 switch ctx.Mode() {
Jingwen Chen33832f92021-01-24 22:55:54 -0500498 case Bp2Build:
Jingwen Chen310bc8f2021-09-20 10:54:27 +0000499 // There are two main ways of converting a Soong module to Bazel:
500 // 1) Manually handcrafting a Bazel target and associating the module with its label
501 // 2) Automatically generating with bp2build converters
502 //
503 // bp2build converters are used for the majority of modules.
Liz Kammerba3ea162021-02-17 13:22:03 -0500504 if b, ok := m.(android.Bazelable); ok && b.HasHandcraftedLabel() {
Jingwen Chen310bc8f2021-09-20 10:54:27 +0000505 // Handle modules converted to handcrafted targets.
506 //
507 // Since these modules are associated with some handcrafted
Cole Faustea602c52022-08-31 14:48:26 -0700508 // target in a BUILD file, we don't autoconvert them.
Jingwen Chen310bc8f2021-09-20 10:54:27 +0000509
510 // Log the module.
Chris Parsons39a16972023-06-08 14:28:51 +0000511 metrics.AddUnconvertedModule(m, moduleType, dir,
512 android.UnconvertedReason{
513 ReasonType: int(bp2build_metrics_proto.UnconvertedReasonType_DEFINED_IN_BUILD_FILE),
514 })
Liz Kammer2ada09a2021-08-11 00:17:36 -0400515 } else if aModule, ok := m.(android.Module); ok && aModule.IsConvertedByBp2build() {
Jingwen Chen310bc8f2021-09-20 10:54:27 +0000516 // Handle modules converted to generated targets.
517
518 // Log the module.
Chris Parsons39a16972023-06-08 14:28:51 +0000519 metrics.AddConvertedModule(aModule, moduleType, dir)
Jingwen Chen310bc8f2021-09-20 10:54:27 +0000520
521 // Handle modules with unconverted deps. By default, emit a warning.
Liz Kammer6eff3232021-08-26 08:37:59 -0400522 if unconvertedDeps := aModule.GetUnconvertedBp2buildDeps(); len(unconvertedDeps) > 0 {
Sasha Smundakf2bb26f2022-08-04 11:28:15 -0700523 msg := fmt.Sprintf("%s %s:%s depends on unconverted modules: %s",
524 moduleType, bpCtx.ModuleDir(m), m.Name(), strings.Join(unconvertedDeps, ", "))
Usta Shresthac6057152022-09-24 00:23:31 -0400525 switch ctx.unconvertedDepMode {
526 case warnUnconvertedDeps:
Liz Kammer6eff3232021-08-26 08:37:59 -0400527 metrics.moduleWithUnconvertedDepsMsgs = append(metrics.moduleWithUnconvertedDepsMsgs, msg)
Usta Shresthac6057152022-09-24 00:23:31 -0400528 case errorModulesUnconvertedDeps:
Liz Kammer6eff3232021-08-26 08:37:59 -0400529 errs = append(errs, fmt.Errorf(msg))
530 return
531 }
532 }
Liz Kammerdaa09ef2021-12-15 15:35:38 -0500533 if unconvertedDeps := aModule.GetMissingBp2buildDeps(); len(unconvertedDeps) > 0 {
Sasha Smundakf2bb26f2022-08-04 11:28:15 -0700534 msg := fmt.Sprintf("%s %s:%s depends on missing modules: %s",
535 moduleType, bpCtx.ModuleDir(m), m.Name(), strings.Join(unconvertedDeps, ", "))
Usta Shresthac6057152022-09-24 00:23:31 -0400536 switch ctx.unconvertedDepMode {
537 case warnUnconvertedDeps:
Liz Kammerdaa09ef2021-12-15 15:35:38 -0500538 metrics.moduleWithMissingDepsMsgs = append(metrics.moduleWithMissingDepsMsgs, msg)
Usta Shresthac6057152022-09-24 00:23:31 -0400539 case errorModulesUnconvertedDeps:
Liz Kammerdaa09ef2021-12-15 15:35:38 -0500540 errs = append(errs, fmt.Errorf(msg))
541 return
542 }
543 }
Alix94e26032022-08-16 20:37:33 +0000544 targets, targetErrs = generateBazelTargets(bpCtx, aModule)
545 errs = append(errs, targetErrs...)
Liz Kammer2ada09a2021-08-11 00:17:36 -0400546 for _, t := range targets {
Jingwen Chen310bc8f2021-09-20 10:54:27 +0000547 // A module can potentially generate more than 1 Bazel
548 // target, each of a different rule class.
549 metrics.IncrementRuleClassCount(t.ruleClass)
Liz Kammer2ada09a2021-08-11 00:17:36 -0400550 }
MarkDacek9c094ca2023-03-16 19:15:19 +0000551 } else if _, ok := ctx.Config().BazelModulesForceEnabledByFlag()[m.Name()]; ok && m.Name() != "" {
552 err := fmt.Errorf("Force Enabled Module %s not converted", m.Name())
553 errs = append(errs, err)
Chris Parsons39a16972023-06-08 14:28:51 +0000554 } else if aModule, ok := m.(android.Module); ok {
555 reason := aModule.GetUnconvertedReason()
556 if reason == nil {
557 panic(fmt.Errorf("module '%s' was neither converted nor marked unconvertible with bp2build", aModule.Name()))
558 } else {
559 metrics.AddUnconvertedModule(m, moduleType, dir, *reason)
560 }
561 return
Spandan Dasea2abba2023-06-14 21:30:38 +0000562 } else if glib, ok := m.(*bootstrap.GoPackage); ok {
563 targets, targetErrs = generateBazelTargetsGoPackage(bpCtx, glib, nameToGoLibMap)
564 errs = append(errs, targetErrs...)
565 metrics.IncrementRuleClassCount("go_library")
Spandan Das2a55cea2023-06-14 17:56:10 +0000566 } else if gbin, ok := m.(*bootstrap.GoBinary); ok {
Spandan Dasde623292023-06-14 21:30:38 +0000567 targets, targetErrs = generateBazelTargetsGoBinary(bpCtx, gbin, nameToGoLibMap)
568 errs = append(errs, targetErrs...)
569 metrics.IncrementRuleClassCount("go_binary")
Liz Kammerfc46bc12021-02-19 11:06:17 -0500570 } else {
Chris Parsons39a16972023-06-08 14:28:51 +0000571 metrics.AddUnconvertedModule(m, moduleType, dir, android.UnconvertedReason{
572 ReasonType: int(bp2build_metrics_proto.UnconvertedReasonType_TYPE_UNSUPPORTED),
573 })
Liz Kammerba3ea162021-02-17 13:22:03 -0500574 return
Jingwen Chen73850672020-12-14 08:25:34 -0500575 }
Jingwen Chen33832f92021-01-24 22:55:54 -0500576 case QueryView:
Jingwen Chen96af35b2021-02-08 00:49:32 -0500577 // Blocklist certain module types from being generated.
Jingwen Chen164e0862021-02-19 00:48:40 -0500578 if canonicalizeModuleType(bpCtx.ModuleType(m)) == "package" {
Jingwen Chen96af35b2021-02-08 00:49:32 -0500579 // package module name contain slashes, and thus cannot
580 // be mapped cleanly to a bazel label.
581 return
582 }
Alix94e26032022-08-16 20:37:33 +0000583 t, err := generateSoongModuleTarget(bpCtx, m)
584 if err != nil {
585 errs = append(errs, err)
586 }
Liz Kammer2ada09a2021-08-11 00:17:36 -0400587 targets = append(targets, t)
Spandan Das5af0bd32022-09-28 20:43:08 +0000588 case ApiBp2build:
589 if aModule, ok := m.(android.Module); ok && aModule.IsConvertedByBp2build() {
590 targets, errs = generateBazelTargets(bpCtx, aModule)
591 }
Jingwen Chen33832f92021-01-24 22:55:54 -0500592 default:
Liz Kammer6eff3232021-08-26 08:37:59 -0400593 errs = append(errs, fmt.Errorf("Unknown code-generation mode: %s", ctx.Mode()))
594 return
Jingwen Chen73850672020-12-14 08:25:34 -0500595 }
596
Spandan Dasabedff02023-03-07 19:24:34 +0000597 for _, target := range targets {
598 targetDir := target.PackageName()
599 buildFileToTargets[targetDir] = append(buildFileToTargets[targetDir], target)
600 }
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800601 })
Liz Kammer6eff3232021-08-26 08:37:59 -0400602
603 if len(errs) > 0 {
604 return conversionResults{}, errs
605 }
606
Rupert Shuttleworth2a4fc3e2021-04-21 07:10:09 -0400607 if generateFilegroups {
608 // Add a filegroup target that exposes all sources in the subtree of this package
609 // NOTE: This also means we generate a BUILD file for every Android.bp file (as long as it has at least one module)
Cole Faust324a92e2022-08-23 15:29:05 -0700610 //
611 // This works because: https://bazel.build/reference/be/functions#exports_files
612 // "As a legacy behaviour, also files mentioned as input to a rule are exported with the
613 // default visibility until the flag --incompatible_no_implicit_file_export is flipped. However, this behavior
614 // should not be relied upon and actively migrated away from."
615 //
616 // TODO(b/198619163): We should change this to export_files(glob(["**/*"])) instead, but doing that causes these errors:
617 // "Error in exports_files: generated label '//external/avb:avbtool' conflicts with existing py_binary rule"
618 // So we need to solve all the "target ... is both a rule and a file" warnings first.
Usta Shresthac6057152022-09-24 00:23:31 -0400619 for dir := range dirs {
Rupert Shuttleworth2a4fc3e2021-04-21 07:10:09 -0400620 buildFileToTargets[dir] = append(buildFileToTargets[dir], BazelTarget{
621 name: "bp2build_all_srcs",
622 content: `filegroup(name = "bp2build_all_srcs", srcs = glob(["**/*"]))`,
623 ruleClass: "filegroup",
624 })
625 }
626 }
Jingwen Chen164e0862021-02-19 00:48:40 -0500627
Liz Kammer6eff3232021-08-26 08:37:59 -0400628 return conversionResults{
629 buildFileToTargets: buildFileToTargets,
630 metrics: metrics,
Liz Kammer6eff3232021-08-26 08:37:59 -0400631 }, errs
Jingwen Chen164e0862021-02-19 00:48:40 -0500632}
633
Alix94e26032022-08-16 20:37:33 +0000634func generateBazelTargets(ctx bpToBuildContext, m android.Module) ([]BazelTarget, []error) {
Liz Kammer2ada09a2021-08-11 00:17:36 -0400635 var targets []BazelTarget
Alix94e26032022-08-16 20:37:33 +0000636 var errs []error
Liz Kammer2ada09a2021-08-11 00:17:36 -0400637 for _, m := range m.Bp2buildTargets() {
Alix94e26032022-08-16 20:37:33 +0000638 target, err := generateBazelTarget(ctx, m)
639 if err != nil {
640 errs = append(errs, err)
641 return targets, errs
642 }
643 targets = append(targets, target)
Liz Kammer2ada09a2021-08-11 00:17:36 -0400644 }
Alix94e26032022-08-16 20:37:33 +0000645 return targets, errs
Liz Kammer2ada09a2021-08-11 00:17:36 -0400646}
647
648type bp2buildModule interface {
649 TargetName() string
650 TargetPackage() string
651 BazelRuleClass() string
652 BazelRuleLoadLocation() string
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux447f6c92021-08-31 20:30:36 +0000653 BazelAttributes() []interface{}
Liz Kammer2ada09a2021-08-11 00:17:36 -0400654}
655
Alix94e26032022-08-16 20:37:33 +0000656func generateBazelTarget(ctx bpToBuildContext, m bp2buildModule) (BazelTarget, error) {
Liz Kammer2ada09a2021-08-11 00:17:36 -0400657 ruleClass := m.BazelRuleClass()
658 bzlLoadLocation := m.BazelRuleLoadLocation()
Jingwen Chen40067de2021-01-26 21:58:43 -0500659
Jingwen Chen73850672020-12-14 08:25:34 -0500660 // extract the bazel attributes from the module.
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux447f6c92021-08-31 20:30:36 +0000661 attrs := m.BazelAttributes()
Alix94e26032022-08-16 20:37:33 +0000662 props, err := extractModuleProperties(attrs, true)
663 if err != nil {
664 return BazelTarget{}, err
665 }
Jingwen Chen73850672020-12-14 08:25:34 -0500666
Liz Kammer0eae52e2021-10-06 10:32:26 -0400667 // name is handled in a special manner
668 delete(props.Attrs, "name")
Jingwen Chen77e8b7b2021-02-05 03:03:24 -0500669
Jingwen Chen73850672020-12-14 08:25:34 -0500670 // Return the Bazel target with rule class and attributes, ready to be
671 // code-generated.
672 attributes := propsToAttributes(props.Attrs)
Sasha Smundakfb589492022-08-04 11:13:27 -0700673 var content string
Liz Kammer2ada09a2021-08-11 00:17:36 -0400674 targetName := m.TargetName()
Sasha Smundakfb589492022-08-04 11:13:27 -0700675 if targetName != "" {
676 content = fmt.Sprintf(ruleTargetTemplate, ruleClass, targetName, attributes)
677 } else {
678 content = fmt.Sprintf(unnamedRuleTargetTemplate, ruleClass, attributes)
679 }
Jingwen Chen73850672020-12-14 08:25:34 -0500680 return BazelTarget{
Jingwen Chen40067de2021-01-26 21:58:43 -0500681 name: targetName,
Liz Kammer2ada09a2021-08-11 00:17:36 -0400682 packageName: m.TargetPackage(),
Jingwen Chen40067de2021-01-26 21:58:43 -0500683 ruleClass: ruleClass,
684 bzlLoadLocation: bzlLoadLocation,
Sasha Smundakfb589492022-08-04 11:13:27 -0700685 content: content,
Alix94e26032022-08-16 20:37:33 +0000686 }, nil
Jingwen Chen73850672020-12-14 08:25:34 -0500687}
688
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800689// Convert a module and its deps and props into a Bazel macro/rule
690// representation in the BUILD file.
Alix94e26032022-08-16 20:37:33 +0000691func generateSoongModuleTarget(ctx bpToBuildContext, m blueprint.Module) (BazelTarget, error) {
692 props, err := getBuildProperties(ctx, m)
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800693
694 // TODO(b/163018919): DirectDeps can have duplicate (module, variant)
695 // items, if the modules are added using different DependencyTag. Figure
696 // out the implications of that.
697 depLabels := map[string]bool{}
698 if aModule, ok := m.(android.Module); ok {
Jingwen Chendaa54bc2020-12-14 02:58:54 -0500699 ctx.VisitDirectDeps(aModule, func(depModule blueprint.Module) {
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800700 depLabels[qualifiedTargetLabel(ctx, depModule)] = true
701 })
702 }
Liz Kammer0eae52e2021-10-06 10:32:26 -0400703
Usta Shresthadb46a9b2022-07-11 11:29:56 -0400704 for p := range ignoredPropNames {
Liz Kammer0eae52e2021-10-06 10:32:26 -0400705 delete(props.Attrs, p)
706 }
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800707 attributes := propsToAttributes(props.Attrs)
708
709 depLabelList := "[\n"
Usta Shresthadb46a9b2022-07-11 11:29:56 -0400710 for depLabel := range depLabels {
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800711 depLabelList += fmt.Sprintf(" %q,\n", depLabel)
712 }
713 depLabelList += " ]"
714
715 targetName := targetNameWithVariant(ctx, m)
716 return BazelTarget{
Spandan Dasabedff02023-03-07 19:24:34 +0000717 name: targetName,
718 packageName: ctx.ModuleDir(m),
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800719 content: fmt.Sprintf(
Sasha Smundakfb589492022-08-04 11:13:27 -0700720 soongModuleTargetTemplate,
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800721 targetName,
722 ctx.ModuleName(m),
723 canonicalizeModuleType(ctx.ModuleType(m)),
724 ctx.ModuleSubDir(m),
725 depLabelList,
726 attributes),
Alix94e26032022-08-16 20:37:33 +0000727 }, err
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800728}
729
Alix94e26032022-08-16 20:37:33 +0000730func getBuildProperties(ctx bpToBuildContext, m blueprint.Module) (BazelAttributes, error) {
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800731 // TODO: this omits properties for blueprint modules (blueprint_go_binary,
732 // bootstrap_go_binary, bootstrap_go_package), which will have to be handled separately.
733 if aModule, ok := m.(android.Module); ok {
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux447f6c92021-08-31 20:30:36 +0000734 return extractModuleProperties(aModule.GetProperties(), false)
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800735 }
736
Alix94e26032022-08-16 20:37:33 +0000737 return BazelAttributes{}, nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800738}
739
740// Generically extract module properties and types into a map, keyed by the module property name.
Alix94e26032022-08-16 20:37:33 +0000741func extractModuleProperties(props []interface{}, checkForDuplicateProperties bool) (BazelAttributes, error) {
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800742 ret := map[string]string{}
743
744 // Iterate over this android.Module's property structs.
Liz Kammer2ada09a2021-08-11 00:17:36 -0400745 for _, properties := range props {
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800746 propertiesValue := reflect.ValueOf(properties)
747 // Check that propertiesValue is a pointer to the Properties struct, like
748 // *cc.BaseLinkerProperties or *java.CompilerProperties.
749 //
750 // propertiesValue can also be type-asserted to the structs to
751 // manipulate internal props, if needed.
752 if isStructPtr(propertiesValue.Type()) {
753 structValue := propertiesValue.Elem()
Alix94e26032022-08-16 20:37:33 +0000754 ok, err := extractStructProperties(structValue, 0)
755 if err != nil {
756 return BazelAttributes{}, err
757 }
758 for k, v := range ok {
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux447f6c92021-08-31 20:30:36 +0000759 if existing, exists := ret[k]; checkForDuplicateProperties && exists {
Alix94e26032022-08-16 20:37:33 +0000760 return BazelAttributes{}, fmt.Errorf(
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux447f6c92021-08-31 20:30:36 +0000761 "%s (%v) is present in properties whereas it should be consolidated into a commonAttributes",
Alix94e26032022-08-16 20:37:33 +0000762 k, existing)
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux447f6c92021-08-31 20:30:36 +0000763 }
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800764 ret[k] = v
765 }
766 } else {
Alix94e26032022-08-16 20:37:33 +0000767 return BazelAttributes{},
768 fmt.Errorf(
769 "properties must be a pointer to a struct, got %T",
770 propertiesValue.Interface())
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800771 }
772 }
773
Liz Kammer2ada09a2021-08-11 00:17:36 -0400774 return BazelAttributes{
775 Attrs: ret,
Alix94e26032022-08-16 20:37:33 +0000776 }, nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800777}
778
779func isStructPtr(t reflect.Type) bool {
780 return t.Kind() == reflect.Ptr && t.Elem().Kind() == reflect.Struct
781}
782
783// prettyPrint a property value into the equivalent Starlark representation
784// recursively.
Jingwen Chen58ff6802021-11-17 12:14:41 +0000785func prettyPrint(propertyValue reflect.Value, indent int, emitZeroValues bool) (string, error) {
786 if !emitZeroValues && isZero(propertyValue) {
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800787 // A property value being set or unset actually matters -- Soong does set default
788 // values for unset properties, like system_shared_libs = ["libc", "libm", "libdl"] at
789 // https://cs.android.com/android/platform/superproject/+/master:build/soong/cc/linker.go;l=281-287;drc=f70926eef0b9b57faf04c17a1062ce50d209e480
790 //
Jingwen Chenfc490bd2021-03-30 10:24:19 +0000791 // In Bazel-parlance, we would use "attr.<type>(default = <default
792 // value>)" to set the default value of unset attributes. In the cases
793 // where the bp2build converter didn't set the default value within the
794 // mutator when creating the BazelTargetModule, this would be a zero
Jingwen Chen63930982021-03-24 10:04:33 -0400795 // value. For those cases, we return an empty string so we don't
796 // unnecessarily generate empty values.
797 return "", nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800798 }
799
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800800 switch propertyValue.Kind() {
801 case reflect.String:
Liz Kammer72beb342022-02-03 08:42:10 -0500802 return fmt.Sprintf("\"%v\"", escapeString(propertyValue.String())), nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800803 case reflect.Bool:
Liz Kammer72beb342022-02-03 08:42:10 -0500804 return starlark_fmt.PrintBool(propertyValue.Bool()), nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800805 case reflect.Int, reflect.Uint, reflect.Int64:
Liz Kammer72beb342022-02-03 08:42:10 -0500806 return fmt.Sprintf("%v", propertyValue.Interface()), nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800807 case reflect.Ptr:
Jingwen Chen58ff6802021-11-17 12:14:41 +0000808 return prettyPrint(propertyValue.Elem(), indent, emitZeroValues)
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800809 case reflect.Slice:
Liz Kammer72beb342022-02-03 08:42:10 -0500810 elements := make([]string, 0, propertyValue.Len())
811 for i := 0; i < propertyValue.Len(); i++ {
812 val, err := prettyPrint(propertyValue.Index(i), indent, emitZeroValues)
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800813 if err != nil {
814 return "", err
815 }
Liz Kammer72beb342022-02-03 08:42:10 -0500816 if val != "" {
817 elements = append(elements, val)
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800818 }
819 }
Sam Delmerico932c01c2022-03-25 16:33:26 +0000820 return starlark_fmt.PrintList(elements, indent, func(s string) string {
821 return "%s"
822 }), nil
Jingwen Chenb4628eb2021-04-08 14:40:57 +0000823
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800824 case reflect.Struct:
Jingwen Chen5d864492021-02-24 07:20:12 -0500825 // Special cases where the bp2build sends additional information to the codegenerator
826 // by wrapping the attributes in a custom struct type.
Jingwen Chenc1c26502021-04-05 10:35:13 +0000827 if attr, ok := propertyValue.Interface().(bazel.Attribute); ok {
828 return prettyPrintAttribute(attr, indent)
Liz Kammer356f7d42021-01-26 09:18:53 -0500829 } else if label, ok := propertyValue.Interface().(bazel.Label); ok {
830 return fmt.Sprintf("%q", label.Label), nil
831 }
832
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800833 // Sort and print the struct props by the key.
Alix94e26032022-08-16 20:37:33 +0000834 structProps, err := extractStructProperties(propertyValue, indent)
835
836 if err != nil {
837 return "", err
838 }
839
Jingwen Chen3d383bb2021-06-09 07:18:37 +0000840 if len(structProps) == 0 {
841 return "", nil
842 }
Liz Kammer72beb342022-02-03 08:42:10 -0500843 return starlark_fmt.PrintDict(structProps, indent), nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800844 case reflect.Interface:
845 // TODO(b/164227191): implement pretty print for interfaces.
846 // Interfaces are used for for arch, multilib and target properties.
847 return "", nil
Spandan Das6a448ec2023-04-19 17:36:12 +0000848 case reflect.Map:
849 if v, ok := propertyValue.Interface().(bazel.StringMapAttribute); ok {
850 return starlark_fmt.PrintStringStringDict(v, indent), nil
851 }
852 return "", fmt.Errorf("bp2build expects map of type map[string]string for field: %s", propertyValue)
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800853 default:
854 return "", fmt.Errorf(
855 "unexpected kind for property struct field: %s", propertyValue.Kind())
856 }
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800857}
858
859// Converts a reflected property struct value into a map of property names and property values,
860// which each property value correctly pretty-printed and indented at the right nest level,
861// since property structs can be nested. In Starlark, nested structs are represented as nested
862// dicts: https://docs.bazel.build/skylark/lib/dict.html
Alix94e26032022-08-16 20:37:33 +0000863func extractStructProperties(structValue reflect.Value, indent int) (map[string]string, error) {
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800864 if structValue.Kind() != reflect.Struct {
Alix94e26032022-08-16 20:37:33 +0000865 return map[string]string{}, fmt.Errorf("Expected a reflect.Struct type, but got %s", structValue.Kind())
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800866 }
867
Alix94e26032022-08-16 20:37:33 +0000868 var err error
869
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800870 ret := map[string]string{}
871 structType := structValue.Type()
872 for i := 0; i < structValue.NumField(); i++ {
873 field := structType.Field(i)
874 if shouldSkipStructField(field) {
875 continue
876 }
877
878 fieldValue := structValue.Field(i)
879 if isZero(fieldValue) {
880 // Ignore zero-valued fields
881 continue
882 }
Liz Kammer7a210ac2021-09-22 15:52:58 -0400883
Liz Kammer32a03392021-09-14 11:17:21 -0400884 // if the struct is embedded (anonymous), flatten the properties into the containing struct
885 if field.Anonymous {
886 if field.Type.Kind() == reflect.Ptr {
887 fieldValue = fieldValue.Elem()
888 }
889 if fieldValue.Type().Kind() == reflect.Struct {
Alix94e26032022-08-16 20:37:33 +0000890 propsToMerge, err := extractStructProperties(fieldValue, indent)
891 if err != nil {
892 return map[string]string{}, err
893 }
Liz Kammer32a03392021-09-14 11:17:21 -0400894 for prop, value := range propsToMerge {
895 ret[prop] = value
896 }
897 continue
898 }
899 }
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800900
901 propertyName := proptools.PropertyNameForField(field.Name)
Alix94e26032022-08-16 20:37:33 +0000902 var prettyPrintedValue string
903 prettyPrintedValue, err = prettyPrint(fieldValue, indent+1, false)
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800904 if err != nil {
Alix94e26032022-08-16 20:37:33 +0000905 return map[string]string{}, fmt.Errorf(
906 "Error while parsing property: %q. %s",
907 propertyName,
908 err)
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800909 }
910 if prettyPrintedValue != "" {
911 ret[propertyName] = prettyPrintedValue
912 }
913 }
914
Alix94e26032022-08-16 20:37:33 +0000915 return ret, nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800916}
917
918func isZero(value reflect.Value) bool {
919 switch value.Kind() {
920 case reflect.Func, reflect.Map, reflect.Slice:
921 return value.IsNil()
922 case reflect.Array:
923 valueIsZero := true
924 for i := 0; i < value.Len(); i++ {
925 valueIsZero = valueIsZero && isZero(value.Index(i))
926 }
927 return valueIsZero
928 case reflect.Struct:
929 valueIsZero := true
930 for i := 0; i < value.NumField(); i++ {
Lukacs T. Berki1353e592021-04-30 15:35:09 +0200931 valueIsZero = valueIsZero && isZero(value.Field(i))
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800932 }
933 return valueIsZero
934 case reflect.Ptr:
935 if !value.IsNil() {
936 return isZero(reflect.Indirect(value))
937 } else {
938 return true
939 }
Liz Kammer46fb7ab2021-12-01 10:09:34 -0500940 // Always print bool/strings, if you want a bool/string attribute to be able to take the default value, use a
941 // pointer instead
942 case reflect.Bool, reflect.String:
Liz Kammerd366c902021-06-03 13:43:01 -0400943 return false
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800944 default:
Rupert Shuttleworthc194ffb2021-05-19 06:49:02 -0400945 if !value.IsValid() {
946 return true
947 }
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800948 zeroValue := reflect.Zero(value.Type())
949 result := value.Interface() == zeroValue.Interface()
950 return result
951 }
952}
953
954func escapeString(s string) string {
955 s = strings.ReplaceAll(s, "\\", "\\\\")
Jingwen Chen58a12b82021-03-30 13:08:36 +0000956
957 // b/184026959: Reverse the application of some common control sequences.
958 // These must be generated literally in the BUILD file.
959 s = strings.ReplaceAll(s, "\t", "\\t")
960 s = strings.ReplaceAll(s, "\n", "\\n")
961 s = strings.ReplaceAll(s, "\r", "\\r")
962
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800963 return strings.ReplaceAll(s, "\"", "\\\"")
964}
965
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800966func targetNameWithVariant(c bpToBuildContext, logicModule blueprint.Module) string {
967 name := ""
968 if c.ModuleSubDir(logicModule) != "" {
969 // TODO(b/162720883): Figure out a way to drop the "--" variant suffixes.
970 name = c.ModuleName(logicModule) + "--" + c.ModuleSubDir(logicModule)
971 } else {
972 name = c.ModuleName(logicModule)
973 }
974
975 return strings.Replace(name, "//", "", 1)
976}
977
978func qualifiedTargetLabel(c bpToBuildContext, logicModule blueprint.Module) string {
979 return fmt.Sprintf("//%s:%s", c.ModuleDir(logicModule), targetNameWithVariant(c, logicModule))
980}