168 lines
5.9 KiB
Python
Executable File
168 lines
5.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
import errno
|
||
import time
|
||
import sys
|
||
import os
|
||
import telebot
|
||
from telebot import types
|
||
import subprocess
|
||
import yaml
|
||
import colorama
|
||
|
||
import hlna
|
||
|
||
list_config = hlna.list_config
|
||
|
||
|
||
def start():
|
||
if not os.path.exists(f"{hlna.home_dir}/.config/hlna/bot/hlna_bot"):
|
||
key_api = input("Укажите ключ бота\n")
|
||
os.chdir(f"{hlna.home_dir}/.config/hlna/")
|
||
try:
|
||
os.mkdir("bot")
|
||
except OSError as e:
|
||
if e.errno == errno.EEXIST:
|
||
print('Каталог создан')
|
||
else:
|
||
raise
|
||
create_config(key_api)
|
||
else:
|
||
x = input("Уже есть запомненный ключа api. Запустить его? Y/n \n")
|
||
if x == "" or x.lower == 'y':
|
||
read_config()
|
||
|
||
elif x == "n":
|
||
os.remove(f"{hlna.home_dir}/.config/hlna/bot/hlna_bot")
|
||
start()
|
||
|
||
|
||
def check_status(m):
|
||
for i in list_config:
|
||
data = hlna.read_yaml(i)
|
||
hlna.print_line(data)
|
||
x = os.system(f"lsof -w -i :{data['Port']}")
|
||
n_srv = data['SessionName']
|
||
if x != 0:
|
||
hlna_bot.send_message(m.chat.id, text=f"Сервер {n_srv} не работает 🛑")
|
||
elif x == 0:
|
||
hlna_bot.send_message(m.chat.id, text=f"Сервер {n_srv} запустился 🟢")
|
||
|
||
|
||
def read_config():
|
||
with open(f"{hlna.home_dir}/.config/hlna/bot/hlna_bot", "r") as yamlfile:
|
||
data = yaml.load(yamlfile, Loader=yaml.FullLoader)
|
||
return data[0]
|
||
|
||
|
||
def create_config(key_api):
|
||
config_hlna_bot = [
|
||
{
|
||
'key_api': key_api,
|
||
}
|
||
]
|
||
with open(f"{hlna.home_dir}/.config/hlna/bot/hlna_bot", 'w') as yamlfile:
|
||
yaml.dump(config_hlna_bot, yamlfile)
|
||
print(colorama.Fore.GREEN + "Конфиг бота создан" + colorama.Style.RESET_ALL)
|
||
|
||
|
||
start()
|
||
|
||
|
||
data = read_config()
|
||
k_api = data['key_api']
|
||
hlna_bot = telebot.TeleBot(f"{k_api}", parse_mode=None)
|
||
|
||
|
||
@hlna_bot.message_handler(commands=["start"])
|
||
def m_mainmenu(m):
|
||
menu = types.ReplyKeyboardMarkup(resize_keyboard=True)
|
||
k_start = types.KeyboardButton("🦖 Старт")
|
||
k_stop = types.KeyboardButton("🦇 Выключить")
|
||
menu.add(k_start, k_stop)
|
||
|
||
k_restart = types.KeyboardButton("🦕 Перезагрузить")
|
||
k_update = types.KeyboardButton("🔸 Обновить")
|
||
menu.add(k_restart, k_update)
|
||
|
||
k_status = types.KeyboardButton("🧾 Статус")
|
||
menu.add(k_status)
|
||
|
||
k_mods = types.KeyboardButton("⏬ Установка модов")
|
||
menu.add(k_mods)
|
||
|
||
hlna_bot.send_message(m.chat.id, text="Привет, {0.first_name}! Это панель управления сервером".format(m.from_user),
|
||
reply_markup=menu)
|
||
|
||
|
||
@hlna_bot.message_handler(content_types=["text"])
|
||
def m_chose(m):
|
||
if m.text.strip() == '🦖 Старт':
|
||
if list_config == []:
|
||
menu_install = types.ReplyKeyboardMarkup(resize_keyboard=True)
|
||
k_install = types.KeyboardButton("Установить")
|
||
k_back = types.KeyboardButton("Назад")
|
||
menu_install.add(k_back, k_install)
|
||
hlna_bot.send_message(m.chat.id, text="{0.first_name}, ниодна карта не установлена".format(m.from_user),
|
||
reply_markup=menu_install)
|
||
hlna_bot.register_next_step_handler(m, m_install)
|
||
else:
|
||
hlna.read_yaml(list_config[-1], flag=True)
|
||
menu_start = types.ReplyKeyboardMarkup(resize_keyboard=True)
|
||
for i in list_config:
|
||
k_mapname = types.KeyboardButton(i)
|
||
menu_start.add(k_mapname)
|
||
k_back = types.KeyboardButton("Назад")
|
||
menu_start.add(k_back)
|
||
hlna_bot.send_message(m.chat.id, text="{0.first_name}, выбери карту".format(m.from_user), reply_markup=menu_start)
|
||
elif m.text.strip() == '🧾 Статус':
|
||
check_status(m)
|
||
elif m.text.strip() == 'Назад':
|
||
m_mainmenu(m)
|
||
for i in list_config:
|
||
if m.text.strip() == f'{i}':
|
||
hlna.print_line(f"Старт карты {i}")
|
||
hlna.start()
|
||
|
||
# @hlna_bot.message_handler(content_types=["text"])
|
||
# def m_mapstart(m):
|
||
# hlna.print_line("Старт карты")
|
||
# for i in list_config:
|
||
# if m.text.strip() == f'{i}':
|
||
# hlna.print_line(f"Старт карты {i}")
|
||
# hlna.start()
|
||
|
||
|
||
def m_install(m):
|
||
if m.text.strip() == 'Установить':
|
||
hlna_bot.send_message(m.chat.id, subprocess.check_output(hlna.config(), shell=True))
|
||
|
||
|
||
print("Бот запущен")
|
||
hlna_bot.polling(none_stop=True, interval=0, timeout=999)
|
||
|
||
# subprocess.run(["arkmanager", "start", "--alwaysrestart", "@all"])
|
||
# elif m.text.strip() == '🦇 Выключить':
|
||
# subprocess.run(["arkmanager", "stop", "@all"])
|
||
# elif m.text.strip() == '🦕 Перезагрузить':
|
||
# subprocess.run(["arkmanager","restart", "--alwaysrestart", "--warn", "@all"])
|
||
# elif (m.text.strip() == '🔸 Обновить'):
|
||
# subprocess.run(["arkmanager", "update", "--update-mods", "@all"])
|
||
#
|
||
#
|
||
# elif (m.text.strip() == '⏬ Установка модов'):
|
||
# bot.send_message(m.from_user.id, "Введите id модификации")
|
||
# bot.register_next_step_handler(m, mods)
|
||
# else:
|
||
# pass
|
||
# else:
|
||
# if m.text.strip() == 'Zxx843Aj':
|
||
# th = Thread(target=check, args=(m,))
|
||
# th.start()
|
||
# id.append(m.chat.id)
|
||
# bot.send_message(m.chat.id, text="{0.first_name} вы авторизованы, нажмите /start для работы с ботом".format(m.from_user))
|
||
# return id
|
||
# def mods(m):
|
||
# id = m.text.strip()
|
||
# subprocess.run(["arkmanager", "installmods", id, "@all"])
|
||
# subprocess.run(["arkmanager", "enablemod", id, "@all"])
|