From d9796a9d82f92a0d3af1d5d2e3308420eb89606f Mon Sep 17 00:00:00 2001
From: Mykola Grymalyuk <khronokernel@icloud.com>
Date: Thu, 2 Nov 2023 13:52:02 -0600
Subject: [PATCH] CI: Reset hdiutil on error

---
 .github/workflows/build.yml |   4 +-
 Generate-DMG.command        | 123 ++++++++++++++++++++++++++++++++++++
 2 files changed, 125 insertions(+), 2 deletions(-)
 create mode 100644 Generate-DMG.command

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index fde617c..fb3509c 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -6,7 +6,7 @@ on:
   release:
     types: [published]
 
-concurrency:  
+concurrency:
     cancel-in-progress: true
     group: ${{ github.workflow }}-${{ github.event_name }}-${{ github.ref }}
 
@@ -25,7 +25,7 @@ jobs:
       - name: Tag folder
         run: touch Universal-Binaries/.signed
       - name: Build DMG
-        run: ./build-dmg.sh
+        run: python3 Generate-DMG.command
       - name: Generate Checksums
         run: shasum -a 256 Universal-Binaries.dmg > sha256sum.txt
       - name: Upload Universal Binaries to Artifacts
diff --git a/Generate-DMG.command b/Generate-DMG.command
new file mode 100644
index 0000000..1f91ec5
--- /dev/null
+++ b/Generate-DMG.command
@@ -0,0 +1,123 @@
+"""
+Build and sign PatcherSupportPkg Disk Image.
+Note password encryption required to pass Apple's notarization.
+"""
+
+import os
+import subprocess
+
+UB_DIRECTORY:     str = "Universal-Binaries"
+DMG_NAME:         str = "Universal-Binaries.dmg"
+DMG_VOLNAME:      str = "OpenCore Patcher Resources (Root Patching)"
+DMG_SIZE:         str = "4096"
+DMG_FORMAT:       str = "UDRO"
+DMG_PASSPHRASE:   str = "password"
+SIGNING_IDENTITY: str = "OpenCore Legacy Patcher Software Signing"
+
+
+class GenerateDiskImage:
+
+    def __init__(self) -> None:
+        print("Generating DMG")
+        self._set_working_directory()
+        self._strip_extended_attributes()
+        self._remove_ds_store()
+        self._create_dmg()
+        self._convert_dmg()
+        self._sign_dmg()
+        self._remove_tmp_dmg()
+
+
+    def _set_working_directory(self) -> None:
+        print("  - Setting working directory")
+        os.chdir(os.path.dirname(os.path.realpath(__file__)))
+
+
+    def _reset_hdiutil(self) -> None:
+        """
+        Attempt to reset hdiutil
+        On some instances, "hdiutil: create failed - Resource busy" is thrown
+        """
+        print("  - Resetting hdiutil")
+        subprocess.run(["killall", "hdiutil"], capture_output=True)
+
+
+    def _strip_extended_attributes(self) -> None:
+        print("  - Stripping extended attributes")
+        subprocess.run(["/usr/bin/xattr", "-rc", UB_DIRECTORY], capture_output=True)
+
+
+    def _remove_ds_store(self) -> None:
+        print("  - Removing .DS_Store files")
+        subprocess.run(["find", UB_DIRECTORY, "-name", ".DS_Store", "-delete"], capture_output=True)
+
+
+    def _create_dmg(self, raise_on_error: bool = False) -> None:
+        """
+        raise_on_error: If false, attempts to reset hdiutil and try again
+        """
+        print("  - Creating DMG")
+        result = subprocess.run([
+            "hdiutil", "create",
+            "-srcfolder", UB_DIRECTORY, "tmp.dmg",
+            "-volname", DMG_VOLNAME,
+            "-fs", "HFS+", "-ov",
+            "-format", DMG_FORMAT,
+            "-megabytes", DMG_SIZE
+        ], capture_output=True)
+        if result.returncode != 0:
+            print("    - Failed to create DMG")
+            print(f"STDOUT:\n{result.stdout.decode('utf-8')}")
+            print(f"STDERR:\n{result.stderr.decode('utf-8')}")
+            if raise_on_error:
+                raise Exception("Failed to create DMG")
+            if "Resource busy" in result.stderr.decode("utf-8"):
+                self._reset_hdiutil()
+                self._create_dmg(raise_on_error=True)
+            else:
+                raise Exception("Failed to create DMG")
+
+
+    def _convert_dmg(self, raise_on_error: bool = False) -> None:
+        """
+        raise_on_error: If false, attempts to reset hdiutil and try again
+        """
+        print("  - Converting DMG")
+        result = subprocess.run([
+            "hdiutil", "convert",
+            "-format", "ULMO", "tmp.dmg",
+            "-o", DMG_NAME,
+            "-passphrase", DMG_PASSPHRASE,
+            "-encryption", "-ov"
+        ], capture_output=True)
+        if result.returncode != 0:
+            print("    - Failed to convert DMG")
+            print(f"STDOUT:\n{result.stdout.decode('utf-8')}")
+            print(f"STDERR:\n{result.stderr.decode('utf-8')}")
+            if raise_on_error:
+                raise Exception(f"Failed to convert DMG")
+            if "Resource busy" in result.stderr.decode("utf-8"):
+                self._reset_hdiutil()
+                self._convert_dmg(raise_on_error=True)
+            else:
+                raise Exception("Failed to convert DMG")
+
+
+    def _sign_dmg(self) -> None:
+        print("  - Signing DMG")
+        result = subprocess.run([
+            "codesign", "-s", SIGNING_IDENTITY, DMG_NAME
+        ], capture_output=True)
+        if result.returncode != 0:
+            print("    - Failed to sign DMG")
+            print(f"STDOUT:\n{result.stdout.decode('utf-8')}")
+            print(f"STDERR:\n{result.stderr.decode('utf-8')}")
+            raise Exception("Failed to sign DMG")
+
+
+    def _remove_tmp_dmg(self) -> None:
+        subprocess.run(["rm", "tmp.dmg"], capture_output=True)
+
+
+if __name__ == "__main__":
+    GenerateDiskImage()
\ No newline at end of file