Files
OpenCore-Legacy-Patcher/resources/wx_gui/gui_entry.py
2023-11-01 21:55:29 -06:00

95 lines
2.7 KiB
Python

# Entry point for the wxPython GUI
import wx
import sys
import atexit
import logging
from resources import constants
from resources.wx_gui import (
gui_cache_os_update,
gui_main_menu,
gui_build,
gui_install_oc,
gui_sys_patch_start,
gui_update,
)
from resources.sys_patch import sys_patch_detect
class SupportedEntryPoints:
"""
Enum for supported entry points
"""
MAIN_MENU = gui_main_menu.MainFrame
BUILD_OC = gui_build.BuildFrame
INSTALL_OC = gui_install_oc.InstallOCFrame
SYS_PATCH = gui_sys_patch_start.SysPatchStartFrame
UPDATE_APP = gui_update.UpdateFrame
OS_CACHE = gui_cache_os_update.OSUpdateFrame
class EntryPoint:
def __init__(self, global_constants: constants.Constants) -> None:
self.app: wx.App = None
self.main_menu_frame: gui_main_menu.MainFrame = None
self.constants: constants.Constants = global_constants
self.constants.gui_mode = True
def _generate_base_data(self) -> None:
self.app = wx.App()
self.app.SetAppName(self.constants.patcher_name)
def start(self, entry: SupportedEntryPoints = gui_main_menu.MainFrame) -> None:
"""
Launches entry point for the wxPython GUI
"""
self._generate_base_data()
if "--gui_patch" in sys.argv or "--gui_unpatch" in sys.argv:
entry = gui_sys_patch_start.SysPatchStartFrame
patches = sys_patch_detect.DetectRootPatch(self.constants.computer.real_model, self.constants).detect_patch_set()
logging.info(f"Entry point set: {entry.__name__}")
# Normally set by main.py, but transitions from CLI mode may not have this set
self.constants.gui_mode = True
self.frame: wx.Frame = entry(
None,
title=f"{self.constants.patcher_name} {self.constants.patcher_version}{' (Nightly)' if not self.constants.commit_info[0].startswith('refs/tags') else ''}",
global_constants=self.constants,
screen_location=None,
**({"patches": patches} if "--gui_patch" in sys.argv or "--gui_unpatch" in sys.argv else {})
)
atexit.register(self.OnCloseFrame)
if "--gui_patch" in sys.argv:
self.frame.start_root_patching()
elif "--gui_unpatch" in sys.argv:
self.frame.revert_root_patching()
self.app.MainLoop()
def OnCloseFrame(self, event: wx.Event = None) -> None:
"""
Closes the wxPython GUI
"""
if not self.frame:
return
logging.info("Cleaning up wxPython GUI")
self.frame.SetTransparent(0)
wx.Yield()
self.frame.DestroyChildren()
self.frame.Destroy()
self.app.ExitMainLoop()