kdk_handler: Delete unused KDKs after supported is found

resources/constants.py

@@ -114,6 +114,7 @@ class Constants:
         self.wxpython_variant = False #     Determine if using wxPython variant
         self.unpack_thread = None  #        Determine if unpack thread finished
         self.cli_mode = False  #            Determine if running in CLI mode
+        self.should_nuke_kdks = True  #       Determine if KDKs should be nuked if unused in /L*/D*/KDKs
 
         ## Hardware
         self.computer: device_probe.Computer = None  # type: ignore

resources/defaults.py

@@ -69,6 +69,10 @@ class generate_defaults:
         if global_settings.global_settings().read_property("Force_Web_Drivers") is True:
             self.constants.force_nv_web = True
 
+        result = global_settings.global_settings().read_property("ShouldNukeKDKs")
+        if result is False:
+            self.constants.should_nuke_kdks = False
+
     def smbios_probe(self):
         if not self.host_is_target:
             if self.model in ["MacPro4,1", "MacPro5,1"]:

resources/kdk_handler.py

@@ -9,6 +9,8 @@ from typing import cast
 import packaging.version
 import requests
 
+import subprocess
+
 from resources import utilities
 from resources.constants import Constants
 
@@ -111,8 +113,9 @@ class kernel_debug_kit_handler:
     def download_kdk(self, version: str, build: str):
         detected_build = build
 
-        if self.is_kdk_installed(build) is True:
+        if self.is_kdk_installed(detected_build) is True:
             print("- KDK is already installed")
+            self.remove_unused_kdks(detected_build)
             return True, "", detected_build
 
         download_link = None
@@ -148,6 +151,7 @@ class kernel_debug_kit_handler:
             print("- Could not find KDK, finding closest match")
 
             if self.is_kdk_installed(closest_build) is True:
+                self.remove_unused_kdks(closest_build)
                 return True, "", closest_build
 
             if closest_match_download_link is None:
@@ -168,6 +172,7 @@ class kernel_debug_kit_handler:
             return False, "Could not contact Apple download servers", ""
 
         if utilities.download_apple_developer_portal(download_link, self.constants.kdk_download_path):
+            self.remove_unused_kdks(detected_build)
             return True, "", detected_build
         return False, "Failed to download KDK", ""
 
@@ -178,3 +183,18 @@ class kernel_debug_kit_handler:
                     if file.name.endswith(f"{build}.kdk"):
                         return True
         return False
+
+    def remove_unused_kdks(self, exclude_build):
+        if self.constants.should_nuke_kdks is False:
+            return
+
+        if not Path("/Library/Developer/KDKs").exists():
+            return
+
+        print("- Removing unused KDKs")
+        for kdk_folder in Path("/Library/Developer/KDKs").iterdir():
+            if kdk_folder.is_dir():
+                if kdk_folder.name.endswith(".kdk"):
+                    if not kdk_folder.name.endswith(f"{exclude_build}.kdk"):
+                        print(f"  - Removing {kdk_folder.name}")
+                        utilities.elevated(["sudo", "rm", "-rf", kdk_folder], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)