9ca39d94ab
Signed-off-by: loathingKernel <142770+loathingKernel@users.noreply.github.com>
337 lines
13 KiB
Python
337 lines
13 KiB
Python
import json
|
|
import os
|
|
from dataclasses import dataclass
|
|
from enum import IntEnum
|
|
from logging import getLogger
|
|
from pathlib import Path
|
|
from typing import List, Tuple, Optional
|
|
|
|
from PyQt5.QtCore import Qt, QModelIndex, pyqtSignal, QRunnable, QObject, QThreadPool, pyqtSlot
|
|
from PyQt5.QtGui import QStandardItemModel
|
|
from PyQt5.QtWidgets import QFileDialog, QGroupBox, QCompleter, QTreeView, QHeaderView, QMessageBox
|
|
|
|
from rare.lgndr.cli import LegendaryCLI
|
|
from rare.lgndr.glue.arguments import LgndrImportGameArgs
|
|
from rare.lgndr.glue.monkeys import LgndrIndirectStatus
|
|
from rare.shared import LegendaryCoreSingleton, GlobalSignalsSingleton, ApiResultsSingleton
|
|
from rare.ui.components.tabs.games.integrations.import_group import Ui_ImportGroup
|
|
from rare.utils.extra_widgets import IndicatorLineEdit, PathEdit
|
|
from rare.widgets.elide_label import ElideLabel
|
|
|
|
logger = getLogger("Import")
|
|
|
|
|
|
def find_app_name(path: str, core) -> Optional[str]:
|
|
if os.path.exists(os.path.join(path, ".egstore")):
|
|
for i in os.listdir(os.path.join(path, ".egstore")):
|
|
if i.endswith(".mancpn"):
|
|
with open(os.path.join(path, ".egstore", i)) as file:
|
|
app_name = json.load(file).get("AppName")
|
|
return app_name
|
|
elif app_name := LegendaryCLI(core).resolve_aliases(os.path.basename(os.path.normpath(path))):
|
|
# return None if game does not exist (Workaround for overlay)
|
|
if not core.get_game(app_name):
|
|
return None
|
|
return app_name
|
|
else:
|
|
logger.warning(f"Could not find AppName for {path}")
|
|
return None
|
|
|
|
|
|
class ImportResult(IntEnum):
|
|
ERROR = 0
|
|
FAILED = 1
|
|
SUCCESS = 2
|
|
|
|
|
|
@dataclass
|
|
class ImportedGame:
|
|
result: ImportResult
|
|
path: Optional[str] = None
|
|
app_name: Optional[str] = None
|
|
app_title: Optional[str] = None
|
|
message: Optional[str] = None
|
|
|
|
|
|
class ImportWorker(QRunnable):
|
|
class Signals(QObject):
|
|
finished = pyqtSignal(list)
|
|
progress = pyqtSignal(int)
|
|
|
|
def __init__(self, path: str, app_name: str = None, import_folder: bool = False, import_dlcs: bool = False):
|
|
super(ImportWorker, self).__init__()
|
|
self.signals = self.Signals()
|
|
self.core = LegendaryCoreSingleton()
|
|
|
|
self.path = Path(path)
|
|
self.app_name = app_name
|
|
self.import_folder = import_folder
|
|
self.import_dlcs = import_dlcs
|
|
|
|
def run(self) -> None:
|
|
result_list: List = []
|
|
if self.import_folder:
|
|
folders = [i for i in self.path.iterdir() if i.is_dir()]
|
|
number_of_folders = len(folders)
|
|
for i, child in enumerate(folders):
|
|
if not child.is_dir():
|
|
continue
|
|
result = self.__try_import(child, None)
|
|
result_list.append(result)
|
|
self.signals.progress.emit(int(100 * i // number_of_folders))
|
|
else:
|
|
result = self.__try_import(self.path, self.app_name)
|
|
result_list.append(result)
|
|
self.signals.finished.emit(result_list)
|
|
|
|
def __try_import(self, path: Path, app_name: str = None) -> ImportedGame:
|
|
result = ImportedGame(ImportResult.ERROR)
|
|
result.path = str(path)
|
|
if app_name or (app_name := find_app_name(str(path), self.core)):
|
|
result.app_name = app_name
|
|
result.app_title = app_title = self.core.get_game(app_name).app_title
|
|
success, message = self.__import_game(path, app_name, app_title)
|
|
if not success:
|
|
result.result = ImportResult.FAILED
|
|
result.message = message
|
|
else:
|
|
result.result = ImportResult.SUCCESS
|
|
return result
|
|
|
|
def __import_game(self, path: Path, app_name: str, app_title: str):
|
|
cli = LegendaryCLI(self.core)
|
|
status = LgndrIndirectStatus()
|
|
args = LgndrImportGameArgs(
|
|
app_path=str(path),
|
|
app_name=app_name,
|
|
indirect_status=status,
|
|
get_boolean_choice=lambda prompt, default=True: self.import_dlcs
|
|
)
|
|
cli.import_game(args)
|
|
return status.success, status.message
|
|
|
|
|
|
class AppNameCompleter(QCompleter):
|
|
activated = pyqtSignal(str)
|
|
|
|
def __init__(self, app_names: List, parent=None):
|
|
super(AppNameCompleter, self).__init__(parent)
|
|
# pylint: disable=E1136
|
|
super(AppNameCompleter, self).activated[QModelIndex].connect(self.__activated_idx)
|
|
|
|
model = QStandardItemModel(len(app_names), 2)
|
|
for idx, game in enumerate(app_names):
|
|
app_name, app_title = game
|
|
model.setData(model.index(idx, 0), app_title)
|
|
model.setData(model.index(idx, 1), app_name)
|
|
self.setModel(model)
|
|
|
|
treeview = QTreeView()
|
|
self.setPopup(treeview)
|
|
treeview.setRootIsDecorated(False)
|
|
treeview.header().hide()
|
|
treeview.header().setSectionResizeMode(0, QHeaderView.Stretch)
|
|
treeview.header().setSectionResizeMode(1, QHeaderView.Stretch)
|
|
|
|
# listview = QListView()
|
|
# self.setPopup(listview)
|
|
# # listview.setModelColumn(1)
|
|
|
|
self.setFilterMode(Qt.MatchContains)
|
|
self.setCaseSensitivity(Qt.CaseInsensitive)
|
|
# self.setCompletionMode(QCompleter.UnfilteredPopupCompletion)
|
|
|
|
def __activated_idx(self, idx):
|
|
# lk: don't even look at this in a funny way, it will die of shame
|
|
# lk: Note to self, the completer and popup models are different.
|
|
# lk: Getting the index from the popup and trying to use it in the completer will return invalid results
|
|
if isinstance(idx, QModelIndex):
|
|
self.activated.emit(self.popup().model().data(self.popup().model().index(idx.row(), 1)))
|
|
# TODO: implement conversion from app_name to app_title (signal loop here)
|
|
# if isinstance(idx_str, str):
|
|
# self.activated.emit(idx_str)
|
|
|
|
|
|
class ImportGroup(QGroupBox):
|
|
def __init__(self, parent=None):
|
|
super(ImportGroup, self).__init__(parent=parent)
|
|
self.ui = Ui_ImportGroup()
|
|
self.ui.setupUi(self)
|
|
self.core = LegendaryCoreSingleton()
|
|
self.signals = GlobalSignalsSingleton()
|
|
self.api_results = ApiResultsSingleton()
|
|
|
|
self.app_name_list = [game.app_name for game in self.api_results.game_list]
|
|
self.install_dir_list = [
|
|
game.metadata.get("customAttributes", {})
|
|
.get("FolderName", {})
|
|
.get("value", game.app_name)
|
|
for game in self.api_results.game_list
|
|
if not game.is_dlc
|
|
]
|
|
|
|
self.path_edit = PathEdit(
|
|
self.core.get_default_install_dir(),
|
|
QFileDialog.DirectoryOnly,
|
|
edit_func=self.path_edit_callback,
|
|
parent=self,
|
|
)
|
|
self.path_edit.textChanged.connect(self.path_changed)
|
|
self.ui.path_edit_layout.addWidget(self.path_edit)
|
|
|
|
self.app_name_edit = IndicatorLineEdit(
|
|
placeholder=self.tr("Use in case the app name was not found automatically"),
|
|
completer=AppNameCompleter(
|
|
app_names=[(i.app_name, i.app_title) for i in self.api_results.game_list]
|
|
),
|
|
edit_func=self.app_name_edit_callback,
|
|
parent=self,
|
|
)
|
|
self.app_name_edit.textChanged.connect(self.app_name_changed)
|
|
self.ui.app_name_layout.addWidget(self.app_name_edit)
|
|
|
|
self.ui.import_folder_check.stateChanged.connect(self.import_folder_changed)
|
|
self.ui.import_dlcs_check.setEnabled(False)
|
|
self.ui.import_dlcs_check.stateChanged.connect(self.import_dlcs_changed)
|
|
|
|
self.ui.import_button.setEnabled(False)
|
|
self.ui.import_button.clicked.connect(
|
|
lambda: self.import_pressed(self.path_edit.text())
|
|
)
|
|
|
|
self.info_label = ElideLabel(text="", parent=self)
|
|
self.ui.button_info_layout.addWidget(self.info_label)
|
|
|
|
self.threadpool = QThreadPool.globalInstance()
|
|
|
|
def path_edit_callback(self, path) -> Tuple[bool, str, str]:
|
|
if os.path.exists(path):
|
|
if os.path.exists(os.path.join(path, ".egstore")):
|
|
return True, path, ""
|
|
elif os.path.basename(path) in self.install_dir_list:
|
|
return True, path, ""
|
|
else:
|
|
return False, path, PathEdit.reasons.dir_not_exist
|
|
return False, path, ""
|
|
|
|
@pyqtSlot(str)
|
|
def path_changed(self, path: str):
|
|
self.info_label.setText("")
|
|
self.ui.import_folder_check.setCheckState(Qt.Unchecked)
|
|
if self.path_edit.is_valid:
|
|
self.app_name_edit.setText(find_app_name(path, self.core))
|
|
else:
|
|
self.app_name_edit.setText("")
|
|
|
|
def app_name_edit_callback(self, text) -> Tuple[bool, str, str]:
|
|
if not text:
|
|
return False, text, ""
|
|
if text in self.app_name_list:
|
|
return True, text, ""
|
|
else:
|
|
return False, text, IndicatorLineEdit.reasons.game_not_installed
|
|
|
|
@pyqtSlot(str)
|
|
def app_name_changed(self, app_name: str):
|
|
self.info_label.setText("")
|
|
self.ui.import_dlcs_check.setCheckState(Qt.Unchecked)
|
|
if self.app_name_edit.is_valid:
|
|
self.ui.import_dlcs_check.setEnabled(
|
|
bool(self.core.get_dlc_for_game(app_name))
|
|
)
|
|
self.ui.import_button.setEnabled(True)
|
|
else:
|
|
self.ui.import_dlcs_check.setEnabled(False)
|
|
self.ui.import_button.setEnabled(False)
|
|
|
|
@pyqtSlot(int)
|
|
def import_folder_changed(self, state: Qt.CheckState):
|
|
self.app_name_edit.setEnabled(not state)
|
|
self.ui.import_dlcs_check.setCheckState(Qt.Unchecked)
|
|
self.ui.import_dlcs_check.setEnabled(
|
|
state
|
|
or (self.app_name_edit.is_valid and bool(self.core.get_dlc_for_game(self.app_name_edit.text())))
|
|
)
|
|
self.ui.import_button.setEnabled(state or (not state and self.app_name_edit.is_valid))
|
|
|
|
@pyqtSlot(int)
|
|
def import_dlcs_changed(self, state: Qt.CheckState):
|
|
self.ui.import_button.setEnabled(self.ui.import_folder_check.isChecked() or self.app_name_edit.is_valid)
|
|
|
|
def import_pressed(self, path=None):
|
|
if not path:
|
|
path = self.path_edit.text()
|
|
worker = ImportWorker(
|
|
path,
|
|
self.app_name_edit.text(),
|
|
self.ui.import_folder_check.isChecked(),
|
|
self.ui.import_dlcs_check.isChecked(),
|
|
)
|
|
worker.signals.finished.connect(self.import_finished)
|
|
worker.signals.progress.connect(self.import_progress)
|
|
self.threadpool.start(worker)
|
|
self.info_label.setText(self.tr("Importing games"))
|
|
self.ui.import_button.setDisabled(True)
|
|
|
|
@pyqtSlot(list)
|
|
def import_finished(self, result: List):
|
|
logger.info(f"Import finished: {result}")
|
|
self.info_label.setText("")
|
|
|
|
self.signals.game.installed.emit([r.app_name for r in result if r.result == ImportResult.SUCCESS])
|
|
|
|
for failed in (f for f in result if f.result == ImportResult.FAILED):
|
|
igame = self.core.get_installed_game(failed.app_name)
|
|
if igame and igame.version != self.core.get_asset(igame.app_name, igame.platform, False).build_version:
|
|
# update available
|
|
self.signals.download.enqueue_game.emit(igame.app_name)
|
|
self.signals.download.update_tab.emit()
|
|
|
|
if len(result) == 1:
|
|
res = result[0]
|
|
if res.result == ImportResult.SUCCESS:
|
|
self.info_label.setText(
|
|
self.tr("Success: <b>{}</b> imported").format(res.app_title)
|
|
)
|
|
elif res.result == ImportResult.FAILED:
|
|
self.info_label.setText(
|
|
self.tr("Failed: <b>{}</b> - {}").format(res.app_title, res.message)
|
|
)
|
|
else:
|
|
self.info_label.setText(
|
|
self.tr("Error: Could not find AppName for <b>{}</b>").format(res.path)
|
|
)
|
|
else:
|
|
success = [r for r in result if r.result == ImportResult.SUCCESS]
|
|
failure = [r for r in result if r.result == ImportResult.FAILED]
|
|
errored = [r for r in result if r.result == ImportResult.ERROR]
|
|
# pylint: disable=E1101
|
|
messagebox = QMessageBox(
|
|
QMessageBox.Information,
|
|
self.tr("Import summary"),
|
|
self.tr(
|
|
"Tried to import {} folders.\n\n"
|
|
"Successfully imported {} games, failed to import {} games and {} errors occurred"
|
|
).format(len(success) + len(failure) + len(errored), len(success), len(failure), len(errored)),
|
|
buttons=QMessageBox.StandardButton.Close,
|
|
parent=self,
|
|
)
|
|
messagebox.setWindowModality(Qt.NonModal)
|
|
details: List = []
|
|
for res in success:
|
|
details.append(
|
|
self.tr("Success: {} imported").format(res.app_title)
|
|
)
|
|
for res in failure:
|
|
details.append(
|
|
self.tr("Failed: {} - {}").format(res.app_title, res.message)
|
|
)
|
|
for res in errored:
|
|
details.append(
|
|
self.tr("Error: Could not find AppName for {}").format(res.path)
|
|
)
|
|
messagebox.setDetailedText("\n".join(details))
|
|
messagebox.show()
|
|
|
|
def import_progress(self, progress: int):
|
|
pass
|