Refactor: Implement Telegram bot service manager in Python

core/scripts/telegrambot/runbot.py

@@ -0,0 +1,74 @@
+#!/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)