forked from mirror/Archipelago
Jonathan Tinney
7971961166
Some checks failed
Analyze modified files / flake8 (push) Failing after 2m28s
Build / build-win (push) Has been cancelled
Build / build-ubuntu2204 (push) Has been cancelled
ctest / Test C++ ubuntu-latest (push) Has been cancelled
ctest / Test C++ windows-latest (push) Has been cancelled
Analyze modified files / mypy (push) Has been cancelled
Build and Publish Docker Images / Push Docker image to Docker Hub (push) Successful in 5m4s
Native Code Static Analysis / scan-build (push) Failing after 5m2s
type check / pyright (push) Successful in 1m7s
unittests / Test Python 3.11.2 ubuntu-latest (push) Failing after 16m23s
unittests / Test Python 3.12 ubuntu-latest (push) Failing after 28m19s
unittests / Test Python 3.13 ubuntu-latest (push) Failing after 14m49s
unittests / Test hosting with 3.13 on ubuntu-latest (push) Successful in 5m0s
unittests / Test Python 3.13 macos-latest (push) Has been cancelled
unittests / Test Python 3.11 windows-latest (push) Has been cancelled
unittests / Test Python 3.13 windows-latest (push) Has been cancelled
75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
import io
|
|
import os
|
|
import sys
|
|
import tempfile
|
|
import zipfile
|
|
|
|
from typing import Dict
|
|
from settings import get_settings
|
|
from .Data import Rels
|
|
|
|
|
|
def setup_gclib_path():
|
|
"""Extracts gclib files from .apworld zip to temp directory if needed."""
|
|
base_path = os.path.dirname(__file__)
|
|
lib_path = os.path.join(base_path, "lib", "gclib")
|
|
|
|
if ".apworld" in __file__:
|
|
# Find the .apworld file path
|
|
zip_file_path = __file__
|
|
while not zip_file_path.lower().endswith(".apworld"):
|
|
zip_file_path = os.path.dirname(zip_file_path)
|
|
|
|
# Set up temporary extraction directory
|
|
temp_base_dir = tempfile.gettempdir()
|
|
target_dir_path = os.path.join(temp_base_dir, "ttyd_temp_gclib")
|
|
temp_lib_path = os.path.join(target_dir_path, "ttyd", "lib", "gclib")
|
|
|
|
# Clean and recreate directory
|
|
if os.path.exists(target_dir_path):
|
|
import shutil
|
|
shutil.rmtree(target_dir_path)
|
|
os.makedirs(target_dir_path, exist_ok=True)
|
|
|
|
# Extract gclib files from .apworld zip
|
|
with zipfile.ZipFile(zip_file_path, "r") as zip_ref:
|
|
for member in zip_ref.namelist():
|
|
if "gclib" in member:
|
|
zip_ref.extract(member, target_dir_path)
|
|
|
|
# Add lib directory to Python path for imports
|
|
lib_parent = os.path.join(target_dir_path, "ttyd", "lib")
|
|
if lib_parent not in sys.path:
|
|
sys.path.insert(0, lib_parent)
|
|
|
|
return temp_lib_path
|
|
else:
|
|
# For non-apworld case, add the lib directory to path
|
|
lib_parent = os.path.dirname(lib_path)
|
|
if lib_parent not in sys.path:
|
|
sys.path.insert(0, lib_parent)
|
|
return lib_path
|
|
|
|
|
|
class TTYDPatcher:
|
|
rels: Dict[Rels, io.BytesIO] = {}
|
|
|
|
def __init__(self):
|
|
setup_gclib_path()
|
|
from gclib.gcm import GCM
|
|
from gclib.dol import DOL
|
|
|
|
self.iso = GCM(get_settings().ttyd_options.rom_file)
|
|
self.iso.read_entire_disc()
|
|
self.dol = DOL()
|
|
self.dol.read(self.iso.read_file_data("sys/main.dol"))
|
|
for rel in Rels:
|
|
if rel == Rels.dol:
|
|
continue
|
|
path = get_rel_path(rel)
|
|
self.rels[rel] = self.iso.read_file_data(path)
|
|
|
|
|
|
def get_rel_path(rel: Rels):
|
|
return f'files/rel/{rel.value}.rel'
|