Extract baseline-prof.txt from AARs in aar_import

aar_import now extracts baseline-prof.txt and provides
BaselineProfileProvider, so baseline profiles from maven AARs
propagate through the dependency graph to android_binary.

Author: snazhmudinov

rules/aar_import/impl.bzl

@@ -13,7 +13,7 @@
 # limitations under the License.
 """Implementation."""
 
-load("//providers:providers.bzl", "AndroidLintRulesInfo", "AndroidNativeLibsInfo")
+load("@rules_java//java/common:java_common.bzl", "java_common")
 load(
     "//rules:acls.bzl",
     _acls = "acls",
@@ -42,6 +42,7 @@ load("//rules:visibility.bzl", "PROJECT_VISIBILITY")
 load("@rules_java//java/common:java_common.bzl", "java_common")
 load("@rules_java//java/common:java_info.bzl", "JavaInfo")
 load("@rules_java//java/common:proguard_spec_info.bzl", "ProguardSpecInfo")
+load("//providers:providers.bzl", "AndroidLintRulesInfo", "AndroidNativeLibsInfo", "BaselineProfileProvider")
 
 visibility(PROJECT_VISIBILITY)
 
@@ -575,6 +576,18 @@ def impl(ctx):
     )
     providers.extend(lint_providers)
 
+    # Extract baseline-prof.txt from the AAR if present, otherwise create an empty file.
+    baseline_prof = create_aar_artifact(ctx, "baseline-prof.txt")
+    extract_single_file(
+        ctx,
+        baseline_prof,
+        aar,
+        "baseline-prof.txt",
+        unzip_tool,
+        create_empty_file = True,
+    )
+    providers.append(BaselineProfileProvider(files = depset([baseline_prof])))
+
     validation_outputs.append(_validate_rule(
         ctx,
         aar = aar,

rules/aar_import/rule.bzl

@@ -13,7 +13,7 @@
 # limitations under the License.
 """aar_import rule."""
 
-load("//providers:providers.bzl", "AndroidIdeInfo", "AndroidLibraryResourceClassJarProvider", "AndroidNativeLibsInfo")
+load("//providers:providers.bzl", "AndroidIdeInfo", "AndroidLibraryResourceClassJarProvider", "AndroidNativeLibsInfo", "BaselineProfileProvider")
 load(
     "//rules:utils.bzl",
     "ANDROID_SDK_TOOLCHAIN_TYPE",
@@ -60,6 +60,7 @@ aar_import = rule(
         AndroidIdeInfo,
         AndroidLibraryResourceClassJarProvider,
         AndroidNativeLibsInfo,
+        BaselineProfileProvider,
         JavaInfo,
     ],
     toolchains = [