bivashy/Blitz-Proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(core): implement full database cleanup in uninstaller

Browse Source
This commit is contained in:
Whispering Wind
2025-09-08 00:05:48 +03:30
committed by GitHub
parent ed37def474
commit 09fc8540b6
1 changed files with 78 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

133
core/scripts/hysteria2/uninstall.py
View File

@ -1,11 +1,16 @@
#!/usr/bin/env python3
import os import os
import subprocess import subprocess
import sys import sys
from pathlib import Path
try:
import pymongo
except ImportError:
pymongo = None
SERVICES = [ SERVICES = [
"hysteria-server.service", "hysteria-server.service",
"hysteria-auth.service",
"hysteria-webpanel.service", "hysteria-webpanel.service",
"hysteria-caddy.service", "hysteria-caddy.service",
"hysteria-telegram-bot.service", "hysteria-telegram-bot.service",
@ -15,81 +20,99 @@ SERVICES = [
"hysteria-scheduler.service", "hysteria-scheduler.service",
] ]
DB_NAME = "blitz_panel"
def run_command(command, error_message): def run_command(command, error_message):
"""Runs a command and prints an error message if it fails."""
try: try:
subprocess.run(command, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL) subprocess.run(command, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
return 0 return True
except subprocess.CalledProcessError: except subprocess.CalledProcessError:
print(error_message) print(f"Warning: {error_message}")
return 1 return False
except FileNotFoundError: except FileNotFoundError:
print(f"Error: Command not found: {command[0]}") print(f"Warning: Command not found: {command[0]}")
return 1 return False
def drop_mongodb_database():
if not pymongo:
print("Warning: pymongo library not found. Skipping database cleanup.")
return
print(f"Attempting to drop MongoDB database: '{DB_NAME}'...")
try:
client = pymongo.MongoClient("mongodb://localhost:27017/", serverSelectionTimeoutMS=5000)
client.server_info()
client.drop_database(DB_NAME)
print(f"Database '{DB_NAME}' dropped successfully.")
except pymongo.errors.ConnectionFailure:
print("Warning: Could not connect to MongoDB. Skipping database cleanup.")
except Exception as e:
print(f"An error occurred during database cleanup: {e}")
def uninstall_hysteria(): def uninstall_hysteria():
"""Uninstalls Hysteria2.""" print("Uninstalling Hysteria2 and all related components...")
print("Uninstalling Hysteria2...")
print("Running uninstallation script...") print("\nStep 1: Stopping and disabling all Hysteria services...")
run_command(["bash", "-c", "curl -fsSL https://get.hy2.sh/ | bash -- --remove"], "Error running the official uninstallation script.") for service in SERVICES:
run_command(["systemctl", "stop", service], f"Failed to stop {service}.")
run_command(["systemctl", "disable", service], f"Failed to disable {service}.")
print("Removing WARP") print("\nStep 2: Removing systemd service files...")
systemd_path = Path("/etc/systemd/system")
for service in SERVICES:
service_file = systemd_path / service
if service_file.exists():
run_command(["rm", "-f", str(service_file)], f"Failed to remove {service_file}")
print("Reloading systemd daemon...")
run_command(["systemctl", "daemon-reload"], "Failed to reload systemd daemon.")
print("\nStep 3: Removing Hysteria binaries...")
run_command(["bash", "-c", "curl -fsSL https://get.hy2.sh/ | bash -- --remove"], "Failed to run the official Hysteria uninstallation script.")
print("\nStep 4: Cleaning MongoDB database...")
drop_mongodb_database()
print("\nStep 5: Removing related components (WARP)...")
cli_path = "/etc/hysteria/core/cli.py" cli_path = "/etc/hysteria/core/cli.py"
if os.path.exists(cli_path): if os.path.exists(cli_path):
run_command([sys.executable, cli_path, "uninstall-warp"], "Error during WARP removal.") run_command([sys.executable, cli_path, "uninstall-warp"], "Failed during WARP removal.")
else: else:
print("Skipping WARP removal (CLI path not found)") print("Skipping WARP removal (CLI path not found)")
print("Removing Hysteria folder...") print("\nStep 6: Removing all Hysteria panel files...")
run_command(["rm", "-rf", "/etc/hysteria"], "Error removing the Hysteria folder.") run_command(["rm", "-rf", "/etc/hysteria"], "Failed to remove the /etc/hysteria folder.")
print("Deleting hysteria user...") print("\nStep 7: Deleting the 'hysteria' user...")
run_command(["userdel", "-r", "hysteria"], "Error deleting the hysteria user.") run_command(["userdel", "-r", "hysteria"], "Failed to delete the 'hysteria' user.")
print("Stop/Disabling Hysteria Services...") print("\nStep 8: Removing cron jobs...")
for service in SERVICES + ["hysteria-server@*.service"]:
print(f"Stopping and disabling {service}...")
run_command(["systemctl", "stop", service], f"Error stopping {service}.")
run_command(["systemctl", "disable", service], f"Error disabling {service}.")
print("Removing systemd service files...")
for service in SERVICES + ["hysteria-server@*.service"]:
print(f"Removing service file: {service}")
run_command(["rm", "-f", f"/etc/systemd/system/{service}", f"/etc/systemd/system/multi-user.target.wants/{service}"], f"Error removing service files for {service}.")
print("Reloading systemd daemon...")
run_command(["systemctl", "daemon-reload"], "Error reloading systemd daemon.")
print("Removing cron jobs...")
try: try:
crontab_list = subprocess.run(["crontab", "-l"], capture_output=True, text=True, check=False) crontab_list_proc = subprocess.run(["crontab", "-l"], capture_output=True, text=True, check=False)
if "hysteria" in crontab_list.stdout: if "hysteria" in crontab_list_proc.stdout:
new_crontab = "\n".join(line for line in crontab_list.stdout.splitlines() if "hysteria" not in line) new_crontab = "\n".join(line for line in crontab_list_proc.stdout.splitlines() if "hysteria" not in line)
process = subprocess.run(["crontab", "-"], input=new_crontab.encode(), check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL) subprocess.run(["crontab", "-"], input=new_crontab.encode(), check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
except FileNotFoundError: print("Hysteria cron jobs removed.")
print("Warning: crontab command not found.") except (FileNotFoundError, subprocess.CalledProcessError):
except subprocess.CalledProcessError: print("Warning: Could not access or modify crontab.")
print("Warning: Could not access crontab.")
print("Removing alias 'hys2' from .bashrc...") print("\nStep 9: Removing 'hys2' alias from .bashrc...")
bashrc_path = os.path.expanduser("~/.bashrc") bashrc_path = os.path.expanduser("~/.bashrc")
if os.path.exists(bashrc_path): if os.path.exists(bashrc_path):
try: try:
with open(bashrc_path, 'r') as f: with open(bashrc_path, 'r') as f_in:
lines = f.readlines() lines = [line for line in f_in if 'alias hys2=' not in line]
with open(bashrc_path, 'w') as f: with open(bashrc_path, 'w') as f_out:
for line in lines: f_out.writelines(lines)
if 'alias hys2=' not in line: print("Alias 'hys2' removed from .bashrc.")
f.write(line)
except IOError: except IOError:
print(f"Warning: Could not access or modify {bashrc_path}.") print(f"Warning: Could not access or modify {bashrc_path}.")
else:
print(f"Warning: {bashrc_path} not found.") print("\nUninstallation of Hysteria2 panel is complete.")
print("It is recommended to reboot the server to ensure all changes take effect.")
print("Hysteria2 uninstalled!")
print("Rebooting server...")
run_command(["reboot"], "Error initiating reboot.")
if __name__ == "__main__": if __name__ == "__main__":
if os.geteuid() != 0:
print("Error: This script must be run as root.")
sys.exit(1)
uninstall_hysteria() uninstall_hysteria()