Migrate CreateDeployInfoAction from Native to Starlark.

PiperOrigin-RevId: 611167708
Change-Id: I15d17a4850df5647921b8f2d61eadaf7136c1b16

rules/android_binary_internal/impl.bzl

@@ -597,18 +597,27 @@ def finalize(
         if apk_packaging_ctx.v4_signature_file:
             files.append(apk_packaging_ctx.v4_signature_file)
 
+        providers.extend(
+            [
+                DefaultInfo(
+                    files = depset(files),
+                ),
+                OutputGroupInfo(
+                    android_deploy_info = [
+                        apk_packaging_ctx.deploy_info,
+                        packaged_resources_ctx.processed_manifest,
+                    ],
+                    _validation = depset(validation_outputs),
+                ),
+            ],
+        )
+    else:
         providers.append(
-            DefaultInfo(
-                files = depset(files),
+            OutputGroupInfo(
+                _validation = depset(validation_outputs),
             ),
         )
 
-    providers.append(
-        OutputGroupInfo(
-            _validation = depset(validation_outputs),
-        ),
-    )
-
     # Add the AndroidApplicationResourceInfo provider from resource shrinking if it was performed.
     # TODO(ahumesky): This can be cleaned up after the rules are fully migrated to Starlark.
     # Packaging will be the final step in the pipeline, and that step can be responsible for picking
@@ -910,6 +919,7 @@ def _process_apk_packaging(ctx, packaged_resources_ctx, native_libs_ctx, dex_ctx
             deterministic_signing = False,
             java_toolchain = common.get_java_toolchain(ctx),
             resource_extractor = get_android_toolchain(ctx).resource_extractor.files_to_run,
+            deploy_info_writer = get_android_toolchain(ctx).deploy_info_writer.files_to_run,
             zip_aligner = get_android_sdk(ctx).zip_align,
             apk_signer = get_android_sdk(ctx).apk_signer,
             toolchain_type = ANDROID_TOOLCHAIN_TYPE,

rules/apk_packaging.bzl

@@ -17,12 +17,14 @@ load(":java.bzl", "java")
 
 _PROVIDERS = "providers"
 _V4_SIGNATURE_FILE = "v4_signature_file"
+_DEPLOY_INFO = "deploy_info"
 
 _ApkContextInfo = provider(
     "Apk Context Info",
     fields = {
         _PROVIDERS: "The list of all providers to propagate.",
         _V4_SIGNATURE_FILE: "The v4 signature file.",
+        _DEPLOY_INFO: "A proto providing information about how to deploy and launch the APK",
     },
 )
 
@@ -49,6 +51,7 @@ def _process(
         deterministic_signing = False,
         java_toolchain = None,
         resource_extractor = None,
+        deploy_info_writer = None,
         zip_aligner = None,
         apk_signer = None,
         toolchain_type = None):
@@ -77,6 +80,7 @@ def _process(
         deterministic_signing: Boolean. Whether to enable deterministic DSA signing.
         java_toolchain: The JavaToolchain target.
         resource_extractor: FilesToRunProvider. The executable to extract resources from java_resources_zip.
+        deploy_info_writer: FilesToRunProvider. The executable to write the deploy info proto file.
         zip_aligner: FilesToRunProvider. The executable to zipalign the APK.
         apk_signer: FilesToRunProvider. The executable to sign the APK.
         toolchain_type: String. The Android toolchain type.
@@ -129,6 +133,16 @@ def _process(
         toolchain_type = toolchain_type,
     )
 
+    deploy_info = ctx.actions.declare_file(ctx.label.name + "_files/deploy_info.deployinfo.pb")
+    _create_deploy_info(
+        ctx,
+        deploy_info,
+        manifest = merged_manifest,
+        apks_to_deploy = [signed_apk] + ([v4_signature_file] if v4_signature_file else []),
+        deploy_info_writer = deploy_info_writer,
+        toolchain_type = toolchain_type,
+    )
+
     apk_packaging_ctx[_PROVIDERS].append(
         ApkInfo(
             signed_apk = signed_apk,
@@ -142,6 +156,7 @@ def _process(
         ),
     )
     apk_packaging_ctx[_V4_SIGNATURE_FILE] = v4_signature_file
+    apk_packaging_ctx[_DEPLOY_INFO] = deploy_info
 
     return _ApkContextInfo(**apk_packaging_ctx)
 
@@ -332,6 +347,28 @@ def _sign_apk(
         toolchain = toolchain_type,
     )
 
+def _create_deploy_info(
+        ctx,
+        deploy_info,
+        manifest = None,
+        apks_to_deploy = [],
+        deploy_info_writer = None,
+        toolchain_type = None):
+    """Creates a deploy info proto."""
+    args = ctx.actions.args()
+    args.add("--manifest", manifest)
+    args.add_joined("--apk", apks_to_deploy, join_with = ",")
+    args.add("--deploy_info", deploy_info)
+
+    ctx.actions.run(
+        executable = deploy_info_writer,
+        arguments = [args],
+        outputs = [deploy_info],
+        mnemonic = "WriteDeployInfo",
+        progress_message = "Writing Deploy info proto file %s" % deploy_info.short_path,
+        toolchain = toolchain_type,
+    )
+
 apk_packaging = struct(
     process = _process,
 )

src/tools/deploy_info/BUILD

@@ -0,0 +1,21 @@
+load("@io_bazel_rules_go//proto:def.bzl", "go_proto_library")
+load("@io_bazel_rules_go//go:def.bzl", "go_binary", "go_library", "go_test")
+
+package(default_applicable_licenses = ["//:license"])
+
+licenses(["notice"])
+
+go_binary(
+    name = "deploy_info",
+    srcs = ["deploy_info.go"],
+    visibility = [
+        "//test/mobile_install:__subpackages__",
+        "//toolchains/android:__subpackages__",
+    ],
+    deps = [
+        "//src/common/golang:flagfile",
+        "//src/common/golang:flags",
+        "//src/tools/deploy_info/proto:android_deploy_info_go_proto",
+        "@org_golang_google_protobuf//proto",
+    ],
+)

src/tools/deploy_info/deploy_info.go

@@ -0,0 +1,64 @@
+// Copyright 2018 The Bazel Authors. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//    http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// Tool to generate the deploy_info.pb that ASwB needs in order to deploy apps.
+package main
+
+import (
+	"flag"
+	"io/ioutil"
+	"log"
+
+	pb "src/tools/deploy_info/proto/android_deploy_info_go_proto"
+	_ "src/common/golang/flagfile"
+	"src/common/golang/flags"
+	"google.golang.org/protobuf/proto"
+)
+
+var apk = flags.NewStringList("apk", "Path to the apk(s).")
+var manifest = flag.String("manifest", "", "Path to the Android manifest.")
+var deployInfo = flag.String("deploy_info", "", "Deploy info pb output path")
+
+func main() {
+	flag.Parse()
+
+	if *apk == nil {
+		log.Fatalf("-apk needs to be specified.")
+	}
+
+	if *manifest == "" {
+		log.Fatalf("-manifest needs to be specified.")
+	}
+
+	if *deployInfo == "" {
+		log.Fatalf("-deploy_info needs to be specified.")
+	}
+
+	manifestArtifact := &pb.Artifact{ExecRootPath: *manifest}
+	splitArtifacts := []*pb.Artifact{}
+	for _, split := range *apk {
+		splitArtifacts = append(splitArtifacts, &pb.Artifact{ExecRootPath: split})
+	}
+
+	info := &pb.AndroidDeployInfo{
+		MergedManifest: manifestArtifact,
+		ApksToDeploy:   splitArtifacts}
+	bits, err := proto.Marshal(info)
+	if err != nil {
+		log.Fatal(err)
+	}
+	if err := ioutil.WriteFile(*deployInfo, bits, 0644); err != nil {
+		log.Fatal(err)
+	}
+}

src/tools/deploy_info/proto/BUILD

@@ -0,0 +1,13 @@
+load("@io_bazel_rules_go//go:def.bzl", "go_binary", "go_library", "go_test")
+load("@io_bazel_rules_go//proto:def.bzl", "go_proto_library")
+
+go_proto_library(
+    name = "android_deploy_info_go_proto",
+    importpath = "src/tools/deploy_info/proto/android_deploy_info_go_proto",
+    protos = [
+        "@bazel_tools//src/main/protobuf:android_deploy_info_proto",
+    ],
+    visibility = [
+        "//src/tools/deploy_info:__pkg__",
+    ],
+)

toolchains/android/toolchain.bzl

@@ -307,6 +307,11 @@ _ATTRS = dict(
         default = "@bazel_tools//tools/android:resource_extractor",
         executable = True,
     ),
+    deploy_info_writer = attr.label(
+      cfg = "exec",
+      default = Label("//src/tools/deploy_info"),
+      executable = True,
+    )
 
 )