75 lines
2.7 KiB
Python
75 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
core_scripts_dir = Path(__file__).resolve().parents[1]
|
|
if str(core_scripts_dir) not in sys.path:
|
|
sys.path.append(str(core_scripts_dir))
|
|
|
|
from paths import TELEGRAM_ENV
|
|
|
|
|
|
|
|
|
|
def update_env_file(api_token, admin_user_ids):
|
|
TELEGRAM_ENV.write_text(f"""API_TOKEN={api_token}
|
|
ADMIN_USER_IDS=[{admin_user_ids}]
|
|
""")
|
|
|
|
def create_service_file():
|
|
Path("/etc/systemd/system/hysteria-telegram-bot.service").write_text("""[Unit]
|
|
Description=Hysteria Telegram Bot
|
|
After=network.target
|
|
|
|
[Service]
|
|
ExecStart=/bin/bash -c 'source /etc/hysteria/hysteria2_venv/bin/activate && /etc/hysteria/hysteria2_venv/bin/python /etc/hysteria/core/scripts/telegrambot/tbot.py'
|
|
WorkingDirectory=/etc/hysteria/core/scripts/telegrambot
|
|
Restart=always
|
|
|
|
[Install]
|
|
WantedBy=multi-user.target
|
|
""")
|
|
|
|
def start_service(api_token, admin_user_ids):
|
|
if subprocess.run(["systemctl", "is-active", "--quiet", "hysteria-telegram-bot.service"]).returncode == 0:
|
|
print("The hysteria-telegram-bot.service is already running.")
|
|
return
|
|
|
|
update_env_file(api_token, admin_user_ids)
|
|
create_service_file()
|
|
|
|
subprocess.run(["systemctl", "daemon-reload"])
|
|
subprocess.run(["systemctl", "enable", "hysteria-telegram-bot.service"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
subprocess.run(["systemctl", "start", "hysteria-telegram-bot.service"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
|
|
if subprocess.run(["systemctl", "is-active", "--quiet", "hysteria-telegram-bot.service"]).returncode == 0:
|
|
print("Hysteria bot setup completed. The service is now running.\n")
|
|
else:
|
|
print("Hysteria bot setup completed. The service failed to start.")
|
|
|
|
def stop_service():
|
|
subprocess.run(["systemctl", "stop", "hysteria-telegram-bot.service"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
subprocess.run(["systemctl", "disable", "hysteria-telegram-bot.service"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
TELEGRAM_ENV.unlink(missing_ok=True)
|
|
print("\nHysteria bot service stopped and disabled. .env file removed.")
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) < 2:
|
|
print("Usage: python3 runbot.py {start|stop} <API_TOKEN> <ADMIN_USER_IDS>")
|
|
sys.exit(1)
|
|
|
|
action = sys.argv[1]
|
|
|
|
if action == "start":
|
|
if len(sys.argv) != 4:
|
|
print("Usage: python3 runbot.py start <API_TOKEN> <ADMIN_USER_IDS>")
|
|
sys.exit(1)
|
|
start_service(sys.argv[2], sys.argv[3])
|
|
elif action == "stop":
|
|
stop_service()
|
|
else:
|
|
print("Usage: python3 runbot.py {start|stop} <API_TOKEN> <ADMIN_USER_IDS>")
|
|
sys.exit(1)
|