diff --git a/Resources/Constants.py b/Resources/Constants.py index b4cc199eb..72e9d575f 100644 --- a/Resources/Constants.py +++ b/Resources/Constants.py @@ -186,6 +186,8 @@ class Constants: @property def icon_path_sd(self): return self.payload_path / Path("Icon/SD-Card/.VolumeIcon.icns") @property + def icon_path_ssd(self): return self.payload_path / Path("Icon/SSD/.VolumeIcon.icns") + @property def gui_path(self): return self.payload_path / Path("Icon/Resources.zip") # Apple Paylods Paths diff --git a/Resources/build.py b/Resources/build.py index 0c3e9d481..5a5cfb7ed 100644 --- a/Resources/build.py +++ b/Resources/build.py @@ -591,6 +591,10 @@ Please build OpenCore first!""" drive_host_info = plistlib.loads(subprocess.run(f"diskutil info -plist {disk_identifier}".split(), stdout=subprocess.PIPE).stdout.decode().strip().encode()) partition_info = plistlib.loads(subprocess.run(f"diskutil info -plist {disk_identifier}s{response}".split(), stdout=subprocess.PIPE).stdout.decode().strip().encode()) sd_type = drive_host_info["MediaName"] + try: + ssd_type = drive_host_info["SolidState"] + except KeyError: + ssd_type = False mount_path = Path(partition_info["MountPoint"]) disk_type = partition_info["BusProtocol"] utilities.cls() @@ -611,13 +615,15 @@ Please build OpenCore first!""" if sd_type in ["SD Card Reader", "SD/MMC"]: print("- Adding SD Card icon") shutil.copy(self.constants.icon_path_sd, mount_path) + elif ssd_type is True: + print("- Adding SSD icon") + shutil.copy(self.constants.icon_path_ssd, mount_path) + elif disk_type == "USB": + print("- Adding External USB Drive icon") + shutil.copy(self.constants.icon_path_external, mount_path) else: - if disk_type == "USB": - print("- Adding External USB Drive icon") - shutil.copy(self.constants.icon_path_external, mount_path) - else: - print("- Adding Internal Drive icon") - shutil.copy(self.constants.icon_path_internal, mount_path) + print("- Adding Internal Drive icon") + shutil.copy(self.constants.icon_path_internal, mount_path) print("- Cleaning install location") subprocess.run(f"dot_clean '{mount_path}'".split(), stdout=subprocess.PIPE, stderr=subprocess.PIPE) print("- OpenCore transfer complete") diff --git a/images/OC-Picker-SSD.png b/images/OC-Picker-SSD.png new file mode 100644 index 000000000..084d3e054 Binary files /dev/null and b/images/OC-Picker-SSD.png differ diff --git a/payloads/Icon/SSD/.VolumeIcon.icns b/payloads/Icon/SSD/.VolumeIcon.icns new file mode 100644 index 000000000..e942a9471 Binary files /dev/null and b/payloads/Icon/SSD/.VolumeIcon.icns differ