[codex] Split build-script feature resolution

rs/BUILD.bazel

@@ -44,6 +44,7 @@ bzl_library(
     srcs = ["extensions.bzl"],
     visibility = ["//visibility:public"],
     deps = [
+        "//rs/platforms:triples",
         "//rs/private:annotations",
         "//rs/private:cargo_credentials",
         "//rs/private:cargo_workspace_graph",

rs/extensions.bzl

@@ -2,6 +2,7 @@ load("@aspect_tools_telemetry_report//:defs.bzl", "TELEMETRY")  # buildifier: di
 load("@bazel_lib//lib:repo_utils.bzl", "repo_utils")
 load("@bazel_skylib//lib:paths.bzl", "paths")
 load("@rs_rust_host_tools//:defs.bzl", "RS_HOST_CARGO_LABEL")
+load("//rs/platforms:triples.bzl", "SUPPORTED_EXEC_TRIPLES")
 load("//rs/private:annotations.bzl", "annotation_for", "build_annotation_map", "well_known_annotation_snippet_paths")
 load("//rs/private:cargo_credentials.bzl", "load_cargo_credentials")
 load(
@@ -267,6 +268,7 @@ def _generate_hub_and_spokes(
         annotations = annotations,
         platform_triples = platform_triples,
         materialize_workspace_members = False,
+        exec_platform_triples = SUPPORTED_EXEC_TRIPLES,
         validate_lockfile = validate_lockfile,
         debug = debug,
         dep_label_prefix = "@%s//:" % hub_name,
@@ -277,6 +279,7 @@ def _generate_hub_and_spokes(
     platform_cfg_attrs = workspace_resolution.platform_cfg_attrs
     workspace_dep_labels_by_triple = workspace_resolution.workspace_dep_labels_by_triple
     workspace_dep_versions_by_name = workspace_resolution.workspace_dep_versions_by_name
+    exec_feature_resolutions_by_fq_crate = workspace_resolution.exec_feature_resolutions_by_fq_crate
 
     _date(mctx, "set up initial deps!")
 
@@ -290,6 +293,7 @@ def _generate_hub_and_spokes(
         source = package["source"]
 
         feature_resolutions = feature_resolutions_by_fq_crate[_fq_crate(crate_name, version)]
+        exec_feature_resolutions = exec_feature_resolutions_by_fq_crate[_fq_crate(crate_name, version)]
 
         annotation = annotation_for(annotations, crate_name, version, hub_name)
         suggested_annotation = None
@@ -323,6 +327,7 @@ crate.annotation(
             gen_build_script = annotation.gen_build_script,
             build_script_deps = [],
             build_script_deps_select = _select(feature_resolutions.build_deps),
+            exec_build_script_deps_select = _select(exec_feature_resolutions.build_deps),
             build_script_data = annotation.build_script_data,
             build_script_data_select = annotation.build_script_data_select,
             build_script_env = annotation.build_script_env,
@@ -338,9 +343,14 @@ crate.annotation(
             deps = annotation.deps,
             crate_tags = annotation.tags,
             deps_select = _select(feature_resolutions.deps),
+            exec_deps_select = _select(exec_feature_resolutions.deps),
             aliases = feature_resolutions.aliases,
+            exec_aliases = exec_feature_resolutions.aliases,
             crate_features = annotation.crate_features,
             crate_features_select = _select(feature_resolutions.features_enabled),
+            exec_crate_features_select = _select(exec_feature_resolutions.features_enabled),
+            target_active = bool(feature_resolutions.active),
+            exec_active = bool(exec_feature_resolutions.active),
             use_legacy_rules_rust_platforms = use_legacy_rules_rust_platforms,
         )
 

rs/private/cargo_workspace_graph.bzl

@@ -1,6 +1,6 @@
 load("@bazel_skylib//lib:paths.bzl", "paths")
 load("//rs/private:cfg_parser.bzl", "cfg_matches_expr_for_cfg_attrs", "triple_to_cfg_attrs")
-load("//rs/private:resolver.bzl", "resolve")
+load("//rs/private:resolver.bzl", "resolve", "seed_exec_build_dependencies")
 load("//rs/private:select_utils.bzl", "compute_select")
 load("//rs/private:semver.bzl", "select_matching_version")
 
@@ -64,6 +64,7 @@ def cfg_match_info_for_target(target, platform_cfg_attrs, cfg_match_cache):
 
 def new_feature_resolutions(package_index, possible_deps, possible_features, platform_triples):
     return struct(
+        active = set(),
         features_enabled = {triple: set() for triple in platform_triples},
         build_deps = {triple: set() for triple in platform_triples},
         deps = {triple: set() for triple in platform_triples},
@@ -472,6 +473,11 @@ def _resolve_possible_deps(
             else:
                 dep["target"] = set(match_info.matches)
 
+def _apply_annotation_features(feature_resolutions, annotation):
+    for triple, features in feature_resolutions.features_enabled.items():
+        features.update(annotation.crate_features)
+        features.update(annotation.crate_features_select.get(triple, []))
+
 def resolve_cargo_workspace_members(
         ctx,
         *,
@@ -483,19 +489,20 @@ def resolve_cargo_workspace_members(
         annotations,
         platform_triples,
         materialize_workspace_members,
+        exec_platform_triples = [],
         validate_lockfile = True,
         debug = False,
         dep_label_prefix = "//:",
         skip_internal_rustc_placeholder_crates = True,
         watch_manifests = False,
         use_legacy_rules_rust_platforms = False):
     platform_cfg_attrs = [triple_to_cfg_attrs(triple) for triple in platform_triples]
-    platform_cfg_attrs_by_triple = {}
-    for cfg_attr in platform_cfg_attrs:
-        platform_cfg_attrs_by_triple[cfg_attr["_triple"]] = cfg_attr
+    platform_cfg_attrs_by_triple = {cfg_attr["_triple"]: cfg_attr for cfg_attr in platform_cfg_attrs}
 
     cfg_match_cache = {None: struct(matches = platform_triples, uses_feature_cfg = False)}
 
+    exec_platform_cfg_attrs_by_triple = {triple: triple_to_cfg_attrs(triple) for triple in exec_platform_triples}
+
     workspace_member_keys = {}
     for package in cargo_metadata["packages"]:
         workspace_member_keys[(package["name"], package["version"])] = True
@@ -535,6 +542,21 @@ def resolve_cargo_workspace_members(
 
         resolver_packages.append(resolver_package)
 
+    exec_feature_resolutions_by_fq_crate = {}
+    exec_resolver_packages = []
+    if exec_platform_triples:
+        for package in resolver_packages:
+            target_resolution = package["feature_resolutions"]
+            target_resolution.build_deps.update({triple: set() for triple in exec_platform_triples})
+            exec_resolution = new_feature_resolutions(
+                target_resolution.package_index,
+                [dict(dep) for dep in target_resolution.possible_deps],
+                target_resolution.possible_features,
+                exec_platform_triples,
+            )
+            exec_feature_resolutions_by_fq_crate[fq_crate(package["name"], package["version"])] = exec_resolution
+            exec_resolver_packages.append(dict(package, feature_resolutions = exec_resolution))
+
     _resolve_possible_deps(
         resolver_packages,
         resolver_versions_by_name,
@@ -545,6 +567,17 @@ def resolve_cargo_workspace_members(
         dep_label_prefix,
     )
 
+    if exec_platform_triples:
+        _resolve_possible_deps(
+            exec_resolver_packages,
+            resolver_versions_by_name,
+            exec_feature_resolutions_by_fq_crate,
+            exec_platform_triples,
+            exec_platform_cfg_attrs_by_triple.values(),
+            {None: struct(matches = exec_platform_triples, uses_feature_cfg = False)},
+            dep_label_prefix,
+        )
+
     workspace_fq_deps = compute_workspace_fq_deps(workspace_members, resolver_versions_by_name)
     workspace_dep_versions_by_name = {}
     workspace_dep_labels_by_triple = {triple: set() for triple in platform_triples}
@@ -554,6 +587,7 @@ def resolve_cargo_workspace_members(
             ctx.watch(package["manifest_path"])
 
         package_feature_resolutions = feature_resolutions_by_fq_crate[fq_crate(package["name"], package["version"])]
+        package_feature_resolutions.active.update(platform_triples)
         if "default" in package.get("features", {}):
             for triple in platform_triples:
                 package_feature_resolutions.features_enabled[triple].add("default")
@@ -611,8 +645,11 @@ def resolve_cargo_workspace_members(
             match_info = cfg_match_info_for_target(target, platform_cfg_attrs, cfg_match_cache)
 
             for triple in match_info.matches:
+                if dep.get("kind", "normal") == "build":
+                    continue
                 if not is_first_party_dep or materialize_workspace_members:
                     workspace_dep_labels_by_triple[triple].add(":" + dep_name)
+                feature_resolutions.active.add(triple)
                 feature_resolutions.features_enabled[triple].update(features)
 
     for crate, annotation_versions in annotations.items():
@@ -625,15 +662,17 @@ def resolve_cargo_workspace_members(
             if not annotation.crate_features and not annotation.crate_features_select:
                 continue
             for version in target_versions:
-                features_enabled = feature_resolutions_by_fq_crate[fq_crate(crate, version)].features_enabled
-                if annotation.crate_features:
-                    for triple in platform_triples:
-                        features_enabled[triple].update(annotation.crate_features)
-                for triple, features in annotation.crate_features_select.items():
-                    if triple in features_enabled:
-                        features_enabled[triple].update(features)
+                fq = fq_crate(crate, version)
+                _apply_annotation_features(feature_resolutions_by_fq_crate[fq], annotation)
+
+                if exec_platform_triples:
+                    _apply_annotation_features(exec_feature_resolutions_by_fq_crate[fq], annotation)
+
+    resolve(ctx, resolver_packages, feature_resolutions_by_fq_crate, platform_cfg_attrs_by_triple, debug, include_build_dependencies = not exec_platform_triples)
 
-    resolve(ctx, resolver_packages, feature_resolutions_by_fq_crate, platform_cfg_attrs_by_triple, debug)
+    if exec_platform_triples:
+        seed_exec_build_dependencies(resolver_packages, exec_resolver_packages, exec_platform_cfg_attrs_by_triple)
+        resolve(ctx, exec_resolver_packages, exec_feature_resolutions_by_fq_crate, exec_platform_cfg_attrs_by_triple, debug)
 
     for package in packages:
         feature_resolutions = package["feature_resolutions"]
@@ -651,6 +690,7 @@ def resolve_cargo_workspace_members(
 
     return struct(
         cfg_match_cache = cfg_match_cache,
+        exec_feature_resolutions_by_fq_crate = exec_feature_resolutions_by_fq_crate,
         feature_resolutions_by_fq_crate = feature_resolutions_by_fq_crate,
         platform_cfg_attrs = platform_cfg_attrs,
         platform_cfg_attrs_by_triple = platform_cfg_attrs_by_triple,