blob: 09a9d04f37f0eb442f89f4bc1c21ce5daacc5aa4 [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
Spandan Das682e7862023-06-22 22:22:11 +0000356 Data bazel.LabelListAttribute
Spandan Dasea2abba2023-06-14 21:30:38 +0000357 Target_compatible_with bazel.LabelListAttribute
Spandan Das682e7862023-06-22 22:22:11 +0000358
359 // attributes for the dynamically generated go_test target
360 Embed bazel.LabelListAttribute
Spandan Dasea2abba2023-06-14 21:30:38 +0000361}
362
Spandan Das682e7862023-06-22 22:22:11 +0000363type goTestProperties struct {
364 name string
365 dir string
366 testSrcs []string
367 linuxTestSrcs []string
368 darwinTestSrcs []string
369 testData []string
370 // Name of the target that should be compiled together with the test
371 embedName string
372}
373
374// Creates a go_test target for bootstrap_go_package / blueprint_go_binary
375func generateBazelTargetsGoTest(ctx *android.Context, goModulesMap nameToGoLibraryModule, gp goTestProperties) (BazelTarget, error) {
376 ca := android.CommonAttributes{
377 Name: gp.name,
378 }
379 ga := goAttributes{
380 Srcs: goSrcLabels(ctx.Config(), gp.dir, gp.testSrcs, gp.linuxTestSrcs, gp.darwinTestSrcs),
381 Data: goSrcLabels(ctx.Config(), gp.dir, gp.testData, []string{}, []string{}),
382 Embed: bazel.MakeLabelListAttribute(
383 bazel.MakeLabelList(
384 []bazel.Label{bazel.Label{Label: ":" + gp.embedName}},
385 ),
386 ),
387 Target_compatible_with: targetNotCompatibleWithAndroid(),
388 }
389
390 libTest := goBazelTarget{
391 targetName: gp.name,
392 targetPackage: gp.dir,
393 bazelRuleClass: "go_test",
394 bazelRuleLoadLocation: "@io_bazel_rules_go//go:def.bzl",
395 bazelAttributes: []interface{}{&ca, &ga},
396 }
397 return generateBazelTarget(ctx, libTest)
398}
399
400// TODO - b/288491147: testSrcs of certain bootstrap_go_package/blueprint_go_binary are not hermetic and depend on
401// testdata checked into the filesystem.
402// Denylist the generation of go_test targets for these Soong modules.
403// The go_library/go_binary will still be generated, since those are hermitic.
404var (
405 goTestsDenylist = []string{
406 "android-archive-zip",
407 "bazel_notice_gen",
408 "blueprint-bootstrap-bpdoc",
409 "blueprint-microfactory",
410 "blueprint-pathtools",
411 "bssl_ar",
412 "compliance_checkmetadata",
413 "compliance_checkshare",
414 "compliance_dumpgraph",
415 "compliance_dumpresolutions",
416 "compliance_listshare",
417 "compliance-module",
418 "compliancenotice_bom",
419 "compliancenotice_shippedlibs",
420 "compliance_rtrace",
421 "compliance_sbom",
422 "golang-protobuf-internal-fuzz-jsonfuzz",
423 "golang-protobuf-internal-fuzz-textfuzz",
424 "golang-protobuf-internal-fuzz-wirefuzz",
425 "htmlnotice",
426 "protoc-gen-go",
427 "rbcrun-module",
428 "spdx-tools-builder",
429 "spdx-tools-builder2v1",
430 "spdx-tools-builder2v2",
431 "spdx-tools-builder2v3",
432 "spdx-tools-idsearcher",
433 "spdx-tools-spdx-json",
434 "spdx-tools-utils",
435 "soong-ui-build",
436 "textnotice",
437 "xmlnotice",
438 }
439)
440
Spandan Das89aa0f72023-06-30 20:18:39 +0000441func testOfGoPackageIsIncompatible(g *bootstrap.GoPackage) bool {
442 return android.InList(g.Name(), goTestsDenylist) ||
443 // Denylist tests of soong_build
444 // Theses tests have a guard that prevent usage outside a test environment
445 // The guard (`ensureTestOnly`) looks for a `-test` in os.Args, which is present in soong's gotestrunner, but missing in `b test`
446 g.IsPluginFor("soong_build") ||
447 // soong-android is a dep of soong_build
448 // This dependency is created by soong_build by listing it in its deps explicitly in Android.bp, and not via `plugin_for` in `soong-android`
449 g.Name() == "soong-android"
450}
451
452func testOfGoBinaryIsIncompatible(g *bootstrap.GoBinary) bool {
453 return android.InList(g.Name(), goTestsDenylist)
454}
455
Spandan Dasea2abba2023-06-14 21:30:38 +0000456func generateBazelTargetsGoPackage(ctx *android.Context, g *bootstrap.GoPackage, goModulesMap nameToGoLibraryModule) ([]BazelTarget, []error) {
457 ca := android.CommonAttributes{
458 Name: g.Name(),
459 }
Spandan Dasde623292023-06-14 21:30:38 +0000460
461 // For this bootstrap_go_package dep chain,
462 // A --> B --> C ( ---> depends on)
463 // Soong provides the convenience of only listing B as deps of A even if a src file of A imports C
464 // Bazel OTOH
465 // 1. requires C to be listed in `deps` expllicity.
466 // 2. does not require C to be listed if src of A does not import C
467 //
468 // 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
469 transitiveDeps := transitiveGoDeps(g.Deps(), goModulesMap)
470
Spandan Dasea2abba2023-06-14 21:30:38 +0000471 ga := goAttributes{
472 Importpath: bazel.StringAttribute{
473 Value: proptools.StringPtr(g.GoPkgPath()),
474 },
Spandan Das0a8a2752023-06-21 01:50:33 +0000475 Srcs: goSrcLabels(ctx.Config(), ctx.ModuleDir(g), g.Srcs(), g.LinuxSrcs(), g.DarwinSrcs()),
476 Deps: goDepLabels(
477 android.FirstUniqueStrings(transitiveDeps),
478 goModulesMap,
479 ),
Spandan Dasea2abba2023-06-14 21:30:38 +0000480 Target_compatible_with: targetNotCompatibleWithAndroid(),
481 }
482
483 lib := goBazelTarget{
484 targetName: g.Name(),
485 targetPackage: ctx.ModuleDir(g),
486 bazelRuleClass: "go_library",
487 bazelRuleLoadLocation: "@io_bazel_rules_go//go:def.bzl",
488 bazelAttributes: []interface{}{&ca, &ga},
489 }
Spandan Das682e7862023-06-22 22:22:11 +0000490 retTargets := []BazelTarget{}
491 var retErrs []error
492 if libTarget, err := generateBazelTarget(ctx, lib); err == nil {
493 retTargets = append(retTargets, libTarget)
494 } else {
495 retErrs = []error{err}
Spandan Dasea2abba2023-06-14 21:30:38 +0000496 }
Spandan Das682e7862023-06-22 22:22:11 +0000497
498 // If the library contains test srcs, create an additional go_test target
Spandan Das89aa0f72023-06-30 20:18:39 +0000499 if !testOfGoPackageIsIncompatible(g) && (len(g.TestSrcs()) > 0 || len(g.LinuxTestSrcs()) > 0 || len(g.DarwinTestSrcs()) > 0) {
Spandan Das682e7862023-06-22 22:22:11 +0000500 gp := goTestProperties{
501 name: g.Name() + "-test",
502 dir: ctx.ModuleDir(g),
503 testSrcs: g.TestSrcs(),
504 linuxTestSrcs: g.LinuxTestSrcs(),
505 darwinTestSrcs: g.DarwinTestSrcs(),
506 testData: g.TestData(),
507 embedName: g.Name(), // embed the source go_library in the test so that its .go files are included in the compilation unit
508 }
509 if libTestTarget, err := generateBazelTargetsGoTest(ctx, goModulesMap, gp); err == nil {
510 retTargets = append(retTargets, libTestTarget)
511 } else {
512 retErrs = append(retErrs, err)
513 }
514 }
515
516 return retTargets, retErrs
Spandan Dasea2abba2023-06-14 21:30:38 +0000517}
518
519type goLibraryModule struct {
520 Dir string
521 Deps []string
522}
523
524type nameToGoLibraryModule map[string]goLibraryModule
525
526// Visit each module in the graph
527// If a module is of type `bootstrap_go_package`, return a map containing metadata like its dir and deps
528func createGoLibraryModuleMap(ctx *android.Context) nameToGoLibraryModule {
529 ret := nameToGoLibraryModule{}
530 ctx.VisitAllModules(func(m blueprint.Module) {
531 moduleType := ctx.ModuleType(m)
532 // We do not need to store information about blueprint_go_binary since it does not have any rdeps
533 if moduleType == "bootstrap_go_package" {
534 ret[m.Name()] = goLibraryModule{
535 Dir: ctx.ModuleDir(m),
536 Deps: m.(*bootstrap.GoPackage).Deps(),
537 }
538 }
539 })
540 return ret
541}
542
Spandan Dasde623292023-06-14 21:30:38 +0000543// Returns the deps in the transitive closure of a go target
544func transitiveGoDeps(directDeps []string, goModulesMap nameToGoLibraryModule) []string {
545 allDeps := directDeps
546 i := 0
547 for i < len(allDeps) {
548 curr := allDeps[i]
549 allDeps = append(allDeps, goModulesMap[curr].Deps...)
550 i += 1
551 }
552 allDeps = android.SortedUniqueStrings(allDeps)
553 return allDeps
554}
555
556func generateBazelTargetsGoBinary(ctx *android.Context, g *bootstrap.GoBinary, goModulesMap nameToGoLibraryModule) ([]BazelTarget, []error) {
557 ca := android.CommonAttributes{
558 Name: g.Name(),
559 }
560
Spandan Das682e7862023-06-22 22:22:11 +0000561 retTargets := []BazelTarget{}
562 var retErrs []error
563
Spandan Dasde623292023-06-14 21:30:38 +0000564 // For this bootstrap_go_package dep chain,
565 // A --> B --> C ( ---> depends on)
566 // Soong provides the convenience of only listing B as deps of A even if a src file of A imports C
567 // Bazel OTOH
568 // 1. requires C to be listed in `deps` expllicity.
569 // 2. does not require C to be listed if src of A does not import C
570 //
571 // 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
572 transitiveDeps := transitiveGoDeps(g.Deps(), goModulesMap)
573
Spandan Das682e7862023-06-22 22:22:11 +0000574 goSource := ""
575 // If the library contains test srcs, create an additional go_test target
576 // The go_test target will embed a go_source containining the source .go files it tests
Spandan Das89aa0f72023-06-30 20:18:39 +0000577 if !testOfGoBinaryIsIncompatible(g) && (len(g.TestSrcs()) > 0 || len(g.LinuxTestSrcs()) > 0 || len(g.DarwinTestSrcs()) > 0) {
Spandan Das682e7862023-06-22 22:22:11 +0000578 // Create a go_source containing the source .go files of go_library
579 // This target will be an `embed` of the go_binary and go_test
580 goSource = g.Name() + "-source"
581 ca := android.CommonAttributes{
582 Name: goSource,
583 }
584 ga := goAttributes{
585 Srcs: goSrcLabels(ctx.Config(), ctx.ModuleDir(g), g.Srcs(), g.LinuxSrcs(), g.DarwinSrcs()),
586 Deps: goDepLabels(transitiveDeps, goModulesMap),
587 Target_compatible_with: targetNotCompatibleWithAndroid(),
588 }
589 libTestSource := goBazelTarget{
590 targetName: goSource,
591 targetPackage: ctx.ModuleDir(g),
592 bazelRuleClass: "go_source",
593 bazelRuleLoadLocation: "@io_bazel_rules_go//go:def.bzl",
594 bazelAttributes: []interface{}{&ca, &ga},
595 }
596 if libSourceTarget, err := generateBazelTarget(ctx, libTestSource); err == nil {
597 retTargets = append(retTargets, libSourceTarget)
598 } else {
599 retErrs = append(retErrs, err)
600 }
601
602 // Create a go_test target
603 gp := goTestProperties{
604 name: g.Name() + "-test",
605 dir: ctx.ModuleDir(g),
606 testSrcs: g.TestSrcs(),
607 linuxTestSrcs: g.LinuxTestSrcs(),
608 darwinTestSrcs: g.DarwinTestSrcs(),
609 testData: g.TestData(),
610 // embed the go_source in the test
611 embedName: g.Name() + "-source",
612 }
613 if libTestTarget, err := generateBazelTargetsGoTest(ctx, goModulesMap, gp); err == nil {
614 retTargets = append(retTargets, libTestTarget)
615 } else {
616 retErrs = append(retErrs, err)
617 }
618
619 }
620
621 // Create a go_binary target
Spandan Dasde623292023-06-14 21:30:38 +0000622 ga := goAttributes{
Spandan Dasde623292023-06-14 21:30:38 +0000623 Deps: goDepLabels(transitiveDeps, goModulesMap),
624 Target_compatible_with: targetNotCompatibleWithAndroid(),
625 }
626
Spandan Das682e7862023-06-22 22:22:11 +0000627 // If the binary has testSrcs, embed the common `go_source`
628 if goSource != "" {
629 ga.Embed = bazel.MakeLabelListAttribute(
630 bazel.MakeLabelList(
631 []bazel.Label{bazel.Label{Label: ":" + goSource}},
632 ),
633 )
634 } else {
635 ga.Srcs = goSrcLabels(ctx.Config(), ctx.ModuleDir(g), g.Srcs(), g.LinuxSrcs(), g.DarwinSrcs())
636 }
637
Spandan Dasde623292023-06-14 21:30:38 +0000638 bin := goBazelTarget{
639 targetName: g.Name(),
640 targetPackage: ctx.ModuleDir(g),
641 bazelRuleClass: "go_binary",
642 bazelRuleLoadLocation: "@io_bazel_rules_go//go:def.bzl",
643 bazelAttributes: []interface{}{&ca, &ga},
644 }
Spandan Das682e7862023-06-22 22:22:11 +0000645
646 if binTarget, err := generateBazelTarget(ctx, bin); err == nil {
647 retTargets = append(retTargets, binTarget)
648 } else {
649 retErrs = []error{err}
Spandan Dasde623292023-06-14 21:30:38 +0000650 }
Spandan Das682e7862023-06-22 22:22:11 +0000651
652 return retTargets, retErrs
Spandan Dasde623292023-06-14 21:30:38 +0000653}
654
Liz Kammer6eff3232021-08-26 08:37:59 -0400655func GenerateBazelTargets(ctx *CodegenContext, generateFilegroups bool) (conversionResults, []error) {
Jingwen Chen40067de2021-01-26 21:58:43 -0500656 buildFileToTargets := make(map[string]BazelTargets)
Jingwen Chen164e0862021-02-19 00:48:40 -0500657
658 // Simple metrics tracking for bp2build
usta4f5d2c12022-10-28 23:32:01 -0400659 metrics := CreateCodegenMetrics()
Jingwen Chen164e0862021-02-19 00:48:40 -0500660
Rupert Shuttleworth2a4fc3e2021-04-21 07:10:09 -0400661 dirs := make(map[string]bool)
662
Liz Kammer6eff3232021-08-26 08:37:59 -0400663 var errs []error
664
Spandan Dasea2abba2023-06-14 21:30:38 +0000665 // Visit go libraries in a pre-run and store its state in a map
666 // The time complexity remains O(N), and this does not add significant wall time.
667 nameToGoLibMap := createGoLibraryModuleMap(ctx.Context())
668
Jingwen Chen164e0862021-02-19 00:48:40 -0500669 bpCtx := ctx.Context()
670 bpCtx.VisitAllModules(func(m blueprint.Module) {
671 dir := bpCtx.ModuleDir(m)
Chris Parsons492bd912022-01-20 12:55:05 -0500672 moduleType := bpCtx.ModuleType(m)
Rupert Shuttleworth2a4fc3e2021-04-21 07:10:09 -0400673 dirs[dir] = true
674
Liz Kammer2ada09a2021-08-11 00:17:36 -0400675 var targets []BazelTarget
Spandan Dasea2abba2023-06-14 21:30:38 +0000676 var targetErrs []error
Jingwen Chen73850672020-12-14 08:25:34 -0500677
Jingwen Chen164e0862021-02-19 00:48:40 -0500678 switch ctx.Mode() {
Jingwen Chen33832f92021-01-24 22:55:54 -0500679 case Bp2Build:
Jingwen Chen310bc8f2021-09-20 10:54:27 +0000680 // There are two main ways of converting a Soong module to Bazel:
681 // 1) Manually handcrafting a Bazel target and associating the module with its label
682 // 2) Automatically generating with bp2build converters
683 //
684 // bp2build converters are used for the majority of modules.
Liz Kammerba3ea162021-02-17 13:22:03 -0500685 if b, ok := m.(android.Bazelable); ok && b.HasHandcraftedLabel() {
Jingwen Chen310bc8f2021-09-20 10:54:27 +0000686 // Handle modules converted to handcrafted targets.
687 //
688 // Since these modules are associated with some handcrafted
Cole Faustea602c52022-08-31 14:48:26 -0700689 // target in a BUILD file, we don't autoconvert them.
Jingwen Chen310bc8f2021-09-20 10:54:27 +0000690
691 // Log the module.
Chris Parsons39a16972023-06-08 14:28:51 +0000692 metrics.AddUnconvertedModule(m, moduleType, dir,
693 android.UnconvertedReason{
694 ReasonType: int(bp2build_metrics_proto.UnconvertedReasonType_DEFINED_IN_BUILD_FILE),
695 })
Liz Kammer2ada09a2021-08-11 00:17:36 -0400696 } else if aModule, ok := m.(android.Module); ok && aModule.IsConvertedByBp2build() {
Jingwen Chen310bc8f2021-09-20 10:54:27 +0000697 // Handle modules converted to generated targets.
698
699 // Log the module.
Chris Parsons39a16972023-06-08 14:28:51 +0000700 metrics.AddConvertedModule(aModule, moduleType, dir)
Jingwen Chen310bc8f2021-09-20 10:54:27 +0000701
702 // Handle modules with unconverted deps. By default, emit a warning.
Liz Kammer6eff3232021-08-26 08:37:59 -0400703 if unconvertedDeps := aModule.GetUnconvertedBp2buildDeps(); len(unconvertedDeps) > 0 {
Sasha Smundakf2bb26f2022-08-04 11:28:15 -0700704 msg := fmt.Sprintf("%s %s:%s depends on unconverted modules: %s",
705 moduleType, bpCtx.ModuleDir(m), m.Name(), strings.Join(unconvertedDeps, ", "))
Usta Shresthac6057152022-09-24 00:23:31 -0400706 switch ctx.unconvertedDepMode {
707 case warnUnconvertedDeps:
Liz Kammer6eff3232021-08-26 08:37:59 -0400708 metrics.moduleWithUnconvertedDepsMsgs = append(metrics.moduleWithUnconvertedDepsMsgs, msg)
Usta Shresthac6057152022-09-24 00:23:31 -0400709 case errorModulesUnconvertedDeps:
Liz Kammer6eff3232021-08-26 08:37:59 -0400710 errs = append(errs, fmt.Errorf(msg))
711 return
712 }
713 }
Liz Kammerdaa09ef2021-12-15 15:35:38 -0500714 if unconvertedDeps := aModule.GetMissingBp2buildDeps(); len(unconvertedDeps) > 0 {
Sasha Smundakf2bb26f2022-08-04 11:28:15 -0700715 msg := fmt.Sprintf("%s %s:%s depends on missing modules: %s",
716 moduleType, bpCtx.ModuleDir(m), m.Name(), strings.Join(unconvertedDeps, ", "))
Usta Shresthac6057152022-09-24 00:23:31 -0400717 switch ctx.unconvertedDepMode {
718 case warnUnconvertedDeps:
Liz Kammerdaa09ef2021-12-15 15:35:38 -0500719 metrics.moduleWithMissingDepsMsgs = append(metrics.moduleWithMissingDepsMsgs, msg)
Usta Shresthac6057152022-09-24 00:23:31 -0400720 case errorModulesUnconvertedDeps:
Liz Kammerdaa09ef2021-12-15 15:35:38 -0500721 errs = append(errs, fmt.Errorf(msg))
722 return
723 }
724 }
Alix94e26032022-08-16 20:37:33 +0000725 targets, targetErrs = generateBazelTargets(bpCtx, aModule)
726 errs = append(errs, targetErrs...)
Liz Kammer2ada09a2021-08-11 00:17:36 -0400727 for _, t := range targets {
Jingwen Chen310bc8f2021-09-20 10:54:27 +0000728 // A module can potentially generate more than 1 Bazel
729 // target, each of a different rule class.
730 metrics.IncrementRuleClassCount(t.ruleClass)
Liz Kammer2ada09a2021-08-11 00:17:36 -0400731 }
MarkDacek9c094ca2023-03-16 19:15:19 +0000732 } else if _, ok := ctx.Config().BazelModulesForceEnabledByFlag()[m.Name()]; ok && m.Name() != "" {
733 err := fmt.Errorf("Force Enabled Module %s not converted", m.Name())
734 errs = append(errs, err)
Chris Parsons39a16972023-06-08 14:28:51 +0000735 } else if aModule, ok := m.(android.Module); ok {
736 reason := aModule.GetUnconvertedReason()
737 if reason == nil {
738 panic(fmt.Errorf("module '%s' was neither converted nor marked unconvertible with bp2build", aModule.Name()))
739 } else {
740 metrics.AddUnconvertedModule(m, moduleType, dir, *reason)
741 }
742 return
Spandan Dasea2abba2023-06-14 21:30:38 +0000743 } else if glib, ok := m.(*bootstrap.GoPackage); ok {
744 targets, targetErrs = generateBazelTargetsGoPackage(bpCtx, glib, nameToGoLibMap)
745 errs = append(errs, targetErrs...)
746 metrics.IncrementRuleClassCount("go_library")
Spandan Das41f1eee2023-08-01 22:28:16 +0000747 metrics.AddConvertedModule(glib, "go_library", dir)
Spandan Das2a55cea2023-06-14 17:56:10 +0000748 } else if gbin, ok := m.(*bootstrap.GoBinary); ok {
Spandan Dasde623292023-06-14 21:30:38 +0000749 targets, targetErrs = generateBazelTargetsGoBinary(bpCtx, gbin, nameToGoLibMap)
750 errs = append(errs, targetErrs...)
751 metrics.IncrementRuleClassCount("go_binary")
Spandan Das41f1eee2023-08-01 22:28:16 +0000752 metrics.AddConvertedModule(gbin, "go_binary", dir)
Liz Kammerfc46bc12021-02-19 11:06:17 -0500753 } else {
Chris Parsons39a16972023-06-08 14:28:51 +0000754 metrics.AddUnconvertedModule(m, moduleType, dir, android.UnconvertedReason{
755 ReasonType: int(bp2build_metrics_proto.UnconvertedReasonType_TYPE_UNSUPPORTED),
756 })
Liz Kammerba3ea162021-02-17 13:22:03 -0500757 return
Jingwen Chen73850672020-12-14 08:25:34 -0500758 }
Jingwen Chen33832f92021-01-24 22:55:54 -0500759 case QueryView:
Jingwen Chen96af35b2021-02-08 00:49:32 -0500760 // Blocklist certain module types from being generated.
Jingwen Chen164e0862021-02-19 00:48:40 -0500761 if canonicalizeModuleType(bpCtx.ModuleType(m)) == "package" {
Jingwen Chen96af35b2021-02-08 00:49:32 -0500762 // package module name contain slashes, and thus cannot
763 // be mapped cleanly to a bazel label.
764 return
765 }
Alix94e26032022-08-16 20:37:33 +0000766 t, err := generateSoongModuleTarget(bpCtx, m)
767 if err != nil {
768 errs = append(errs, err)
769 }
Liz Kammer2ada09a2021-08-11 00:17:36 -0400770 targets = append(targets, t)
Spandan Das5af0bd32022-09-28 20:43:08 +0000771 case ApiBp2build:
772 if aModule, ok := m.(android.Module); ok && aModule.IsConvertedByBp2build() {
773 targets, errs = generateBazelTargets(bpCtx, aModule)
774 }
Jingwen Chen33832f92021-01-24 22:55:54 -0500775 default:
Liz Kammer6eff3232021-08-26 08:37:59 -0400776 errs = append(errs, fmt.Errorf("Unknown code-generation mode: %s", ctx.Mode()))
777 return
Jingwen Chen73850672020-12-14 08:25:34 -0500778 }
779
Spandan Dasabedff02023-03-07 19:24:34 +0000780 for _, target := range targets {
781 targetDir := target.PackageName()
782 buildFileToTargets[targetDir] = append(buildFileToTargets[targetDir], target)
783 }
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800784 })
Liz Kammer6eff3232021-08-26 08:37:59 -0400785
786 if len(errs) > 0 {
787 return conversionResults{}, errs
788 }
789
Rupert Shuttleworth2a4fc3e2021-04-21 07:10:09 -0400790 if generateFilegroups {
791 // Add a filegroup target that exposes all sources in the subtree of this package
792 // 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 -0700793 //
794 // This works because: https://bazel.build/reference/be/functions#exports_files
795 // "As a legacy behaviour, also files mentioned as input to a rule are exported with the
796 // default visibility until the flag --incompatible_no_implicit_file_export is flipped. However, this behavior
797 // should not be relied upon and actively migrated away from."
798 //
799 // TODO(b/198619163): We should change this to export_files(glob(["**/*"])) instead, but doing that causes these errors:
800 // "Error in exports_files: generated label '//external/avb:avbtool' conflicts with existing py_binary rule"
801 // So we need to solve all the "target ... is both a rule and a file" warnings first.
Usta Shresthac6057152022-09-24 00:23:31 -0400802 for dir := range dirs {
Rupert Shuttleworth2a4fc3e2021-04-21 07:10:09 -0400803 buildFileToTargets[dir] = append(buildFileToTargets[dir], BazelTarget{
804 name: "bp2build_all_srcs",
805 content: `filegroup(name = "bp2build_all_srcs", srcs = glob(["**/*"]))`,
806 ruleClass: "filegroup",
807 })
808 }
809 }
Jingwen Chen164e0862021-02-19 00:48:40 -0500810
Liz Kammer6eff3232021-08-26 08:37:59 -0400811 return conversionResults{
812 buildFileToTargets: buildFileToTargets,
813 metrics: metrics,
Liz Kammer6eff3232021-08-26 08:37:59 -0400814 }, errs
Jingwen Chen164e0862021-02-19 00:48:40 -0500815}
816
Alix94e26032022-08-16 20:37:33 +0000817func generateBazelTargets(ctx bpToBuildContext, m android.Module) ([]BazelTarget, []error) {
Liz Kammer2ada09a2021-08-11 00:17:36 -0400818 var targets []BazelTarget
Alix94e26032022-08-16 20:37:33 +0000819 var errs []error
Liz Kammer2ada09a2021-08-11 00:17:36 -0400820 for _, m := range m.Bp2buildTargets() {
Alix94e26032022-08-16 20:37:33 +0000821 target, err := generateBazelTarget(ctx, m)
822 if err != nil {
823 errs = append(errs, err)
824 return targets, errs
825 }
826 targets = append(targets, target)
Liz Kammer2ada09a2021-08-11 00:17:36 -0400827 }
Alix94e26032022-08-16 20:37:33 +0000828 return targets, errs
Liz Kammer2ada09a2021-08-11 00:17:36 -0400829}
830
831type bp2buildModule interface {
832 TargetName() string
833 TargetPackage() string
834 BazelRuleClass() string
835 BazelRuleLoadLocation() string
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux447f6c92021-08-31 20:30:36 +0000836 BazelAttributes() []interface{}
Liz Kammer2ada09a2021-08-11 00:17:36 -0400837}
838
Alix94e26032022-08-16 20:37:33 +0000839func generateBazelTarget(ctx bpToBuildContext, m bp2buildModule) (BazelTarget, error) {
Liz Kammer2ada09a2021-08-11 00:17:36 -0400840 ruleClass := m.BazelRuleClass()
841 bzlLoadLocation := m.BazelRuleLoadLocation()
Jingwen Chen40067de2021-01-26 21:58:43 -0500842
Jingwen Chen73850672020-12-14 08:25:34 -0500843 // extract the bazel attributes from the module.
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux447f6c92021-08-31 20:30:36 +0000844 attrs := m.BazelAttributes()
Alix94e26032022-08-16 20:37:33 +0000845 props, err := extractModuleProperties(attrs, true)
846 if err != nil {
847 return BazelTarget{}, err
848 }
Jingwen Chen73850672020-12-14 08:25:34 -0500849
Liz Kammer0eae52e2021-10-06 10:32:26 -0400850 // name is handled in a special manner
851 delete(props.Attrs, "name")
Jingwen Chen77e8b7b2021-02-05 03:03:24 -0500852
Jingwen Chen73850672020-12-14 08:25:34 -0500853 // Return the Bazel target with rule class and attributes, ready to be
854 // code-generated.
855 attributes := propsToAttributes(props.Attrs)
Sasha Smundakfb589492022-08-04 11:13:27 -0700856 var content string
Liz Kammer2ada09a2021-08-11 00:17:36 -0400857 targetName := m.TargetName()
Sasha Smundakfb589492022-08-04 11:13:27 -0700858 if targetName != "" {
859 content = fmt.Sprintf(ruleTargetTemplate, ruleClass, targetName, attributes)
860 } else {
861 content = fmt.Sprintf(unnamedRuleTargetTemplate, ruleClass, attributes)
862 }
Jingwen Chen73850672020-12-14 08:25:34 -0500863 return BazelTarget{
Jingwen Chen40067de2021-01-26 21:58:43 -0500864 name: targetName,
Liz Kammer2ada09a2021-08-11 00:17:36 -0400865 packageName: m.TargetPackage(),
Jingwen Chen40067de2021-01-26 21:58:43 -0500866 ruleClass: ruleClass,
867 bzlLoadLocation: bzlLoadLocation,
Sasha Smundakfb589492022-08-04 11:13:27 -0700868 content: content,
Alix94e26032022-08-16 20:37:33 +0000869 }, nil
Jingwen Chen73850672020-12-14 08:25:34 -0500870}
871
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800872// Convert a module and its deps and props into a Bazel macro/rule
873// representation in the BUILD file.
Alix94e26032022-08-16 20:37:33 +0000874func generateSoongModuleTarget(ctx bpToBuildContext, m blueprint.Module) (BazelTarget, error) {
875 props, err := getBuildProperties(ctx, m)
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800876
877 // TODO(b/163018919): DirectDeps can have duplicate (module, variant)
878 // items, if the modules are added using different DependencyTag. Figure
879 // out the implications of that.
880 depLabels := map[string]bool{}
881 if aModule, ok := m.(android.Module); ok {
Jingwen Chendaa54bc2020-12-14 02:58:54 -0500882 ctx.VisitDirectDeps(aModule, func(depModule blueprint.Module) {
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800883 depLabels[qualifiedTargetLabel(ctx, depModule)] = true
884 })
885 }
Liz Kammer0eae52e2021-10-06 10:32:26 -0400886
Usta Shresthadb46a9b2022-07-11 11:29:56 -0400887 for p := range ignoredPropNames {
Liz Kammer0eae52e2021-10-06 10:32:26 -0400888 delete(props.Attrs, p)
889 }
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800890 attributes := propsToAttributes(props.Attrs)
891
892 depLabelList := "[\n"
Usta Shresthadb46a9b2022-07-11 11:29:56 -0400893 for depLabel := range depLabels {
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800894 depLabelList += fmt.Sprintf(" %q,\n", depLabel)
895 }
896 depLabelList += " ]"
897
898 targetName := targetNameWithVariant(ctx, m)
899 return BazelTarget{
Spandan Dasabedff02023-03-07 19:24:34 +0000900 name: targetName,
901 packageName: ctx.ModuleDir(m),
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800902 content: fmt.Sprintf(
Sasha Smundakfb589492022-08-04 11:13:27 -0700903 soongModuleTargetTemplate,
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800904 targetName,
905 ctx.ModuleName(m),
906 canonicalizeModuleType(ctx.ModuleType(m)),
907 ctx.ModuleSubDir(m),
908 depLabelList,
909 attributes),
Alix94e26032022-08-16 20:37:33 +0000910 }, err
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800911}
912
Alix94e26032022-08-16 20:37:33 +0000913func getBuildProperties(ctx bpToBuildContext, m blueprint.Module) (BazelAttributes, error) {
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800914 // TODO: this omits properties for blueprint modules (blueprint_go_binary,
915 // bootstrap_go_binary, bootstrap_go_package), which will have to be handled separately.
916 if aModule, ok := m.(android.Module); ok {
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux447f6c92021-08-31 20:30:36 +0000917 return extractModuleProperties(aModule.GetProperties(), false)
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800918 }
919
Alix94e26032022-08-16 20:37:33 +0000920 return BazelAttributes{}, nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800921}
922
923// Generically extract module properties and types into a map, keyed by the module property name.
Alix94e26032022-08-16 20:37:33 +0000924func extractModuleProperties(props []interface{}, checkForDuplicateProperties bool) (BazelAttributes, error) {
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800925 ret := map[string]string{}
926
927 // Iterate over this android.Module's property structs.
Liz Kammer2ada09a2021-08-11 00:17:36 -0400928 for _, properties := range props {
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800929 propertiesValue := reflect.ValueOf(properties)
930 // Check that propertiesValue is a pointer to the Properties struct, like
931 // *cc.BaseLinkerProperties or *java.CompilerProperties.
932 //
933 // propertiesValue can also be type-asserted to the structs to
934 // manipulate internal props, if needed.
935 if isStructPtr(propertiesValue.Type()) {
936 structValue := propertiesValue.Elem()
Alix94e26032022-08-16 20:37:33 +0000937 ok, err := extractStructProperties(structValue, 0)
938 if err != nil {
939 return BazelAttributes{}, err
940 }
941 for k, v := range ok {
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux447f6c92021-08-31 20:30:36 +0000942 if existing, exists := ret[k]; checkForDuplicateProperties && exists {
Alix94e26032022-08-16 20:37:33 +0000943 return BazelAttributes{}, fmt.Errorf(
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux447f6c92021-08-31 20:30:36 +0000944 "%s (%v) is present in properties whereas it should be consolidated into a commonAttributes",
Alix94e26032022-08-16 20:37:33 +0000945 k, existing)
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux447f6c92021-08-31 20:30:36 +0000946 }
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800947 ret[k] = v
948 }
949 } else {
Alix94e26032022-08-16 20:37:33 +0000950 return BazelAttributes{},
951 fmt.Errorf(
952 "properties must be a pointer to a struct, got %T",
953 propertiesValue.Interface())
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800954 }
955 }
956
Liz Kammer2ada09a2021-08-11 00:17:36 -0400957 return BazelAttributes{
958 Attrs: ret,
Alix94e26032022-08-16 20:37:33 +0000959 }, nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800960}
961
962func isStructPtr(t reflect.Type) bool {
963 return t.Kind() == reflect.Ptr && t.Elem().Kind() == reflect.Struct
964}
965
966// prettyPrint a property value into the equivalent Starlark representation
967// recursively.
Jingwen Chen58ff6802021-11-17 12:14:41 +0000968func prettyPrint(propertyValue reflect.Value, indent int, emitZeroValues bool) (string, error) {
969 if !emitZeroValues && isZero(propertyValue) {
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800970 // A property value being set or unset actually matters -- Soong does set default
971 // values for unset properties, like system_shared_libs = ["libc", "libm", "libdl"] at
972 // https://cs.android.com/android/platform/superproject/+/master:build/soong/cc/linker.go;l=281-287;drc=f70926eef0b9b57faf04c17a1062ce50d209e480
973 //
Jingwen Chenfc490bd2021-03-30 10:24:19 +0000974 // In Bazel-parlance, we would use "attr.<type>(default = <default
975 // value>)" to set the default value of unset attributes. In the cases
976 // where the bp2build converter didn't set the default value within the
977 // mutator when creating the BazelTargetModule, this would be a zero
Jingwen Chen63930982021-03-24 10:04:33 -0400978 // value. For those cases, we return an empty string so we don't
979 // unnecessarily generate empty values.
980 return "", nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800981 }
982
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800983 switch propertyValue.Kind() {
984 case reflect.String:
Liz Kammer72beb342022-02-03 08:42:10 -0500985 return fmt.Sprintf("\"%v\"", escapeString(propertyValue.String())), nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800986 case reflect.Bool:
Liz Kammer72beb342022-02-03 08:42:10 -0500987 return starlark_fmt.PrintBool(propertyValue.Bool()), nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800988 case reflect.Int, reflect.Uint, reflect.Int64:
Liz Kammer72beb342022-02-03 08:42:10 -0500989 return fmt.Sprintf("%v", propertyValue.Interface()), nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800990 case reflect.Ptr:
Jingwen Chen58ff6802021-11-17 12:14:41 +0000991 return prettyPrint(propertyValue.Elem(), indent, emitZeroValues)
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800992 case reflect.Slice:
Liz Kammer72beb342022-02-03 08:42:10 -0500993 elements := make([]string, 0, propertyValue.Len())
994 for i := 0; i < propertyValue.Len(); i++ {
995 val, err := prettyPrint(propertyValue.Index(i), indent, emitZeroValues)
Liz Kammer2dd9ca42020-11-25 16:06:39 -0800996 if err != nil {
997 return "", err
998 }
Liz Kammer72beb342022-02-03 08:42:10 -0500999 if val != "" {
1000 elements = append(elements, val)
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001001 }
1002 }
Sam Delmerico932c01c2022-03-25 16:33:26 +00001003 return starlark_fmt.PrintList(elements, indent, func(s string) string {
1004 return "%s"
1005 }), nil
Jingwen Chenb4628eb2021-04-08 14:40:57 +00001006
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001007 case reflect.Struct:
Jingwen Chen5d864492021-02-24 07:20:12 -05001008 // Special cases where the bp2build sends additional information to the codegenerator
1009 // by wrapping the attributes in a custom struct type.
Jingwen Chenc1c26502021-04-05 10:35:13 +00001010 if attr, ok := propertyValue.Interface().(bazel.Attribute); ok {
1011 return prettyPrintAttribute(attr, indent)
Liz Kammer356f7d42021-01-26 09:18:53 -05001012 } else if label, ok := propertyValue.Interface().(bazel.Label); ok {
1013 return fmt.Sprintf("%q", label.Label), nil
1014 }
1015
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001016 // Sort and print the struct props by the key.
Alix94e26032022-08-16 20:37:33 +00001017 structProps, err := extractStructProperties(propertyValue, indent)
1018
1019 if err != nil {
1020 return "", err
1021 }
1022
Jingwen Chen3d383bb2021-06-09 07:18:37 +00001023 if len(structProps) == 0 {
1024 return "", nil
1025 }
Liz Kammer72beb342022-02-03 08:42:10 -05001026 return starlark_fmt.PrintDict(structProps, indent), nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001027 case reflect.Interface:
1028 // TODO(b/164227191): implement pretty print for interfaces.
1029 // Interfaces are used for for arch, multilib and target properties.
1030 return "", nil
Spandan Das6a448ec2023-04-19 17:36:12 +00001031 case reflect.Map:
1032 if v, ok := propertyValue.Interface().(bazel.StringMapAttribute); ok {
1033 return starlark_fmt.PrintStringStringDict(v, indent), nil
1034 }
1035 return "", fmt.Errorf("bp2build expects map of type map[string]string for field: %s", propertyValue)
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001036 default:
1037 return "", fmt.Errorf(
1038 "unexpected kind for property struct field: %s", propertyValue.Kind())
1039 }
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001040}
1041
1042// Converts a reflected property struct value into a map of property names and property values,
1043// which each property value correctly pretty-printed and indented at the right nest level,
1044// since property structs can be nested. In Starlark, nested structs are represented as nested
1045// dicts: https://docs.bazel.build/skylark/lib/dict.html
Alix94e26032022-08-16 20:37:33 +00001046func extractStructProperties(structValue reflect.Value, indent int) (map[string]string, error) {
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001047 if structValue.Kind() != reflect.Struct {
Alix94e26032022-08-16 20:37:33 +00001048 return map[string]string{}, fmt.Errorf("Expected a reflect.Struct type, but got %s", structValue.Kind())
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001049 }
1050
Alix94e26032022-08-16 20:37:33 +00001051 var err error
1052
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001053 ret := map[string]string{}
1054 structType := structValue.Type()
1055 for i := 0; i < structValue.NumField(); i++ {
1056 field := structType.Field(i)
1057 if shouldSkipStructField(field) {
1058 continue
1059 }
1060
1061 fieldValue := structValue.Field(i)
1062 if isZero(fieldValue) {
1063 // Ignore zero-valued fields
1064 continue
1065 }
Liz Kammer7a210ac2021-09-22 15:52:58 -04001066
Liz Kammer32a03392021-09-14 11:17:21 -04001067 // if the struct is embedded (anonymous), flatten the properties into the containing struct
1068 if field.Anonymous {
1069 if field.Type.Kind() == reflect.Ptr {
1070 fieldValue = fieldValue.Elem()
1071 }
1072 if fieldValue.Type().Kind() == reflect.Struct {
Alix94e26032022-08-16 20:37:33 +00001073 propsToMerge, err := extractStructProperties(fieldValue, indent)
1074 if err != nil {
1075 return map[string]string{}, err
1076 }
Liz Kammer32a03392021-09-14 11:17:21 -04001077 for prop, value := range propsToMerge {
1078 ret[prop] = value
1079 }
1080 continue
1081 }
1082 }
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001083
1084 propertyName := proptools.PropertyNameForField(field.Name)
Alix94e26032022-08-16 20:37:33 +00001085 var prettyPrintedValue string
1086 prettyPrintedValue, err = prettyPrint(fieldValue, indent+1, false)
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001087 if err != nil {
Alix94e26032022-08-16 20:37:33 +00001088 return map[string]string{}, fmt.Errorf(
1089 "Error while parsing property: %q. %s",
1090 propertyName,
1091 err)
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001092 }
1093 if prettyPrintedValue != "" {
1094 ret[propertyName] = prettyPrintedValue
1095 }
1096 }
1097
Alix94e26032022-08-16 20:37:33 +00001098 return ret, nil
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001099}
1100
1101func isZero(value reflect.Value) bool {
1102 switch value.Kind() {
1103 case reflect.Func, reflect.Map, reflect.Slice:
1104 return value.IsNil()
1105 case reflect.Array:
1106 valueIsZero := true
1107 for i := 0; i < value.Len(); i++ {
1108 valueIsZero = valueIsZero && isZero(value.Index(i))
1109 }
1110 return valueIsZero
1111 case reflect.Struct:
1112 valueIsZero := true
1113 for i := 0; i < value.NumField(); i++ {
Lukacs T. Berki1353e592021-04-30 15:35:09 +02001114 valueIsZero = valueIsZero && isZero(value.Field(i))
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001115 }
1116 return valueIsZero
1117 case reflect.Ptr:
1118 if !value.IsNil() {
1119 return isZero(reflect.Indirect(value))
1120 } else {
1121 return true
1122 }
Liz Kammer46fb7ab2021-12-01 10:09:34 -05001123 // Always print bool/strings, if you want a bool/string attribute to be able to take the default value, use a
1124 // pointer instead
1125 case reflect.Bool, reflect.String:
Liz Kammerd366c902021-06-03 13:43:01 -04001126 return false
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001127 default:
Rupert Shuttleworthc194ffb2021-05-19 06:49:02 -04001128 if !value.IsValid() {
1129 return true
1130 }
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001131 zeroValue := reflect.Zero(value.Type())
1132 result := value.Interface() == zeroValue.Interface()
1133 return result
1134 }
1135}
1136
1137func escapeString(s string) string {
1138 s = strings.ReplaceAll(s, "\\", "\\\\")
Jingwen Chen58a12b82021-03-30 13:08:36 +00001139
1140 // b/184026959: Reverse the application of some common control sequences.
1141 // These must be generated literally in the BUILD file.
1142 s = strings.ReplaceAll(s, "\t", "\\t")
1143 s = strings.ReplaceAll(s, "\n", "\\n")
1144 s = strings.ReplaceAll(s, "\r", "\\r")
1145
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001146 return strings.ReplaceAll(s, "\"", "\\\"")
1147}
1148
Liz Kammer2dd9ca42020-11-25 16:06:39 -08001149func targetNameWithVariant(c bpToBuildContext, logicModule blueprint.Module) string {
1150 name := ""
1151 if c.ModuleSubDir(logicModule) != "" {
1152 // TODO(b/162720883): Figure out a way to drop the "--" variant suffixes.
1153 name = c.ModuleName(logicModule) + "--" + c.ModuleSubDir(logicModule)
1154 } else {
1155 name = c.ModuleName(logicModule)
1156 }
1157
1158 return strings.Replace(name, "//", "", 1)
1159}
1160
1161func qualifiedTargetLabel(c bpToBuildContext, logicModule blueprint.Module) string {
1162 return fmt.Sprintf("//%s:%s", c.ModuleDir(logicModule), targetNameWithVariant(c, logicModule))
1163}