Allow looking for multiple executable names

This commit is contained in:
PapaTutuWawa 2023-12-23 14:04:18 +01:00
parent b9fc712534
commit 94582f1547
5 changed files with 26 additions and 15 deletions

View File

@ -134,8 +134,12 @@ class BaldursGate3Game(ProtonGame):
return overlays
@property
def game_executable(self) -> Path:
return self.installation_path / "bin" / "bg3_dx11.exe"
def game_executables(self) -> list[Path]:
# Handle both the Dx11 version and the Vulkan version.
return [
self.installation_path / "bin" / "bg3_dx11.exe",
self.installation_path / "bin" / "bg3.exe",
]
def to_dict(self) -> dict[str, Any]:
return {

View File

@ -91,7 +91,7 @@ class ProtonGame(Game, abc.ABC):
raise NotImplementedError()
@property
def game_executable(self) -> Path:
def game_executables(self) -> list[Path]:
raise NotImplementedError()
@property
@ -101,9 +101,10 @@ class ProtonGame(Game, abc.ABC):
def wait(self):
# Wait until we started the game.
proc = Path("/proc")
windows_path = str(self.game_executable).replace("/", "\\")
wine_exe_path = f"Z:{windows_path}"
print(wine_exe_path)
windows_paths = [str(executable).replace("/", "\\") for executable in self.game_executables]
wine_exe_paths = [f"Z:{windows_path}" for windows_path in windows_paths]
executables = [str(exe) for exe in self.game_executables]
print(wine_exe_paths)
print("Polling for game process...")
while True:
for dir in os.listdir(proc):
@ -127,9 +128,9 @@ class ProtonGame(Game, abc.ABC):
abs_cmdline = os.path.abspath(cmdline).replace("\x00", "")
if (
cmdline.startswith(wine_exe_path)
or abs_cmdline == str(self.game_executable)
or abs_cmdline.startswith(str(self.game_executable))
any(cmdline.startswith(wine_exe_path) for wine_exe_path in wine_exe_paths)
or abs_cmdline in executables
or any(abs_cmdline.startswith(exe) for exe in executables)
):
self._pid = dir
break

View File

@ -35,8 +35,10 @@ class LethalCompanyGame(ProtonGame):
)
@property
def game_executable(self) -> Path:
return self.installation_path / "Lethal Company.exe"
def game_executables(self) -> list[Path]:
return [
self.installation_path / "Lethal Company.exe",
]
def prepare_overlays(
self, profile: LethalCompanyProfile

View File

@ -35,8 +35,10 @@ class ProjectDivaMegaMixGame(ProtonGame):
)
@property
def game_executable(self) -> Path:
return self.installation_path / "DivaMegaMix.exe"
def game_executables(self) -> list[Path]:
return [
self.installation_path / "DivaMegaMix.exe",
]
def prepare_overlays(
self, profile: ProjectDivaMegaMixProfile

View File

@ -46,8 +46,10 @@ class ReadyOrNotGame(ProtonGame):
]
@property
def game_executable(self) -> Path:
return self.installation_path / "ReadyOrNot.exe"
def game_executables(self) -> list[Path]:
return [
self.installation_path / "ReadyOrNot.exe",
]
def to_dict(self) -> dict[str, Any]:
return {