Auto updater system

compile.py

@@ -1,4 +1,5 @@
 import os
+import sys
 import platform
 import subprocess
 import plistlib
@@ -25,7 +26,7 @@ def build():
     icon = "data/icon-win.png" if system == "Windows" else "data/icon.png"
 
     args = [
-        "pyinstaller",
+        sys.executable, "-m", "PyInstaller",
         f"--name={name}",
         "--onefile",
         "--clean",

ghost.py

@@ -15,7 +15,7 @@
 
 from utils.files import get_application_support
 from utils.config import Config
-from utils import startup_check, check_fonts, console, load_fonts, is_admin, run_elevated, relaunch_normal, send_telemetry_ping
+from utils import startup_check, check_fonts, console, load_fonts, is_admin, run_elevated, relaunch_normal, send_telemetry_ping, get_update_info
 from bot.controller import BotController
 
 
@@ -34,13 +34,13 @@ def parse_args():
     )
     return parser.parse_args()
 
-def run_gui():
+def run_gui(update_info=None):
     from gui.main import GhostGUI
 
     cfg = Config()
     controller = BotController()
 
-    GhostGUI(controller).run()
+    GhostGUI(controller, update_info=update_info).run()
 
 
 def run_cli():
@@ -70,6 +70,7 @@ def main():
     headless = args.headless
 
     get_application_support()
+    update_info = get_update_info()
     startup_check.check()
     cfg = Config()
     cfg.check()
@@ -82,11 +83,15 @@ def main():
 
     console.info("Running in GUI mode.")
 
+    if update_info and update_info.has_update:
+        run_gui(update_info=update_info)
+        return
+
     if cfg.get_skip_fonts():
         cfg.set_skip_fonts(False)
-        run_gui()
+        run_gui(update_info=update_info)
     elif check_fonts():
-        run_gui()
+        run_gui(update_info=update_info)
     elif args.install_fonts:
         load_fonts()
         if check_fonts():
@@ -96,7 +101,7 @@ def main():
             run_elevated()
             return
         load_fonts()
-        run_gui()
+        run_gui(update_info=update_info)
 
 if __name__ == "__main__":
     main()

gui/main.py

@@ -11,16 +11,18 @@
 from utils.config import Config
 import utils.console as logging
 from utils.files import resource_path
+from utils import UpdateInfo
 
-from gui.pages import HomePage, LoadingPage, SettingsPage, OnboardingPage, ScriptsPage, ToolsPage
+from gui.pages import HomePage, LoadingPage, SettingsPage, OnboardingPage, ScriptsPage, ToolsPage, UpdatePage
 from gui.components import Sidebar, Console, Titlebar, RoundedFrame
 from gui.helpers import Images, Layout, Style, apply_theme
 
 class GhostGUI:
-    def __init__(self, bot_controller):
+    def __init__(self, bot_controller, update_info=None):
         self.resize_grip_size = 5
         self.size = (750, 530)
         self.bot_controller = bot_controller
+        self.update_info = update_info
         self.resize_grips = {}
 
         self.root = ttk.tk.Tk()
@@ -84,6 +86,7 @@ def __init__(self, bot_controller):
         self.settings_page   = SettingsPage(self.root, self.bot_controller, self.draw_settings)
         self.scripts_page    = ScriptsPage(self, self.bot_controller, self.images)
         self.tools_page      = ToolsPage(self.root, self.bot_controller, self.images, self.layout, self._position_resize_grips)
+        self.update_page     = UpdatePage(self.root, self)
 
         logging.set_gui(self)
 
@@ -249,6 +252,33 @@ def draw_onboarding(self):
         main = self.layout.main(scrollable=False)
         self.onboarding_page.draw(main)
         self.root.after(150, self._position_resize_grips)
+
+    def draw_update(self):
+        self.sidebar.set_current_page("onboarding")
+        self.layout.clear()
+        main = self.layout.main(scrollable=False)
+        self.update_page.draw(main)
+        self.root.after(150, self._position_resize_grips)
+
+    def _continue_after_update_prompt(self):
+        self.update_info = None
+
+        if self.cfg.get("token") == "":
+            self.draw_onboarding()
+            return
+
+        if not self.bot_controller.running:
+            self.bot_controller.start()
+
+        self.root.after(25, self.sidebar.disable)
+        self.draw_home(start=True)
+        self.root.after(100, self._pre_load_images)
+
+        if sys.platform == "win32":
+            self.root.after(50, lambda: hPyT.window_frame.restore(self.root))
+            self.root.after(75, lambda: hPyT.window_frame.center(self.root))
+
+        self.root.after(100, self._check_bot_started)
 
     # def draw_console(self):
     #     self.sidebar.set_current_page("console")
@@ -328,29 +358,13 @@ def run(self, preserve_position=False):
         if not preserve_position:
             self.layout.center_window(self.size[0], self.size[1])
 
-        if self.cfg.get("token") == "":
-        # if True:
-            self.draw_onboarding()
+        if self.update_info and self.update_info.has_update:
+            self.layout.resize(self.size[0], self.size[1] + 50)
+            self.layout.center_window(self.size[0], self.size[1] + 50)
+            self.draw_update()
             self.root.mainloop()
             return
-
-        if not self.bot_controller.running:
-            self.bot_controller.start()
-
-        # self.layout.hide_titlebar()
-        # self.layout.stick_window()
-        # self.layout.resize(400, 90)
-        # self.layout.center_window(400, 90)
-        # self.loading_page.draw()
-        self.root.after(25, self.sidebar.disable)
-        self.draw_home(start=True)
-        self.root.after(100, self._pre_load_images)
-
-        if sys.platform == "win32":
-            self.root.after(50, lambda: hPyT.window_frame.restore(self.root))
-            self.root.after(75, lambda: hPyT.window_frame.center(self.root))
-
-        self.root.after(100, self._check_bot_started)
+        self._continue_after_update_prompt()
         self.root.mainloop()
 
     def quit(self):

gui/pages/__init__.py

@@ -4,4 +4,5 @@
 from .onboarding import OnboardingPage
 from .scripts import ScriptsPage
 from .script import ScriptPage
-from .tools import ToolsPage
+from .tools import ToolsPage
+from .update import UpdatePage

gui/pages/update.py

@@ -0,0 +1,214 @@
+import re
+import threading
+import webbrowser
+
+import ttkbootstrap as ttk
+
+from gui.components import RoundedFrame, RoundedButton
+from gui.helpers import Style
+
+class UpdatePage:
+    def __init__(self, root, master):
+        self.root = root
+        self.master = master
+        self.installing = False
+
+    def _build_markdown_tags(self, text_widget):
+        base_font = ("Host Grotesk", 12)
+        text_widget.tag_configure("heading1", font=("Host Grotesk", 16, "bold"), spacing1=10, spacing3=8)
+        text_widget.tag_configure("heading2", font=("Host Grotesk", 14, "bold"), spacing1=8, spacing3=6)
+        text_widget.tag_configure("heading3", font=("Host Grotesk", 13, "bold"), spacing1=6, spacing3=4)
+        text_widget.tag_configure("bold", font=("Host Grotesk", 12, "bold"))
+        text_widget.tag_configure("italic", font=("Host Grotesk", 12, "italic"))
+        text_widget.tag_configure("underline", font=("Host Grotesk", 12, "underline"))
+        text_widget.tag_configure("inline_code", font=("JetBrains Mono", 11), background=self.root.style.colors.get("secondary"), foreground=self.root.style.colors.get("text"))
+        text_widget.tag_configure("blockquote", lmargin1=0, lmargin2=0, foreground="#b3b3b3")
+        text_widget.tag_configure("bullet", lmargin1=18, lmargin2=36)
+        text_widget.tag_configure("plain", font=base_font)
+        text_widget.tag_configure("warning", font=("Host Grotesk", 13, "bold"), foreground="#facc15", spacing1=6, spacing3=4)
+        text_widget.tag_configure("note", font=("Host Grotesk", 13, "bold"), foreground="#158bfa", spacing1=6, spacing3=4)
+
+    def _insert_inline_markdown(self, text_widget, line, base_tags=()):
+        patterns = [
+            (r"<ins>(.*?)</ins>", "underline"),
+            (r"`([^`]+)`", "inline_code"),
+            (r"\*\*([^*]+)\*\*", "bold"),
+            (r"(?<!\*)\*([^*]+)\*(?!\*)", "italic"),
+            (r"_([^_]+)_", "italic"),
+            (r"\[([^\]]+)\]\(([^)]+)\)", "link"),
+        ]
+
+        matches = []
+        for pattern, tag in patterns:
+            for match in re.finditer(pattern, line):
+                matches.append((match.start(), match.end(), tag, match))
+
+        matches.sort(key=lambda item: item[0])
+
+        index = 0
+        while index < len(line):
+            overlapping = next((item for item in matches if item[0] == index), None)
+            if overlapping:
+                _, end, tag, match = overlapping
+                if tag == "link":
+                    display = match.group(1)
+                    text_widget.insert("end", display, base_tags + ("link",))
+                else:
+                    text_widget.insert("end", match.group(1), base_tags + (tag,))
+                index = end
+                continue
+
+            text_widget.insert("end", line[index], base_tags)
+            index += 1
+
+    def _render_markdown(self, text_widget, markdown_text):
+        self._build_markdown_tags(text_widget)
+        text_widget.tag_configure("link", foreground="#8b5cf6", underline=True)
+
+        in_code_block = False
+        for raw_line in markdown_text.splitlines():
+            line = raw_line.rstrip()
+
+            if line.strip().startswith("```"):
+                in_code_block = not in_code_block
+                if not in_code_block:
+                    text_widget.insert("end", "\n")
+                continue
+
+            if in_code_block:
+                text_widget.insert("end", raw_line + "\n", ("inline_code",))
+                continue
+
+            stripped = line.strip()
+            if not stripped:
+                text_widget.insert("end", "\n")
+                continue
+
+            if stripped.startswith("# "):
+                text_widget.insert("end", stripped[2:] + "\n", ("heading1",))
+                continue
+            if stripped.startswith("## "):
+                text_widget.insert("end", stripped[3:] + "\n", ("heading2",))
+                continue
+            if stripped.startswith("### "):
+                text_widget.insert("end", stripped[4:] + "\n", ("heading3",))
+                continue
+            admonition = re.match(r"^>\s*(WARNING|NOTE)\s*$", stripped)
+            if admonition:
+                tag = "warning" if admonition.group(1) == "WARNING" else "note"
+                text_widget.insert("end", admonition.group(1) + "\n", (tag,))
+                continue
+            if stripped.startswith(("- ", "* ")):
+                prefix = "• " if stripped.startswith(("- ", "* ")) else ""
+                text_widget.insert("end", prefix)
+                self._insert_inline_markdown(text_widget, stripped[2:])
+                text_widget.insert("end", "\n", ("bullet",))
+                continue
+            if stripped.startswith(">"):
+                blockquote_content = stripped[1:].strip()
+                self._insert_inline_markdown(text_widget, blockquote_content, ("blockquote",))
+                text_widget.insert("end", "\n", ("blockquote",))
+                continue
+
+            self._insert_inline_markdown(text_widget, stripped)
+            text_widget.insert("end", "\n", ("plain",))
+
+    def _set_install_progress(self, status, progress):
+        if not self.installing:
+            return
+
+        self.install_status.configure(text=status)
+        if progress is None:
+            if self.install_progress.cget("mode") != "indeterminate":
+                self.install_progress.configure(mode="indeterminate")
+                self.install_progress.start(12)
+            return
+
+        if self.install_progress.cget("mode") == "indeterminate":
+            self.install_progress.stop()
+            self.install_progress.configure(mode="determinate")
+        self.install_progress.configure(value=progress)
+
+    def _install_update(self, update_info):
+        self.installing = True
+        self.install_button.set_state("disabled")
+        self.skip_button.set_state("disabled")
+        self._set_install_progress("Starting update", 0)
+
+        def report(status, progress):
+            self.root.after(0, self._set_install_progress, status, progress)
+
+        def install():
+            try:
+                installed = update_info.install(progress_callback=report, exit_on_success=False)
+            except Exception as exc:
+                self.root.after(0, self._update_failed, str(exc))
+                return
+
+            if installed:
+                self.root.after(0, self._finish_install)
+            else:
+                self.root.after(0, self._update_failed, "Could not install the update. Check the console for details.")
+
+        threading.Thread(target=install, daemon=True).start()
+
+    def _finish_install(self):
+        self.install_status.configure(text="Closing Ghost")
+        self.root.after(100, self.master.quit)
+
+    def _update_failed(self, message):
+        self.installing = False
+        self.install_progress.stop()
+        self.install_progress.configure(mode="determinate", value=0)
+        self.install_status.configure(text=message)
+        self.install_button.set_state("normal")
+        self.skip_button.set_state("normal")
+
+    def draw(self, wrapper):
+        update_info = getattr(self.master, "update_info", None)
+
+        ttk.Label(wrapper, text="New Update Available", font=("Host Grotesk", 24, "bold")).pack(anchor="w", padx=24, pady=(24, 10))
+
+        version_text = "A new version is ready to install."
+        if update_info:
+            version_text = f"Version {update_info.latest_version} is ready to install."
+
+        ttk.Label(wrapper, text=version_text, wraplength=500, justify="left").pack(anchor="w", padx=24)
+
+        card = RoundedFrame(wrapper, radius=(18, 18, 0, 0), background=self.root.style.colors.get("secondary"))
+        card.pack(fill=ttk.BOTH, expand=True, padx=24, pady=24)
+
+        changelog_label = ttk.Label(card, text="Changelog", font=("Host Grotesk", 14, "bold"))
+        changelog_label.configure(background=self.root.style.colors.get("secondary"))
+        changelog_label.pack(anchor="w", pady=(10, 10), padx=10)
+
+        if update_info and update_info.changelog:
+            changelog_frame = RoundedFrame(card, radius=(0, 0, 18, 18), background=self.root.style.colors.get("dark"), parent_background=self.root.style.colors.get("bg"))
+            changelog_frame.pack(fill=ttk.BOTH, expand=True)
+
+            changelog = ttk.tk.Text(changelog_frame, height=7, wrap="word", borderwidth=0, highlightthickness=0)
+            self._render_markdown(changelog, update_info.changelog.strip())
+            changelog.configure(state="disabled", background=self.root.style.colors.get("dark"), foreground=self.root.style.colors.get("text"), borderwidth=0, highlightthickness=0, font=("Host Grotesk", 12))
+            changelog.pack(fill=ttk.BOTH, expand=True, padx=10, pady=10)
+
+        button_row = ttk.Frame(wrapper)
+        button_row.pack(fill=ttk.X, padx=24, pady=(8, 24))
+
+        self.install_status = ttk.Label(button_row, text="", font=("Host Grotesk", 11))
+        self.install_status.pack(side=ttk.TOP, anchor="w", pady=(0, 6))
+        self.install_progress = ttk.Progressbar(button_row, mode="determinate", maximum=100, value=0, bootstyle="primary")
+        self.install_progress.pack(side=ttk.TOP, fill=ttk.X, pady=(0, 12))
+
+        full_changelog_button = RoundedButton(
+            button_row,
+            text="Full Changelog",
+            command=lambda e: webbrowser.open(update_info.full_changelog_url) if update_info else None,
+            bootstyle="secondary.TButton",
+        )
+        full_changelog_button.pack(side=ttk.LEFT)
+
+        self.install_button = RoundedButton(button_row, text="Install Update", command=lambda e: self._install_update(update_info) if update_info else None, bootstyle="primary.TButton")
+        self.install_button.pack(side=ttk.RIGHT)
+
+        self.skip_button = RoundedButton(button_row, text="Skip", command=lambda e: self.master._continue_after_update_prompt(), bootstyle="secondary.TButton")
+        self.skip_button.pack(side=ttk.RIGHT, padx=(0, 10))