main.py

import os
import sys
import ctypes
import platform
import json
import winreg
import winapi
import installer
import uninstaller
import logger
from PyQt5 import QtWidgets


class App:
    def __init__(self) -> None:
        self.encoding = sys.getdefaultencoding()
        self.cwd = os.path.dirname(__file__) or os.getcwd()
        os.chdir(self.cwd)
        self.files_dir = os.path.join(self.cwd, 'files')
        self.temp_dir = os.getenv('temp')
        self.exec_fn = os.path.basename(sys.executable)
        self.is_compiled = not self.exec_fn.lower().split('.')[0] == 'python'
        self.spawn_args = [sys.executable] if self.is_compiled else [sys.executable, __file__]
        self.spawn_str = '"' + '"'.join(self.spawn_args) + '"'
        self.exec_script = self.spawn_args[-1]
        self.locale = {}
        self.is_dark = False
        self.reg_path = 'Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\GDLocalisation'
        self.logger = logger.Logger(self)
        self.logger.log('GDL Log')
        self.logger.log('CWD', self.cwd)
        self.logger.log('Executable', sys.executable)
        self.logger.log('Python', sys.version)
        self.logger.log('Version', sys.version_info)
        self.logger.log('Encoding', self.encoding)
        self.logger.log('Platform', platform.platform())
        self.logger.log('Files dir', self.files_dir)
        self.logger.log('Temp dir', self.temp_dir)
        self.logger.log('REG Path', self.reg_path)
        self.logger.log('Spawn args', self.spawn_args)
        self.child_app = None
        self.main()

    def check_dark_theme(self) -> None:
        if os.getenv('GDL_ENABLE_DARK_THEME'):
            self.is_dark = os.environ['GDL_ENABLE_DARK_THEME'] == '1'
            return
        if not winapi.ShouldUseDarkMode:
            return self.check_dark_theme_reg()
        self.is_dark = bool(winapi.ShouldUseDarkMode())

    def check_dark_theme_reg(self) -> None:
        try:
            key = winreg.OpenKeyEx(
                winreg.HKEY_CURRENT_USER, 'Software\\Microsoft\\Windows\\CurrentVersion\\Themes\\Personalize'
            )
        except FileNotFoundError:
            return
        try:
            self.is_dark = not winreg.QueryValueEx(key, 'AppsUseLightTheme')[0]
        except (FileNotFoundError, TypeError, IndexError, AttributeError):
            pass
        winreg.CloseKey(key)

    def apply_dark(self, hwnd: int) -> None:
        if not hwnd or not winapi.DwmSetWindowAttribute:
            return
        self.logger.log(
            f'Dark Theme 1 for hwnd {hwnd}', not winapi.DwmSetWindowAttribute(hwnd, 19, ctypes.c_buffer(b'\0\0\0\1'), 4)
        )
        self.logger.log(
            f'Dark Theme 2 for hwnd {hwnd}', not winapi.DwmSetWindowAttribute(hwnd, 20, ctypes.c_buffer(b'\0\0\0\1'), 4)
        )

    def read_binary(self, fn: str) -> bytes:
        self.logger.log('Reading binary file', fn)
        f = open(fn, 'rb')
        result = f.read()
        f.close()
        return result

    def read_text(self, fn: str) -> str:
        self.logger.log('Reading text file', fn)
        f = open(fn, 'r', encoding=self.encoding)
        result = f.read()
        f.close()
        return result

    def write_text(self, fn: str, content: str) -> int:
        self.logger.log('Writing text file', fn)
        f = open(fn, 'w', encoding=self.encoding)
        result = f.write(content)
        f.close()
        return result

    def write_binary(self, fn: str, content: bytes) -> int:
        self.logger.log('Writing binary file', fn)
        f = open(fn, 'wb')
        result = f.write(content)
        f.close()
        return result

    def round_point(self, number: float, count: int = 2) -> None:
        self.logger.log('Rounding point', number, count)
        count_10 = 10 ** count
        return round(number * count_10) / count_10

    def show_error(self, window: any, caption: str, text: str, cb: any = None) -> QtWidgets.QMessageBox:
        self.logger.log('Showing error', caption, text)
        box = QtWidgets.QMessageBox(window)
        if self.is_dark:
            self.apply_dark(int(box.winId()))
        box.setIcon(box.Icon.Critical)
        box.setWindowTitle(caption)
        box.setText(text)
        box.addButton(self.locale['data'][44], box.ActionRole).clicked.connect(cb or box.hide)  # noqa
        box.show()
        return box

    def show_question(
            self, window: any, caption: str, text: str, yes_cb: any, no_cb: any = None
    ) -> QtWidgets.QMessageBox:
        self.logger.log('Showing question', caption, text)
        box = QtWidgets.QMessageBox(window)
        if self.is_dark:
            self.apply_dark(int(box.winId()))
        box.setIcon(box.Icon.Question)
        box.setWindowTitle(caption)
        box.setText(text)
        box.addButton(self.locale['data'][45], box.ActionRole).clicked.connect(yes_cb)  # noqa
        box.addButton(self.locale['data'][46], box.ActionRole).clicked.connect(no_cb or box.hide)  # noqa
        box.show()
        return box

    def run_installer(self, json_data: dict = None) -> None:
        self.logger.log('Running installer')
        self.child_app = installer.Installer(self, json_data)
        self.logger.destroy()
        sys.exit(self.child_app.exit_code)

    def run_uninstaller(self, json_data: dict = None) -> None:
        self.logger.log('Running uninstaller')
        self.child_app = uninstaller.Uninstaller(self, json_data)
        self.logger.destroy()
        sys.exit(self.child_app.exit_code)

    def is_gd_path(self, path: str) -> bool:
        counter = 0
        try:
            for fn in os.listdir(path):
                if fn.lower() in ('fmod.dll', 'libcocos2d.dll', 'libextensions.dll', 'sqlite3.dll', 'glew32.dll'):
                    counter += 1
        except Exception as _err:
            self.logger.error(f'Failed to check is path {path} for gd {_err}')
        return counter > 3

    def main(self) -> None:
        settings_path = os.path.join(os.path.dirname(self.spawn_args[-1]), 'gdl-installer.json')
        self.logger.log('Settings path', settings_path)
        if not os.path.isfile(settings_path):
            return self.run_installer()
        json_data = json.loads(self.read_text(settings_path))
        self.locale = json.loads(self.read_text(os.path.join(self.files_dir, f'locale_{json_data["locale"]}.json')))
        if sys.argv[-1] == '--modify':
            return self.run_installer(json_data)
        if sys.argv[-1] == '--remove':
            return self.run_uninstaller(json_data)
        self.logger.log('Showing msg about install')
        msg_result = winapi.MessageBoxW(
            0,
            self.locale['data'][42],
            self.locale['data'][43],
            0x00000003 | 0x00000020
        )
        if msg_result == 6:
            return self.run_installer(json_data)
        if msg_result == 7:
            return self.run_uninstaller(json_data)


if __name__ == '__main__':
    App()