From a398bbd5dc57c2b8394b130181b4cc798d325bd6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Maty=C3=A1=C5=A1=20Caras?= <contact@hernikplays.cz>
Date: Wed, 23 Mar 2022 14:21:44 +0100
Subject: [PATCH] Main menu

---
 __main__.py | 31 +++++++++++++++++++++++++++++--
 1 file changed, 29 insertions(+), 2 deletions(-)

diff --git a/__main__.py b/__main__.py
index 590f478..b75d148 100644
--- a/__main__.py
+++ b/__main__.py
@@ -3,8 +3,24 @@ from lib.game import *
 from colorama import init, Back, Fore
 from os import mkdir, listdir
 
+def lang():
+    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"
+    data = ""
+    with open(f"./lib/lang/{lang}.yml",encoding="utf-8") as f:
+        data = yaml.load(f,Loader=SafeLoader)
+    return data
 
 def main(): # TODO: Maybe a menu for available text games?
+    l = lang()
     init()
     if(not isdir("./games")):
         mkdir("./games")
@@ -14,11 +30,22 @@ def main(): # TODO: Maybe a menu for available text games?
             # finds available games
             try:
                 # try parsing
-                g = load(f"./games/{file}")
+                g = load(f"./games/{file}",l)
                 games.append(g)
             except Exception as e:
-                print(f"{Back.RED}{Fore.RED}An exception has occured while loading {file}:")
+                print(f"{Back.RED}{Fore.RED}{l['error_loading']} {file}:")
                 print(e)
+    # PRINT OUT GAME SELECT
+    # TODO SWITCH TO MENU MANAGER
+    print("     TEXTY   ")
+    print(l['available'])
+    if len(games) < 1:
+        print(l['no_games'])
+    else:
+        for i,g in enumerate(games):
+            print(f"{i} - {g.name}")
+    print(f"{len(games)} - {l['quit']}")
+
 
 if __name__ == "__main__":
     main()