215 lines
7.7 KiB
Python
215 lines
7.7 KiB
Python
import platform
|
|
from logging import getLogger
|
|
|
|
from PyQt5.QtCore import Qt, pyqtSignal, QRunnable, QObject, QThreadPool, QSettings
|
|
from PyQt5.QtWidgets import QDialog, QApplication
|
|
from requests.exceptions import ConnectionError, HTTPError
|
|
|
|
from rare.components.dialogs.login import LoginDialog
|
|
from rare.models.apiresults import ApiResults
|
|
from rare.shared import LegendaryCoreSingleton, ArgumentsSingleton, ApiResultsSingleton
|
|
from rare.shared.image_manager import ImageManagerSingleton
|
|
from rare.ui.components.dialogs.launch_dialog import Ui_LaunchDialog
|
|
from rare.utils.utils import CloudWorker
|
|
|
|
logger = getLogger("Login")
|
|
|
|
|
|
class LaunchWorker(QRunnable):
|
|
class Signals(QObject):
|
|
progress = pyqtSignal(int)
|
|
result = pyqtSignal(object, str)
|
|
|
|
def __init__(self):
|
|
super(LaunchWorker, self).__init__()
|
|
self.setAutoDelete(True)
|
|
self.signals = LaunchWorker.Signals()
|
|
self.core = LegendaryCoreSingleton()
|
|
|
|
def run(self):
|
|
pass
|
|
|
|
|
|
class ImageWorker(LaunchWorker):
|
|
def __init__(self):
|
|
super(ImageWorker, self).__init__()
|
|
self.image_manager = ImageManagerSingleton()
|
|
|
|
def run(self):
|
|
# Download Images
|
|
games, dlcs = self.core.get_game_and_dlc_list(update_assets=True, skip_ue=False)
|
|
self.signals.result.emit((games, dlcs), "gamelist")
|
|
dlc_list = [dlc[0] for dlc in dlcs.values()]
|
|
|
|
na_games, na_dlcs = self.core.get_non_asset_library_items(force_refresh=False, skip_ue=False)
|
|
self.signals.result.emit(na_games, "no_assets")
|
|
na_dlc_list = [dlc[0] for dlc in na_dlcs.values()]
|
|
|
|
game_list = games + dlc_list + na_games + na_dlc_list
|
|
fetched = [False] * len(game_list)
|
|
|
|
def set_fetched(idx):
|
|
fetched[idx] = True
|
|
self.signals.progress.emit(sum(fetched))
|
|
|
|
for i, game in enumerate(game_list):
|
|
if game.app_title == "Unreal Engine":
|
|
game.app_title += f" {game.app_name.split('_')[-1]}"
|
|
self.core.lgd.set_game_meta(game.app_name, game)
|
|
self.image_manager.download_image_blocking(game)
|
|
self.signals.progress.emit(int(i / len(game_list) * 100))
|
|
# self.image_manager.download_image(
|
|
# game,
|
|
# load_callback=lambda: set_fetched(i),
|
|
# priority=i
|
|
# )
|
|
# while not all(fetched):
|
|
# continue
|
|
|
|
self.signals.progress.emit(100)
|
|
|
|
|
|
class ApiRequestWorker(LaunchWorker):
|
|
def __init__(self):
|
|
super(ApiRequestWorker, self).__init__()
|
|
self.settings = QSettings()
|
|
|
|
def run(self) -> None:
|
|
if self.settings.value("mac_meta", platform.system() == "Darwin", bool):
|
|
try:
|
|
result = self.core.get_game_and_dlc_list(update_assets=True, platform="Mac")
|
|
except HTTPError:
|
|
result = [], {}
|
|
else:
|
|
result = [], {}
|
|
self.signals.result.emit(result, "mac")
|
|
|
|
if self.settings.value("win32_meta", False, bool):
|
|
try:
|
|
result = self.core.get_game_and_dlc_list(update_assets=True, platform="Win32")
|
|
except HTTPError:
|
|
result = [], {}
|
|
else:
|
|
result = [], {}
|
|
self.signals.result.emit(result, "32bit")
|
|
|
|
|
|
class LaunchDialog(QDialog, Ui_LaunchDialog):
|
|
quit_app = pyqtSignal(int)
|
|
start_app = pyqtSignal()
|
|
completed = 0
|
|
|
|
def __init__(self, parent=None):
|
|
super(LaunchDialog, self).__init__(parent=parent)
|
|
self.setupUi(self)
|
|
self.setAttribute(Qt.WA_DeleteOnClose, True)
|
|
self.setWindowFlags(Qt.Dialog | Qt.CustomizeWindowHint | Qt.WindowTitleHint)
|
|
self.setWindowModality(Qt.WindowModal)
|
|
|
|
self.core = LegendaryCoreSingleton()
|
|
self.args = ArgumentsSingleton()
|
|
self.thread_pool = QThreadPool().globalInstance()
|
|
self.api_results = ApiResults()
|
|
|
|
def login(self):
|
|
do_launch = True
|
|
try:
|
|
if self.args.offline:
|
|
pass
|
|
else:
|
|
QApplication.processEvents()
|
|
if self.core.login():
|
|
logger.info("You are logged in")
|
|
else:
|
|
raise ValueError("You are not logged in. Open Login Window")
|
|
except ValueError as e:
|
|
logger.info(str(e))
|
|
# Do not set parent, because it won't show a task bar icon
|
|
do_launch = LoginDialog(core=self.core).login()
|
|
except ConnectionError as e:
|
|
logger.warning(e)
|
|
self.args.offline = True
|
|
finally:
|
|
if do_launch:
|
|
if not self.args.silent:
|
|
self.show()
|
|
self.launch()
|
|
else:
|
|
self.quit_app.emit(0)
|
|
|
|
def launch(self):
|
|
|
|
if not self.args.offline:
|
|
self.image_info.setText(self.tr("Downloading Images"))
|
|
image_worker = ImageWorker()
|
|
image_worker.signals.progress.connect(self.update_image_progbar)
|
|
image_worker.signals.result.connect(self.handle_api_worker_result)
|
|
self.thread_pool.start(image_worker)
|
|
|
|
# gamelist and no_asset games are from Image worker
|
|
api_worker = ApiRequestWorker()
|
|
api_worker.signals.result.connect(self.handle_api_worker_result)
|
|
self.thread_pool.start(api_worker)
|
|
|
|
# cloud save from another worker, because it is used in cloud_save_utils too
|
|
cloud_worker = CloudWorker()
|
|
cloud_worker.signals.result_ready.connect(lambda x: self.handle_api_worker_result(x, "saves"))
|
|
self.thread_pool.start(cloud_worker)
|
|
|
|
else:
|
|
self.completed = 2
|
|
if self.core.lgd.assets:
|
|
(
|
|
self.api_results.game_list,
|
|
self.api_results.dlcs,
|
|
) = self.core.get_game_and_dlc_list(False)
|
|
self.api_results.bit32_games = list(
|
|
map(lambda i: i.app_name, self.core.get_game_list(False, "Win32"))
|
|
)
|
|
self.api_results.mac_games = list(
|
|
map(lambda i: i.app_name, self.core.get_game_list(False, "Mac"))
|
|
)
|
|
else:
|
|
logger.warning("No assets found. Falling back to empty game lists")
|
|
self.api_results.game_list, self.api_results.dlcs = [], {}
|
|
self.api_results.mac_games = self.api_results.bit32_games = []
|
|
self.finish()
|
|
|
|
def handle_api_worker_result(self, result, text):
|
|
logger.debug(f"Api Request got from {text}")
|
|
if text == "gamelist":
|
|
if result:
|
|
self.api_results.game_list, self.api_results.dlcs = result
|
|
else:
|
|
(
|
|
self.api_results.game_list,
|
|
self.api_results.dlcs,
|
|
) = self.core.get_game_and_dlc_list(False)
|
|
elif text == "32bit":
|
|
self.api_results.bit32_games = [i.app_name for i in result[0]] if result else []
|
|
elif text == "mac":
|
|
self.api_results.mac_games = [i.app_name for i in result[0]] if result else []
|
|
|
|
elif text == "no_assets":
|
|
self.api_results.no_asset_games = result if result else []
|
|
|
|
elif text == "saves":
|
|
self.api_results.saves = result
|
|
|
|
if self.api_results:
|
|
self.finish()
|
|
|
|
def update_image_progbar(self, i: int):
|
|
self.image_prog_bar.setValue(i)
|
|
if i == 100:
|
|
self.finish()
|
|
|
|
def finish(self):
|
|
if self.completed == 1:
|
|
logger.info("App starting")
|
|
self.image_info.setText(self.tr("Starting..."))
|
|
ApiResultsSingleton(self.api_results)
|
|
self.start_app.emit()
|
|
else:
|
|
self.completed += 1
|