#!/usr/bin/env python3
import os
import yaml
#map_server = "TheIsland"
#dir_server = "~/ARK_Servers/"
#session_name = "Dodo server"
#Port = 7777
#QueryPort = 27015
#ServerPassword = ""
#MaxPlayers = 70
#ModsId = []
#listen = True
#Cluster = False

def read_yaml():
    with open("config.yaml", "r") as f:
        return yaml.safe_load(f)

if not os.path.exists(dir_server):
    os.mkdir(dir_server)

def install(map_server):
    print("""Укажите требуется ли кластер?
    1. Да
    2. Нет""")
    count_cluster = input(":\n")
    if count_cluster == "1":
        Cluster = True
    else:
        Cluster = False

    while True:
        count_maps = input("Укажите количество карт:\n")
        if count_maps.isdigit:
            break
        print("Введите цифры")
    
    for i in range(int(count_maps)):
        print("""Выберите карту из списка указав номер
        1. The Island
        2. The Center
        3. Scorched Earth
        4. Ragnarok
        5. Aberration
        6. Extinction
        7. Valguero
        8. Genesis: Part 1  
        9. Crystal Isles    
        10. Genesis: Part 2 
        11. Lost Island    
        12. Fjordur""")
        while True:
            count_map = input("")
            if count_maps.isdigit:
                break
        if  count_map == "1":
            map_server = "TheIsland"
        elif count_map == "2":
            map_server = "TheCenter"
        elif count_map == "3":
            map_server = "ScorchedEarth_P"
        elif count_map == "4":
            map_server = "Ragnarok"
        elif count_map == "5":
            map_server = "Aberration_P"
        elif count_map == "6":
            map_server = "Extinction"
        elif count_map == "7":
            map_server = "Valguero_P"
        elif count_map == "8":
            map_server = "Genesis"
        elif count_map == "9":
            map_server = "CrystalIsles"
        elif count_map == "10":
            map_server = "Gen2"
        elif count_map == "11":
            map_server = "LostIsland"
        elif count_map == "12":
            map_server = "Fjordur"
            print("Введите цифры")
        if not os.path.isdir(dir_server + map_server):
            os.mkdir(dir_server + map_server)
        else:
            os.system(f"steamcmd +force_install_dir {dir_server + map_server} +login anonymous +app_update 376030 +quit")
print("Укажите название Сервера")
session_name = input("")
print("Укажите порт Сервера <7777>")
Port = input("")
print("Укажите Query-порт Сервера <27015>")
QueryPort = input("")
install(map_server)
os.chdir(dir_server + map_server + "/" + "ShooterGame/Binaries/Linux/")
def start():
    if Cluster == False:
        os.system(f"./ShooterGameServer {map_server}?SessionName={session_name}?Port={Port}?QueryPort={QueryPort}?MaxPlayers={MaxPlayers}?GameModIds={ModsId}?listen={listen}")
    #os.system("ShooterGameServer")
start()