1058 lines
46 KiB
Python
1058 lines
46 KiB
Python
import os
|
|
import shlex
|
|
import glob
|
|
import shutil
|
|
import subprocess
|
|
import threading
|
|
import logging
|
|
import re
|
|
import orjson
|
|
import vdf
|
|
from PySide6.QtWidgets import QMessageBox, QDialog, QMenu, QFileDialog
|
|
from PySide6.QtCore import QUrl, QPoint, QObject, Signal, Qt
|
|
from PySide6.QtGui import QDesktopServices
|
|
from portprotonqt.localization import _
|
|
from portprotonqt.config_utils import parse_desktop_entry, read_favorites, save_favorites
|
|
from portprotonqt.steam_api import is_game_in_steam, add_to_steam, remove_from_steam, get_steam_home, get_last_steam_user, convert_steam_id
|
|
from portprotonqt.egs_api import add_egs_to_steam, get_egs_executable
|
|
from portprotonqt.dialogs import AddGameDialog, generate_thumbnail
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class ContextMenuSignals(QObject):
|
|
"""Signals for thread-safe UI updates from worker threads."""
|
|
show_status_message = Signal(str, int)
|
|
show_warning_dialog = Signal(str, str)
|
|
show_info_dialog = Signal(str, str)
|
|
|
|
class ContextMenuManager:
|
|
"""Manages context menu actions for game management in PortProtonQt."""
|
|
|
|
def __init__(self, parent, portproton_location, theme, load_games_callback, update_game_grid_callback):
|
|
"""
|
|
Initialize the ContextMenuManager.
|
|
|
|
Args:
|
|
parent: The parent widget (MainWindow instance).
|
|
portproton_location: Path to the PortProton directory.
|
|
theme: The current theme object.
|
|
load_games_callback: Callback to reload games list.
|
|
update_game_grid_callback: Callback to update the game grid UI.
|
|
"""
|
|
self.parent = parent
|
|
self.portproton_location = portproton_location
|
|
self.theme = theme
|
|
self.load_games = load_games_callback
|
|
self.update_game_grid = update_game_grid_callback
|
|
self.legendary_path = os.path.join(
|
|
os.getenv("XDG_CACHE_HOME", os.path.join(os.path.expanduser("~"), ".cache")),
|
|
"PortProtonQt", "legendary_cache", "legendary"
|
|
)
|
|
self.legendary_config_path = os.path.join(
|
|
os.getenv("XDG_CACHE_HOME", os.path.join(os.path.expanduser("~"), ".cache")),
|
|
"PortProtonQt", "legendary_cache"
|
|
)
|
|
self.signals = ContextMenuSignals()
|
|
if self.parent.statusBar() is None:
|
|
logger.warning("Status bar is not initialized in MainWindow")
|
|
else:
|
|
self.signals.show_status_message.connect(
|
|
self.parent.statusBar().showMessage,
|
|
Qt.ConnectionType.QueuedConnection
|
|
)
|
|
logger.debug("Connected show_status_message signal to statusBar")
|
|
self.signals.show_warning_dialog.connect(
|
|
self._show_warning_dialog,
|
|
Qt.ConnectionType.QueuedConnection
|
|
)
|
|
self.signals.show_info_dialog.connect(
|
|
self._show_info_dialog,
|
|
Qt.ConnectionType.QueuedConnection
|
|
)
|
|
|
|
def _show_warning_dialog(self, title: str, message: str):
|
|
"""Show a warning dialog in the main thread."""
|
|
logger.debug("Showing warning dialog: %s - %s", title, message)
|
|
QMessageBox.warning(self.parent, title, message)
|
|
|
|
def _show_info_dialog(self, title: str, message: str):
|
|
"""Show an info dialog in the main thread."""
|
|
logger.debug("Showing info dialog: %s - %s", title, message)
|
|
QMessageBox.information(self.parent, title, message)
|
|
|
|
def _show_status_message(self, message: str, timeout: int = 3000):
|
|
"""Show a status message on the status bar if available."""
|
|
if self.parent.statusBar():
|
|
self.parent.statusBar().showMessage(message, timeout)
|
|
logger.debug("Direct status message: %s", message)
|
|
else:
|
|
logger.warning("Status bar not available for message: %s", message)
|
|
|
|
def _check_portproton(self):
|
|
"""Check if PortProton is available."""
|
|
if self.portproton_location is None:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("PortProton is not found")
|
|
)
|
|
return False
|
|
return True
|
|
|
|
def _is_egs_game_installed(self, app_name: str) -> bool:
|
|
"""
|
|
Check if an EGS game is installed by reading installed.json.
|
|
|
|
Args:
|
|
app_name: The Legendary app_name (unique identifier for the game).
|
|
|
|
Returns:
|
|
bool: True if the game is installed, False otherwise.
|
|
"""
|
|
installed_json_path = os.path.join(self.legendary_config_path, "installed.json")
|
|
if not os.path.exists(installed_json_path):
|
|
logger.debug("installed.json not found at %s", installed_json_path)
|
|
return False
|
|
try:
|
|
with open(installed_json_path, "rb") as f:
|
|
installed_games = orjson.loads(f.read())
|
|
return app_name in installed_games
|
|
except (OSError, orjson.JSONDecodeError) as e:
|
|
logger.error("Failed to read installed.json: %s", e)
|
|
return False
|
|
|
|
def show_context_menu(self, game_card, pos: QPoint):
|
|
"""
|
|
Show the context menu for a game card at the specified position.
|
|
|
|
Args:
|
|
game_card: The GameCard instance requesting the context menu.
|
|
pos: The position (in widget coordinates) where the menu should appear.
|
|
"""
|
|
menu = QMenu(self.parent)
|
|
menu.setStyleSheet(self.theme.CONTEXT_MENU_STYLE)
|
|
|
|
favorites = read_favorites()
|
|
is_favorite = game_card.name in favorites
|
|
favorite_action = menu.addAction(
|
|
_("Remove from Favorites") if is_favorite else _("Add to Favorites")
|
|
)
|
|
favorite_action.triggered.connect(lambda: self.toggle_favorite(game_card, not is_favorite))
|
|
|
|
if game_card.game_source == "epic":
|
|
import_action = menu.addAction(_("Import to Legendary"))
|
|
import_action.triggered.connect(
|
|
lambda: self.import_to_legendary(game_card.name, game_card.appid)
|
|
)
|
|
if self._is_egs_game_installed(game_card.appid):
|
|
is_in_steam = is_game_in_steam(game_card.name)
|
|
steam_action = menu.addAction(
|
|
_("Remove from Steam") if is_in_steam else _("Add to Steam")
|
|
)
|
|
steam_action.triggered.connect(
|
|
lambda: self.remove_from_steam(game_card.name, game_card.exec_line, game_card.game_source)
|
|
if is_in_steam
|
|
else self.add_egs_to_steam(game_card.name, game_card.appid)
|
|
)
|
|
open_folder_action = menu.addAction(_("Open Game Folder"))
|
|
open_folder_action.triggered.connect(
|
|
lambda: self.open_egs_game_folder(game_card.appid)
|
|
)
|
|
desktop_dir = subprocess.check_output(['xdg-user-dir', 'DESKTOP']).decode('utf-8').strip()
|
|
desktop_path = os.path.join(desktop_dir, f"{game_card.name}.desktop")
|
|
desktop_action = menu.addAction(
|
|
_("Remove from Desktop") if os.path.exists(desktop_path) else _("Add to Desktop")
|
|
)
|
|
desktop_action.triggered.connect(
|
|
lambda: self.remove_egs_from_desktop(game_card.name)
|
|
if os.path.exists(desktop_path)
|
|
else self.add_egs_to_desktop(game_card.name, game_card.appid)
|
|
)
|
|
applications_dir = os.path.join(os.path.expanduser("~"), ".local", "share", "applications")
|
|
menu_path = os.path.join(applications_dir, f"{game_card.name}.desktop")
|
|
menu_action = menu.addAction(
|
|
_("Remove from Menu") if os.path.exists(menu_path) else _("Add to Menu")
|
|
)
|
|
menu_action.triggered.connect(
|
|
lambda: self.remove_egs_from_menu(game_card.name)
|
|
if os.path.exists(menu_path)
|
|
else self.add_egs_to_menu(game_card.name, game_card.appid)
|
|
)
|
|
|
|
if game_card.game_source not in ("steam", "epic"):
|
|
desktop_dir = subprocess.check_output(['xdg-user-dir', 'DESKTOP']).decode('utf-8').strip()
|
|
desktop_path = os.path.join(desktop_dir, f"{game_card.name}.desktop")
|
|
desktop_action = menu.addAction(
|
|
_("Remove from Desktop") if os.path.exists(desktop_path) else _("Add to Desktop")
|
|
)
|
|
desktop_action.triggered.connect(
|
|
lambda: self.remove_from_desktop(game_card.name)
|
|
if os.path.exists(desktop_path)
|
|
else self.add_to_desktop(game_card.name, game_card.exec_line)
|
|
)
|
|
edit_action = menu.addAction(_("Edit Shortcut"))
|
|
edit_action.triggered.connect(
|
|
lambda: self.edit_game_shortcut(game_card.name, game_card.exec_line, game_card.cover_path)
|
|
)
|
|
delete_action = menu.addAction(_("Delete from PortProton"))
|
|
delete_action.triggered.connect(lambda: self.delete_game(game_card.name, game_card.exec_line))
|
|
open_folder_action = menu.addAction(_("Open Game Folder"))
|
|
open_folder_action.triggered.connect(
|
|
lambda: self.open_game_folder(game_card.name, game_card.exec_line)
|
|
)
|
|
applications_dir = os.path.join(os.path.expanduser("~"), ".local", "share", "applications")
|
|
menu_path = os.path.join(applications_dir, f"{game_card.name}.desktop")
|
|
menu_action = menu.addAction(
|
|
_("Remove from Menu") if os.path.exists(menu_path) else _("Add to Menu")
|
|
)
|
|
menu_action.triggered.connect(
|
|
lambda: self.remove_from_menu(game_card.name)
|
|
if os.path.exists(menu_path)
|
|
else self.add_to_menu(game_card.name, game_card.exec_line)
|
|
)
|
|
is_in_steam = is_game_in_steam(game_card.name)
|
|
steam_action = menu.addAction(
|
|
_("Remove from Steam") if is_in_steam else _("Add to Steam")
|
|
)
|
|
steam_action.triggered.connect(
|
|
lambda: self.remove_from_steam(game_card.name, game_card.exec_line, game_card.game_source)
|
|
if is_in_steam
|
|
else self.add_to_steam(game_card.name, game_card.exec_line, game_card.cover_path)
|
|
)
|
|
|
|
menu.exec(game_card.mapToGlobal(pos))
|
|
|
|
def add_egs_to_steam(self, game_name: str, app_name: str):
|
|
"""
|
|
Adds an EGS game to Steam using the egs_api.
|
|
|
|
Args:
|
|
game_name: The display name of the game.
|
|
app_name: The Legendary app_name (unique identifier for the game).
|
|
"""
|
|
if not self._check_portproton():
|
|
return
|
|
if not os.path.exists(self.legendary_path):
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Legendary executable not found at {path}").format(path=self.legendary_path)
|
|
)
|
|
return
|
|
|
|
def on_add_to_steam_result(result: tuple[bool, str]):
|
|
success, message = result
|
|
self.signals.show_info_dialog.emit(
|
|
_("Success"),
|
|
_("'{game_name}' was added to Steam. Please restart Steam for changes to take effect.").format(game_name=game_name)
|
|
)
|
|
|
|
logger.debug("Adding EGS game '%s' to Steam", game_name)
|
|
add_egs_to_steam(app_name, game_name, self.legendary_path, on_add_to_steam_result)
|
|
|
|
def open_egs_game_folder(self, app_name: str):
|
|
"""
|
|
Opens the folder containing the EGS game's executable.
|
|
|
|
Args:
|
|
app_name: The Legendary app_name (unique identifier for the game).
|
|
"""
|
|
if not self._check_portproton():
|
|
return
|
|
exe_path = get_egs_executable(app_name, self.legendary_config_path)
|
|
if not exe_path or not os.path.exists(exe_path):
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Executable not found for game: {game_name}").format(game_name=app_name)
|
|
)
|
|
return
|
|
try:
|
|
folder_path = os.path.dirname(os.path.abspath(exe_path))
|
|
QDesktopServices.openUrl(QUrl.fromLocalFile(folder_path))
|
|
self._show_status_message(_("Opened folder for '{game_name}'").format(game_name=app_name))
|
|
except Exception as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to open folder: {error}").format(error=str(e))
|
|
)
|
|
|
|
def import_to_legendary(self, game_name, app_name):
|
|
"""
|
|
Imports an installed Epic Games Store game to Legendary asynchronously.
|
|
|
|
Args:
|
|
game_name: The display name of the game.
|
|
app_name: The Legendary app_name (unique identifier for the game).
|
|
"""
|
|
if not self._check_portproton():
|
|
return
|
|
folder_path = QFileDialog.getExistingDirectory(
|
|
self.parent, _("Select Game Installation Folder"), os.path.expanduser("~")
|
|
)
|
|
if not folder_path:
|
|
self._show_status_message(_("No folder selected"))
|
|
return
|
|
if not os.path.exists(self.legendary_path):
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Legendary executable not found at {path}").format(path=self.legendary_path)
|
|
)
|
|
return
|
|
def run_import():
|
|
cmd = [self.legendary_path, "import", app_name, folder_path]
|
|
try:
|
|
subprocess.run(cmd, capture_output=True, text=True, check=True)
|
|
self.signals.show_info_dialog.emit(
|
|
_("Success"),
|
|
_("Imported '{game_name}' to Legendary").format(game_name=game_name)
|
|
)
|
|
except subprocess.CalledProcessError as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to import '{game_name}' to Legendary: {error}").format(
|
|
game_name=game_name, error=e.stderr
|
|
)
|
|
)
|
|
self._show_status_message(_("Importing '{game_name}' to Legendary...").format(game_name=game_name))
|
|
threading.Thread(target=run_import, daemon=True).start()
|
|
|
|
def toggle_favorite(self, game_card, add: bool):
|
|
"""
|
|
Toggle the favorite status of a game and update its icon.
|
|
|
|
Args:
|
|
game_card: The GameCard instance to toggle.
|
|
add: True to add to favorites, False to remove.
|
|
"""
|
|
favorites = read_favorites()
|
|
if add and game_card.name not in favorites:
|
|
favorites.append(game_card.name)
|
|
game_card.is_favorite = True
|
|
message = _("Added '{game_name}' to favorites").format(game_name=game_card.name)
|
|
elif not add and game_card.name in favorites:
|
|
favorites.remove(game_card.name)
|
|
game_card.is_favorite = False
|
|
message = _("Removed '{game_name}' from favorites").format(game_name=game_card.name)
|
|
else:
|
|
return
|
|
save_favorites(favorites)
|
|
game_card.update_favorite_icon()
|
|
self._show_status_message(message)
|
|
|
|
def _get_desktop_path(self, game_name):
|
|
"""Construct the .desktop file path, trying both original and sanitized game names."""
|
|
desktop_path = os.path.join(self.portproton_location, f"{game_name}.desktop")
|
|
if not os.path.exists(desktop_path):
|
|
sanitized_name = game_name.replace("/", "_").replace(":", "_").replace(" ", "_")
|
|
desktop_path = os.path.join(self.portproton_location, f"{sanitized_name}.desktop")
|
|
return desktop_path
|
|
|
|
def _get_egs_desktop_path(self, game_name):
|
|
"""Construct the .desktop file path for EGS games."""
|
|
return os.path.join(self.portproton_location, "egs_desktops", f"{game_name}.desktop")
|
|
|
|
def _create_egs_desktop_file(self, game_name: str, app_name: str) -> bool:
|
|
"""
|
|
Creates a .desktop file for an EGS game in the PortProton egs_desktops directory.
|
|
|
|
Args:
|
|
game_name: The display name of the game.
|
|
app_name: The Legendary app_name (unique identifier for the game).
|
|
|
|
Returns:
|
|
bool: True if the .desktop file was created successfully, False otherwise.
|
|
"""
|
|
if not self._check_portproton():
|
|
return False
|
|
if not os.path.exists(self.legendary_path):
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Legendary executable not found at {path}").format(path=self.legendary_path)
|
|
)
|
|
return False
|
|
wrapper = "flatpak run ru.linux_gaming.PortProton"
|
|
start_sh_path = os.path.join(self.portproton_location, "data", "scripts", "start.sh")
|
|
if self.portproton_location and ".var" not in self.portproton_location:
|
|
wrapper = start_sh_path
|
|
if not os.path.exists(start_sh_path):
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("start.sh not found at {path}").format(path=start_sh_path)
|
|
)
|
|
return False
|
|
icon_path = os.path.join(self.portproton_location, "data", "img", f"{game_name}.png")
|
|
os.makedirs(os.path.dirname(icon_path), exist_ok=True)
|
|
|
|
# Generate 128x128 icon from exe only
|
|
exe_path = get_egs_executable(app_name, self.legendary_config_path)
|
|
if exe_path and os.path.exists(exe_path):
|
|
if not generate_thumbnail(exe_path, icon_path, size=128):
|
|
logger.error(f"Failed to generate thumbnail from exe: {exe_path}")
|
|
icon_path = ""
|
|
else:
|
|
logger.error(f"No executable found for EGS game: {app_name}")
|
|
icon_path = ""
|
|
|
|
egs_desktop_dir = os.path.join(self.portproton_location, "egs_desktops")
|
|
os.makedirs(egs_desktop_dir, exist_ok=True)
|
|
desktop_path = self._get_egs_desktop_path(game_name)
|
|
comment = _('Launch game "{name}" with PortProton').format(name=game_name)
|
|
desktop_entry = f"""[Desktop Entry]
|
|
Type=Application
|
|
Name={game_name}
|
|
Comment={comment}
|
|
Terminal=false
|
|
StartupNotify=true
|
|
Exec="{self.legendary_path}" launch {app_name} --no-wine --wrapper "env START_FROM_STEAM=1 {wrapper}"
|
|
Icon={icon_path}
|
|
Categories=Game
|
|
"""
|
|
try:
|
|
with open(desktop_path, "w", encoding="utf-8") as f:
|
|
f.write(desktop_entry)
|
|
os.chmod(desktop_path, 0o755)
|
|
logger.info("Created .desktop file for EGS game: %s", desktop_path)
|
|
return True
|
|
except Exception as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to create .desktop file: {error}").format(error=str(e))
|
|
)
|
|
return False
|
|
|
|
def add_egs_to_desktop(self, game_name: str, app_name: str):
|
|
"""
|
|
Copies the .desktop file for an EGS game to the Desktop folder.
|
|
|
|
Args:
|
|
game_name: The display name of the game.
|
|
app_name: The Legendary app_name (unique identifier for the game).
|
|
"""
|
|
if not self._check_portproton():
|
|
return
|
|
desktop_path = self._get_egs_desktop_path(game_name)
|
|
if not os.path.exists(desktop_path):
|
|
if not self._create_egs_desktop_file(game_name, app_name):
|
|
return
|
|
desktop_dir = subprocess.check_output(['xdg-user-dir', 'DESKTOP']).decode('utf-8').strip()
|
|
os.makedirs(desktop_dir, exist_ok=True)
|
|
dest_path = os.path.join(desktop_dir, f"{game_name}.desktop")
|
|
try:
|
|
shutil.copyfile(desktop_path, dest_path)
|
|
os.chmod(dest_path, 0o755)
|
|
self._show_status_message(_("Added '{game_name}' to {location}").format(
|
|
game_name=game_name, location=_("Desktop")
|
|
))
|
|
except OSError as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to add '{game_name}' to {location}: {error}").format(
|
|
game_name=game_name, location=_("Desktop"), error=str(e)
|
|
)
|
|
)
|
|
|
|
def remove_egs_from_desktop(self, game_name: str):
|
|
"""
|
|
Removes the .desktop file for an EGS game from the Desktop folder.
|
|
|
|
Args:
|
|
game_name: The display name of the game.
|
|
"""
|
|
desktop_dir = subprocess.check_output(['xdg-user-dir', 'DESKTOP']).decode('utf-8').strip()
|
|
desktop_path = os.path.join(desktop_dir, f"{game_name}.desktop")
|
|
self._remove_file(
|
|
desktop_path,
|
|
_("Failed to remove '{game_name}' from {location}: {error}"),
|
|
_("Removed '{game_name}' from {location}"),
|
|
game_name,
|
|
location=_("Desktop")
|
|
)
|
|
|
|
def add_egs_to_menu(self, game_name: str, app_name: str):
|
|
"""
|
|
Copies the .desktop file for an EGS game to ~/.local/share/applications.
|
|
|
|
Args:
|
|
game_name: The display name of the game.
|
|
app_name: The Legendary app_name (unique identifier for the game).
|
|
"""
|
|
if not self._check_portproton():
|
|
return
|
|
desktop_path = self._get_egs_desktop_path(game_name)
|
|
if not os.path.exists(desktop_path):
|
|
if not self._create_egs_desktop_file(game_name, app_name):
|
|
return
|
|
applications_dir = os.path.join(os.path.expanduser("~"), ".local", "share", "applications")
|
|
os.makedirs(applications_dir, exist_ok=True)
|
|
dest_path = os.path.join(applications_dir, f"{game_name}.desktop")
|
|
try:
|
|
shutil.copyfile(desktop_path, dest_path)
|
|
os.chmod(dest_path, 0o755)
|
|
self._show_status_message(_("Added '{game_name}' to {location}").format(
|
|
game_name=game_name, location=_("Menu")
|
|
))
|
|
except OSError as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to add '{game_name}' to {location}: {error}").format(
|
|
game_name=game_name, location=_("Menu"), error=str(e)
|
|
)
|
|
)
|
|
|
|
def remove_egs_from_menu(self, game_name: str):
|
|
"""
|
|
Removes the .desktop file for an EGS game from ~/.local/share/applications.
|
|
|
|
Args:
|
|
game_name: The display name of the game.
|
|
"""
|
|
applications_dir = os.path.join(os.path.expanduser("~"), ".local", "share", "applications")
|
|
desktop_path = os.path.join(applications_dir, f"{game_name}.desktop")
|
|
self._remove_file(
|
|
desktop_path,
|
|
_("Failed to remove '{game_name}' from {location}: {error}"),
|
|
_("Removed '{game_name}' from {location}"),
|
|
game_name,
|
|
location=_("Menu")
|
|
)
|
|
|
|
def _get_exec_line(self, game_name, exec_line):
|
|
"""Retrieve and validate exec_line from .desktop file if necessary."""
|
|
if exec_line and exec_line.strip() != "full":
|
|
return exec_line
|
|
desktop_path = self._get_desktop_path(game_name)
|
|
if os.path.exists(desktop_path):
|
|
try:
|
|
entry = parse_desktop_entry(desktop_path)
|
|
if entry:
|
|
exec_line = entry.get("Exec", entry.get("exec", "")).strip()
|
|
if not exec_line:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("No executable command in .desktop file for '{game_name}'").format(game_name=game_name)
|
|
)
|
|
return None
|
|
else:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to parse .desktop file for '{game_name}'").format(game_name=game_name)
|
|
)
|
|
return None
|
|
except Exception as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to read .desktop file: {error}").format(error=str(e))
|
|
)
|
|
return None
|
|
else:
|
|
for file in glob.glob(os.path.join(self.portproton_location, "*.desktop")):
|
|
entry = parse_desktop_entry(file)
|
|
if entry:
|
|
exec_line = entry.get("Exec", entry.get("exec", "")).strip()
|
|
if exec_line:
|
|
return exec_line
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("No .desktop file found for '{game_name}'").format(game_name=game_name)
|
|
)
|
|
return None
|
|
return exec_line
|
|
|
|
def _parse_exe_path(self, exec_line, game_name):
|
|
"""Parse the executable path from exec_line."""
|
|
try:
|
|
entry_exec_split = shlex.split(exec_line)
|
|
if not entry_exec_split:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Invalid executable command: {exec_line}").format(exec_line=exec_line)
|
|
)
|
|
return None
|
|
if entry_exec_split[0] == "env" and len(entry_exec_split) >= 3:
|
|
exe_path = entry_exec_split[2]
|
|
elif entry_exec_split[0] == "flatpak" and len(entry_exec_split) >= 4:
|
|
exe_path = entry_exec_split[3]
|
|
else:
|
|
exe_path = entry_exec_split[-1]
|
|
if not exe_path or not os.path.exists(exe_path):
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Executable not found: {path}").format(path=exe_path or "None")
|
|
)
|
|
return None
|
|
return exe_path
|
|
except Exception as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to parse executable: {error}").format(error=str(e))
|
|
)
|
|
return None
|
|
|
|
def _remove_file(self, file_path, error_message, success_message, game_name, location=""):
|
|
"""Remove a file and handle errors."""
|
|
try:
|
|
os.remove(file_path)
|
|
self._show_status_message(_(success_message).format(game_name=game_name, location=location))
|
|
return True
|
|
except OSError as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_(error_message).format(game_name=game_name, location=location, error=str(e))
|
|
)
|
|
return False
|
|
|
|
def delete_game(self, game_name, exec_line):
|
|
"""Delete the .desktop file and associated custom data for the game."""
|
|
reply = QMessageBox.question(
|
|
self.parent,
|
|
_("Confirm Deletion"),
|
|
_("Are you sure you want to delete '{game_name}'? This will remove the .desktop file and custom data.").format(game_name=game_name),
|
|
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
|
QMessageBox.StandardButton.No
|
|
)
|
|
if reply != QMessageBox.StandardButton.Yes:
|
|
return
|
|
if not self._check_portproton():
|
|
return
|
|
desktop_path = self._get_desktop_path(game_name)
|
|
if not os.path.exists(desktop_path):
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("No .desktop file found for '{game_name}'").format(game_name=game_name)
|
|
)
|
|
return
|
|
exec_line = self._get_exec_line(game_name, exec_line)
|
|
if not exec_line:
|
|
return
|
|
exe_path = self._parse_exe_path(exec_line, game_name)
|
|
exe_name = os.path.splitext(os.path.basename(exe_path))[0] if exe_path else None
|
|
if not self._remove_file(
|
|
desktop_path,
|
|
_("Failed to delete .desktop file: {error}"),
|
|
_("Deleted '{game_name}' successfully"),
|
|
game_name
|
|
):
|
|
return
|
|
if exe_name:
|
|
xdg_data_home = os.getenv(
|
|
"XDG_DATA_HOME",
|
|
os.path.join(os.path.expanduser("~"), ".local", "share")
|
|
)
|
|
custom_folder = os.path.join(xdg_data_home, "PortProtonQt", "custom_data", exe_name)
|
|
if os.path.exists(custom_folder):
|
|
try:
|
|
shutil.rmtree(custom_folder)
|
|
except OSError as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to delete custom data: {error}").format(error=str(e))
|
|
)
|
|
|
|
def add_to_menu(self, game_name, exec_line):
|
|
"""Copy the .desktop file to ~/.local/share/applications."""
|
|
if not self._check_portproton():
|
|
return
|
|
desktop_path = self._get_desktop_path(game_name)
|
|
if not os.path.exists(desktop_path):
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("No .desktop file found for '{game_name}'").format(game_name=game_name)
|
|
)
|
|
return
|
|
applications_dir = os.path.join(os.path.expanduser("~"), ".local", "share", "applications")
|
|
os.makedirs(applications_dir, exist_ok=True)
|
|
dest_path = os.path.join(applications_dir, f"{game_name}.desktop")
|
|
try:
|
|
shutil.copyfile(desktop_path, dest_path)
|
|
os.chmod(dest_path, 0o755)
|
|
self._show_status_message(_("Added '{game_name}' to {location}").format(
|
|
game_name=game_name, location=_("Menu")
|
|
))
|
|
except OSError as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to add '{game_name}' to {location}: {error}").format(
|
|
game_name=game_name, location=_("Menu"), error=str(e)
|
|
)
|
|
)
|
|
|
|
def remove_from_menu(self, game_name):
|
|
"""
|
|
Removes the game from the menu by removing its .desktop file from ~/.local/share/applications.
|
|
|
|
Args:
|
|
game_name: The display name of the game.
|
|
"""
|
|
applications_dir = os.path.join(os.path.expanduser("~"), ".local", "share", "applications")
|
|
desktop_path = os.path.join(applications_dir, f"{game_name}.desktop")
|
|
self._remove_file(
|
|
desktop_path,
|
|
_("Failed to remove '{game_name}' from {location}: {error}"),
|
|
_("Removed '{game_name}' from {location}"),
|
|
game_name,
|
|
location=_("Menu")
|
|
)
|
|
|
|
def add_to_desktop(self, game_name, exec_line):
|
|
"""
|
|
Copies the .desktop file to the user's Desktop folder.
|
|
|
|
Args:
|
|
game_name: The display name of the game.
|
|
exec_line: The executable command line.
|
|
"""
|
|
|
|
if not self._check_portproton():
|
|
return
|
|
desktop_path = self._get_desktop_path(game_name)
|
|
if not os.path.exists(desktop_path):
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("No .desktop file found for '{game_name}'").format(game_name=game_name)
|
|
)
|
|
return
|
|
# Ensure icon exists
|
|
exec_line = self._get_exec_line(game_name, exec_line)
|
|
if not exec_line:
|
|
return
|
|
exe_path = self._parse_exe_path(exec_line, game_name)
|
|
if not exe_path:
|
|
return
|
|
icon_path = os.path.join(self.portproton_location, "data", "img", f"{game_name}.png")
|
|
if not os.path.exists(icon_path):
|
|
if not generate_thumbnail(exe_path, icon_path, size=128):
|
|
logger.error(f"Failed to generate thumbnail for {exe_path}")
|
|
|
|
desktop_dir = subprocess.check_output(['xdg-user-dir', 'DESKTOP']).decode('utf-8').strip()
|
|
os.makedirs(desktop_dir, exist_ok=True)
|
|
dest_path = os.path.join(desktop_dir, f"{game_name}.desktop")
|
|
try:
|
|
shutil.copyfile(desktop_path, dest_path)
|
|
os.chmod(dest_path, 0o755)
|
|
self._show_status_message(_("Added '{game_name}' to {location}").format(
|
|
game_name=game_name, location=_("Desktop")
|
|
))
|
|
except OSError as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to add '{game_name}' to {location}: {error}").format(
|
|
game_name=game_name, location=_("Desktop"), error=str(e)
|
|
)
|
|
)
|
|
|
|
def remove_from_desktop(self, game_name):
|
|
"""
|
|
Removes the game from the Desktop folder by removing its .desktop file.
|
|
|
|
Args:
|
|
game_name: The display name of the game.
|
|
"""
|
|
desktop_dir = subprocess.check_output(['xdg-user-dir', 'DESKTOP']).decode('utf-8').strip()
|
|
desktop_path = os.path.join(desktop_dir, f"{game_name}.desktop")
|
|
self._remove_file(
|
|
desktop_path,
|
|
_("Failed to remove '{game_name}' from {location}: {error}"),
|
|
_("Removed '{game_name}' from {location}"),
|
|
game_name,
|
|
location=_("Desktop")
|
|
)
|
|
|
|
def edit_game_shortcut(self, game_name, exec_line, cover_path):
|
|
"""
|
|
Opens a dialog allowing the user to edit a game shortcut in edit mode to modify an existing .desktop file.
|
|
|
|
Args:
|
|
game_name: The display name of the game.
|
|
exec_line: The executable command line of the game.
|
|
cover_path: The path to the game's cover image.
|
|
"""
|
|
if not self._check_portproton():
|
|
return
|
|
exec_line = self._get_exec_line(game_name, exec_line)
|
|
if not exec_line:
|
|
return
|
|
exe_path = self._parse_exe_path(exec_line, game_name)
|
|
if not exe_path:
|
|
return
|
|
dialog = AddGameDialog(
|
|
parent=self.parent,
|
|
theme=self.theme,
|
|
edit_mode=True,
|
|
game_name=game_name,
|
|
exe_path=exe_path,
|
|
cover_path=cover_path,
|
|
)
|
|
|
|
if dialog.exec() == QDialog.DialogCode.Accepted:
|
|
new_name = dialog.nameEdit.text().strip()
|
|
new_exe_path = dialog.exeEdit.text().strip()
|
|
new_cover_path = dialog.coverEdit.text().strip()
|
|
if not new_name or not new_exe_path:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Game name and executable path are required")
|
|
)
|
|
return
|
|
desktop_entry, new_desktop_path = dialog.getDesktopEntryData()
|
|
if not desktop_entry or not new_desktop_path:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to generate .desktop file data")
|
|
)
|
|
return
|
|
|
|
old_path = self._get_desktop_path(game_name)
|
|
if game_name != new_name and os.path.exists(old_path):
|
|
self._remove_file(
|
|
old_path,
|
|
_("Failed to delete old .desktop file: {error}"),
|
|
_("Removed old .desktop file for '{game_name}'"),
|
|
game_name
|
|
)
|
|
|
|
try:
|
|
with open(new_desktop_path, "w", encoding="utf-8") as f:
|
|
f.write(desktop_entry)
|
|
os.chmod(new_desktop_path, 0o755)
|
|
except OSError as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to save .desktop file: {error}").format(error=str(e))
|
|
)
|
|
return
|
|
|
|
if os.path.isfile(new_cover_path):
|
|
exe_name = os.path.splitext(os.path.basename(new_exe_path))[0]
|
|
xdg_data_home = os.getenv(
|
|
"XDG_DATA_HOME",
|
|
os.path.join(os.path.expanduser("~"), ".local", "share")
|
|
)
|
|
custom_folder = os.path.join(xdg_data_home, "PortProtonQt", "custom_data", exe_name)
|
|
os.makedirs(custom_folder, exist_ok=True)
|
|
ext = os.path.splitext(new_cover_path)[1].lower()
|
|
if ext in [".png", ".jpg", ".jpeg", ".bmp"]:
|
|
try:
|
|
shutil.copyfile(new_cover_path, os.path.join(custom_folder, f"cover{ext}"))
|
|
except OSError as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to copy cover image: {error}").format(error=str(e))
|
|
)
|
|
return
|
|
|
|
def add_to_steam(self, game_name, exec_line, cover_path):
|
|
"""
|
|
Adds a non-Steam game to Steam using steam_api.
|
|
|
|
Args:
|
|
game_name: The display name of the game.
|
|
exec_line: The executable command line.
|
|
cover_path: Path to the cover image.
|
|
"""
|
|
if not self._check_portproton():
|
|
return
|
|
exec_line = self._get_exec_line(game_name, exec_line)
|
|
if not exec_line:
|
|
return
|
|
exe_path = self._parse_exe_path(exec_line, game_name)
|
|
if not exe_path:
|
|
return
|
|
logger.debug("Adding '%s' to Steam", game_name)
|
|
try:
|
|
success, message = add_to_steam(game_name, exec_line, cover_path)
|
|
self.signals.show_info_dialog.emit(
|
|
_("Success"),
|
|
_("'{game_name}' was added to Steam. Please restart Steam for changes to take effect.").format(game_name=game_name)
|
|
)
|
|
except Exception as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to add '{game_name}' to Steam: {error}").format(
|
|
game_name=game_name, error=str(e)
|
|
)
|
|
)
|
|
|
|
def remove_from_steam(self, game_name, exec_line, game_source):
|
|
"""Handle removing a game from Steam via steam_api, supporting both EGS and non-EGS games."""
|
|
if not self._check_portproton():
|
|
return
|
|
|
|
def on_remove_from_steam_result(result: tuple[bool, str]):
|
|
success, message = result
|
|
if success:
|
|
self.signals.show_info_dialog.emit(
|
|
_("Success"),
|
|
_("'{game_name}' was removed from Steam. Please restart Steam for changes to take effect.").format(game_name=game_name)
|
|
)
|
|
else:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_(message).format(game_name=game_name)
|
|
)
|
|
|
|
if game_source == "epic":
|
|
# For EGS games, construct the script path used in Steam shortcuts.vdf
|
|
if not self.portproton_location:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("PortProton directory not found")
|
|
)
|
|
return
|
|
steam_scripts_dir = os.path.join(self.portproton_location, "steam_scripts")
|
|
safe_game_name = re.sub(r'[<>:"/\\|?*]', '_', game_name.strip())
|
|
script_path = os.path.join(steam_scripts_dir, f"{safe_game_name}_egs.sh")
|
|
quoted_script_path = f'"{script_path}"'
|
|
|
|
# Directly remove the shortcut by matching AppName and Exe
|
|
try:
|
|
steam_home = get_steam_home()
|
|
if not steam_home:
|
|
self.signals.show_warning_dialog.emit(_("Error"), _("Steam directory not found"))
|
|
return
|
|
|
|
last_user = get_last_steam_user(steam_home)
|
|
if not last_user or 'SteamID' not in last_user:
|
|
self.signals.show_warning_dialog.emit(_("Error"), _("Failed to get Steam user ID"))
|
|
return
|
|
|
|
userdata_dir = os.path.join(steam_home, "userdata")
|
|
user_id = last_user['SteamID']
|
|
unsigned_id = convert_steam_id(user_id)
|
|
user_dir = os.path.join(userdata_dir, str(unsigned_id))
|
|
steam_shortcuts_path = os.path.join(user_dir, "config", "shortcuts.vdf")
|
|
backup_path = f"{steam_shortcuts_path}.backup"
|
|
|
|
if not os.path.exists(steam_shortcuts_path):
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Steam shortcuts file not found")
|
|
)
|
|
return
|
|
|
|
# Backup shortcuts.vdf
|
|
try:
|
|
shutil.copy2(steam_shortcuts_path, backup_path)
|
|
logger.info("Created backup of shortcuts.vdf at %s", backup_path)
|
|
except Exception as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to create backup of shortcuts.vdf: {error}").format(error=str(e))
|
|
)
|
|
return
|
|
|
|
# Load shortcuts.vdf
|
|
try:
|
|
with open(steam_shortcuts_path, 'rb') as f:
|
|
shortcuts_data = vdf.binary_load(f)
|
|
except Exception as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to load shortcuts.vdf: {error}").format(error=str(e))
|
|
)
|
|
return
|
|
|
|
shortcuts = shortcuts_data.get("shortcuts", {})
|
|
modified = False
|
|
new_shortcuts = {}
|
|
index = 0
|
|
|
|
for _key, entry in shortcuts.items():
|
|
if entry.get("AppName") == game_name and entry.get("Exe") == quoted_script_path:
|
|
modified = True
|
|
logger.info("Removing EGS game '%s' from Steam shortcuts", game_name)
|
|
continue
|
|
new_shortcuts[str(index)] = entry
|
|
index += 1
|
|
|
|
if not modified:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Game '{game_name}' not found in Steam shortcuts").format(game_name=game_name)
|
|
)
|
|
return
|
|
|
|
# Save updated shortcuts.vdf
|
|
try:
|
|
with open(steam_shortcuts_path, 'wb') as f:
|
|
vdf.binary_dump({"shortcuts": new_shortcuts}, f)
|
|
logger.info("Updated shortcuts.vdf, removed '%s'", game_name)
|
|
on_remove_from_steam_result((True, "Game '{game_name}' was removed from Steam. Please restart Steam for changes to take effect."))
|
|
except Exception as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to update shortcuts.vdf: {error}").format(error=str(e))
|
|
)
|
|
if os.path.exists(backup_path):
|
|
try:
|
|
shutil.copy2(backup_path, steam_shortcuts_path)
|
|
logger.info("Restored shortcuts.vdf from backup")
|
|
except Exception as restore_err:
|
|
logger.error("Failed to restore shortcuts.vdf: %s", restore_err)
|
|
on_remove_from_steam_result((False, "Failed to update shortcuts.vdf: {error}"))
|
|
return
|
|
|
|
# Optionally, remove the script file
|
|
if os.path.exists(script_path):
|
|
try:
|
|
os.remove(script_path)
|
|
logger.info("Removed EGS script: %s", script_path)
|
|
except OSError as e:
|
|
logger.warning(f"Failed to remove EGS script '{script_path}': {str(e)}")
|
|
|
|
except Exception as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to remove EGS game '{game_name}' from Steam: {error}").format(
|
|
game_name=game_name, error=str(e)
|
|
)
|
|
)
|
|
on_remove_from_steam_result((False, "Failed to remove EGS game '{game_name}' from Steam: {error}"))
|
|
return
|
|
|
|
else:
|
|
# For non-EGS games, use steam_api
|
|
exec_line = self._get_exec_line(game_name, exec_line)
|
|
if not exec_line:
|
|
return
|
|
exe_path = self._parse_exe_path(exec_line, game_name)
|
|
if not exe_path:
|
|
return
|
|
logger.debug("Removing non-EGS game '%s' from Steam", game_name)
|
|
try:
|
|
success, message = remove_from_steam(game_name, exec_line)
|
|
self.signals.show_info_dialog.emit(
|
|
_("Success"),
|
|
_("'{game_name}' was removed from Steam. Please restart Steam for changes to take effect.").format(game_name=game_name)
|
|
)
|
|
except Exception as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to remove game '{game_name}' from Steam: {error}").format(
|
|
game_name=game_name, error=str(e)
|
|
)
|
|
)
|
|
|
|
def open_game_folder(self, game_name, exec_line):
|
|
"""
|
|
Opens the folder containing the game's executable.
|
|
|
|
Args:
|
|
game_name: The display name of the game.
|
|
exec_line: The executable command line.
|
|
"""
|
|
if not self._check_portproton():
|
|
return
|
|
exec_line = self._get_exec_line(game_name, exec_line)
|
|
if not exec_line:
|
|
return
|
|
exe_path = self._parse_exe_path(exec_line, game_name)
|
|
if not exe_path:
|
|
return
|
|
try:
|
|
folder_path = os.path.dirname(os.path.abspath(exe_path))
|
|
QDesktopServices.openUrl(QUrl.fromLocalFile(folder_path))
|
|
self._show_status_message(_("Opened folder for '{game_name}'").format(game_name=game_name))
|
|
except Exception as e:
|
|
self.signals.show_warning_dialog.emit(
|
|
_("Error"),
|
|
_("Failed to open folder: {error}").format(error=str(e))
|
|
)
|