The Messenger: content update

worlds/messenger/client_setup.py

@@ -0,0 +1,151 @@
+import io
+import logging
+import os.path
+import subprocess
+import urllib.request
+from pathlib import Path
+from shutil import which
+from tkinter.messagebox import askyesnocancel
+from typing import Any, Optional
+from zipfile import ZipFile
+from Utils import open_file
+
+import requests
+
+from Utils import is_linux, is_windows, messagebox, tuplize_version
+
+
+def launch_game(url: Optional[str] = None) -> None:
+    """Check the game installation, then launch it"""
+    if not (is_linux or is_windows):
+        return
+
+    def courier_installed() -> bool:
+        """Check if Courier is installed"""
+        return os.path.exists(os.path.join(folder, "TheMessenger_Data", "Managed", "Assembly-CSharp.Courier.mm.dll"))
+
+    def mod_installed() -> bool:
+        """Check if the mod is installed"""
+        return os.path.exists(os.path.join(folder, "Mods", "TheMessengerRandomizerAP", "courier.toml"))
+
+    def request_data(request_url: str) -> Any:
+        """Fetches json response from given url"""
+        logging.info(f"requesting {request_url}")
+        response = requests.get(request_url)
+        if response.status_code == 200:  # success
+            try:
+                data = response.json()
+            except requests.exceptions.JSONDecodeError:
+                raise RuntimeError(f"Unable to fetch data. (status code {response.status_code})")
+        else:
+            raise RuntimeError(f"Unable to fetch data. (status code {response.status_code})")
+        return data
+
+    def install_courier() -> None:
+        """Installs latest version of Courier"""
+        # can't use latest since courier uses pre-release tags
+        courier_url = "https://api.github.com/repos/Brokemia/Courier/releases"
+        latest_download = request_data(courier_url)[0]["assets"][-1]["browser_download_url"]
+
+        with urllib.request.urlopen(latest_download) as download:
+            with ZipFile(io.BytesIO(download.read()), "r") as zf:
+                for member in zf.infolist():
+                    zf.extract(member, path=folder)
+
+        working_directory = os.getcwd()
+        os.chdir(folder)
+        # linux handling
+        if is_linux:
+            mono_exe = which("mono")
+            if not mono_exe:
+                # download and use mono kickstart
+                # this allows steam deck support
+                mono_kick_url = "https://github.com/flibitijibibo/MonoKickstart/archive/refs/heads/master.zip"
+                target = os.path.join(folder, "monoKickstart")
+                os.makedirs(target, exist_ok=True)
+                with urllib.request.urlopen(mono_kick_url) as download:
+                    with ZipFile(io.BytesIO(download.read()), "r") as zf:
+                        for member in zf.infolist():
+                            zf.extract(member, path=target)
+                installer = subprocess.Popen([os.path.join(target, "precompiled"),
+                                              os.path.join(folder, "MiniInstaller.exe")], shell=False)
+                os.remove(target)
+            else:
+                installer = subprocess.Popen([mono_exe, os.path.join(folder, "MiniInstaller.exe")], shell=False)
+        else:
+            installer = subprocess.Popen(os.path.join(folder, "MiniInstaller.exe"), shell=False)
+
+        failure = installer.wait()
+        if failure:
+            messagebox("Failure", "Failed to install Courier", True)
+            os.chdir(working_directory)
+            raise RuntimeError("Failed to install Courier")
+        os.chdir(working_directory)
+
+        if courier_installed():
+            messagebox("Success!", "Courier successfully installed!")
+            return
+        messagebox("Failure", "Failed to install Courier", True)
+        raise RuntimeError("Failed to install Courier")
+
+    def install_mod() -> None:
+        """Installs latest version of the mod"""
+        get_url = "https://api.github.com/repos/alwaysintreble/TheMessengerRandomizerModAP/releases/latest"
+        release_url = request_data(get_url)["assets"][0]["browser_download_url"]
+
+        mod_folder = os.path.join(folder, "Mods")
+        os.makedirs(mod_folder, exist_ok=True)
+        with urllib.request.urlopen(release_url) as download:
+            with ZipFile(io.BytesIO(download.read()), "r") as zf:
+                for member in zf.infolist():
+                    zf.extract(member, path=mod_folder)
+
+        messagebox("Success!", "Latest mod successfully installed!")
+
+    def available_mod_update() -> bool:
+        """Check if there's an available update"""
+        get_url = "https://api.github.com/repos/alwaysintreble/TheMessengerRandomizerModAP/releases/latest"
+        latest_version: str = request_data(get_url)["tag_name"].lstrip("v")
+
+        toml_path = os.path.join(folder, "Mods", "TheMessengerRandomizerAP", "courier.toml")
+        with open(toml_path, "r") as f:
+            installed_version = f.read().splitlines()[1].strip("version = \"")
+
+        logging.info(f"Installed version: {installed_version}. Latest version: {latest_version}")
+        # one of the alpha builds
+        return not latest_version.isnumeric() or tuplize_version(latest_version) > tuplize_version(installed_version)
+
+    from . import MessengerWorld
+    folder = os.path.dirname(MessengerWorld.settings.game_path)
+    if not courier_installed():
+        should_install = askyesnocancel("Install Courier",
+                                        "No Courier installation detected. Would you like to install now?")
+        if not should_install:
+            return
+        logging.info("Installing Courier")
+        install_courier()
+    if not mod_installed():
+        should_install = askyesnocancel("Install Mod",
+                                        "No randomizer mod detected. Would you like to install now?")
+        if not should_install:
+            return
+        logging.info("Installing Mod")
+        install_mod()
+    else:
+        if available_mod_update():
+            should_update = askyesnocancel("Update Mod",
+                                           "Old mod version detected. Would you like to update now?")
+            if should_update:
+                logging.info("Updating mod")
+                install_mod()
+    if is_linux:
+        if url:
+            open_file(f"steam://rungameid/764790//{url}/")
+        else:
+            open_file("steam://rungameid/764790")
+    else:
+        os.chdir(Path(MessengerWorld.settings.game_path).parent)
+        if url:
+            subprocess.Popen([MessengerWorld.settings.game_path, str(url)])
+        else:
+            subprocess.Popen(MessengerWorld.settings.game_path)