84 lines
3.1 KiB
Python
84 lines
3.1 KiB
Python
|
import os
|
||
|
|
from getpass import getuser
|
||
|
|
from logging import getLogger
|
||
|
|
|
||
|
|
from PyQt5.QtCore import pyqtSignal
|
||
|
|
from PyQt5.QtWidgets import QWidget
|
||
|
|
from custom_legendary.core import LegendaryCore
|
||
|
|
|
||
|
|
from rare.ui.components.dialogs.login.import_login import Ui_ImportLogin
|
||
|
|
|
||
|
|
logger = getLogger("ImportLogin")
|
||
|
|
|
||
|
|
|
||
|
|
class ImportLogin(QWidget, Ui_ImportLogin):
|
||
|
|
success = pyqtSignal()
|
||
|
|
changed = pyqtSignal()
|
||
|
|
if os.name == "nt":
|
||
|
|
localappdata = os.path.expandvars("%LOCALAPPDATA%")
|
||
|
|
else:
|
||
|
|
localappdata = os.path.join("drive_c/users", getuser(), "Local Settings/Application Data")
|
||
|
|
appdata_path = os.path.join(localappdata, "EpicGamesLauncher/Saved/Config/Windows")
|
||
|
|
found = False
|
||
|
|
|
||
|
|
def __init__(self, core: LegendaryCore, parent=None):
|
||
|
|
super(ImportLogin, self).__init__(parent=parent)
|
||
|
|
self.setupUi(self)
|
||
|
|
|
||
|
|
self.core = core
|
||
|
|
|
||
|
|
if os.name == "nt":
|
||
|
|
if not self.core.egl.appdata_path and os.path.exists(self.egl_data_path):
|
||
|
|
self.core.egl.appdata_path = self.appdata_path
|
||
|
|
if not self.core.egl.appdata_path:
|
||
|
|
self.status_label.setText(self.tr("Could not find EGL Program Data."))
|
||
|
|
else:
|
||
|
|
self.status_label.setText(self.tr("Found EGL Program Data. Click 'Next' to import them."))
|
||
|
|
self.found = True
|
||
|
|
else:
|
||
|
|
self.info_label.setText(
|
||
|
|
self.tr(
|
||
|
|
"Please select the Wine prefix where Epic Games Launcher is installed. ") + self.info_label.text()
|
||
|
|
)
|
||
|
|
prefixes = self.get_wine_prefixes()
|
||
|
|
if len(prefixes):
|
||
|
|
self.prefix_combo.addItems(prefixes)
|
||
|
|
self.status_label.setText(self.tr("Select the Wine prefix you want to import."))
|
||
|
|
else:
|
||
|
|
self.status_label.setText(self.tr("Could not any EGL Program Data."))
|
||
|
|
self.prefix_combo.editTextChanged.connect(self.changed.emit)
|
||
|
|
|
||
|
|
def get_wine_prefixes(self):
|
||
|
|
possible_prefixes = [
|
||
|
|
os.path.expanduser("~/.wine"),
|
||
|
|
os.path.expanduser("~/Games/epic-games-store"),
|
||
|
|
os.path.expanduser("~/Wine/Stores")
|
||
|
|
]
|
||
|
|
prefixes = []
|
||
|
|
for prefix in possible_prefixes:
|
||
|
|
if os.path.exists(os.path.join(prefix, self.appdata_path)):
|
||
|
|
prefixes.append(prefix)
|
||
|
|
return prefixes
|
||
|
|
|
||
|
|
def is_valid(self):
|
||
|
|
if os.name == "nt":
|
||
|
|
return self.found
|
||
|
|
else:
|
||
|
|
return os.path.exists(os.path.join(self.prefix_combo.currentText(), self.appdata_path))
|
||
|
|
|
||
|
|
def do_login(self):
|
||
|
|
self.status_label.setText(self.tr("Loading..."))
|
||
|
|
if os.name == "nt":
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
self.core.egl.appdata_path = os.path.join(self.prefix_combo.currentText(), self.appdata_path)
|
||
|
|
try:
|
||
|
|
if self.core.auth_import():
|
||
|
|
logger.info(f"Logged in as {self.core.lgd.userdata['displayName']}")
|
||
|
|
self.success.emit()
|
||
|
|
else:
|
||
|
|
self.status_label.setText(self.tr("Login failed."))
|
||
|
|
logger.warning("Failed to import existing session")
|
||
|
|
except Exception as e:
|
||
|
|
logger.warning(e)
|