d3dbdb4491
* Kivy: Add a button prompt box * auto format the buttons to display 2 per row to look nicer * update to kivymd * have the uri popup use the new API * have messenger use the new API * make the buttonprompt import even more lazy * messenger needs to be lazy too * make the buttons take up the full dialog width --------- Co-authored-by: NewSoupVi <57900059+NewSoupVi@users.noreply.github.com>
259 lines
11 KiB
Python
259 lines
11 KiB
Python
import argparse
|
|
import io
|
|
import logging
|
|
import os.path
|
|
import requests
|
|
import subprocess
|
|
import urllib.request
|
|
from shutil import which
|
|
from typing import Any, Callable, TYPE_CHECKING
|
|
from zipfile import ZipFile
|
|
from Utils import is_windows, messagebox, open_file, tuplize_version
|
|
|
|
if TYPE_CHECKING:
|
|
from kvui import ButtonsPrompt
|
|
|
|
|
|
MOD_URL = "https://api.github.com/repos/alwaysintreble/TheMessengerRandomizerModAP/releases/latest"
|
|
|
|
|
|
def create_yes_no_popup(title: str, text: str, callback: Callable[[str], None]) -> "ButtonsPrompt":
|
|
from kvui import ButtonsPrompt
|
|
buttons = ["Yes", "No", "Cancel"]
|
|
|
|
prompt = ButtonsPrompt(title, text, callback, *buttons)
|
|
prompt.open()
|
|
return prompt
|
|
|
|
|
|
def launch_game(*args) -> None:
|
|
"""Check the game installation, then launch it"""
|
|
def courier_installed() -> bool:
|
|
"""Check if Courier is installed"""
|
|
assembly_path = os.path.join(game_folder, "TheMessenger_Data", "Managed", "Assembly-CSharp.dll")
|
|
with open(assembly_path, "rb") as assembly:
|
|
for line in assembly:
|
|
if b"Courier" in line:
|
|
return True
|
|
return False
|
|
|
|
def mod_installed() -> bool:
|
|
"""Check if the mod is installed"""
|
|
return os.path.exists(os.path.join(game_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=game_folder)
|
|
|
|
os.chdir(game_folder)
|
|
# linux and mac handling
|
|
if not is_windows:
|
|
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/716f0a2bd5d75138969090494a76328f39a6dd78.zip"
|
|
files = []
|
|
with urllib.request.urlopen(mono_kick_url) as download:
|
|
with ZipFile(io.BytesIO(download.read()), "r") as zf:
|
|
for member in zf.infolist():
|
|
if "precompiled/" not in member.filename or member.filename.endswith("/"):
|
|
continue
|
|
member.filename = member.filename.split("/")[-1]
|
|
if member.filename.endswith("bin.x86_64"):
|
|
member.filename = "MiniInstaller.bin.x86_64"
|
|
zf.extract(member, path=game_folder)
|
|
files.append(member.filename)
|
|
mono_installer = os.path.join(game_folder, "MiniInstaller.bin.x86_64")
|
|
os.chmod(mono_installer, 0o755)
|
|
installer = subprocess.Popen(mono_installer, shell=False)
|
|
failure = installer.wait()
|
|
for file in files:
|
|
os.remove(file)
|
|
else:
|
|
installer = subprocess.Popen([mono_exe, os.path.join(game_folder, "MiniInstaller.exe")], shell=True)
|
|
failure = installer.wait()
|
|
else:
|
|
installer = subprocess.Popen(os.path.join(game_folder, "MiniInstaller.exe"), shell=True)
|
|
failure = installer.wait()
|
|
|
|
print(failure)
|
|
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"""
|
|
assets = request_data(MOD_URL)["assets"]
|
|
if len(assets) == 1:
|
|
release_url = assets[0]["browser_download_url"]
|
|
else:
|
|
for asset in assets:
|
|
if "TheMessengerRandomizerAP" in asset["name"]:
|
|
release_url = asset["browser_download_url"]
|
|
break
|
|
else:
|
|
messagebox("Failure", "Failed to find latest mod download", True)
|
|
raise RuntimeError("Failed to install Mod")
|
|
|
|
mod_folder = os.path.join(game_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(latest_version: str) -> bool:
|
|
"""Check if there's an available update"""
|
|
latest_version = latest_version.lstrip("v")
|
|
toml_path = os.path.join(game_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 "alpha" in latest_version or tuplize_version(latest_version) > tuplize_version(installed_version)
|
|
|
|
def after_courier_install_popup(answer: str) -> None:
|
|
"""Gets called if the user doesn't have courier installed. Handle the button they pressed."""
|
|
nonlocal prompt
|
|
|
|
prompt.dismiss()
|
|
if answer in ("No", "Cancel"):
|
|
return
|
|
logging.info("Installing Courier")
|
|
install_courier()
|
|
prompt = create_yes_no_popup("Install Mod",
|
|
"No randomizer mod detected. Would you like to install now?",
|
|
after_mod_install_popup)
|
|
|
|
def after_mod_install_popup(answer: str) -> None:
|
|
"""Gets called if the user has courier but mod isn't installed, or there's an available update."""
|
|
nonlocal prompt
|
|
|
|
prompt.dismiss()
|
|
if answer in ("No", "Cancel"):
|
|
return
|
|
logging.info("Installing Mod")
|
|
install_mod()
|
|
prompt = create_yes_no_popup("Launch Game",
|
|
"Courier and Game mod installed successfully. Launch game now?",
|
|
launch)
|
|
|
|
def after_mod_update_popup(answer: str) -> None:
|
|
"""Gets called if there's an available update."""
|
|
nonlocal prompt
|
|
|
|
prompt.dismiss()
|
|
if answer == "Cancel":
|
|
return
|
|
if answer == "Yes":
|
|
logging.info("Updating Mod")
|
|
install_mod()
|
|
prompt = create_yes_no_popup("Launch Game",
|
|
"Courier and Game mod installed successfully. Launch game now?",
|
|
launch)
|
|
else:
|
|
prompt = create_yes_no_popup("Launch Game",
|
|
"Game Mod not updated. Launch game now?",
|
|
launch)
|
|
|
|
def launch(answer: str | None = None) -> None:
|
|
"""Launch the game."""
|
|
nonlocal args
|
|
|
|
if prompt:
|
|
prompt.dismiss()
|
|
if answer and answer in ("No", "Cancel"):
|
|
return
|
|
|
|
parser = argparse.ArgumentParser(description="Messenger Client Launcher")
|
|
parser.add_argument("url", type=str, nargs="?", help="Archipelago Webhost uri to auto connect to.")
|
|
args = parser.parse_args(args)
|
|
|
|
if not is_windows:
|
|
if args.url:
|
|
open_file(f"steam://rungameid/764790//{args.url}/")
|
|
else:
|
|
open_file("steam://rungameid/764790")
|
|
else:
|
|
os.chdir(game_folder)
|
|
if args.url:
|
|
subprocess.Popen([MessengerWorld.settings.game_path, str(args.url)])
|
|
else:
|
|
subprocess.Popen(MessengerWorld.settings.game_path)
|
|
os.chdir(working_directory)
|
|
|
|
from . import MessengerWorld
|
|
try:
|
|
game_folder = os.path.dirname(MessengerWorld.settings.game_path)
|
|
except ValueError as e:
|
|
logging.error(e)
|
|
messagebox("Invalid File", "Selected file did not match expected hash. "
|
|
"Please try again and ensure you select The Messenger.exe.")
|
|
return
|
|
working_directory = os.getcwd()
|
|
# setup ssl context
|
|
try:
|
|
import certifi
|
|
import ssl
|
|
context = ssl.create_default_context(ssl.Purpose.SERVER_AUTH, cafile=certifi.where())
|
|
context.set_alpn_protocols(["http/1.1"])
|
|
https_handler = urllib.request.HTTPSHandler(context=context)
|
|
opener = urllib.request.build_opener(https_handler)
|
|
urllib.request.install_opener(opener)
|
|
except ImportError:
|
|
pass
|
|
if not courier_installed():
|
|
prompt = create_yes_no_popup("Install Courier",
|
|
"No Courier installation detected. Would you like to install now?",
|
|
after_courier_install_popup)
|
|
return
|
|
if not mod_installed():
|
|
prompt = create_yes_no_popup("Install Mod",
|
|
"No randomizer mod detected. Would you like to install now?",
|
|
after_mod_install_popup)
|
|
return
|
|
else:
|
|
latest = request_data(MOD_URL)["tag_name"]
|
|
if available_mod_update(latest):
|
|
prompt = create_yes_no_popup("Update Mod",
|
|
f"New mod version detected. Would you like to update to {latest} now?",
|
|
after_mod_update_popup)
|
|
return
|
|
|
|
if not args:
|
|
prompt = create_yes_no_popup("Launch Game",
|
|
"Mod installed and up to date. Would you like to launch the game now?",
|
|
launch)
|