feat: Add check for validating symlinks in pfx/dosdevices

Author: shymega

src/wemod.py

@@ -560,6 +560,24 @@ def init(proton: str, iswine: bool = False) -> None:
                 [None],
                 [None],
             )
+            # Ensure dosdevices exists and `drive_c`/`drive_z` are correctly symlinked.
+            dosdevices = os.path.join(WINEPREFIX, "dosdevices")
+
+            # Validate symlinks.
+            if os.path.isdir(dosdevices):
+                from mainutils import create_symlink
+                c_drive = os.path.join(dosdevices, "c:")
+                z_drive = os.path.join(dosdevices, "z:")
+                # Check c_drive is a symlink.
+                if not os.path.islink(c_drive) or os.readlink(c_drive) != '../drive_c':
+                    create_symlink("../drive_c", c_drive)
+                # Check z_drive is a symlink.
+                if not os.path.islink(z_drive) or os.readlink(z_drive) != '/':
+                    create_symlink("/", z_drive)
+
+                assert os.path.islink(c_drive) and os.readlink(c_drive) == '../drive_c'
+                assert os.path.islink(z_drive) and os.readlink(z_drive) == '/'
+
             syncwemod()  # Sync WeMod data
             log(
                 f"Copied Proton version {cut_version[0]}.{cut_version[1]} prefix to game prefix that was on version {current_version_parts[0]}.{current_version_parts[1]}"