From 4c8305b24a70f63e99e78fa606f0db037b8307f4 Mon Sep 17 00:00:00 2001
From: Ted Xie <tedx@google.com>
Date: Thu, 24 Aug 2023 14:30:35 -0700
Subject: [PATCH] Align android_binary mobile-install adapter more with
 internal impl

Part of #12

PiperOrigin-RevId: 559871080
Change-Id: I3af1904105473db13cff875ef380698e6dcf4c53
---
 mobile_install/adapters/android_binary.bzl | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/mobile_install/adapters/android_binary.bzl b/mobile_install/adapters/android_binary.bzl
index 8ea2c047..76f00ef3 100644
--- a/mobile_install/adapters/android_binary.bzl
+++ b/mobile_install/adapters/android_binary.bzl
@@ -46,6 +46,7 @@ def extract(target, ctx):
     Returns:
       Input for process method
     """
+    extension_registry_class_jar = utils.get_extension_registry_class_jar(target)
     return dict(
         debug_key = utils.only(ctx.rule.files.debug_key, allow_empty = True),
         debug_signing_keys = ctx.rule.files.debug_signing_keys,
@@ -81,6 +82,9 @@ def extract(target, ctx):
                     target[JavaInfo].runtime_output_jars,
                 ) +
                 (
+                    [
+                        extension_registry_class_jar,
+                    ] if extension_registry_class_jar else []
                 ),
                 target[JavaInfo].transitive_compile_time_jars,
             ),