blob: baa7380571eda5090d2c0e22ee3ec9b4d9e641af [file] [log] [blame]
Colin Cross3f40fa42015-01-30 17:27:36 -08001// Copyright 2015 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 main
16
17import (
Usta Shrestha2ba28a32022-10-24 11:33:09 -040018 "bytes"
Colin Cross3f40fa42015-01-30 17:27:36 -080019 "flag"
20 "fmt"
21 "os"
22 "path/filepath"
Jingwen Cheneb76c432021-01-28 08:22:12 -050023 "strings"
Lukacs T. Berkic99c9472021-03-24 10:50:06 +010024 "time"
Colin Cross3f40fa42015-01-30 17:27:36 -080025
Dan Willemsen66213a62021-09-21 17:50:30 -070026 "android/soong/android"
Spandan Das5af0bd32022-09-28 20:43:08 +000027 "android/soong/bazel"
Lukacs T. Berkif8e24282021-04-14 10:31:00 +020028 "android/soong/bp2build"
Lukacs T. Berki7690c092021-02-26 14:27:36 +010029 "android/soong/shared"
Chris Parsons715b08f2022-03-22 19:23:40 -040030 "android/soong/ui/metrics/bp2build_metrics_proto"
Lukacs T. Berkie571dc32021-08-25 14:14:13 +020031
Colin Cross70b40592015-03-23 12:57:34 -070032 "github.com/google/blueprint/bootstrap"
Lukacs T. Berkid518e1a2021-04-14 13:49:50 +020033 "github.com/google/blueprint/deptools"
Chris Parsons715b08f2022-03-22 19:23:40 -040034 "github.com/google/blueprint/metrics"
Dan Willemsen66213a62021-09-21 17:50:30 -070035 androidProtobuf "google.golang.org/protobuf/android"
Colin Cross3f40fa42015-01-30 17:27:36 -080036)
37
Colin Crosse87040b2017-12-11 15:52:26 -080038var (
Lukacs T. Berkif8e24282021-04-14 10:31:00 +020039 topDir string
Lukacs T. Berkif8e24282021-04-14 10:31:00 +020040 availableEnvFile string
41 usedEnvFile string
42
Lukacs T. Berki809d2ed2021-08-18 10:55:32 +020043 globFile string
44 globListDir string
Lukacs T. Berkif8e24282021-04-14 10:31:00 +020045 delveListen string
46 delvePath string
47
Sasha Smundakaf5ca922022-12-12 21:23:34 -080048 cmdlineArgs android.CmdArgs
Colin Crosse87040b2017-12-11 15:52:26 -080049)
50
51func init() {
Lukacs T. Berkif8e24282021-04-14 10:31:00 +020052 // Flags that make sense in every mode
Lukacs T. Berki7690c092021-02-26 14:27:36 +010053 flag.StringVar(&topDir, "top", "", "Top directory of the Android source tree")
Sasha Smundakaf5ca922022-12-12 21:23:34 -080054 flag.StringVar(&cmdlineArgs.SoongOutDir, "soong_out", "", "Soong output directory (usually $TOP/out/soong)")
Lukacs T. Berkif8e24282021-04-14 10:31:00 +020055 flag.StringVar(&availableEnvFile, "available_env", "", "File containing available environment variables")
56 flag.StringVar(&usedEnvFile, "used_env", "", "File containing used environment variables")
Lukacs T. Berkib078ade2021-08-31 10:42:08 +020057 flag.StringVar(&globFile, "globFile", "build-globs.ninja", "the Ninja file of globs to output")
58 flag.StringVar(&globListDir, "globListDir", "", "the directory containing the glob list files")
Sasha Smundakaf5ca922022-12-12 21:23:34 -080059 flag.StringVar(&cmdlineArgs.OutDir, "out", "", "the ninja builddir directory")
Lukacs T. Berkib078ade2021-08-31 10:42:08 +020060 flag.StringVar(&cmdlineArgs.ModuleListFile, "l", "", "file that lists filepaths to parse")
Lukacs T. Berkif8e24282021-04-14 10:31:00 +020061
62 // Debug flags
Lukacs T. Berki7d613bf2021-03-02 10:09:41 +010063 flag.StringVar(&delveListen, "delve_listen", "", "Delve port to listen on for debugging")
64 flag.StringVar(&delvePath, "delve_path", "", "Path to Delve. Only used if --delve_listen is set")
Lukacs T. Berkib078ade2021-08-31 10:42:08 +020065 flag.StringVar(&cmdlineArgs.Cpuprofile, "cpuprofile", "", "write cpu profile to file")
66 flag.StringVar(&cmdlineArgs.TraceFile, "trace", "", "write trace to file")
67 flag.StringVar(&cmdlineArgs.Memprofile, "memprofile", "", "write memory profile to file")
68 flag.BoolVar(&cmdlineArgs.NoGC, "nogc", false, "turn off GC for debugging")
Lukacs T. Berkif8e24282021-04-14 10:31:00 +020069
70 // Flags representing various modes soong_build can run in
Sasha Smundakaf5ca922022-12-12 21:23:34 -080071 flag.StringVar(&cmdlineArgs.ModuleGraphFile, "module_graph_file", "", "JSON module graph file to output")
72 flag.StringVar(&cmdlineArgs.ModuleActionsFile, "module_actions_file", "", "JSON file to output inputs/outputs of actions of modules")
73 flag.StringVar(&cmdlineArgs.DocFile, "soong_docs", "", "build documentation file to output")
74 flag.StringVar(&cmdlineArgs.BazelQueryViewDir, "bazel_queryview_dir", "", "path to the bazel queryview directory relative to --top")
75 flag.StringVar(&cmdlineArgs.BazelApiBp2buildDir, "bazel_api_bp2build_dir", "", "path to the bazel api_bp2build directory relative to --top")
76 flag.StringVar(&cmdlineArgs.Bp2buildMarker, "bp2build_marker", "", "If set, run bp2build, touch the specified marker file then exit")
77 flag.StringVar(&cmdlineArgs.SymlinkForestMarker, "symlink_forest_marker", "", "If set, create the bp2build symlink forest, touch the specified marker file, then exit")
Lukacs T. Berkif9008072021-08-16 15:24:48 +020078 flag.StringVar(&cmdlineArgs.OutFile, "o", "build.ninja", "the Ninja file to output")
MarkDacekd06db5d2022-11-29 00:47:59 +000079 flag.StringVar(&cmdlineArgs.BazelForceEnabledModules, "bazel-force-enabled-modules", "", "additional modules to build with Bazel. Comma-delimited")
Lukacs T. Berkib078ade2021-08-31 10:42:08 +020080 flag.BoolVar(&cmdlineArgs.EmptyNinjaFile, "empty-ninja-file", false, "write out a 0-byte ninja file")
Chris Parsonsef615e52022-08-18 22:04:11 -040081 flag.BoolVar(&cmdlineArgs.BazelMode, "bazel-mode", false, "use bazel for analysis of certain modules")
Jingwen Chene647db82022-11-01 11:28:29 +000082 flag.BoolVar(&cmdlineArgs.BazelModeStaging, "bazel-mode-staging", false, "use bazel for analysis of certain near-ready modules")
Chris Parsonsef615e52022-08-18 22:04:11 -040083 flag.BoolVar(&cmdlineArgs.BazelModeDev, "bazel-mode-dev", false, "use bazel for analysis of a large number of modules (less stable)")
Lukacs T. Berkib078ade2021-08-31 10:42:08 +020084
Sasha Smundakaf5ca922022-12-12 21:23:34 -080085 // Flags that probably shouldn't be flags of soong_build, but we haven't found
Lukacs T. Berkib078ade2021-08-31 10:42:08 +020086 // the time to remove them yet
Sasha Smundakaf5ca922022-12-12 21:23:34 -080087 flag.BoolVar(&cmdlineArgs.RunGoTests, "t", false, "build and run go tests during bootstrap")
Dan Willemsen66213a62021-09-21 17:50:30 -070088
89 // Disable deterministic randomization in the protobuf package, so incremental
90 // builds with unrelated Soong changes don't trigger large rebuilds (since we
91 // write out text protos in command lines, and command line changes trigger
92 // rebuilds).
93 androidProtobuf.DisableRand()
Colin Crosse87040b2017-12-11 15:52:26 -080094}
95
Jeff Gaston088e29e2017-11-29 16:47:17 -080096func newNameResolver(config android.Config) *android.NameResolver {
Paul Duffin3f7bf9f2022-11-08 12:21:15 +000097 return android.NewNameResolver(config)
Jeff Gaston088e29e2017-11-29 16:47:17 -080098}
99
Lukacs T. Berkiffc9e8d2021-09-07 17:54:38 +0200100func newContext(configuration android.Config) *android.Context {
Colin Crossae8600b2020-10-29 17:09:13 -0700101 ctx := android.NewContext(configuration)
Chris Parsonsf3c96ef2020-09-29 02:23:17 -0400102 ctx.SetNameInterface(newNameResolver(configuration))
103 ctx.SetAllowMissingDependencies(configuration.AllowMissingDependencies())
Spandan Dasc5763832022-11-08 18:42:16 +0000104 ctx.AddIncludeTags(configuration.IncludeTags()...)
Chris Parsonsf3c96ef2020-09-29 02:23:17 -0400105 return ctx
106}
107
Chris Parsonsf874e462022-05-10 13:50:12 -0400108// Bazel-enabled mode. Attaches a mutator to queue Bazel requests, adds a
109// BeforePrepareBuildActionsHook to invoke Bazel, and then uses Bazel metadata
110// for modules that should be handled by Bazel.
Sasha Smundak1845f422022-12-13 14:18:58 -0800111func runMixedModeBuild(ctx *android.Context, extraNinjaDeps []string) string {
Chris Parsonsf874e462022-05-10 13:50:12 -0400112 ctx.EventHandler.Begin("mixed_build")
113 defer ctx.EventHandler.End("mixed_build")
Lukacs T. Berkid518e1a2021-04-14 13:49:50 +0200114
Chris Parsonsf874e462022-05-10 13:50:12 -0400115 bazelHook := func() error {
Sasha Smundak1845f422022-12-13 14:18:58 -0800116 return ctx.Config().BazelContext.InvokeBazel(ctx.Config(), ctx)
Lukacs T. Berki6790ebc2021-04-01 17:55:58 +0200117 }
Chris Parsonsf874e462022-05-10 13:50:12 -0400118 ctx.SetBeforePrepareBuildActionsHook(bazelHook)
Sasha Smundak1845f422022-12-13 14:18:58 -0800119 ninjaDeps := bootstrap.RunBlueprint(cmdlineArgs.Args, bootstrap.DoEverything, ctx.Context, ctx.Config())
Chris Parsons027881c2022-05-24 15:38:38 -0400120 ninjaDeps = append(ninjaDeps, extraNinjaDeps...)
Lukacs T. Berki809d2ed2021-08-18 10:55:32 +0200121
Sasha Smundak1845f422022-12-13 14:18:58 -0800122 bazelPaths, err := readFileLines(ctx.Config().Getenv("BAZEL_DEPS_FILE"))
MarkDacek0d5bca52022-10-10 20:07:48 +0000123 if err != nil {
124 panic("Bazel deps file not found: " + err.Error())
125 }
126 ninjaDeps = append(ninjaDeps, bazelPaths...)
Sasha Smundak1845f422022-12-13 14:18:58 -0800127 ninjaDeps = append(ninjaDeps, writeBuildGlobsNinjaFile(ctx)...)
Lukacs T. Berki809d2ed2021-08-18 10:55:32 +0200128
Paul Duffin780a1852022-11-05 10:17:12 +0000129 writeDepFile(cmdlineArgs.OutFile, ctx.EventHandler, ninjaDeps)
Paul Duffin0c09a432022-11-05 15:28:04 +0000130 return cmdlineArgs.OutFile
Lukacs T. Berki6790ebc2021-04-01 17:55:58 +0200131}
132
133// Run the code-generation phase to convert BazelTargetModules to BUILD files.
Sasha Smundak1845f422022-12-13 14:18:58 -0800134func runQueryView(queryviewDir, queryviewMarker string, ctx *android.Context) {
Chris Parsons715b08f2022-03-22 19:23:40 -0400135 ctx.EventHandler.Begin("queryview")
136 defer ctx.EventHandler.End("queryview")
Cole Faustb85d1a12022-11-08 18:14:01 -0800137 codegenContext := bp2build.NewCodegenContext(ctx.Config(), ctx, bp2build.QueryView, topDir)
Sasha Smundak1845f422022-12-13 14:18:58 -0800138 err := createBazelWorkspace(codegenContext, shared.JoinPath(topDir, queryviewDir))
139 maybeQuit(err, "")
Lukacs T. Berki3a821692021-09-06 17:08:02 +0200140 touch(shared.JoinPath(topDir, queryviewMarker))
Lukacs T. Berki6790ebc2021-04-01 17:55:58 +0200141}
142
Spandan Das5af0bd32022-09-28 20:43:08 +0000143// Run the code-generation phase to convert API contributions to BUILD files.
144// Return marker file for the new synthetic workspace
Sasha Smundak1845f422022-12-13 14:18:58 -0800145func runApiBp2build(ctx *android.Context, extraNinjaDeps []string) string {
Spandan Das5af0bd32022-09-28 20:43:08 +0000146 ctx.EventHandler.Begin("api_bp2build")
147 defer ctx.EventHandler.End("api_bp2build")
Paul Duffinb4e8d912022-11-04 13:35:50 +0000148 // Do not allow missing dependencies.
149 ctx.SetAllowMissingDependencies(false)
Spandan Das5af0bd32022-09-28 20:43:08 +0000150 ctx.RegisterForApiBazelConversion()
151
152 // Register the Android.bp files in the tree
153 // Add them to the workspace's .d file
154 ctx.SetModuleListFile(cmdlineArgs.ModuleListFile)
155 if paths, err := ctx.ListModulePaths("."); err == nil {
156 extraNinjaDeps = append(extraNinjaDeps, paths...)
157 } else {
158 panic(err)
159 }
160
161 // Run the loading and analysis phase
Sasha Smundakaf5ca922022-12-12 21:23:34 -0800162 ninjaDeps := bootstrap.RunBlueprint(cmdlineArgs.Args,
Spandan Das5af0bd32022-09-28 20:43:08 +0000163 bootstrap.StopBeforePrepareBuildActions,
164 ctx.Context,
Sasha Smundak1845f422022-12-13 14:18:58 -0800165 ctx.Config())
Spandan Das5af0bd32022-09-28 20:43:08 +0000166 ninjaDeps = append(ninjaDeps, extraNinjaDeps...)
167
168 // Add the globbed dependencies
Sasha Smundak1845f422022-12-13 14:18:58 -0800169 ninjaDeps = append(ninjaDeps, writeBuildGlobsNinjaFile(ctx)...)
Spandan Das5af0bd32022-09-28 20:43:08 +0000170
171 // Run codegen to generate BUILD files
Cole Faustb85d1a12022-11-08 18:14:01 -0800172 codegenContext := bp2build.NewCodegenContext(ctx.Config(), ctx, bp2build.ApiBp2build, topDir)
Sasha Smundakaf5ca922022-12-12 21:23:34 -0800173 absoluteApiBp2buildDir := shared.JoinPath(topDir, cmdlineArgs.BazelApiBp2buildDir)
Sasha Smundak1845f422022-12-13 14:18:58 -0800174 err := createBazelWorkspace(codegenContext, absoluteApiBp2buildDir)
175 maybeQuit(err, "")
Spandan Das5af0bd32022-09-28 20:43:08 +0000176 ninjaDeps = append(ninjaDeps, codegenContext.AdditionalNinjaDeps()...)
177
178 // Create soong_injection repository
Sasha Smundak1845f422022-12-13 14:18:58 -0800179 soongInjectionFiles := bp2build.CreateSoongInjectionFiles(ctx.Config(), bp2build.CreateCodegenMetrics())
180 absoluteSoongInjectionDir := shared.JoinPath(topDir, ctx.Config().SoongOutDir(), bazel.SoongInjectionDirName)
Spandan Das5af0bd32022-09-28 20:43:08 +0000181 for _, file := range soongInjectionFiles {
Spandan Das067210f2022-12-07 01:14:52 +0000182 // The API targets in api_bp2build workspace do not have any dependency on api_bp2build.
183 // But we need to create these files to prevent errors during Bazel analysis.
184 // These need to be created in Read-Write mode.
185 // This is because the subsequent step (bp2build in api domain analysis) creates them in Read-Write mode
186 // to allow users to edit/experiment in the synthetic workspace.
187 writeReadWriteFile(absoluteSoongInjectionDir, file)
Spandan Das5af0bd32022-09-28 20:43:08 +0000188 }
189
Sasha Smundak1845f422022-12-13 14:18:58 -0800190 workspace := shared.JoinPath(ctx.Config().SoongOutDir(), "api_bp2build")
Spandan Das5af0bd32022-09-28 20:43:08 +0000191 // Create the symlink forest
192 symlinkDeps := bp2build.PlantSymlinkForest(
Sasha Smundak1845f422022-12-13 14:18:58 -0800193 ctx.Config().IsEnvTrue("BP2BUILD_VERBOSE"),
Spandan Das5af0bd32022-09-28 20:43:08 +0000194 topDir,
195 workspace,
Sasha Smundakaf5ca922022-12-12 21:23:34 -0800196 cmdlineArgs.BazelApiBp2buildDir,
Sasha Smundak1845f422022-12-13 14:18:58 -0800197 apiBuildFileExcludes(ctx))
Spandan Das5af0bd32022-09-28 20:43:08 +0000198 ninjaDeps = append(ninjaDeps, symlinkDeps...)
199
200 workspaceMarkerFile := workspace + ".marker"
Paul Duffin780a1852022-11-05 10:17:12 +0000201 writeDepFile(workspaceMarkerFile, ctx.EventHandler, ninjaDeps)
Spandan Das5af0bd32022-09-28 20:43:08 +0000202 touch(shared.JoinPath(topDir, workspaceMarkerFile))
203 return workspaceMarkerFile
204}
205
206// With some exceptions, api_bp2build does not have any dependencies on the checked-in BUILD files
207// Exclude them from the generated workspace to prevent unrelated errors during the loading phase
Sasha Smundak1845f422022-12-13 14:18:58 -0800208func apiBuildFileExcludes(ctx *android.Context) []string {
209 ret := bazelArtifacts()
Spandan Das5af0bd32022-09-28 20:43:08 +0000210 srcs, err := getExistingBazelRelatedFiles(topDir)
Sasha Smundak1845f422022-12-13 14:18:58 -0800211 maybeQuit(err, "Error determining existing Bazel-related files")
Spandan Das5af0bd32022-09-28 20:43:08 +0000212 for _, src := range srcs {
Sasha Smundak1845f422022-12-13 14:18:58 -0800213 // Exclude all src BUILD files
Spandan Das5af0bd32022-09-28 20:43:08 +0000214 if src != "WORKSPACE" &&
215 src != "BUILD" &&
216 src != "BUILD.bazel" &&
217 !strings.HasPrefix(src, "build/bazel") &&
Spandan Das6b91a382022-11-30 02:17:06 +0000218 !strings.HasPrefix(src, "external/bazel-skylib") &&
Spandan Das5af0bd32022-09-28 20:43:08 +0000219 !strings.HasPrefix(src, "prebuilts/clang") {
220 ret = append(ret, src)
221 }
222 }
Sasha Smundak1845f422022-12-13 14:18:58 -0800223 // Android.bp files for api surfaces are mounted to out/, but out/ should not be a
224 // dep for api_bp2build. Otherwise, api_bp2build will be run every single time
225 ret = append(ret, ctx.Config().OutDir())
Spandan Das5af0bd32022-09-28 20:43:08 +0000226 return ret
227}
228
Paul Duffin780a1852022-11-05 10:17:12 +0000229func writeMetrics(configuration android.Config, eventHandler *metrics.EventHandler, metricsDir string) {
Chris Parsons715b08f2022-03-22 19:23:40 -0400230 if len(metricsDir) < 1 {
231 fmt.Fprintf(os.Stderr, "\nMissing required env var for generating soong metrics: LOG_DIR\n")
232 os.Exit(1)
233 }
234 metricsFile := filepath.Join(metricsDir, "soong_build_metrics.pb")
235 err := android.WriteMetrics(configuration, eventHandler, metricsFile)
Sasha Smundak1845f422022-12-13 14:18:58 -0800236 maybeQuit(err, "error writing soong_build metrics %s", metricsFile)
Lukacs T. Berki6790ebc2021-04-01 17:55:58 +0200237}
238
Sasha Smundakaf5ca922022-12-12 21:23:34 -0800239func writeJsonModuleGraphAndActions(ctx *android.Context, cmdArgs android.CmdArgs) {
240 graphFile, graphErr := os.Create(shared.JoinPath(topDir, cmdArgs.ModuleGraphFile))
Sasha Smundak1845f422022-12-13 14:18:58 -0800241 maybeQuit(graphErr, "graph err")
kgui67007242022-01-25 13:50:25 +0800242 defer graphFile.Close()
Sasha Smundak1845f422022-12-13 14:18:58 -0800243 actionsFile, actionsErr := os.Create(shared.JoinPath(topDir, cmdArgs.ModuleActionsFile))
244 maybeQuit(actionsErr, "actions err")
kgui67007242022-01-25 13:50:25 +0800245 defer actionsFile.Close()
246 ctx.Context.PrintJSONGraphAndActions(graphFile, actionsFile)
Lukacs T. Berki97bb9f12021-04-01 18:28:45 +0200247}
248
Sasha Smundak1845f422022-12-13 14:18:58 -0800249func writeBuildGlobsNinjaFile(ctx *android.Context) []string {
Chris Parsons715b08f2022-03-22 19:23:40 -0400250 ctx.EventHandler.Begin("globs_ninja_file")
251 defer ctx.EventHandler.End("globs_ninja_file")
252
Sasha Smundak1845f422022-12-13 14:18:58 -0800253 globDir := bootstrap.GlobDirectory(ctx.Config().SoongOutDir(), globListDir)
Lukacs T. Berki809d2ed2021-08-18 10:55:32 +0200254 bootstrap.WriteBuildGlobsNinjaFile(&bootstrap.GlobSingleton{
Chris Parsons715b08f2022-03-22 19:23:40 -0400255 GlobLister: ctx.Globs,
Lukacs T. Berki809d2ed2021-08-18 10:55:32 +0200256 GlobFile: globFile,
257 GlobDir: globDir,
Chris Parsons715b08f2022-03-22 19:23:40 -0400258 SrcDir: ctx.SrcDir(),
Sasha Smundak1845f422022-12-13 14:18:58 -0800259 }, ctx.Config())
Lukacs T. Berki809d2ed2021-08-18 10:55:32 +0200260 return bootstrap.GlobFileListFiles(globDir)
261}
262
Paul Duffin780a1852022-11-05 10:17:12 +0000263func writeDepFile(outputFile string, eventHandler *metrics.EventHandler, ninjaDeps []string) {
Chris Parsons715b08f2022-03-22 19:23:40 -0400264 eventHandler.Begin("ninja_deps")
265 defer eventHandler.End("ninja_deps")
Lukacs T. Berkie571dc32021-08-25 14:14:13 +0200266 depFile := shared.JoinPath(topDir, outputFile+".d")
267 err := deptools.WriteDepFile(depFile, outputFile, ninjaDeps)
Sasha Smundak1845f422022-12-13 14:18:58 -0800268 maybeQuit(err, "error writing depfile '%s'", depFile)
Paul Duffin0c09a432022-11-05 15:28:04 +0000269}
Lukacs T. Berki6790ebc2021-04-01 17:55:58 +0200270
Paul Duffin0c09a432022-11-05 15:28:04 +0000271// runSoongOnlyBuild runs the standard Soong build in a number of different modes.
Sasha Smundak1845f422022-12-13 14:18:58 -0800272func runSoongOnlyBuild(ctx *android.Context, extraNinjaDeps []string) string {
Paul Duffin39eae8f2022-11-05 14:59:52 +0000273 ctx.EventHandler.Begin("soong_build")
274 defer ctx.EventHandler.End("soong_build")
275
Paul Duffin0c09a432022-11-05 15:28:04 +0000276 var stopBefore bootstrap.StopBefore
Sasha Smundak1845f422022-12-13 14:18:58 -0800277 switch ctx.Config().BuildMode {
278 case android.GenerateModuleGraph:
Paul Duffin0c09a432022-11-05 15:28:04 +0000279 stopBefore = bootstrap.StopBeforeWriteNinja
Usta Shrestha7fae6952022-12-21 11:46:28 -0500280 case android.GenerateQueryView, android.GenerateDocFile:
Paul Duffin0c09a432022-11-05 15:28:04 +0000281 stopBefore = bootstrap.StopBeforePrepareBuildActions
Sasha Smundak1845f422022-12-13 14:18:58 -0800282 default:
Paul Duffin0c09a432022-11-05 15:28:04 +0000283 stopBefore = bootstrap.DoEverything
284 }
285
Sasha Smundak1845f422022-12-13 14:18:58 -0800286 ninjaDeps := bootstrap.RunBlueprint(cmdlineArgs.Args, stopBefore, ctx.Context, ctx.Config())
Paul Duffin0c09a432022-11-05 15:28:04 +0000287 ninjaDeps = append(ninjaDeps, extraNinjaDeps...)
288
Sasha Smundak1845f422022-12-13 14:18:58 -0800289 globListFiles := writeBuildGlobsNinjaFile(ctx)
Paul Duffin0c09a432022-11-05 15:28:04 +0000290 ninjaDeps = append(ninjaDeps, globListFiles...)
291
292 // Convert the Soong module graph into Bazel BUILD files.
Sasha Smundak1845f422022-12-13 14:18:58 -0800293 switch ctx.Config().BuildMode {
294 case android.GenerateQueryView:
Sasha Smundakaf5ca922022-12-12 21:23:34 -0800295 queryviewMarkerFile := cmdlineArgs.BazelQueryViewDir + ".marker"
Sasha Smundak1845f422022-12-13 14:18:58 -0800296 runQueryView(cmdlineArgs.BazelQueryViewDir, queryviewMarkerFile, ctx)
Paul Duffin0c09a432022-11-05 15:28:04 +0000297 writeDepFile(queryviewMarkerFile, ctx.EventHandler, ninjaDeps)
298 return queryviewMarkerFile
Sasha Smundak1845f422022-12-13 14:18:58 -0800299 case android.GenerateModuleGraph:
Sasha Smundakaf5ca922022-12-12 21:23:34 -0800300 writeJsonModuleGraphAndActions(ctx, cmdlineArgs)
301 writeDepFile(cmdlineArgs.ModuleGraphFile, ctx.EventHandler, ninjaDeps)
302 return cmdlineArgs.ModuleGraphFile
Sasha Smundak1845f422022-12-13 14:18:58 -0800303 case android.GenerateDocFile:
Paul Duffin0c09a432022-11-05 15:28:04 +0000304 // TODO: we could make writeDocs() return the list of documentation files
305 // written and add them to the .d file. Then soong_docs would be re-run
306 // whenever one is deleted.
Sasha Smundak1845f422022-12-13 14:18:58 -0800307 err := writeDocs(ctx, shared.JoinPath(topDir, cmdlineArgs.DocFile))
308 maybeQuit(err, "error building Soong documentation")
Sasha Smundakaf5ca922022-12-12 21:23:34 -0800309 writeDepFile(cmdlineArgs.DocFile, ctx.EventHandler, ninjaDeps)
310 return cmdlineArgs.DocFile
Sasha Smundak1845f422022-12-13 14:18:58 -0800311 default:
Paul Duffin0c09a432022-11-05 15:28:04 +0000312 // The actual output (build.ninja) was written in the RunBlueprint() call
313 // above
314 writeDepFile(cmdlineArgs.OutFile, ctx.EventHandler, ninjaDeps)
Paul Duffinb713ddf2022-11-05 16:14:30 +0000315 return cmdlineArgs.OutFile
Paul Duffin0c09a432022-11-05 15:28:04 +0000316 }
Lukacs T. Berkif8e24282021-04-14 10:31:00 +0200317}
318
319// soong_ui dumps the available environment variables to
320// soong.environment.available . Then soong_build itself is run with an empty
321// environment so that the only way environment variables can be accessed is
322// using Config, which tracks access to them.
323
324// At the end of the build, a file called soong.environment.used is written
325// containing the current value of all used environment variables. The next
326// time soong_ui is run, it checks whether any environment variables that was
327// used had changed and if so, it deletes soong.environment.used to cause a
328// rebuild.
329//
330// The dependency of build.ninja on soong.environment.used is declared in
331// build.ninja.d
332func parseAvailableEnv() map[string]string {
333 if availableEnvFile == "" {
334 fmt.Fprintf(os.Stderr, "--available_env not set\n")
335 os.Exit(1)
336 }
Lukacs T. Berkif8e24282021-04-14 10:31:00 +0200337 result, err := shared.EnvFromFile(shared.JoinPath(topDir, availableEnvFile))
Sasha Smundak1845f422022-12-13 14:18:58 -0800338 maybeQuit(err, "error reading available environment file '%s'", availableEnvFile)
Lukacs T. Berkif8e24282021-04-14 10:31:00 +0200339 return result
Lukacs T. Berki6790ebc2021-04-01 17:55:58 +0200340}
341
Colin Cross3f40fa42015-01-30 17:27:36 -0800342func main() {
343 flag.Parse()
344
Lukacs T. Berki7d613bf2021-03-02 10:09:41 +0100345 shared.ReexecWithDelveMaybe(delveListen, delvePath)
Lukacs T. Berki7690c092021-02-26 14:27:36 +0100346 android.InitSandbox(topDir)
Lukacs T. Berki7690c092021-02-26 14:27:36 +0100347
Lukacs T. Berkif8e24282021-04-14 10:31:00 +0200348 availableEnv := parseAvailableEnv()
Sasha Smundakaf5ca922022-12-12 21:23:34 -0800349 configuration, err := android.NewConfig(cmdlineArgs, availableEnv)
Sasha Smundak1845f422022-12-13 14:18:58 -0800350 maybeQuit(err, "")
Lukacs T. Berkid1e3f1f2021-03-16 08:55:23 +0100351 if configuration.Getenv("ALLOW_MISSING_DEPENDENCIES") == "true" {
352 configuration.SetAllowMissingDependencies()
353 }
354
Sasha Smundakaf5ca922022-12-12 21:23:34 -0800355 extraNinjaDeps := []string{configuration.ProductVariablesFileName, usedEnvFile}
Lukacs T. Berki7d613bf2021-03-02 10:09:41 +0100356 if shared.IsDebugging() {
Colin Crossaa812d12019-06-19 13:33:24 -0700357 // Add a non-existent file to the dependencies so that soong_build will rerun when the debugger is
358 // enabled even if it completed successfully.
Lukacs T. Berki9f6c24a2021-08-26 15:07:24 +0200359 extraNinjaDeps = append(extraNinjaDeps, filepath.Join(configuration.SoongOutDir(), "always_rerun_for_delve"))
Colin Crossaa812d12019-06-19 13:33:24 -0700360 }
Jingwen Chendaa54bc2020-12-14 02:58:54 -0500361
Dan Willemsenccf36aa2022-04-20 23:11:43 -0700362 // Bypass configuration.Getenv, as LOG_DIR does not need to be dependency tracked. By definition, it will
363 // change between every CI build, so tracking it would require re-running Soong for every build.
Sasha Smundak1845f422022-12-13 14:18:58 -0800364 metricsDir := availableEnv["LOG_DIR"]
Dan Willemsenccf36aa2022-04-20 23:11:43 -0700365
Joe Onorato2e5e4012022-06-07 17:16:08 -0700366 ctx := newContext(configuration)
Joe Onorato2e5e4012022-06-07 17:16:08 -0700367
Sasha Smundak1845f422022-12-13 14:18:58 -0800368 var finalOutputFile string
Joe Onorato2e5e4012022-06-07 17:16:08 -0700369
Sasha Smundak1845f422022-12-13 14:18:58 -0800370 // Run Soong for a specific activity, like bp2build, queryview
371 // or the actual Soong build for the build.ninja file.
372 switch configuration.BuildMode {
373 case android.SymlinkForest:
374 finalOutputFile = runSymlinkForestCreation(ctx, extraNinjaDeps, metricsDir)
375 case android.Bp2build:
376 // Run the alternate pipeline of bp2build mutators and singleton to convert
377 // Blueprint to BUILD files before everything else.
378 finalOutputFile = runBp2Build(ctx, extraNinjaDeps, metricsDir)
379 case android.ApiBp2build:
380 finalOutputFile = runApiBp2build(ctx, extraNinjaDeps)
381 writeMetrics(configuration, ctx.EventHandler, metricsDir)
382 default:
383 ctx.Register()
384 if configuration.IsMixedBuildsEnabled() {
385 finalOutputFile = runMixedModeBuild(ctx, extraNinjaDeps)
386 } else {
387 finalOutputFile = runSoongOnlyBuild(ctx, extraNinjaDeps)
388 }
389 writeMetrics(configuration, ctx.EventHandler, metricsDir)
390 }
Lukacs T. Berkif8e24282021-04-14 10:31:00 +0200391 writeUsedEnvironmentFile(configuration, finalOutputFile)
Lukacs T. Berkif0b3b942021-03-23 11:46:47 +0100392}
393
Lukacs T. Berkif8e24282021-04-14 10:31:00 +0200394func writeUsedEnvironmentFile(configuration android.Config, finalOutputFile string) {
395 if usedEnvFile == "" {
396 return
397 }
398
399 path := shared.JoinPath(topDir, usedEnvFile)
Lukacs T. Berkif0b3b942021-03-23 11:46:47 +0100400 data, err := shared.EnvFileContents(configuration.EnvDeps())
Sasha Smundak1845f422022-12-13 14:18:58 -0800401 maybeQuit(err, "error writing used environment file '%s'\n", usedEnvFile)
Lukacs T. Berkif0b3b942021-03-23 11:46:47 +0100402
Usta Shrestha2ba28a32022-10-24 11:33:09 -0400403 if preexistingData, err := os.ReadFile(path); err != nil {
404 if !os.IsNotExist(err) {
Sasha Smundak1845f422022-12-13 14:18:58 -0800405 maybeQuit(err, "error reading used environment file '%s'", usedEnvFile)
Usta Shrestha2ba28a32022-10-24 11:33:09 -0400406 }
407 } else if bytes.Equal(preexistingData, data) {
408 // used environment file is unchanged
409 return
410 }
Sasha Smundak1845f422022-12-13 14:18:58 -0800411 err = os.WriteFile(path, data, 0666)
412 maybeQuit(err, "error writing used environment file '%s'", usedEnvFile)
413
Lukacs T. Berkif8e24282021-04-14 10:31:00 +0200414 // Touch the output file so that it's not older than the file we just
Lukacs T. Berkic99c9472021-03-24 10:50:06 +0100415 // wrote. We can't write the environment file earlier because one an access
416 // new environment variables while writing it.
Lukacs T. Berkif8e24282021-04-14 10:31:00 +0200417 touch(shared.JoinPath(topDir, finalOutputFile))
Colin Cross3f40fa42015-01-30 17:27:36 -0800418}
Jingwen Chen5ba7e472020-07-15 10:06:41 +0000419
Lukacs T. Berkif8e24282021-04-14 10:31:00 +0200420func touch(path string) {
421 f, err := os.OpenFile(path, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0666)
Sasha Smundak1845f422022-12-13 14:18:58 -0800422 maybeQuit(err, "Error touching '%s'", path)
Lukacs T. Berkif8e24282021-04-14 10:31:00 +0200423 err = f.Close()
Sasha Smundak1845f422022-12-13 14:18:58 -0800424 maybeQuit(err, "Error touching '%s'", path)
Lukacs T. Berkif8e24282021-04-14 10:31:00 +0200425
426 currentTime := time.Now().Local()
427 err = os.Chtimes(path, currentTime, currentTime)
Sasha Smundak1845f422022-12-13 14:18:58 -0800428 maybeQuit(err, "error touching '%s'", path)
Rupert Shuttleworth00960792021-05-12 21:20:13 -0400429}
430
Rupert Shuttleworthe03bb612021-05-20 19:34:50 -0400431// Read the bazel.list file that the Soong Finder already dumped earlier (hopefully)
432// It contains the locations of BUILD files, BUILD.bazel files, etc. in the source dir
433func getExistingBazelRelatedFiles(topDir string) ([]string, error) {
Lukacs T. Berkif9008072021-08-16 15:24:48 +0200434 bazelFinderFile := filepath.Join(filepath.Dir(cmdlineArgs.ModuleListFile), "bazel.list")
Rupert Shuttleworthe03bb612021-05-20 19:34:50 -0400435 if !filepath.IsAbs(bazelFinderFile) {
436 // Assume this was a relative path under topDir
437 bazelFinderFile = filepath.Join(topDir, bazelFinderFile)
438 }
Sasha Smundak1845f422022-12-13 14:18:58 -0800439 return readFileLines(bazelFinderFile)
Rupert Shuttleworthe03bb612021-05-20 19:34:50 -0400440}
441
Spandan Das5af0bd32022-09-28 20:43:08 +0000442func bazelArtifacts() []string {
443 return []string{
444 "bazel-bin",
445 "bazel-genfiles",
446 "bazel-out",
447 "bazel-testlogs",
448 "bazel-" + filepath.Base(topDir),
449 }
450}
451
Lukacs T. Berkic541cd22022-10-26 07:26:50 +0000452// This could in theory easily be separated into a binary that generically
453// merges two directories into a symlink tree. The main obstacle is that this
454// function currently depends on both Bazel-specific knowledge (the existence
455// of bazel-* symlinks) and configuration (the set of BUILD.bazel files that
456// should and should not be kept)
457//
458// Ideally, bp2build would write a file that contains instructions to the
459// symlink tree creation binary. Then the latter would not need to depend on
460// the very heavy-weight machinery of soong_build .
Sasha Smundak1845f422022-12-13 14:18:58 -0800461func runSymlinkForestCreation(ctx *android.Context, extraNinjaDeps []string, metricsDir string) string {
Paul Duffinb4e8d912022-11-04 13:35:50 +0000462 ctx.EventHandler.Do("symlink_forest", func() {
Usta (Tsering) Shrestha93b2a9b2022-12-01 05:55:35 +0000463 var ninjaDeps []string
464 ninjaDeps = append(ninjaDeps, extraNinjaDeps...)
Sasha Smundak1845f422022-12-13 14:18:58 -0800465 verbose := ctx.Config().IsEnvTrue("BP2BUILD_VERBOSE")
Usta (Tsering) Shrestha93b2a9b2022-12-01 05:55:35 +0000466
467 // PlantSymlinkForest() returns all the directories that were readdir()'ed.
468 // Such a directory SHOULD be added to `ninjaDeps` so that a child directory
469 // or file created/deleted under it would trigger an update of the symlink forest.
Sasha Smundak1845f422022-12-13 14:18:58 -0800470 generatedRoot := shared.JoinPath(ctx.Config().SoongOutDir(), "bp2build")
471 workspaceRoot := shared.JoinPath(ctx.Config().SoongOutDir(), "workspace")
Usta (Tsering) Shrestha93b2a9b2022-12-01 05:55:35 +0000472 ctx.EventHandler.Do("plant", func() {
473 symlinkForestDeps := bp2build.PlantSymlinkForest(
Sasha Smundak1845f422022-12-13 14:18:58 -0800474 verbose, topDir, workspaceRoot, generatedRoot, excludedFromSymlinkForest(ctx, verbose))
Usta (Tsering) Shrestha93b2a9b2022-12-01 05:55:35 +0000475 ninjaDeps = append(ninjaDeps, symlinkForestDeps...)
476 })
477
Sasha Smundakaf5ca922022-12-12 21:23:34 -0800478 writeDepFile(cmdlineArgs.SymlinkForestMarker, ctx.EventHandler, ninjaDeps)
479 touch(shared.JoinPath(topDir, cmdlineArgs.SymlinkForestMarker))
Usta (Tsering) Shrestha93b2a9b2022-12-01 05:55:35 +0000480 })
usta4f5d2c12022-10-28 23:32:01 -0400481 codegenMetrics := bp2build.ReadCodegenMetrics(metricsDir)
482 if codegenMetrics == nil {
483 m := bp2build.CreateCodegenMetrics()
484 codegenMetrics = &m
485 } else {
486 //TODO (usta) we cannot determine if we loaded a stale file, i.e. from an unrelated prior
487 //invocation of codegen. We should simply use a separate .pb file
488 }
Paul Duffinb4e8d912022-11-04 13:35:50 +0000489 writeBp2BuildMetrics(codegenMetrics, ctx.EventHandler, metricsDir)
Sasha Smundakaf5ca922022-12-12 21:23:34 -0800490 return cmdlineArgs.SymlinkForestMarker
Lukacs T. Berkic541cd22022-10-26 07:26:50 +0000491}
492
Sasha Smundak1845f422022-12-13 14:18:58 -0800493func excludedFromSymlinkForest(ctx *android.Context, verbose bool) []string {
494 excluded := bazelArtifacts()
495 if cmdlineArgs.OutDir[0] != '/' {
496 excluded = append(excluded, cmdlineArgs.OutDir)
497 }
498
499 // Find BUILD files in the srcDir which are not in the allowlist
500 // (android.Bp2BuildConversionAllowlist#ShouldKeepExistingBuildFileForDir)
501 // and return their paths so they can be left out of the Bazel workspace dir (i.e. ignored)
502 existingBazelFiles, err := getExistingBazelRelatedFiles(topDir)
503 maybeQuit(err, "Error determining existing Bazel-related files")
504
505 for _, path := range existingBazelFiles {
506 fullPath := shared.JoinPath(topDir, path)
507 fileInfo, err2 := os.Stat(fullPath)
508 if err2 != nil {
509 // Warn about error, but continue trying to check files
510 fmt.Fprintf(os.Stderr, "WARNING: Error accessing path '%s', err: %s\n", fullPath, err2)
511 continue
512 }
513 // Exclude only files named 'BUILD' or 'BUILD.bazel' and unless forcibly kept
514 if fileInfo.IsDir() ||
515 (fileInfo.Name() != "BUILD" && fileInfo.Name() != "BUILD.bazel") ||
516 ctx.Config().Bp2buildPackageConfig.ShouldKeepExistingBuildFileForDir(filepath.Dir(path)) {
517 // Don't ignore this existing build file
518 continue
519 }
520 if verbose {
521 fmt.Fprintf(os.Stderr, "Ignoring existing BUILD file: %s\n", path)
522 }
523 excluded = append(excluded, path)
524 }
525
526 // Temporarily exclude stuff to make `bazel build //external/...` (and `bazel build //frameworks/...`) work
527 excluded = append(excluded,
528 // FIXME: 'autotest_lib' is a symlink back to external/autotest, and this causes an infinite
529 // symlink expansion error for Bazel
530 "external/autotest/venv/autotest_lib",
531 "external/autotest/autotest_lib",
532 "external/autotest/client/autotest_lib/client",
533
534 // FIXME: The external/google-fruit/extras/bazel_root/third_party/fruit dir is poison
535 // It contains several symlinks back to real source dirs, and those source dirs contain
536 // BUILD files we want to ignore
537 "external/google-fruit/extras/bazel_root/third_party/fruit",
538
539 // FIXME: 'frameworks/compile/slang' has a filegroup error due to an escaping issue
540 "frameworks/compile/slang",
541
542 // FIXME(b/260809113): 'prebuilts/clang/host/linux-x86/clang-dev' is a tool-generated symlink
543 // directory that contains a BUILD file. The bazel files finder code doesn't traverse into symlink dirs,
544 // and hence is not aware of this BUILD file and exclude it accordingly during symlink forest generation
545 // when checking against keepExistingBuildFiles allowlist.
546 //
547 // This is necessary because globs in //prebuilts/clang/host/linux-x86/BUILD
548 // currently assume no subpackages (keepExistingBuildFile is not recursive for that directory).
549 //
550 // This is a bandaid until we the symlink forest logic can intelligently exclude BUILD files found in
551 // source symlink dirs according to the keepExistingBuildFile allowlist.
552 "prebuilts/clang/host/linux-x86/clang-dev",
553 )
554 return excluded
555}
556
Jingwen Chendaa54bc2020-12-14 02:58:54 -0500557// Run Soong in the bp2build mode. This creates a standalone context that registers
558// an alternate pipeline of mutators and singletons specifically for generating
559// Bazel BUILD files instead of Ninja files.
Sasha Smundak1845f422022-12-13 14:18:58 -0800560func runBp2Build(ctx *android.Context, extraNinjaDeps []string, metricsDir string) string {
usta4f5d2c12022-10-28 23:32:01 -0400561 var codegenMetrics *bp2build.CodegenMetrics
Paul Duffinb4e8d912022-11-04 13:35:50 +0000562 ctx.EventHandler.Do("bp2build", func() {
Jingwen Cheneb76c432021-01-28 08:22:12 -0500563
Usta Shrestha5c6b9482022-05-26 12:22:17 -0400564 // Propagate "allow misssing dependencies" bit. This is normally set in
Paul Duffinb4e8d912022-11-04 13:35:50 +0000565 // newContext(), but we create ctx without calling that method.
Sasha Smundak1845f422022-12-13 14:18:58 -0800566 ctx.SetAllowMissingDependencies(ctx.Config().AllowMissingDependencies())
567 ctx.SetNameInterface(newNameResolver(ctx.Config()))
Paul Duffinb4e8d912022-11-04 13:35:50 +0000568 ctx.RegisterForBazelConversion()
569 ctx.SetModuleListFile(cmdlineArgs.ModuleListFile)
Jingwen Cheneb76c432021-01-28 08:22:12 -0500570
Usta Shresthaa117c582022-10-04 12:13:25 -0400571 var ninjaDeps []string
572 ninjaDeps = append(ninjaDeps, extraNinjaDeps...)
573
Usta Shrestha5c6b9482022-05-26 12:22:17 -0400574 // Run the loading and analysis pipeline to prepare the graph of regular
575 // Modules parsed from Android.bp files, and the BazelTargetModules mapped
576 // from the regular Modules.
Paul Duffinb4e8d912022-11-04 13:35:50 +0000577 ctx.EventHandler.Do("bootstrap", func() {
Usta Shresthaa117c582022-10-04 12:13:25 -0400578 blueprintArgs := cmdlineArgs
Sasha Smundak1845f422022-12-13 14:18:58 -0800579 bootstrapDeps := bootstrap.RunBlueprint(blueprintArgs.Args,
580 bootstrap.StopBeforePrepareBuildActions, ctx.Context, ctx.Config())
Usta Shresthaa117c582022-10-04 12:13:25 -0400581 ninjaDeps = append(ninjaDeps, bootstrapDeps...)
582 })
Lukacs T. Berkid518e1a2021-04-14 13:49:50 +0200583
Sasha Smundak1845f422022-12-13 14:18:58 -0800584 globListFiles := writeBuildGlobsNinjaFile(ctx)
Usta Shrestha5c6b9482022-05-26 12:22:17 -0400585 ninjaDeps = append(ninjaDeps, globListFiles...)
Lukacs T. Berkid518e1a2021-04-14 13:49:50 +0200586
Usta Shrestha5c6b9482022-05-26 12:22:17 -0400587 // Run the code-generation phase to convert BazelTargetModules to BUILD files
Usta Shresthaa117c582022-10-04 12:13:25 -0400588 // and print conversion codegenMetrics to the user.
Cole Faustb85d1a12022-11-08 18:14:01 -0800589 codegenContext := bp2build.NewCodegenContext(ctx.Config(), ctx, bp2build.Bp2Build, topDir)
Paul Duffinb4e8d912022-11-04 13:35:50 +0000590 ctx.EventHandler.Do("codegen", func() {
Usta Shresthaa117c582022-10-04 12:13:25 -0400591 codegenMetrics = bp2build.Codegen(codegenContext)
592 })
Lukacs T. Berkib353cca2021-04-16 13:47:36 +0200593
Usta Shrestha5c6b9482022-05-26 12:22:17 -0400594 ninjaDeps = append(ninjaDeps, codegenContext.AdditionalNinjaDeps()...)
Chris Parsons715b08f2022-03-22 19:23:40 -0400595
Sasha Smundakaf5ca922022-12-12 21:23:34 -0800596 writeDepFile(cmdlineArgs.Bp2buildMarker, ctx.EventHandler, ninjaDeps)
597 touch(shared.JoinPath(topDir, cmdlineArgs.Bp2buildMarker))
Usta Shrestha5c6b9482022-05-26 12:22:17 -0400598 })
Chris Parsons715b08f2022-03-22 19:23:40 -0400599
Lukacs T. Berkib353cca2021-04-16 13:47:36 +0200600 // Only report metrics when in bp2build mode. The metrics aren't relevant
601 // for queryview, since that's a total repo-wide conversion and there's a
602 // 1:1 mapping for each module.
Sasha Smundak1845f422022-12-13 14:18:58 -0800603 if ctx.Config().IsEnvTrue("BP2BUILD_VERBOSE") {
Usta Shresthaa117c582022-10-04 12:13:25 -0400604 codegenMetrics.Print()
Sasha Smundak0fd93e02022-05-19 19:34:31 -0700605 }
Paul Duffinb4e8d912022-11-04 13:35:50 +0000606 writeBp2BuildMetrics(codegenMetrics, ctx.EventHandler, metricsDir)
Sasha Smundakaf5ca922022-12-12 21:23:34 -0800607 return cmdlineArgs.Bp2buildMarker
Jingwen Chendaa54bc2020-12-14 02:58:54 -0500608}
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux947fdbf2021-11-10 09:55:20 -0500609
610// Write Bp2Build metrics into $LOG_DIR
Paul Duffinf4906562022-11-05 14:21:16 +0000611func writeBp2BuildMetrics(codegenMetrics *bp2build.CodegenMetrics, eventHandler *metrics.EventHandler, metricsDir string) {
Chris Parsons715b08f2022-03-22 19:23:40 -0400612 for _, event := range eventHandler.CompletedEvents() {
usta4f5d2c12022-10-28 23:32:01 -0400613 codegenMetrics.AddEvent(&bp2build_metrics_proto.Event{
614 Name: event.Id,
615 StartTime: uint64(event.Start.UnixNano()),
616 RealTime: event.RuntimeNanoseconds(),
617 })
Chris Parsons715b08f2022-03-22 19:23:40 -0400618 }
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux947fdbf2021-11-10 09:55:20 -0500619 if len(metricsDir) < 1 {
620 fmt.Fprintf(os.Stderr, "\nMissing required env var for generating bp2build metrics: LOG_DIR\n")
621 os.Exit(1)
622 }
Chris Parsons715b08f2022-03-22 19:23:40 -0400623 codegenMetrics.Write(metricsDir)
Alex Márquez Pérez Muñíz Díaz Púras Thaureaux947fdbf2021-11-10 09:55:20 -0500624}
MarkDacek0d5bca52022-10-10 20:07:48 +0000625
Sasha Smundak1845f422022-12-13 14:18:58 -0800626func readFileLines(path string) ([]string, error) {
627 data, err := os.ReadFile(path)
628 if err == nil {
629 return strings.Split(strings.TrimSpace(string(data)), "\n"), nil
MarkDacek0d5bca52022-10-10 20:07:48 +0000630 }
Sasha Smundak1845f422022-12-13 14:18:58 -0800631 return nil, err
632
633}
634func maybeQuit(err error, format string, args ...interface{}) {
635 if err == nil {
636 return
637 }
638 if format != "" {
639 fmt.Fprintln(os.Stderr, fmt.Sprintf(format, args...)+": "+err.Error())
640 } else {
641 fmt.Fprintln(os.Stderr, err)
642 }
643 os.Exit(1)
MarkDacek0d5bca52022-10-10 20:07:48 +0000644}