stellaris-dlc-unlocker/gui/main_window.py

import os
from time import sleep
from sys import argv, exit
from PyQt5.QtWidgets import QFileDialog, QMessageBox, QMainWindow, QProgressDialog
from gui.cream_api_maker import CreamAPI
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QIcon, QDesktopServices
import design.main_window as main_design
from libs.server_data import gameversion, version, get_remote_file_size, url, server_msg
from libs.game_path import stellaris_path
from gui.DownloadThread import DownloaderThread
from libs.encrypt import decrypt
from subprocess import Popen, run
from zipfile import ZipFile
from shutil import rmtree, copytree
from requests import get
import ctypes


class MainWindow(QMainWindow, main_design.Ui_MainWindow):

    def __init__(self):
        super(MainWindow, self).__init__()
        self.setupUi(self)
        self.setWindowState(Qt.WindowActive)

        # ----------- инициализация кнопок ----------- #

        self.next_button.clicked.connect(self.switch_to_next)
        self.next_button_2.clicked.connect(self.download_file)
        self.next_button_3.clicked.connect(self.switch_to_next)
        self.next_button_4.clicked.connect(self.switch_to_next)
        self.next_button_5.clicked.connect(self.switch_to_next)
        self.cancel_button.clicked.connect(self.cancel)
        self.cancel_button_2.clicked.connect(self.cancel)
        self.cancel_button_3.clicked.connect(self.cancel)
        self.cancel_button_4.clicked.connect(self.cancel)
        self.cancel_button_5.clicked.connect(self.cancel)
        self.cancel_button_6.clicked.connect(self.cancel)
        self.back_button.clicked.connect(self.switch_to_back)
        self.back_button_2.clicked.connect(self.switch_to_back)
        self.back_button_3.clicked.connect(self.switch_to_back)
        self.reinstall_button.clicked.connect(self.reinstall)
        # self.skip_button.clicked.connect(lambda: self.reinstall(skip=True))
        self.finish_button.clicked.connect(self.finish)
        self.locate_folder.clicked.connect(self.browse_folder)
        self.eula_true.toggled.connect(self.on_radio_button_toggled)
        self.eula_true1.toggled.connect(self.on_radio_button_toggled)
        self.eula_false.toggled.connect(self.on_radio_button_toggled)
        self.textBrowser_5.anchorClicked.connect(self.open_link_in_browser)
        self.textBrowser_15.anchorClicked.connect(self.open_link_in_browser)

        self.download_thread = None
        self.is_downloading = False
        self.next_button_5.setEnabled(False)
        # -------------------------------------------- #

        # ----------- запуск необходимых стартовых функций ----------- #
        self.version_check()
        self.version_change()
        self.server_msg()
        self.gameversion_change()
        self.space_req_change()
        self.path_change()
        self.setWindowTitle("Stellaris DLC Unlocker")
        self.parent_directory = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
        self.setWindowIcon(QIcon(f'{self.parent_directory}/design/435345.png'))
        # ------------------------------------------------------------ #

    def switch_to_next(self):
        self.stackedWidget.setCurrentIndex(self.stackedWidget.currentIndex() + 1)

    def switch_to_back(self):
        self.stackedWidget.setCurrentIndex(self.stackedWidget.currentIndex() - 1)

    @staticmethod
    def updateApplication(download_url):
        old_file = argv[0]
        old_dir = os.path.dirname(old_file)
        pid = ctypes.windll.kernel32.GetCurrentProcessId()

        progress_dialog = QProgressDialog("Загрузка обновления...", None, 0, 100)
        progress_dialog.setWindowTitle("Обновление")
        progress_dialog.setWindowModality(2)
        progress_dialog.show()

        response = get(download_url, stream=True)
        total_size_in_bytes = int(response.headers.get('content-length', 0))
        block_size = 1024
        downloaded_bytes = 0
        with open(f'{old_dir}/Stellaris-DLC-Unlocker.load', 'wb') as new_exe:
            for data in response.iter_content(block_size):
                new_exe.write(data)
                downloaded_bytes += len(data)
                progress = int(downloaded_bytes / total_size_in_bytes * 100)
                progress_dialog.setValue(progress)

        with open('updater.bat', 'w') as updater_file:
            updater_file.write('@echo off\n')
            updater_file.write(f'taskkill /pid {pid} /f\n')
            updater_file.write(f'ping 127.0.0.1 -n 1 > nul\n')
            updater_file.write('echo Updating...\n')
            updater_file.write(f'del "{old_file}"\n')
            updater_file.write(f'rename "{old_dir}\\Stellaris-DLC-Unlocker.load" "Stellaris-DLC-Unlocker.exe"\n')
            updater_file.write(f'start "" "{old_dir}\\Stellaris-DLC-Unlocker.exe"\n')
            updater_file.write('ping 127.0.0.1 -n 1 > nul\n')
            updater_file.write('del %0')
        progress_dialog.close()

        sleep(0.5)
        Popen(['cmd.exe', '/c', 'updater.bat'], shell=True)
        exit()

    def version_check(self):
        iversion = '0.6'
        if float(version) > float(iversion):
            if self.ok_dialog('Новая версия',
                              "На сервере обнаружена новая версия!\nНажмите 'ОК' для обновления\nПосле обновления новая версия будет автоматически открыта\n\n"
                              f"Ваша версия: {iversion}  Версия на сервере: {version}",
                              QMessageBox.Critical):
                # os.system(f"start https://github.com/seuyh/stellaris-dlc-unlocker/releases/tag/{str(version)}")
                self.updateApplication(
                    f'https://github.com/seuyh/stellaris-dlc-unlocker/releases/download/{str(version)}/Stellaris-DLC-Unlocker.exe')

    def cancel(self):
        msg_box = QMessageBox()
        msg_box.setIcon(QMessageBox.Information)
        msg_box.setWindowTitle('Выход')
        msg_box.setText(
            'Если вы выйдете, разблокировщик не будет установлен.\n\n\nВыйти из программы установки?')
        msg_box.setStandardButtons(QMessageBox.Yes | QMessageBox.Cancel)
        msg_box.setDefaultButton(QMessageBox.Yes)
        yes_button = msg_box.button(QMessageBox.Yes)
        yes_button.setText('Да')
        cancel_button = msg_box.button(QMessageBox.Cancel)
        cancel_button.setText('Нет')
        reply = msg_box.exec_()
        if reply == QMessageBox.Yes:
            try:
                if os.path.exists(self.save_path):
                    os.remove(self.save_path)
            except:
                pass
            self.close()

    def version_change(self):
        new_text = self.version.text().replace("%nan%", version)
        self.version.setText(new_text)

    def gameversion_change(self):
        new_text = self.hello_msg.toHtml().replace("[unknown]", gameversion)
        self.hello_msg.setText(new_text)
        new_text = self.hello2_msg.toHtml().replace("[unknown]", gameversion)
        self.hello2_msg.setText(new_text)

    def space_req_change(self):
        new_text = self.space_req.text().replace("%nan%", get_remote_file_size(
            decrypt(url, 'LPrVJDjMXGx1ToihooozyFX4-toGjKcCr8pjZFmq62c=')))
        self.space_req.setText(new_text)

    def on_radio_button_toggled(self):
        if self.eula_true.isChecked() or self.eula_true1.isChecked():
            self.next_button_3.setEnabled(True)
            self.next_button_3.setCursor(Qt.PointingHandCursor)
        elif self.eula_false.isChecked():
            self.next_button_3.setEnabled(False)
            self.next_button_3.setCursor(Qt.ForbiddenCursor)

    def open_link_in_browser(self, url):
        content = self.sender().toHtml()
        QDesktopServices.openUrl(url)
        self.sender().setHtml(content)

    def path_change(self):
        path = stellaris_path()
        if path:
            self.path_place.setPlainText(path)

    def path_check(self):
        if os.path.isfile(
                os.path.join(self.path_place.toPlainText(), "stellaris.exe")):
            return 1
        else:
            msg_box = QMessageBox()
            msg_box.setIcon(QMessageBox.Critical)
            msg_box.setWindowTitle('Ошибка!')
            msg_box.setText(
                'Исполняемый файл "stellaris.exe" не найден.\n\n\nВозможно, вы указали неверную директорию.')
            msg_box.addButton(QMessageBox.Ok)
            msg_box.exec_()
            return 0

    def browse_folder(self):
        directory = QFileDialog.getExistingDirectory(self, "Выберите папку Stellaris", self.path_place.toPlainText())
        if directory:
            self.path_place.setPlainText(directory)

    def download_file(self):
        if self.path_check():
            self.is_downloading = True
            if self.stackedWidget.currentIndex() != 4:
                self.stackedWidget.setCurrentIndex(self.stackedWidget.currentIndex() + 1)
            file_url = decrypt(url, 'LPrVJDjMXGx1ToihooozyFX4-toGjKcCr8pjZFmq62c=')
            # self.save_path = os.path.join(os.path.expanduser("~"), "Downloads", 'stellaris_unlocker.zip')
            self.save_path = os.path.join(self.path_place.toPlainText(), 'stellaris_unlocker.zip')
            try:
                if os.path.exists(self.save_path):
                    os.remove(self.save_path)
            except:
                pass

            self.download_thread = DownloaderThread(file_url, self.save_path)
            self.download_thread.progress_signal.connect(self.update_progress)
            self.download_thread.error_signal.connect(self.show_error)
            self.download_thread.speed_signal.connect(self.show_download_speed)

            self.creamapi_maker = CreamAPI()
            self.creamapi_maker.progress_signal.connect(self.update_creamapi_progress)
            self.creamapi_maker.dlc_signal.connect(self.show_dlc_get_message)

            self.download_thread.start()
            self.creamapi_maker.start()
            self.download_text.setText("Загрузка...")
            self.creamapi_label.setText(f"Получение инфо о dlc: подключение к api")

    def update_creamapi_progress(self, value):
        self.creamapi_progressBar_2.setValue(value)
        if value == 100:
            self.creamapi_label.setText("Готово!")
            self.download_complete()

    def show_dlc_get_message(self, dlc_name):
        self.creamapi_label.setText(f"Получение инфо о dlc: {dlc_name}")

    def update_progress(self, value):
        self.download_progressBar.setValue(value)
        if value == 100:
            self.download_text.setText("Готово!")
            self.speed_label.setText(f"")
            self.download_complete()

    def show_download_speed(self, speed):
        self.speed_label.setText(f"Скорость: {speed} Мб/с")

    def show_error(self, error_message):
        QMessageBox.warning(self, "Ошибка", "Ошибка загрузки файла\nСкорее всего в данный момент сервер не доступен")

    def download_complete(self):
        if self.download_progressBar.value() == 100 and self.creamapi_progressBar_2.value() == 100:
            self.next_button_5.setEnabled(True)
            self.cancel_button_5.setEnabled(True)

    @staticmethod
    def paradox_remove():
        user_home = os.path.expanduser("~")
        paradox_folder1 = os.path.join(user_home, "AppData", "Local", "Programs", "Paradox Interactive")
        paradox_folder2 = os.path.join(user_home, "AppData", "Local", "Paradox Interactive")
        paradox_folder3 = os.path.join(user_home, "AppData", "Roaming", "Paradox Interactive")
        paradox_folder4 = os.path.join(user_home, "AppData", "Roaming", "paradox-launcher-v2")
        try:
            if os.path.exists(paradox_folder1):
                rmtree(paradox_folder1)

            if os.path.exists(paradox_folder2):
                rmtree(paradox_folder2)

            if os.path.exists(paradox_folder3):
                rmtree(paradox_folder3)

            if os.path.exists(paradox_folder4):
                rmtree(paradox_folder4)
        except Exception:
            pass

    def reinstall(self, skip=False):
        self.stackedWidget.setCurrentIndex(6)
        if skip:
            user_home = os.path.expanduser("~")
            folder_path = os.path.join(user_home, "AppData", "Local", "Programs", "Paradox Interactive", "launcher")
            launcher_folders = []
            for item in os.listdir(folder_path):
                if item.startswith("launcher"):
                    launcher_folders.append(item)
            sleep(0.2)
            self.replace_files(os.path.join(os.path.join(folder_path, launcher_folders[1])))
        else:
            self.paradox_remove()
            if self.ok_dialog('Внимание',
                              "Сейчас будет открыт инсталятор лаунчера, пожалуйста выберете 'Remove', если будет предложено, либо просто продолжите установку",
                              QMessageBox.Information):
                process = Popen([f"{self.path_place.toPlainText()}/launcher-installer-windows.msi"],
                                shell=True)
                process.wait()

            user_home = os.path.expanduser("~")
            folder_path = os.path.join(user_home, "AppData", "Local", "Programs", "Paradox Interactive", "launcher")
            if not os.path.exists(folder_path):
                if self.ok_dialog('Внимание',
                                  "Сейчас будет открыт инсталятор лаунчера, пожалуйста выполните установку",
                                  QMessageBox.Information):
                    process = Popen([f"{self.path_place.toPlainText()}/launcher-installer-windows.msi"],
                                    shell=True)
                    process.wait()
                sleep(1)
            launcher_folders = []
            for item in os.listdir(folder_path):
                if item.startswith("launcher"):
                    launcher_folders.append(item)
            launcher_folder = os.path.join(os.path.join(folder_path, launcher_folders[0]))
            if self.ok_dialog('Внимание',
                              "Сейчас мы запустим лаунчер, но так как мы не можем отследить когда он обновится вам придется нам помочь, после его открытия нажмите 'SKIP' в правом верхем углу и дождитесь пока лаучнер скажет, что обновление готово, а затем просто закройте его\nУведомления в лаунчере отображаются в колокольчике в правом верхнем углу",
                              QMessageBox.Information):
                process = Popen([os.path.join(folder_path, launcher_folder, "Paradox Launcher.exe")])
                process.wait()
            sleep(1)
            launcher_folders = []
            for item in os.listdir(folder_path):
                if item.startswith("launcher"):
                    launcher_folders.append(item)
                    sleep(0.2)
                    self.switch_to_next()
            self.replace_files(os.path.join(os.path.join(folder_path, launcher_folders[1])))

    def replace_files(self, launcher_folder):
        rmtree(f'{self.path_place.toPlainText()}/dlc')
        self.unzip_and_replace()
        try:
            os.remove(f'{launcher_folder}/resources/app.asar.unpacked/dist/main/steam_api64_o.dll')
        except:
            pass
        os.rename(f'{launcher_folder}/resources/app.asar.unpacked/dist/main/steam_api64.dll',
                  f'{launcher_folder}/resources/app.asar.unpacked/dist/main/steam_api64_o.dll')
        copytree(f'{self.parent_directory}/creamapi_launcher_files',
                 f'{launcher_folder}/resources/app.asar.unpacked/dist/main',
                 dirs_exist_ok=True)
        copytree(f'{self.parent_directory}/creamapi_steam_files', self.path_place.toPlainText(), dirs_exist_ok=True)
        # move(f'{unzipped}/dlc', self.path_place.toPlainText())
        self.finish_text.setPlainText('Все готово!')
        sleep(1)
        self.finish_button.setEnabled(True)

    def unzip_and_replace(self):
        zip_path = self.save_path
        # extract_folder = os.path.splitext(zip_path)[0]
        extract_folder = self.path_place.toPlainText()
        if not os.path.exists(extract_folder):
            os.makedirs(extract_folder)

        with ZipFile(zip_path, 'r') as zip_ref:
            zip_ref.extractall(extract_folder)
        return extract_folder

    def finish(self):
        if self.launch_game.isChecked():
            try:
                run('start steam://run/281990', shell=True, capture_output=True, text=True)
                # Popen([f"{self.path_place.toPlainText()}/stellaris.exe"])
            except:
                pass

        try:
            if os.path.exists(self.save_path):
                os.remove(self.save_path)
        except:
            pass

        self.close()

    def server_msg(self):
        if server_msg:
            if self.ok_dialog('Важное сообщение',
                              f"Сообщение с сервера\n\n{server_msg}",
                              QMessageBox.Information):
                pass

    @staticmethod
    def ok_dialog(title, text, msg_type):
        msg_box = QMessageBox()
        msg_box.setIcon(msg_type)
        msg_box.setWindowTitle(title)
        msg_box.setText(text)
        ok_button = msg_box.addButton(QMessageBox.Ok)
        msg_box.exec_()
        return msg_box.clickedButton() == ok_button