249 lines
10 KiB
Python
Executable File
249 lines
10 KiB
Python
Executable File
|
|
#!/usr/bin/env python3
|
|
import os
|
|
import yaml
|
|
import threading
|
|
from pathlib import Path
|
|
import click
|
|
import colorama
|
|
home_dir = Path.home()
|
|
dir_server = f"{home_dir}/ARK_Servers/"
|
|
config_hlna = f"{home_dir}/.config/hlna/"
|
|
mods_id = ""
|
|
listen_server = True
|
|
@click.group()
|
|
def hlna():
|
|
pass
|
|
list_config = next(os.walk(config_hlna), (None, None, []))[2] # [] if no file
|
|
x = list_config.count('.directory')
|
|
if x>0:
|
|
list_config.remove('.directory')
|
|
def print_line(text):
|
|
"Добавление тире вокруг текста, покраска"
|
|
print(colorama.Fore.YELLOW + "-"*30)
|
|
print(colorama.Fore.GREEN + text)
|
|
print(colorama.Fore.YELLOW + "-"*30 + colorama.Style.RESET_ALL)
|
|
def create_dir(directory):
|
|
"Проверка и создание директории"
|
|
if not os.path.exists(directory):
|
|
os.mkdir(directory)
|
|
def check_int(number=""):
|
|
"Проверка на ввод числа"
|
|
while True:
|
|
try:
|
|
x = input(number)
|
|
if x == "":
|
|
return 0
|
|
x = x.replace(' ', '')
|
|
x = x.replace(',', '')
|
|
x = int(x)
|
|
return x
|
|
except ValueError:
|
|
print("Введите число")
|
|
create_dir(dir_server)
|
|
create_dir(config_hlna)
|
|
@hlna.command(help='Для конфигурирования параметров запускаемого сервера или кластера серверов')
|
|
def config(map_server=list_config):
|
|
query_p = []
|
|
port_s = []
|
|
for i in map_server:
|
|
with open(f"{home_dir}/.config/hlna/{i}", "r") as yamlfile:
|
|
data = yaml.load(yamlfile, Loader=yaml.FullLoader)
|
|
data = data[0]
|
|
port_s.append(data['Port'])
|
|
query_p.append(data['QueryPort'])
|
|
cluster_dir_override = ""
|
|
cluster_id = ""
|
|
count_cluster = check_int("""Укажите требуется ли кластер?
|
|
1. Да
|
|
2. Нет
|
|
: """)
|
|
if count_cluster == 0: # 0 возвращает функция chek_int, если ничего не было введено
|
|
count_cluster = 2
|
|
if count_cluster == 1:
|
|
cluster_server = True
|
|
cluster_id = input("Укажите id для кластера, любое сочетание символов: \n")
|
|
create_dir(dir_server + cluster_id)
|
|
cluster_dir_override = (dir_server + cluster_id)
|
|
else:
|
|
cluster_server = False
|
|
if map_server != []:
|
|
print("Уже установленные карты: ")
|
|
for i in map_server:
|
|
print(i)
|
|
count_maps = check_int("Укажите количество карт: \n")
|
|
if count_maps == 0: # 0 возрвращает check_int когда, ничего не было введено
|
|
count_maps = 1
|
|
for i in range(count_maps):
|
|
while True:
|
|
"Проверка на выбор карты из списка"
|
|
os.system("clear")
|
|
amount_map = check_int("""Выберите карту из списка указав номер
|
|
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
|
|
: """)
|
|
if amount_map == 0: # 0 возрвращает check_int когда, ничего не было введено
|
|
amount_map = i + 1
|
|
if 0<amount_map<=12:
|
|
break
|
|
match amount_map:
|
|
case 1:
|
|
map_server.append("TheIsland")
|
|
case 2:
|
|
map_server.append("TheCenter")
|
|
case 3:
|
|
map_server.append("ScorchedEarth_P")
|
|
case 4:
|
|
map_server.append("Ragnarok")
|
|
case 5:
|
|
map_server.append("Aberration_P")
|
|
case 6:
|
|
map_server.append("Extinction")
|
|
case 7:
|
|
map_server.append("Valguero_P")
|
|
case 8:
|
|
map_server.append("Genesis")
|
|
case 9:
|
|
map_server.append("CrystalIsles")
|
|
case 10:
|
|
map_server.append("Gen2")
|
|
case 11:
|
|
map_server.append("LostIsland")
|
|
case 12:
|
|
map_server.append("Fjordur")
|
|
name_server = input("Укажите название Сервера: \n")
|
|
if name_server == "":
|
|
name_server = map_server[-1]
|
|
while True:
|
|
port_server = check_int("Укажите порт Сервера <7777>: \n")
|
|
if port_server == 0:
|
|
if port_s == []:
|
|
port_server = 7777
|
|
else:
|
|
port_server = max(port_s) + 2
|
|
if port_server is port_s: # если веденный порт есть в списке портов
|
|
print("Этот порт уже занят")
|
|
else:
|
|
port_s.append(port_server)
|
|
break
|
|
while True:
|
|
query_port = check_int("Укажите Query-порт Сервера <27015>: \n")
|
|
if query_port == 0:
|
|
if query_p == []:
|
|
query_port = 27015
|
|
else:
|
|
query_port = max(query_p) + 2
|
|
if query_port is query_p: # если веденный порт есть в списке портов
|
|
print("Этот порт уже занят")
|
|
else:
|
|
port_s.append(query_p)
|
|
break
|
|
password_server = input("Укажите пароль Сервера: \n")
|
|
max_players = check_int("Укажите максимальное количество игроков: \n")
|
|
if max_players == "":
|
|
max_players = 70
|
|
yaml_create(cluster_server, map_server, name_server, port_server, query_port, password_server, max_players, cluster_id, cluster_dir_override)
|
|
def yaml_create(cluster_server, map_server, name_server, port_server, query_port, password_server, max_players, cluster_id, cluster_dir_override):
|
|
settings_hlna = [
|
|
{
|
|
'Cluster' : cluster_server,
|
|
'SessionName' : name_server,
|
|
'Port' : port_server,
|
|
'QueryPort' : query_port,
|
|
'ServerPassword' : password_server,
|
|
'MaxPlayers' : max_players,
|
|
'ModsId' : mods_id,
|
|
'Listen' : listen_server,
|
|
'ServerPath' : dir_server
|
|
}
|
|
]
|
|
print(cluster_id)
|
|
if cluster_id != None:
|
|
settings_hlna[0]['clusterid'] = cluster_id
|
|
settings_hlna[0]['clusterdir'] = cluster_dir_override
|
|
with open(config_hlna + f"{map_server[-1]}", 'w') as yamlfile:
|
|
data = yaml.dump(settings_hlna, yamlfile)
|
|
print(colorama.Fore.GREEN + "Конфиг создан" + colorama.Style.RESET_ALL)
|
|
@hlna.command()
|
|
def servers(map_server=list_config):
|
|
if map_server==[]:
|
|
print("Сервера не установлены")
|
|
else:
|
|
for i in map_server:
|
|
cluster_server, name_server, port_server, query_port, password_server, max_players, dir_server, mods_id, listen_server, cluster_id, cluster_dir_ovverride = read_yaml(i)
|
|
print(f"""
|
|
Сервер: {i}
|
|
Имя сервера: {name_server}
|
|
Кластер: {cluster_server}
|
|
Порт сервера: {port_server}
|
|
Query порт: {query_port}
|
|
Пароль: {password_server}
|
|
Максимальное кол-во игроков: {max_players}
|
|
Моды: {mods_id}
|
|
Кластер id: {cluster_id}""")
|
|
print("\n")
|
|
@hlna.command(help='Для запуска, сконфигурированного сервера или кластера')
|
|
def start(map_server=list_config):
|
|
map_choise = []
|
|
while True:
|
|
"Проверка на выбор имеющихся карт"
|
|
check = True
|
|
for i in range(len(map_server)):
|
|
print(f"{i+1}) {map_server[i]}")
|
|
map_list = input("Выбери карты для запуска в формате 1 2 3 или 1,2,3")
|
|
map_list = map_list.replace(',', ' ')
|
|
map_list = list(set(map_list.split()))
|
|
try:
|
|
for s in map_list:
|
|
if int(s) > len(map_server):
|
|
check = False
|
|
if len(map_list) <= len(map_server) and check == True:
|
|
break
|
|
except:
|
|
print("Неправильный ввод")
|
|
print_line(map_list)
|
|
for j in map_list:
|
|
print_line(j)
|
|
map_choise.append(map_server[int(j) - 1])
|
|
map_server = map_choise
|
|
for i in map_server:
|
|
cluster_server, name_server, port_server, query_port, password_server, max_players, dir_server, mods_id, listen_server, cluster_id, cluster_dir_ovverride = read_yaml(i)
|
|
create_dir(dir_server + i)
|
|
print_line("Качаем карту: " + i)
|
|
os.system(f"steamcmd +force_install_dir {dir_server + i} +login anonymous +app_update 376030 +quit")
|
|
print_line("Карта скачана: " + i)
|
|
def start(i):
|
|
os.chdir(dir_server + i + "/ShooterGame/Binaries/Linux/")
|
|
os.system(f"./ShooterGameServer {i}?SessionName={name_server}?Port={port_server}?QueryPort={query_port}?MaxPlayers={max_players}?GameModIds={mods_id}?listen={listen_server} -clusterid={cluster_id} -ClusterDirOverride={cluster_dir_ovverride}")
|
|
threads = threading.Thread(target=start, args=(i,))
|
|
threads.start()
|
|
def read_yaml(map_server):
|
|
with open(f"{home_dir}/.config/hlna/{map_server}", "r") as yamlfile:
|
|
data = yaml.load(yamlfile, Loader=yaml.FullLoader)
|
|
data = data[0]
|
|
dir_server = data['ServerPath']
|
|
name_server = data['SessionName']
|
|
port_server = data['Port']
|
|
query_port = data['QueryPort']
|
|
password_server = data['ServerPassword']
|
|
max_players = data['MaxPlayers']
|
|
mods_id = data['ModsId']
|
|
listen_server = data['Listen']
|
|
cluster_server = data['Cluster']
|
|
cluster_id = data['clusterid']
|
|
cluster_dir_ovverride = data['clusterdir']
|
|
print(cluster_id)
|
|
print(cluster_dir_ovverride)
|
|
return cluster_server, name_server, port_server, query_port, password_server, max_players, dir_server, mods_id, listen_server, cluster_id, cluster_dir_ovverride
|
|
if __name__ == '__main__':
|
|
hlna() |