#!/bin/bash

# Ensure necessary packages are installed
if ! command -v jq &> /dev/null || ! command -v qrencode &> /dev/null || ! command -v curl &> /dev/null; then
    echo "Necessary packages are not installed. Please wait while they are being installed..."
    apt-get update -qq && apt-get install jq qrencode curl -y >/dev/null 2>&1
fi

# Function to install and configure Hysteria2
install_and_configure() {
    echo "Installing and configuring Hysteria2..."
    bash <(curl -s https://raw.githubusercontent.com/H-Return/Hysteria2/main/install.sh)
    echo -e "\n\n\n"
    echo "Installation and configuration complete."
}

# Function to update Hysteria2
update_core() {
    echo "Starting the update process for Hysteria2..." 
    echo "Backing up the current configuration..."
    cp /etc/hysteria/config.json /etc/hysteria/config_backup.json
    sleep 1
    echo "Downloading and installing the latest version of Hysteria2..."
    bash <(curl -fsSL https://get.hy2.sh/) >/dev/null 2>&1
    echo "Restoring configuration from backup..."
    mv /etc/hysteria/config_backup.json /etc/hysteria/config.json
    echo "Modifying systemd service to use config.json..."
    sed -i 's|/etc/hysteria/config.yaml|/etc/hysteria/config.json|' /etc/systemd/system/hysteria-server.service
    systemctl daemon-reload >/dev/null 2>&1
    systemctl restart hysteria-server.service >/dev/null 2>&1
    sleep 1
    echo "Hysteria2 has been successfully updated."
    echo ""
}


# Function to install TCP Brutal
install_tcp_brutal() {
    echo "Installing TCP Brutal..."
    bash <(curl -fsSL https://tcp.hy2.sh/)
    sleep 3
    clear
    echo "TCP Brutal installation complete."
}

# Function to change port
change_port() {
    while true; do
        read -p "Enter the new port number you want to use: " port
        if ! [[ "$port" =~ ^[0-9]+$ ]] || [ "$port" -lt 1 ] || [ "$port" -gt 65535 ]; then
            echo "Invalid port number. Please enter a number between 1 and 65535."
        else
            break
        fi
    done

    if [ -f "/etc/hysteria/config.json" ]; then
        jq --arg port "$port" '.listen = ":" + $port' /etc/hysteria/config.json > /etc/hysteria/config_temp.json && mv /etc/hysteria/config_temp.json /etc/hysteria/config.json
        systemctl restart hysteria-server.service >/dev/null 2>&1
        echo "Port changed successfully to $port."
    else
        echo "Error: Config file /etc/hysteria/config.json not found."
    fi
}

# Function to show URI if Hysteria2 is installed and active
show_uri() {
    if [ -f "/etc/hysteria/config.json" ]; then
        port=$(jq -r '.listen' /etc/hysteria/config.json | cut -d':' -f2)
        sha256=$(jq -r '.tls.pinSHA256' /etc/hysteria/config.json)
        obfspassword=$(jq -r '.obfs.salamander.password' /etc/hysteria/config.json)
        authpassword=$(jq -r '.auth.password' /etc/hysteria/config.json)

        if systemctl is-active --quiet hysteria-server.service; then
            IP=$(curl -s -4 ip.sb)
            IP6=$(curl -s -6 ip.sb)
            URI="hy2://$authpassword@$IP:$port?obfs=salamander&obfs-password=$obfspassword&pinSHA256=$sha256&insecure=1&sni=bing.com#Hysteria2-IPv4"
            URI6="hy2://$authpassword@[$IP6]:$port?obfs=salamander&obfs-password=$obfspassword&pinSHA256=$sha256&insecure=1&sni=bing.com#Hysteria2-IPv6"

            cols=$(tput cols)
            rows=$(tput lines)
            qr1=$(echo -n "$URI" | qrencode -t UTF8 -s 3 -m 2)
            qr2=$(echo -n "$URI6" | qrencode -t UTF8 -s 3 -m 2)

            echo -e "\nIPv4:\n"
            echo "$qr1" | while IFS= read -r line; do
                printf "%*s\n" $(( (${#line} + cols) / 2)) "$line"
            done

            echo -e "\nIPv6:\n"
            echo "$qr2" | while IFS= read -r line; do
                printf "%*s\n" $(( (${#line} + cols) / 2)) "$line"
            done
            echo
            echo "IPv4: $URI"
            echo
            echo "IPv6: $URI6"
            echo
        else
            echo "Error: Hysteria2 is not active."
        fi
    else
        echo "Error: Config file /etc/hysteria/config.json not found."
    fi
}

# Function to check traffic status
traffic_status() {
    green='\033[0;32m'
    cyan='\033[0;36m'
    NC='\033[0m'

    secret=$(jq -r '.trafficStats.secret' /etc/hysteria/config.json)

    if [ -z "$secret" ]; then
        echo "Error: Secret not found in config.json"
        return
    fi

    response=$(curl -s -H "Authorization: $secret" http://127.0.0.1:25413/traffic)
    if [ -z "$response" ] || [ "$response" = "{}" ]; then
        echo -e "Upload (TX): ${green}0B${NC}"
        echo -e "Download (RX): ${cyan}0B${NC}"
        return
    fi

    tx_bytes=$(echo "$response" | jq -r '.user.tx // 0')
    rx_bytes=$(echo "$response" | jq -r '.user.rx // 0')

    format_bytes() {
        bytes=$1
        if [ "$bytes" -lt 1024 ]; then
            echo "${bytes}B"
        elif [ "$bytes" -lt 1048576 ]; then
            echo "$(bc <<< "scale=2; $bytes / 1024")KB"
        elif [ "$bytes" -lt 1073741824 ]; then
            echo "$(bc <<< "scale=2; $bytes / 1048576")MB"
        elif [ "$bytes" -lt 1099511627776 ]; then
            echo "$(bc <<< "scale=2; $bytes / 1073741824")GB"
        else
            echo "$(bc <<< "scale=2; $bytes / 1099511627776")TB"
        fi
    }

    echo -e "Upload (TX): ${green}$(format_bytes "$tx_bytes")${NC}"
    echo -e "Download (RX): ${cyan}$(format_bytes "$rx_bytes")${NC}"
}

# Function to uninstall Hysteria2
uninstall_hysteria() {
    echo "Uninstalling Hysteria2..."
    sleep 1
    echo "Running uninstallation script..."
    bash <(curl -fsSL https://get.hy2.sh/) --remove >/dev/null 2>&1
    sleep 1
    echo "Removing Hysteria folder..."
    rm -rf /etc/hysteria >/dev/null 2>&1
    sleep 1
    echo "Deleting hysteria user..."
    userdel -r hysteria >/dev/null 2>&1
    sleep 1
    echo "Removing systemd service files..."
    rm -f /etc/systemd/system/multi-user.target.wants/hysteria-server.service >/dev/null 2>&1
    rm -f /etc/systemd/system/multi-user.target.wants/hysteria-server@*.service >/dev/null 2>&1
    sleep 1
    echo "Reloading systemd daemon..."
    systemctl daemon-reload >/dev/null 2>&1
    sleep 1
    echo "Hysteria2 uninstalled!"
    echo ""
}

# Main menu
main_menu() {
    clear
    echo "===== Hysteria2 & TCP Brutal Setup Menu ====="
    echo "1. Install and Configure Hysteria2"
    echo "2. Update Hysteria2"
    echo "3. Install TCP Brutal"
    echo "4. Change Port (Hysteria2)"
    echo "5. Show URI (Hysteria2)"
    echo "6. Check Traffic Status (Hysteria2)"
    echo "7. Uninstall Hysteria2"
    echo "8. Exit"
    echo "============================================="

    read -p "Enter your choice: " choice
    case $choice in
        1) install_and_configure ;;
        2) update_core ;;
        3) install_tcp_brutal ;;
        4) change_port ;;
        5) show_uri ;;
        6) traffic_status ;;
        7) uninstall_hysteria ;;
        8) exit 0 ;;
        *) echo "Invalid option. Please try again." ;;
    esac
    read -p "Press any key to return to the menu..."
}

# Loop to display the menu repeatedly
while true; do
    main_menu
done