From 8e9ae98c281bc755e14d073ee620ebde9f058307 Mon Sep 17 00:00:00 2001
From: Ruairidh MacLeod <ruairidh@rkm.dev>
Date: Mon, 2 Nov 2020 14:01:46 +0000
Subject: [PATCH] fix dotnet build cleanup

---
 pre_commit/languages/dotnet.py | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/pre_commit/languages/dotnet.py b/pre_commit/languages/dotnet.py
index a8abc8611..5d3d897dc 100644
--- a/pre_commit/languages/dotnet.py
+++ b/pre_commit/languages/dotnet.py
@@ -76,9 +76,17 @@ def install_environment(
             ),
         )
 
-        # Cleanup build output
-        for d in ('bin', 'obj', build_dir):
-            rmtree(prefix.path(d))
+        # Remove everything except for the envdir
+        to_keep = (
+            helpers.environment_dir(ENVIRONMENT_DIR, version),
+            '.pre-commit-hooks.yaml',
+        )
+        for f in os.listdir(prefix.prefix_dir):
+            path = prefix.path(f)
+            if os.path.isfile(path):
+                os.remove(path)
+            elif f not in to_keep:
+                rmtree(path)
 
 
 def run_hook(