texty/lib/game.py

import yaml
from yaml.loader import SafeLoader
from colorama import Fore, Back, Style
import re
from .save import SaveManager
from .ascii import AsciiAnimation
from time import sleep
from os import system, path, mkdir

class Game: # the game class keeps information about the loaded game
    def __init__(self,data:dict):
        self.name = data["meta"]["name"]
        self.author = data["meta"]["creator"]
        self.current = "start"
        self.nodes = {}
        self.inventory = []
        self.save = SaveManager()
        self.id = data["meta"]["id"]
        for k in data["game"]:
            self.nodes.update({k:data["game"][k]})

    def main_menu(self): # displays the main menu
        l = self.save.load()
        if not l:
            # New game
            print(self.name)
            print(f"A game by {self.author}")
            print("")
            print("1 - Start")
            print("2 - Options")
            print("0 - Quit")
            selection = self.make_selection(3)
            system("cls||clear")
            if(selection == 1): # start new game
                self.print_text()
            elif(selection == 2):
                self.settings_menu()
            elif(selection == 0):
                print("Quitting")
                exit()
        else: # Display continue
            print(self.name)
            print(f"A game by {self.author}")
            print("")
            print("1 - Continue")
            print("2 - New game")
            print("3 - Options")
            print("0 - Quit")
            selection = self.make_selection(3)
            system("cls||clear")
            if(selection == 1):
                self.current = self.save.currentPrompt
                self.inventory = self.save.inventory
                self.print_text()
            elif(selection == 2):
                self.print_text()
            elif(selection == 3):
                self.settings_menu()
            elif(selection == 0):
                print("Quitting")
                exit()

    def settings_menu(self): # displays the settings menu
        print("Options")
        print("")
        print("1 - Language")
        print("0 - Back")
        selection = self.make_selection(1)
        if(selection == 1):
            print("Language")
            print("")
            print("1 - English")
            print("2 - Czech")
            print("0 - Back")
            selection = self.make_selection(2)
            system("cls||clear")
            if(selection == 1):
                with open("./saves/lang","w") as f:
                    f.write("en")
            elif(selection == 2):
                with open("./saves/lang","w") as f:
                    f.write("cz")
            self.settings_menu()
        else:
            self.main_menu()

    def make_selection(self, length=0) -> int: # this method makes sure a valid selection is made and returns the selection as a number
        # TODO: replace with selection by keyboard(?)
        l = length # sets the length
        if(l == 0): # if no length was set, we get it from nodes
            l = len(self.nodes[self.current]["actions"])-1
        y = False
        selection = 0
        # TODO: Check for "has_item"
        while y == False: # while the selection is not correct
            try:
                selection = int(input("Make a selection (number): ")) # ask for selection
            except ValueError: # handle wrong input type
                print("Not a number selection")
            if(selection > l or selection < 0): # if the number is bigger than the length or smaller than 0, print error
                print("Invalid selection")
            else:
                y = True # else return the selection
        return selection
    
    def print_text(self): # Prints out the current prompt
        system("cls||clear")
        animated = re.search(r"(?!{).+(?=})",self.nodes[self.current]["text"]) # find the animated text
        if(animated != None):
            self.print_animated(animated.group(0))
            self.nodes[self.current]["text"] = self.nodes[self.current]["text"].replace("{"+animated.group(0)+"}","") # remove the animated text from the text prompt
        print(self.parse_colors(self.nodes[self.current]["text"]))
        print("")
        ostring = ""
        if("actions" in self.nodes[self.current].keys()):
            for i,option in enumerate(self.nodes[self.current]["actions"]):
                ostring+=f"{i} - {self.nodes[option]['description']}\n"
            print(ostring)
            sel = self.make_selection()
            if "add_item" in self.nodes[self.current]: # if there is an add_inventory key in the node,
                # add item to inventory
                self.inventory.append(self.nodes[self.current]["add_inventory"])
            self.current = self.nodes[self.current]["actions"][sel]
            self.save.currentPrompt = self.current # save the current prompt
            self.print_text()

    def print_animated(self,animid): # prinst the first found occurence of an ascii animation
        animation = AsciiAnimation()
        animation.load_ascii(animid)
        for frame in animation.frames:
            system("cls||clear")
            print(frame)
            sleep(animation.speed)
        print()
        
    def parse_colors(self,text:str) -> str: # Converts color codes into terminal colors
        newText = text.replace("&b",Fore.CYAN).replace("&c",Fore.RED).replace("&e", Fore.YELLOW).replace("&a",Fore.GREEN).replace("&9",Fore.BLUE).replace("&r",Fore.RESET).replace("&f",Fore.WHITE).replace("&5",Fore.MAGENTA).replace("\n",Fore.RESET + "\n") # replace color codes and newlines with colorama
        newText += Fore.RESET # reset color at the end of the text
        return newText


def load(file_path): # starts to load the game from YAML
    lang = "en"
    if not (path.exists("./saves/lang")):
        mkdir("./saves")
        with open("./saves/lang","w") as f:
            f.write("en")
    else:
        with open("./saves/lang","r") as f:
            lang = f.read()
            if lang == "cz":
                lang = "cz"
    try:
        with open(file_path) as f:
            data = yaml.load(f,Loader=SafeLoader)
            g = Game(data)
            g.lang = lang
            return g
    except Exception as e:
        print(f"{Back.RED}{Fore.WHITE}An exception has occured while loading the game from the YAML file:{Fore.RESET}{Back.RESET}")
        print(e)
        return None