2021-03-18 02:16:33 +13:00
|
|
|
import logging
|
2021-03-19 00:45:59 +13:00
|
|
|
import os
|
2022-02-06 04:03:21 +13:00
|
|
|
import shutil
|
2021-03-25 05:01:12 +13:00
|
|
|
import sys
|
2021-09-11 08:41:30 +12:00
|
|
|
import traceback
|
2022-02-26 06:43:27 +13:00
|
|
|
from argparse import Namespace
|
2022-03-28 08:52:32 +13:00
|
|
|
from datetime import datetime
|
2022-06-19 06:45:36 +12:00
|
|
|
from typing import Optional
|
2021-03-18 02:16:33 +13:00
|
|
|
|
2022-04-13 11:07:48 +12:00
|
|
|
import requests.exceptions
|
2022-09-08 23:58:11 +12:00
|
|
|
from PyQt5.QtCore import QThreadPool, QTimer
|
2022-09-04 10:14:43 +12:00
|
|
|
from PyQt5.QtWidgets import QApplication, QMessageBox
|
2021-10-15 10:05:00 +13:00
|
|
|
from requests import HTTPError
|
2021-03-18 02:16:33 +13:00
|
|
|
|
2021-04-08 08:39:23 +12:00
|
|
|
from rare.components.dialogs.launch_dialog import LaunchDialog
|
|
|
|
from rare.components.main_window import MainWindow
|
2022-09-04 10:14:43 +12:00
|
|
|
from rare.shared import (
|
|
|
|
LegendaryCoreSingleton,
|
|
|
|
GlobalSignalsSingleton,
|
|
|
|
ArgumentsSingleton,
|
|
|
|
)
|
2022-09-05 05:38:24 +12:00
|
|
|
from rare.shared.rare_core import RareCore
|
2022-09-08 10:27:37 +12:00
|
|
|
from rare.utils import legendary_utils, config_helper, paths
|
2022-06-24 07:26:08 +12:00
|
|
|
from rare.widgets.rare_app import RareApp
|
2021-03-18 02:16:33 +13:00
|
|
|
|
|
|
|
logger = logging.getLogger("Rare")
|
|
|
|
|
|
|
|
|
2021-09-11 08:41:30 +12:00
|
|
|
def excepthook(exc_type, exc_value, exc_tb):
|
|
|
|
tb = "".join(traceback.format_exception(exc_type, exc_value, exc_tb))
|
|
|
|
print("Error")
|
2021-10-15 10:05:00 +13:00
|
|
|
if exc_tb == HTTPError:
|
|
|
|
try:
|
2022-02-26 06:43:27 +13:00
|
|
|
if LegendaryCoreSingleton().login():
|
2021-10-15 10:05:00 +13:00
|
|
|
return
|
|
|
|
else:
|
|
|
|
raise ValueError
|
|
|
|
except Exception as e:
|
|
|
|
logger.fatal(str(e))
|
|
|
|
QMessageBox.warning(None, "Error", QApplication.tr("Failed to login"))
|
|
|
|
QApplication.exit(1)
|
|
|
|
return
|
2021-09-11 08:41:30 +12:00
|
|
|
logger.fatal(tb)
|
|
|
|
QMessageBox.warning(None, "Error", tb)
|
|
|
|
QApplication.exit(1)
|
|
|
|
|
|
|
|
|
2022-06-24 07:26:08 +12:00
|
|
|
class App(RareApp):
|
2022-02-26 06:43:27 +13:00
|
|
|
def __init__(self, args: Namespace):
|
2022-09-08 23:58:11 +12:00
|
|
|
log_file = "Rare_{0}.log"
|
|
|
|
super(App, self).__init__(args, log_file)
|
2022-09-05 05:38:24 +12:00
|
|
|
self.rare_core = RareCore(args=args)
|
|
|
|
self.args = ArgumentsSingleton()
|
|
|
|
self.signals = GlobalSignalsSingleton()
|
2022-09-04 10:14:43 +12:00
|
|
|
self.core = LegendaryCoreSingleton()
|
2022-09-05 05:38:24 +12:00
|
|
|
|
|
|
|
config_helper.init_config_handler(self.core)
|
2021-03-27 06:23:22 +13:00
|
|
|
|
2022-06-24 07:26:08 +12:00
|
|
|
lang = self.settings.value("language", self.core.language_code, type=str)
|
|
|
|
self.load_translator(lang)
|
|
|
|
|
2021-04-05 20:57:11 +12:00
|
|
|
# set Application name for settings
|
2022-09-04 10:14:43 +12:00
|
|
|
self.mainwindow: Optional[MainWindow] = None
|
|
|
|
self.launch_dialog: Optional[LaunchDialog] = None
|
2022-09-08 00:35:51 +12:00
|
|
|
self.timer = QTimer()
|
2021-09-30 10:22:47 +13:00
|
|
|
|
2021-04-05 20:57:11 +12:00
|
|
|
# launch app
|
2022-09-02 00:13:58 +12:00
|
|
|
self.launch_dialog = LaunchDialog(parent=None)
|
2021-06-08 08:12:26 +12:00
|
|
|
self.launch_dialog.quit_app.connect(self.launch_dialog.close)
|
|
|
|
self.launch_dialog.quit_app.connect(lambda ec: exit(ec))
|
2021-03-25 05:01:12 +13:00
|
|
|
self.launch_dialog.start_app.connect(self.start_app)
|
2021-06-08 08:12:26 +12:00
|
|
|
self.launch_dialog.start_app.connect(self.launch_dialog.close)
|
2021-04-20 01:44:28 +12:00
|
|
|
|
2021-12-10 08:03:34 +13:00
|
|
|
self.launch_dialog.login()
|
2021-03-25 05:01:12 +13:00
|
|
|
|
2022-03-28 08:52:32 +13:00
|
|
|
dt_exp = datetime.fromisoformat(self.core.lgd.userdata['expires_at'][:-1])
|
|
|
|
dt_now = datetime.utcnow()
|
|
|
|
td = abs(dt_exp - dt_now)
|
|
|
|
self.timer.timeout.connect(self.re_login)
|
2022-04-13 11:07:48 +12:00
|
|
|
self.timer.start(int(td.total_seconds() - 60) * 1000)
|
2022-03-28 08:52:32 +13:00
|
|
|
|
|
|
|
def re_login(self):
|
|
|
|
logger.info("Session expires shortly. Renew session")
|
2022-04-13 11:07:48 +12:00
|
|
|
try:
|
|
|
|
self.core.login()
|
|
|
|
except requests.exceptions.ConnectionError:
|
|
|
|
self.timer.start(3000) # try again if no connection
|
|
|
|
return
|
|
|
|
dt_exp = datetime.fromisoformat(self.core.lgd.userdata['expires_at'][:-1])
|
|
|
|
dt_now = datetime.utcnow()
|
|
|
|
td = abs(dt_exp - dt_now)
|
|
|
|
self.timer.start(int(td.total_seconds() - 60) * 1000)
|
2022-03-28 08:52:32 +13:00
|
|
|
|
2021-10-08 07:19:24 +13:00
|
|
|
def start_app(self):
|
2022-01-14 08:11:20 +13:00
|
|
|
for igame in self.core.get_installed_list():
|
2022-01-16 04:40:31 +13:00
|
|
|
if not os.path.exists(igame.install_path):
|
2022-07-16 06:16:12 +12:00
|
|
|
# lk; since install_path is lost anyway, set keep_files to True
|
|
|
|
# lk: to avoid spamming the log with "file not found" errors
|
|
|
|
legendary_utils.uninstall_game(self.core, igame.app_name, keep_files=True)
|
2022-01-16 04:40:31 +13:00
|
|
|
logger.info(f"Uninstalled {igame.title}, because no game files exist")
|
|
|
|
continue
|
2022-08-03 11:33:50 +12:00
|
|
|
# lk: games that don't have an override and can't find their executable due to case sensitivity
|
|
|
|
# lk: will still erroneously require verification. This might need to be removed completely
|
|
|
|
# lk: or be decoupled from the verification requirement
|
|
|
|
if override_exe := self.core.lgd.config.get(igame.app_name, "override_exe", fallback=""):
|
|
|
|
igame_executable = override_exe
|
|
|
|
else:
|
|
|
|
igame_executable = igame.executable
|
|
|
|
if not os.path.exists(os.path.join(igame.install_path, igame_executable.replace("\\", "/").lstrip("/"))):
|
2022-01-14 08:11:20 +13:00
|
|
|
igame.needs_verification = True
|
|
|
|
self.core.lgd.set_installed_game(igame.app_name, igame)
|
2022-01-16 04:40:31 +13:00
|
|
|
logger.info(f"{igame.title} needs verification")
|
2022-01-14 08:11:20 +13:00
|
|
|
|
2021-10-08 07:19:24 +13:00
|
|
|
self.mainwindow = MainWindow()
|
2022-09-04 10:14:43 +12:00
|
|
|
self.mainwindow.exit_app.connect(self.exit_app)
|
2021-03-18 02:16:33 +13:00
|
|
|
|
2021-04-18 06:00:15 +12:00
|
|
|
if not self.args.silent:
|
2022-09-04 10:14:43 +12:00
|
|
|
self.mainwindow.show()
|
2021-12-10 08:03:34 +13:00
|
|
|
|
2022-02-26 06:43:27 +13:00
|
|
|
if self.args.test_start:
|
2021-12-11 12:18:57 +13:00
|
|
|
self.exit_app(0)
|
|
|
|
|
2021-06-08 08:12:26 +12:00
|
|
|
def exit_app(self, exit_code=0):
|
2021-10-21 06:19:03 +13:00
|
|
|
threadpool = QThreadPool.globalInstance()
|
|
|
|
threadpool.waitForDone()
|
2022-09-02 09:20:04 +12:00
|
|
|
if self.timer is not None:
|
|
|
|
self.timer.stop()
|
|
|
|
self.timer.deleteLater()
|
|
|
|
self.timer = None
|
2022-09-04 10:14:43 +12:00
|
|
|
if self.mainwindow is not None:
|
|
|
|
self.mainwindow.close()
|
|
|
|
self.mainwindow = None
|
2022-09-05 05:38:24 +12:00
|
|
|
self.rare_core.deleteLater()
|
|
|
|
del self.rare_core
|
2021-06-08 08:12:26 +12:00
|
|
|
self.processEvents()
|
2022-09-08 10:27:37 +12:00
|
|
|
shutil.rmtree(paths.tmp_dir())
|
|
|
|
os.makedirs(paths.tmp_dir())
|
2022-02-06 04:03:21 +13:00
|
|
|
|
2021-06-08 08:12:26 +12:00
|
|
|
self.exit(exit_code)
|
|
|
|
|
2021-03-25 05:01:12 +13:00
|
|
|
|
2021-04-09 21:36:27 +12:00
|
|
|
def start(args):
|
2021-09-11 08:41:30 +12:00
|
|
|
# set excepthook to show dialog with exception
|
|
|
|
sys.excepthook = excepthook
|
|
|
|
|
2021-04-07 21:50:35 +12:00
|
|
|
while True:
|
2022-02-26 06:43:27 +13:00
|
|
|
app = App(args)
|
2021-04-07 21:50:35 +12:00
|
|
|
exit_code = app.exec_()
|
|
|
|
# if not restart
|
|
|
|
# restart app
|
2021-09-13 09:28:54 +12:00
|
|
|
del app
|
2021-06-08 08:12:26 +12:00
|
|
|
if exit_code != -133742:
|
|
|
|
break
|
2022-09-04 10:14:43 +12:00
|
|
|
|
|
|
|
|