From 7e6fa744af1710f58828f9cee05bce32075b03e2 Mon Sep 17 00:00:00 2001
From: Leyla Becker <git@jan-leila.com>
Date: Wed, 17 Sep 2025 19:42:15 -0500
Subject: [PATCH] fix: wrapped prostudiomasters in --in-process-gpu flag

---
 modules/common-modules/pkgs/default.nix       |  1 -
 .../common-modules/pkgs/prostudiomasters.nix  | 25 ++++++++++++++++---
 2 files changed, 22 insertions(+), 4 deletions(-)

diff --git a/modules/common-modules/pkgs/default.nix b/modules/common-modules/pkgs/default.nix
index e608350..c97f97c 100644
--- a/modules/common-modules/pkgs/default.nix
+++ b/modules/common-modules/pkgs/default.nix
@@ -10,7 +10,6 @@
         ./webtoon-dl.nix
         {};
     })
-    # TODO: this package always needs to be called with the --in-process-gpu flag for some reason, can we automate that?
     (final: prev: {
       prostudiomasters =
         pkgs.callPackage
diff --git a/modules/common-modules/pkgs/prostudiomasters.nix b/modules/common-modules/pkgs/prostudiomasters.nix
index c1c03fe..1a3ad01 100644
--- a/modules/common-modules/pkgs/prostudiomasters.nix
+++ b/modules/common-modules/pkgs/prostudiomasters.nix
@@ -1,6 +1,7 @@
 {
   fetchurl,
   appimageTools,
+  writeShellScript,
 }: let
   pname = "prostudiomasters";
   version = "2.5.6";
@@ -8,7 +9,25 @@
     url = "https://download.prostudiomasters.com/linux/ProStudioMasters-${version}.AppImage";
     hash = "sha256-7owOwdcucFfl+JsVj+Seau2KOz0J4P/ep7WrBSNSmbs=";
   };
-in
-  appimageTools.wrapType2 {
+
+  # Create the base AppImage wrapper
+  baseApp = appimageTools.wrapType2 {
     inherit pname version src;
-  }
+  };
+
+  # Create a wrapper script that automatically adds the --in-process-gpu flag
+  wrapper = writeShellScript "prostudiomasters-wrapper" ''
+    exec ${baseApp}/bin/prostudiomasters --in-process-gpu "$@"
+  '';
+in
+  # Override the base app to use our wrapper script
+  baseApp.overrideAttrs (oldAttrs: {
+    buildCommand =
+      oldAttrs.buildCommand
+      + ''
+        # Replace the original binary with our wrapper
+        rm $out/bin/prostudiomasters
+        cp ${wrapper} $out/bin/prostudiomasters
+        chmod +x $out/bin/prostudiomasters
+      '';
+  })