add update button and self updating

Author: mastercoms

ThirdPartyLicenses.txt

@@ -0,0 +1 @@
+Icons by css.gg: https://github.com/astrit/css.gg/blob/main/LICENSE.md

gui/index.html

@@ -85,7 +85,6 @@
 
         .btn-secondary {
             background-color: rgb(97, 97, 97);
-            color: white;
         }
 
         .btn-secondary:hover:not(:disabled) {
@@ -141,6 +140,16 @@
             <div class="game-menu">
                 <span class="info-label" style="display: none">Offline mode</span>
                 <button id="launch-btn" class="btn btn-primary btn-lg" disabled>Updating...</button>
+                <button id="update-btn" class="btn btn-secondary btn-lg" title="Check for updates" disabled>
+                    <svg width="24" height="20" viewBox="0 0 24 20" fill="none" xmlns="http://www.w3.org/2000/svg">
+                        <path
+                            d="M4.56079 10.6418L6.35394 3.94971L8.25402 5.84979C11.7312 3.6588 16.3814 4.07764 19.41 7.1063L17.9958 8.52052C15.7536 6.27827 12.3686 5.87519 9.71551 7.31128L11.2529 8.84869L4.56079 10.6418Z"
+                            fill="currentColor" />
+                        <path
+                            d="M19.4392 13.3581L17.646 20.0502L15.7459 18.1501C12.2688 20.3411 7.61857 19.9223 4.58991 16.8936L6.00413 15.4794C8.24638 17.7217 11.6313 18.1247 14.2844 16.6887L12.747 15.1512L19.4392 13.3581Z"
+                            fill="currentColor" />
+                    </svg>
+                </button>
                 <button id="settings-btn" class="btn btn-secondary btn-lg" title="Settings">
                     <svg width="24" height="20" viewBox="0 0 24 20" fill="none" xmlns="http://www.w3.org/2000/svg">
                         <path fill-rule="evenodd" clip-rule="evenodd"
@@ -179,17 +188,35 @@ <h2>Install Folder</h2>
         document.getElementById("folder-btn").addEventListener("click", () => {
             window.pywebview.api.open_install_folder();
         });
+        document.getElementById("update-btn").addEventListener("click", () => {
+            if (document.getElementById("update-btn").disabled) {
+                return;
+            }
+            document.getElementById("update-btn").disabled = true;
+            document.getElementById("launch-btn").innerText = "Updating...";
+            document.getElementById("launch-btn").disabled = true;
+            document.getElementById("folder-btn").disabled = true;
+            window.pywebview.api.check_for_updates();
+        });
         function archiveReady(offline) {
-            document.getElementById("launch-btn").disabled = false;
-            document.getElementById("folder-btn").disabled = false;
-            document.getElementById("launch-btn").innerText = "Launch";
+            const available = offline <= 1;
+            document.getElementById("update-btn").disabled = false;
+            if (available) {
+                document.getElementById("launch-btn").disabled = false;
+                document.getElementById("folder-btn").disabled = false;
+                document.getElementById("launch-btn").innerText = "Launch";
+            } else {
+                document.getElementById("launch-btn").disabled = true;
+                document.getElementById("folder-btn").disabled = true;
+            }
             const online = offline === 0;
-            if (!online) {
-                document.querySelector(".info-label").innerText = "Offline mode";
-                if (offline !== 1) {
+            if (online) {
+                document.querySelector(".info-label").style.display = "none";
+            } else {
+                document.querySelector(".info-label").style.display = "inline";
+                if (offline > 1) {
                     document.getElementById("launch-btn").innerText = "Install failed";
                 }
-                document.querySelector(".info-label").style.display = "inline";
             }
         }
     </script>

tc2_launcher/__main__.py

@@ -1,5 +1,6 @@
 import argparse
 import multiprocessing
+import time
 from pathlib import Path
 import sys
 
@@ -8,11 +9,36 @@
 from tc2_launcher.run import launch_game
 from tc2_launcher.run import set_launch_options
 from tc2_launcher.run import default_dest_dir
+from tc2_launcher.run import update_self
 
 version = "0.1.0"
 
 
 def main():
+    launch_gui = False
+    if len(sys.argv) >= 3 and sys.argv[1] == "--replace":
+        # Replacement mode after self-update
+        original_path = Path(sys.argv[2]).resolve()
+        current_path = Path(sys.argv[0]).resolve()
+        try:
+            # Wait a moment to ensure the original process has exited
+            time.sleep(2)
+            # Replace the original file with the current file
+            original_path.unlink()
+            current_path.replace(original_path)
+            print("Self-update applied successfully.")
+        except Exception as e:
+            print(f"ERROR: Failed to apply self-update: {e}")
+
+        launch_gui = len(sys.argv) == 3
+    else:
+        if update_self(version):
+            sys.exit(0)
+            return
+
+        launch_gui = len(sys.argv) == 1
+
+
     parser = argparse.ArgumentParser(description=f"TC2 Launcher v{version}")
     parser.add_argument(
         "--dest",
@@ -44,8 +70,8 @@ def main():
         nargs=argparse.REMAINDER,
         help="User launch options",
     )
-
-    if len(sys.argv) == 1:
+    
+    if launch_gui:
         start_gui()
         return
 

tc2_launcher/gui.py

@@ -24,6 +24,10 @@ def set_launch_options(self, options):
     def open_install_folder(self):
         open_install_folder()
 
+    def check_for_updates(self):
+        res = update_archive()
+        webview.windows[0].evaluate_js(f"archiveReady({res});")
+
 
 def get_entrypoint():
     def exists(path):

tc2_launcher/run.py

@@ -1,6 +1,7 @@
 import os
 import sys
 import json
+import tempfile
 from pathlib import Path
 from typing import Optional, Tuple
 from shutil import rmtree
@@ -57,8 +58,51 @@ def _find_asset(release: dict, asset_filter: str) -> Optional[Tuple[str, str]]:
     return None
 
 
-def update_self():
-    pass
+def update_self(current_version: str) -> bool:
+    try:
+        release = _get_latest_release(LAUNCHER_REPO)
+        tag = release.get("tag_name", "").lstrip("v")
+    except Exception as e:
+        print(f"ERROR: Failed to get self-update info: {e}")
+        return False
+
+    if tag == current_version:
+        return False
+    
+    print(f"Self-update available: {current_version} -> {tag}")
+
+    if os.name == "nt":
+        asset_filter = ".exe"
+    else:
+        asset_filter = "-linux"
+    asset = _find_asset(release, asset_filter)
+    if not asset:
+        print(f"ERROR: No asset matching '{asset_filter}' found in self-update release {tag}.")
+        return False
+    
+    asset_name, download_url = asset
+
+    # Step 4: Download asset to temp file
+    temp_dir = Path(tempfile.gettempdir())
+    temp_path = temp_dir / f"TC2LauncherUpdate_{tag}" / asset_name
+    print(f"Downloading self-update to {temp_path}...")
+    try:
+        _download(download_url, temp_path)
+    except Exception as e:
+        print(f"ERROR: Failed to download self-update: {e}")
+        return False
+    print("Self-update download complete.")
+
+    current_path = Path(sys.argv[0]).resolve()
+    print("Launching self-update...")
+    try:
+        filtered_args = [arg for arg in sys.argv[1:] if arg != "--replace"]
+        run_non_blocking([temp_path, "--replace", current_path] + filtered_args)
+        sys.exit(0)
+        return True
+    except Exception as e:
+        print(f"Failed to launch self-update: {e}")
+        return False
 
 
 def _read_data(path: Path) -> dict:
@@ -144,7 +188,7 @@ def update_archive(
 
     state = read_state(dest)
     current_tag = state.get("tag")
-    if not force and current_tag == tag:
+    if not force and current_tag == tag and fail_code == 1:
         print("Latest asset already downloaded.")
         return 0
 
@@ -153,7 +197,6 @@ def update_archive(
     else:
         asset_filter = "game-linux.zip"
     asset = _find_asset(release, asset_filter)
-
     if not asset:
         print(f"ERROR: No asset matching '{asset_filter}' found in release {tag}.")
         return fail_code
@@ -168,7 +211,7 @@ def update_archive(
     asset_path = dest / asset_name
     game_dir = dest / "game"
 
-    print("Downloading latest asset...")
+    print(f"Downloading latest asset to {asset_path}...")
     try:
         _download(download_url, asset_path)
     except Exception as e: