set RUSTUP_HOME when using a non-system rust

asottile · asottile · commit 2700a7d62241 · 2023-02-27T20:49:22.000-05:00
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)
 
