protonfixes: move more protonfixes related functions out of the proton script

Author: loathingKernel

__init__.py

@@ -1,6 +1,7 @@
 """Starts the protonfix module and runs fixes after pre-flight-checks"""
 
 import os
+import subprocess
 import sys
 import traceback
 from typing import Callable
@@ -18,6 +19,10 @@
 x86_64_lib_dir: str = f'{os.path.dirname(os.path.realpath(__file__))}/files/lib/x86_64-linux-gnu'
 
 
+# This is needed for protonfixes
+os.environ['PROTON_DLL_COPY'] = '*'
+
+
 def check_conditions() -> bool:
     """Determine, if the actual game was executed and protonfixes isn't deactivated.
 
@@ -49,6 +54,43 @@ def setup(env: dict, bin_path_var: str, lib_path_var: str, func: Callable[[dict,
     func(env, lib_path_var, f'{x86_64_lib_dir}:{i386_lib_dir}', ':')
 
 
+def winetricks(env: dict, wine_bin, wineserver_bin) -> None:
+    if (env.get("UMU_ID")
+        and env.get("EXE", "").endswith("winetricks")
+        and env.get("PROTON_VERB") == "waitforexitandrun"
+    ):
+        wt_verbs = " ".join(sys.argv[2:][2:])
+        env["WINE"] = wine_bin
+        env["WINELOADER"] = wine_bin
+        env["WINESERVER"] = wineserver_bin
+        env["WINETRICKS_LATEST_VERSION_CHECK"] = "disabled"
+        env["LD_PRELOAD"] = ""
+
+        log(f"Running winetricks verbs in prefix: {wt_verbs}")
+        rc = subprocess.run(sys.argv[2:], check=False, env=env).returncode
+
+        sys.exit(rc)
+
+
+def _is_directory_empty(dir_path):
+    # Check if the directory is empty
+    return not any(os.scandir(dir_path))
+
+
+def setup_mount_drives(func: Callable[[str, str, str], None]):
+    if os.environ.get('UMU_ID', ''):
+        drive_map = {
+            '/media': 'u:',
+            '/run/media': 'v:',
+            '/mnt': 'w:',
+            os.path.expanduser('~'): 'x:',  # Current user's home directory
+        }
+
+        for directory in drive_map.keys():
+            if os.path.exists(directory) and not _is_directory_empty(directory):
+                func('gamedrive', drive_map[directory], directory)
+
+
 def execute() -> None:
     """Execute protonfixes"""
     if check_iscriptevaluator():
@@ -65,4 +107,4 @@ def execute() -> None:
             sys.stderr.flush()
 
 
-__all__ = ["setup", "execute"]
+__all__ = ["setup", "execute", "winetricks", "setup_mount_drives"]