196 lines
6.9 KiB
Python
196 lines
6.9 KiB
Python
import os
|
|
from getpass import getuser
|
|
from json import loads
|
|
from logging import getLogger
|
|
|
|
from PyQt5.QtCore import QUrl, pyqtSignal
|
|
from PyQt5.QtWebEngineWidgets import QWebEngineView, QWebEngineProfile, QWebEnginePage
|
|
from PyQt5.QtWidgets import QDialog, QWidget, QVBoxLayout, QLabel, QPushButton, QStackedLayout, QLineEdit, QButtonGroup, \
|
|
QRadioButton
|
|
from legendary.core import LegendaryCore
|
|
|
|
logger = getLogger("LoginWindow")
|
|
|
|
|
|
class LoginBrowser(QWebEngineView):
|
|
def __init__(self):
|
|
super(LoginBrowser, self).__init__()
|
|
self.browser_profile = QWebEngineProfile("storage", self)
|
|
self.webpage = QWebEnginePage(self.browser_profile, self)
|
|
self.setPage(self.webpage)
|
|
|
|
def createWindow(self, webengine_window_type):
|
|
return self
|
|
|
|
|
|
class ImportWidget(QWidget):
|
|
signal = pyqtSignal(bool)
|
|
|
|
def __init__(self, core: LegendaryCore):
|
|
|
|
super(ImportWidget, self).__init__()
|
|
self.wine_paths = []
|
|
self.layout = QVBoxLayout()
|
|
self.core = core
|
|
|
|
self.import_text = QLabel("<h2>Import from existing Epic Games Launcher installation</h2>\nYou will get "
|
|
"logged out there")
|
|
self.import_accept_button = QPushButton("Import")
|
|
self.import_accept_button.clicked.connect(self.auth)
|
|
appdata_paths = self.get_appdata_path()
|
|
|
|
self.layout.addWidget(self.import_text)
|
|
|
|
if len(appdata_paths) == 0:
|
|
self.path = QLineEdit()
|
|
self.path.setPlaceholderText("Path to Wineprefix (Not implemented)")
|
|
self.layout.addWidget(self.path)
|
|
else:
|
|
self.btn_group = QButtonGroup()
|
|
for i in appdata_paths:
|
|
radio_button = QRadioButton(i)
|
|
self.btn_group.addButton(radio_button)
|
|
self.layout.addWidget(radio_button)
|
|
|
|
# for i in appdata_paths:
|
|
|
|
self.appdata_path_text = QLabel(f"Appdata path: {self.core.egl.appdata_path}")
|
|
|
|
self.login_text = QLabel("")
|
|
# self.layout.addWidget(self.btn_group)
|
|
self.layout.addWidget(self.login_text)
|
|
self.layout.addStretch(1)
|
|
self.layout.addWidget(self.import_accept_button)
|
|
|
|
self.setLayout(self.layout)
|
|
|
|
def get_appdata_path(self) -> list:
|
|
if self.core.egl.appdata_path:
|
|
return [self.core.egl.appdata_path]
|
|
|
|
else: # Linux
|
|
wine_paths = []
|
|
possible_wine_paths = [os.path.expanduser('~/Games/epic-games-store/'),
|
|
os.path.expanduser('~/.wine/')]
|
|
|
|
for i in possible_wine_paths:
|
|
if os.path.exists(i):
|
|
if os.path.exists(os.path.join(i, "drive_c/users", getuser(),
|
|
'Local Settings/Application Data/EpicGamesLauncher',
|
|
'Saved/Config/Windows')):
|
|
wine_paths.append(i)
|
|
|
|
if len(wine_paths) > 0:
|
|
appdata_dirs = [
|
|
os.path.join(i, "drive_c/users", getuser(), 'Local Settings/Application Data/EpicGamesLauncher',
|
|
'Saved/Config/Windows') for i in wine_paths]
|
|
return appdata_dirs
|
|
return []
|
|
|
|
def auth(self):
|
|
self.import_accept_button.setDisabled(True)
|
|
if not self.btn_group:
|
|
self.core.egl.appdata_path = self.path.text()
|
|
|
|
for i in self.btn_group.buttons():
|
|
|
|
if i.isChecked():
|
|
self.core.egl.appdata_path = i.text()
|
|
try:
|
|
if self.core.auth_import():
|
|
logger.info(f"Logged in as {self.core.lgd.userdata['displayName']}")
|
|
self.signal.emit(True)
|
|
return
|
|
|
|
except:
|
|
pass
|
|
|
|
self.import_accept_button.setDisabled(False)
|
|
logger.warning("Error: No valid session found")
|
|
self.login_text.setText("Error: No valid session found")
|
|
|
|
|
|
class LoginWindow(QDialog):
|
|
def __init__(self, core: LegendaryCore):
|
|
super(LoginWindow, self).__init__()
|
|
self.core = core
|
|
self.success_code = False
|
|
self.widget = QWidget()
|
|
self.setGeometry(0, 0, 200, 300)
|
|
self.welcome_layout = QVBoxLayout()
|
|
self.title = QLabel(
|
|
"<h2>Welcome to Rare the graphical interface for Legendary, an open source Epic Games alternative.</h2>\n<h3>Select one Option to Login</h3>")
|
|
self.browser_btn = QPushButton("Use browser to login")
|
|
self.browser_btn.clicked.connect(self.browser_login)
|
|
self.import_btn = QPushButton("Import from existing Epic Games installation")
|
|
|
|
self.import_btn.clicked.connect(self.import_login)
|
|
self.text = QLabel("")
|
|
self.exit_btn = QPushButton("Exit App")
|
|
self.exit_btn.clicked.connect(self.exit_login)
|
|
|
|
self.welcome_layout.addWidget(self.title)
|
|
self.welcome_layout.addWidget(self.browser_btn)
|
|
self.welcome_layout.addWidget(self.import_btn)
|
|
self.welcome_layout.addWidget(self.text)
|
|
self.welcome_layout.addWidget(self.exit_btn)
|
|
self.widget.setLayout(self.welcome_layout)
|
|
|
|
self.browser = LoginBrowser()
|
|
self.browser.loadFinished.connect(self.check_for_sid_page)
|
|
|
|
self.import_widget = ImportWidget(self.core)
|
|
self.import_widget.signal.connect(self.import_resp)
|
|
self.layout = QStackedLayout()
|
|
self.layout.addWidget(self.widget)
|
|
self.layout.addWidget(self.browser)
|
|
self.layout.addWidget(self.import_widget)
|
|
self.setLayout(self.layout)
|
|
self.show()
|
|
|
|
def import_resp(self, b: bool):
|
|
if b:
|
|
self.success()
|
|
else:
|
|
self.layout.setCurrentIndex(0)
|
|
self.text.setText("<h4 style='color: red'>No valid session found</h4>")
|
|
|
|
def login(self):
|
|
self.exec_()
|
|
return self.success_code
|
|
|
|
def success(self):
|
|
self.success_code = True
|
|
self.close()
|
|
|
|
def retry(self):
|
|
self.__init__(self.core)
|
|
|
|
def exit_login(self):
|
|
self.code = 1
|
|
self.close()
|
|
|
|
def browser_login(self):
|
|
self.setGeometry(0, 0, 800, 600)
|
|
self.browser.load(QUrl(
|
|
'https://www.epicgames.com/id/login?redirectUrl=https%3A%2F%2Fwww.epicgames.com%2Fid%2Fapi%2Fredirect'))
|
|
self.layout.setCurrentIndex(1)
|
|
|
|
def import_login(self):
|
|
self.layout.setCurrentIndex(2)
|
|
|
|
def check_for_sid_page(self):
|
|
if self.browser.url() == QUrl("https://www.epicgames.com/id/api/redirect"):
|
|
self.browser.page().toPlainText(self.browser_auth)
|
|
|
|
def browser_auth(self, json):
|
|
token = self.core.auth_sid(loads(json)["sid"])
|
|
if self.core.auth_code(token):
|
|
logger.info(f"Successfully logged in as {self.core.lgd.userdata['displayName']}")
|
|
self.success()
|
|
else:
|
|
self.layout.setCurrentIndex(0)
|
|
logger.warning("Login failed")
|
|
self.browser.close()
|
|
self.text.setText("Login Failed")
|