set RUSTUP_HOME when using a non-system rust

pre-commit · Feb 28, 2023 · 2700a7d · 2700a7d
1 parent 294590f
commit 2700a7d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pre_commit/languages/rust.py b/pre_commit/languages/rust.py
@@ -142,10 +142,15 @@ def install_environment(
         else:
             packages_to_install.add((package,))
 
-    with in_env(prefix, version):
+    with contextlib.ExitStack() as ctx:
+        ctx.enter_context(in_env(prefix, version))
+
         if version != 'system':
             install_rust_with_toolchain(_rust_toolchain(version))
 
+            tmpdir = ctx.enter_context(tempfile.TemporaryDirectory())
+            ctx.enter_context(envcontext((('RUSTUP_HOME', tmpdir),)))
+
         if len(lib_deps) > 0:
             _add_dependencies(prefix, lib_deps)