Grinch-AP/worlds/smz3/TotalSMZ3/Text/Texts.py

from typing import Any, List
from worlds.smz3.TotalSMZ3.Region import Region
from worlds.smz3.TotalSMZ3.Regions.Zelda.GanonsTower import GanonsTower
from worlds.smz3.TotalSMZ3.Item import Item, ItemType
from yaml import load, Loader
import random
import os

text_folder = os.path.dirname(__file__)

class Texts:
    @staticmethod
    def ParseYamlScripts(resource: str):
        with open(resource, 'rb') as f:
            yaml = str(f.read(), "utf-8")
        return load(yaml, Loader)

    @staticmethod        
    def ParseTextScript(resource: str):
        with open(resource, 'r', encoding="utf-8-sig") as file:
            return [text.rstrip('\n') for text in file.read().replace("\r", "").split("---\n") if text]

    scripts: Any = ParseYamlScripts.__func__(text_folder + "/Scripts/General.yaml")
    blind: List[str] = ParseTextScript.__func__(text_folder + "/Scripts/Blind.txt")
    ganon: List[str] = ParseTextScript.__func__(text_folder + "/Scripts/Ganon.txt")
    tavernMan: List[str] = ParseTextScript.__func__(text_folder + "/Scripts/TavernMan.txt")
    triforceRoom: List[str] = ParseTextScript.__func__(text_folder + "/Scripts/TriforceRoom.txt")

    @staticmethod
    def SahasrahlaReveal(dungeon: Region):
        text = Texts.scripts["SahasrahlaReveal"]
        return text.replace("<dungeon>", dungeon.Area)

    @staticmethod
    def BombShopReveal(dungeons: List[Region]):
        text = Texts.scripts["BombShopReveal"]
        return text.replace("<first>", dungeons[0].Area).replace("<second>", dungeons[1].Area)

    @staticmethod
    def GanonThirdPhaseSingle(silvers: Region):
        node = Texts.scripts["GanonSilversReveal"]["single"]
        text = node["local" if isinstance(silvers, GanonsTower) else "remote"]
        return text.replace("<region>", silvers.Area)

    @staticmethod
    def GanonThirdPhaseMulti(silvers: Region, myWorld: int, silversWorld: int = None, silversPlayer: str = None):
        node = Texts.scripts["GanonSilversReveal"]["multi"]
        if silvers is None:
            if (silversWorld == myWorld.Id):
                return node["local"]
            player = silversPlayer
        else:
            if (silvers.world == myWorld):
                return node["local"]
            player = silvers.world.Player
        player = player.rjust(7 + len(player) // 2)
        text = node["remote"]
        return text.replace("<player>", player)

    @staticmethod
    def ItemTextbox(item: Item):
        nameMap = {
                    ItemType.BottleWithGoldBee : ItemType.BottleWithBee.name,
                    ItemType.HeartContainerRefill : ItemType.HeartContainer.name,
                    ItemType.OneRupee : "PocketRupees",
                    ItemType.FiveRupees : "PocketRupees",
                    ItemType.TwentyRupees : "CouchRupees",
                    ItemType.TwentyRupees2 : "CouchRupees",
                    ItemType.FiftyRupees : "CouchRupees",
                    ItemType.BombUpgrade5 : "BombUpgrade",
                    ItemType.BombUpgrade10 : "BombUpgrade",
                    ItemType.ArrowUpgrade5 : "ArrowUpgrade",
                    ItemType.ArrowUpgrade10 : "ArrowUpgrade",
                    item.Type : item.Type.name,
                }
        if item.IsMap(): name = "Map"
        elif item.IsCompass(): name = "Compass"
        elif item.IsKeycard(): name = "Keycard"
        else: name = nameMap[item.Type]

        items = Texts.scripts["Items"]
        return items.get(name, None) or items["default"]

    @staticmethod
    def Blind(rnd: random): return Texts.RandomLine(rnd, Texts.blind)

    @staticmethod
    def TavernMan(rnd: random): return Texts.RandomLine(rnd, Texts.tavernMan)

    @staticmethod
    def GanonFirstPhase(rnd: random): return Texts.RandomLine(rnd, Texts.ganon)

    @staticmethod
    def TriforceRoom(rnd: random): return Texts.RandomLine(rnd, Texts.triforceRoom)

    @staticmethod
    def RandomLine(rnd: random, lines: List[str]): return lines[rnd.randrange(0, len(lines))]