Extract setter store from AAR in aar_import

Author: pswaminathan

examples/databinding/MODULE.bazel

@@ -33,6 +33,7 @@ maven.install(
     aar_import_bzl_label = "@rules_android//rules:rules.bzl",
     artifacts = [
         "androidx.annotation:annotation-jvm:1.9.1",
+        "androidx.databinding:databinding-adapters:8.7.0",
         "androidx.databinding:databinding-runtime:8.7.0",
     ],
     repositories = [

examples/databinding/java/com/databindingapp/BUILD

@@ -18,6 +18,7 @@ android_library(
     enable_data_binding = True,
     deps = [
         "@maven//:androidx_annotation_annotation_jvm",
+        "@maven//:androidx_databinding_databinding_adapters",
         "@maven//:androidx_databinding_databinding_runtime",
     ],
 )

rules/aar_import/impl.bzl

@@ -95,17 +95,19 @@ def _extract_resources(
         ctx,
         out_resources_dir,
         out_assets_dir,
+        out_databinding_setter_store_dir,
         aar,
         aar_resources_extractor_tool):
     args = ctx.actions.args()
     args.add("--input_aar", aar)
     args.add("--output_res_dir", out_resources_dir.path)
     args.add("--output_assets_dir", out_assets_dir.path)
+    args.add("--output_databinding_setter_store_dir", out_databinding_setter_store_dir.path)
     ctx.actions.run(
         executable = aar_resources_extractor_tool,
         arguments = [args],
         inputs = [aar],
-        outputs = [out_resources_dir, out_assets_dir],
+        outputs = [out_resources_dir, out_assets_dir, out_databinding_setter_store_dir],
         mnemonic = "AarResourcesExtractor",
         progress_message = "Extracting resources and assets from %s" % aar.basename,
         toolchain = None,
@@ -142,10 +144,12 @@ def _process_resources(
     # Extract resources and assets, if they exist.
     resources = _create_aar_tree_artifact(ctx, "resources")
     assets = _create_aar_tree_artifact(ctx, "assets")
+    data_binding_setter_store = _create_aar_tree_artifact(ctx, "data_binding_setter_store")
     _extract_resources(
         ctx,
         resources,
         assets,
+        data_binding_setter_store,
         aar,
         aar_resources_extractor_tool,
     )
@@ -156,6 +160,7 @@ def _process_resources(
         assets = [assets],
         assets_dir = assets.path,
         resource_files = [resources],
+        data_binding_setter_store = data_binding_setter_store,
         stamp_manifest = False,
         deps = ctx.attr.deps,
         exports = ctx.attr.exports,

rules/data_binding.bzl

@@ -328,6 +328,19 @@ def _process(
 
     return DataBindingContextInfo(**db_info)
 
+def _process_aar(
+    ctx,
+    data_binding_setter_store):
+    """Builds a data binding provider for an AAR's extracted setter store
+    """
+    return DataBindingV2Info(
+        setter_stores = depset(direct = [data_binding_setter_store]),
+        class_infos = depset(),
+        transitive_br_files = depset(),
+        java_packages = [],
+    )
+
 data_binding = struct(
     process = _process,
+    process_aar = _process_aar,
 )

rules/resources.bzl

@@ -15,6 +15,7 @@
 
 load("//providers:providers.bzl", "AndroidLibraryResourceClassJarProvider", "ResourcesNodeInfo", "StarlarkAndroidResourcesInfo")
 load("//rules:acls.bzl", "acls")
+load("//rules:data_binding.bzl", _data_binding = "data_binding")
 load("//rules:min_sdk_version.bzl", _min_sdk_version = "min_sdk_version")
 load("//rules:visibility.bzl", "PROJECT_VISIBILITY")
 load("@rules_java//java/common:java_common.bzl", "java_common")
@@ -1190,6 +1191,7 @@ def _process_starlark(
         resource_files = None,
         neverlink = False,
         enable_data_binding = False,
+        data_binding_setter_store = None,
         fix_resource_transitivity = False,
         aapt = None,
         android_jar = None,
@@ -1369,6 +1371,12 @@ def _process_starlark(
     data_binding_layout_info = None
     processed_resources = resource_files
     processed_manifest = None
+
+    if data_binding_setter_store != None:
+        resources_ctx[_PROVIDERS].append(
+            _data_binding.process_aar(ctx, data_binding_setter_store)
+        )
+
     if not defines_resources:
         if aapt:
             # Generate an empty manifest with the right package