From b3ddd7a7825010692053531e760915a9ccffd17d Mon Sep 17 00:00:00 2001
From: Thiago Cruz <thiago.hmcruz@gmail.com>
Date: Wed, 5 Jun 2024 13:50:00 -0400
Subject: [PATCH] Fix import_middleman in Bazel 7 (#873)

Can be considered a follow up to
https://github.com/bazel-ios/rules_ios/pull/850 to fix
`import_middleman` in Bazel 7 while keeping it backwards compatible with
Bazel 6.
---
 .github/workflows/preflight_env.sh |  2 +-
 rules/BUILD.bazel                  |  1 +
 rules/import_middleman.bzl         | 51 +++++++++++++++++++++++++++---
 3 files changed, 49 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/preflight_env.sh b/.github/workflows/preflight_env.sh
index c82847e..f5cfe3e 100755
--- a/.github/workflows/preflight_env.sh
+++ b/.github/workflows/preflight_env.sh
@@ -8,7 +8,7 @@ set -e
 #   - XCODE_VERSION: The version of Xcode to use.
 
 # GitHub runners are hitting 'module not found pkg_resources' required by prepare_sim.py
-pip3 install setuptools --break-system-packages
+pip3 install setuptools==69.5.1 --break-system-packages
 
 # If flag --no-bzlmod is passed, writes a user.bazelrc file to disable Bzlmod.
 if [[ "$*" == *--no-bzlmod* ]]; then
diff --git a/rules/BUILD.bazel b/rules/BUILD.bazel
index 4e73f94..8e48c04 100644
--- a/rules/BUILD.bazel
+++ b/rules/BUILD.bazel
@@ -104,6 +104,7 @@ bzl_library(
     visibility = ["//visibility:public"],
     deps = [
         ":features",
+        "//rules:utils.bzl",
         "//rules/internal:objc_provider_utils",
         "@build_bazel_rules_apple//apple",
     ],
diff --git a/rules/import_middleman.bzl b/rules/import_middleman.bzl
index d0ca171..aeeaa7f 100644
--- a/rules/import_middleman.bzl
+++ b/rules/import_middleman.bzl
@@ -1,6 +1,8 @@
-load("@build_bazel_rules_apple//apple/internal:providers.bzl", "AppleFrameworkImportInfo", "new_appleframeworkimportinfo")
 load("//rules/internal:objc_provider_utils.bzl", "objc_provider_utils")
+load("//rules:utils.bzl", "is_bazel_7")
+load("@bazel_tools//tools/cpp:toolchain_utils.bzl", "find_cpp_toolchain", "use_cpp_toolchain")
 load("@build_bazel_rules_apple//apple/internal:bundling_support.bzl", "bundling_support")
+load("@build_bazel_rules_apple//apple/internal:providers.bzl", "AppleFrameworkImportInfo", "new_appleframeworkimportinfo")
 
 _FindImportsAspectInfo = provider(fields = {
     "imported_library_file": "",
@@ -159,6 +161,15 @@ def _deduplicate_test_deps(test_deps, deps):
     return filtered
 
 def _file_collector_rule_impl(ctx):
+    cc_toolchain = find_cpp_toolchain(ctx)
+    cc_features = cc_common.configure_features(
+        ctx = ctx,
+        cc_toolchain = cc_toolchain,
+        language = "objc",
+        requested_features = ctx.features,
+        unsupported_features = ctx.disabled_features,
+    )
+
     linker_deps = _merge_linked_inputs(ctx.attr.deps)
     test_linker_deps = _merge_linked_inputs(ctx.attr.test_deps)
     input_static_frameworks = _deduplicate_test_deps(test_linker_deps[0], linker_deps[0])
@@ -255,8 +266,31 @@ def _file_collector_rule_impl(ctx):
     )
 
     # Create the CcInfo provider, linking information from this is used in Bazel 7+.
-    dep_cc_infos = [dep[CcInfo] for dep in ctx.attr.deps if CcInfo in dep]
-    cc_info = cc_common.merge_cc_infos(cc_infos = dep_cc_infos)
+    cc_info = None
+    if is_bazel_7:
+        cc_info = CcInfo(
+            linking_context = cc_common.create_linking_context(
+                linker_inputs = depset([
+                    cc_common.create_linker_input(
+                        owner = ctx.label,
+                        user_link_flags = compat_link_opt if len(all_replaced_frameworks) else [],
+                        libraries = depset([
+                            cc_common.create_library_to_link(
+                                actions = ctx.actions,
+                                cc_toolchain = cc_toolchain,
+                                feature_configuration = cc_features,
+                                static_library = static_library,
+                                alwayslink = False,
+                            )
+                            for static_library in replaced_static_framework.replaced.values()
+                        ]),
+                    ),
+                ]),
+            ),
+        )
+    else:
+        dep_cc_infos = [dep[CcInfo] for dep in ctx.attr.deps if CcInfo in dep]
+        cc_info = cc_common.merge_cc_infos(cc_infos = dep_cc_infos)
 
     return [
         DefaultInfo(files = depset(dynamic_framework_dirs + replaced_frameworks)),
@@ -266,11 +300,20 @@ def _file_collector_rule_impl(ctx):
 
 import_middleman = rule(
     implementation = _file_collector_rule_impl,
-    fragments = ["apple"],
+    fragments = ["apple", "cpp"],
+    toolchains = use_cpp_toolchain(),
     attrs = {
         "deps": attr.label_list(aspects = [find_imports]),
         "test_deps": attr.label_list(aspects = [find_imports], allow_empty = True),
         "update_in_place": attr.label(executable = True, default = Label("//tools/m1_utils:update_in_place"), cfg = "exec"),
+        "_cc_toolchain": attr.label(
+            providers = [cc_common.CcToolchainInfo],
+            default = Label("@bazel_tools//tools/cpp:current_cc_toolchain"),
+            doc = """\
+The C++ toolchain from which linking flags and other tools needed by the Swift
+toolchain (such as `clang`) will be retrieved.
+""",
+        ),
     },
     doc = """
 This rule adds the ability to update the Mach-o header on imported