Add TemplateVariableInfo for build script toolchains attribute. (baze…

…lbuild#2123)
wayveai · Aug 25, 2023 · 1320071 · 1320071
1 parent 7f40480
commit 1320071
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cargo/private/cargo_build_script.bzl b/cargo/private/cargo_build_script.bzl
@@ -188,6 +188,9 @@ def _cargo_build_script_impl(ctx):
             if type(all_files) == "list":
                 all_files = depset(all_files)
             toolchain_tools.append(all_files)
+        if platform_common.TemplateVariableInfo in target:
+            variables = getattr(target[platform_common.TemplateVariableInfo], "variables", depset([]))
+            env.update(variables)
 
     _merge_env_dict(env, expand_dict_value_locations(
         ctx,