#!/bin/sh
#============================================================================#
#  Wireless Report Installer                                                 #
#  Version: 1.2.1                                                            #
#  Author: JB_1366                                                           #
#  Revised: add router/node nicknames                                        #
#============================================================================#

# --- Configuration ---
GITHUB_ROOT="https://raw.githubusercontent.com/JB1366/Wireless_Report/main"
INSTALL_DIR="/jffs/addons/wireless_report"
REPORT_SCRIPT="$INSTALL_DIR/gen_report.sh"
MENU_SCRIPT="$INSTALL_DIR/install_menu.sh"
SSH_KEY="/tmp/home/root/.ssh/id_dropbear"
CONF_FILE="$INSTALL_DIR/webui.conf"

# Colors
CYAN='\033[0;36m'
GREEN='\033[0;32m'
RED='\033[0;31m'
NC='\033[0m'

check_version() {
    local GITHUB_URL="$GITHUB_ROOT/gen_report.sh"
    if [ -f "$REPORT_SCRIPT" ]; then
        LOCAL_VER=$(grep "SCRIPT_VERSION=" "$REPORT_SCRIPT" | head -n 1 | cut -d'"' -f2 2>/dev/null)
    else
        LOCAL_VER="NOT INSTALLED"
    fi
    REMOTE_DATA=$(curl -s --connect-timeout 2 "$GITHUB_URL")
    if [ $? -ne 0 ] || [ -z "$REMOTE_DATA" ]; then
        REMOTE_VER=""
    else
        REMOTE_VER=$(echo "$REMOTE_DATA" | grep "SCRIPT_VERSION=" | head -n 1 | cut -d'"' -f2 2>/dev/null)
    fi
    echo -e "${CYAN}==================================================${NC}"
    echo -e "${CYAN}              WIRELESS REPORT AIMESH              ${NC}"
    echo -e "${CYAN}==================================================${NC}"
    if [ -z "$REMOTE_VER" ]; then
        echo -e " STATUS: ${RED}[Offline]${NC} Could not reach GitHub"
    elif [ "$LOCAL_VER" = "NOT INSTALLED" ]; then
        echo -e " STATUS: [Ready] Latest available is v$REMOTE_VER"
    elif [ "$LOCAL_VER" != "$REMOTE_VER" ]; then
        echo -e " STATUS: ${RED}[UPDATE AVAILABLE] v$REMOTE_VER${NC} (Current: v$LOCAL_VER)"
    else
        echo -e " STATUS: [Up to date] v$LOCAL_VER"
    fi
    echo -e "${CYAN}==================================================${NC}"
}

check_storage() {
    echo -e "${CYAN}[*] Checking for USB Storage...${NC}"
    USB_PATH=$(mount | grep -E "ext2|ext3|ext4|tfat|ntfs|vfat" | grep -v "/jffs" | awk '{print $3}' | head -n 1)
    if [ -n "$USB_PATH" ]; then
        DATA_DIR="$USB_PATH/gen_report"
        echo -e "${GREEN}[+] USB Found: Using $DATA_DIR for history.${NC}"
    else
        DATA_DIR="$INSTALL_DIR/data"
        echo -e "${RED}[!] No USB detected: Using JFFS at $DATA_DIR.${NC}"
    fi
    mkdir -p "$DATA_DIR" 2>/dev/null
}

show_menu() {
    echo -e ""
    echo -e "  (1)  Install / Update Wireless Report"
    echo -e "  (2)  Uninstall Wireless Report"
    echo -e "  (3)  Temperature Unit: °F(def) or °C"
    echo -e "  (4)  Set Router/Node Nicknames"
    echo -e "  (e)  Exit"
    echo -e ""
    echo -e "${CYAN}==================================================${NC}"
    printf " Selection: "
}

check_ssh_environment() {
    mkdir -p "$INSTALL_DIR" 2>/dev/null
    # Ensure the file exists so sed does not error out on first install or upgrade
    [ ! -f "$CONF_FILE" ] && touch "$CONF_FILE"

    echo -e "${CYAN}[*] Verifying Passwordless SSH Environment...${NC}"
    
    if [ ! -f "$SSH_KEY" ]; then
        echo -e "${RED}[!] ERROR: Local SSH Key not found at $SSH_KEY${NC}"
        echo -e "${RED}[!] Please setup passwordless SSH Key(s) and try again.${NC}"
        exit 1
    fi

    SSH_PORT=$(nvram get sshd_port)
    [ -z "$SSH_PORT" ] && SSH_PORT=22  

    # Pull both Alias ($2) and IP ($3) from nvram
    NODE_IPS=$(nvram get asus_device_list | sed 's/</\n/g' | grep '>2$' | awk -F '>' '{print $2 "|" $3}' | sort -t . -k 4,4n)
    NODE_USER=$(nvram get http_username)
    
    if [ -z "$NODE_IPS" ]; then
        echo -e "${RED}[!] No AIMesh Nodes detected. This script requires a AIMesh environment.${NC}"
        echo -e "${RED}[!] Installation aborted.${NC}"
        exit 1
    fi

    any_success=0
    VALID_NODES=""
    
    for line in $NODE_IPS; do
        ALIAS=$(echo "$line" | cut -d'|' -f1)
        IP=$(echo "$line" | cut -d'|' -f2)

        echo -ne "[*] Testing Passwordless SSH to $ALIAS ($IP) on port $SSH_PORT... "
        /usr/bin/ssh -p "$SSH_PORT" -i "$SSH_KEY" -o StrictHostKeyChecking=no -o ConnectTimeout=3 -o BatchMode=yes "${NODE_USER}@${IP}" "exit" >/dev/null 2>&1
        
        if [ $? -eq 0 ]; then
            echo -e "${GREEN}AUTHENTICATED${NC}"
            any_success=1
            VALID_NODES="$VALID_NODES $ALIAS|$IP"
        else
            echo -e "${RED}FAILED${NC}"
        fi
    done

    if [ "$any_success" -eq 0 ]; then
        echo -e "${RED}[!] Error: No nodes authenticated. Setup SSH Keys and try again.${NC}"
        exit 1
    fi

    # Save ONLY the authenticated nodes to webui.conf
    sed -i '/SSH_NODES=/d' "$CONF_FILE"
    echo "SSH_NODES=\"$VALID_NODES\"" >> "$CONF_FILE"
}

do_install() {
    if [ -f "$REPORT_SCRIPT" ]; then
        LOCAL_VER=$(grep "SCRIPT_VERSION=" "$REPORT_SCRIPT" | head -n 1 | cut -d'"' -f2 2>/dev/null)
        REMOTE_DATA=$(curl -s --connect-timeout 2 "$GITHUB_ROOT/gen_report.sh")
        REMOTE_VER=$(echo "$REMOTE_DATA" | grep "SCRIPT_VERSION=" | head -n 1 | cut -d'"' -f2 2>/dev/null)
        
        if [ "$LOCAL_VER" = "$REMOTE_VER" ] && [ -n "$REMOTE_VER" ]; then
            echo -e "\n${GREEN}[i] You are already on the latest version (v$LOCAL_VER).${NC}"
            printf " Do you want to reinstall/overwrite anyway? (y/n): "
            read confirm_reinstall
            if [ "$confirm_reinstall" != "y" ] && [ "$confirm_reinstall" != "Y" ]; then
                return
            fi
        fi
    fi

    if [ "$(nvram get jffs2_scripts)" != "1" ]; then
        echo -e "${RED}[!] ERROR: JFFS custom scripts not enabled.${NC}"
        exit 1
    fi

    check_storage
    check_ssh_environment
    
    echo -e "${CYAN}[*] Processing Wireless Report Files...${NC}"
    
    if ! grep -q "REPORT_UNIT=" "$CONF_FILE" 2>/dev/null; then 
        echo "REPORT_UNIT=F" >> "$CONF_FILE"
    fi

    [ -f "/tmp/menuTree.js" ] && sed -i '/Wireless Report/d' /tmp/menuTree.js 2>/dev/null

    curl -s --connect-timeout 5 "$GITHUB_ROOT/gen_report.sh" -o "$REPORT_SCRIPT"
    curl -s --connect-timeout 5 "$GITHUB_ROOT/install_menu.sh" -o "$MENU_SCRIPT"
    chmod +x "$REPORT_SCRIPT" "$MENU_SCRIPT" 2>/dev/null
    
    if [ -f "$MENU_SCRIPT" ]; then
        sh "$MENU_SCRIPT" >/dev/null 2>&1
        
        [ ! -f "/jffs/scripts/services-start" ] && echo "#!/bin/sh" > /jffs/scripts/services-start
        sed -i "\|$MENU_SCRIPT|d" /jffs/scripts/services-start
        echo "sh $MENU_SCRIPT # Inject Wireless Report" >> /jffs/scripts/services-start
        chmod +x /jffs/scripts/services-start
        
        [ ! -f "/jffs/scripts/service-event" ] && echo "#!/bin/sh" > /jffs/scripts/service-event
        sed -i "/wireless_report/d" /jffs/scripts/service-event
        echo "if [ \"\$1\" = \"restart\" ] && [ \"\$2\" = \"wireless_report\" ]; then sh $REPORT_SCRIPT; fi" >> /jffs/scripts/service-event
        chmod +x /jffs/scripts/service-event
        
        service restart_httpd >/dev/null 2>&1 || killall -HUP httpd >/dev/null 2>&1
        
        sh "$REPORT_SCRIPT" >/dev/null 2>&1 &
        
        echo -e "\n${GREEN}SUCCESS: Installation complete!${NC}"
        echo -e "${CYAN}[i] Use Option 4 if you wish to set custom nicknames.${NC}"
    else
        echo -e "${RED}[!] ERROR: Download failed.${NC}"
    fi
    pause
}

do_uninstall() {
    # 1. THE GATEKEEPER: Don't run if folder is missing
    if [ ! -d "$INSTALL_DIR" ]; then
        echo -e "\n${RED}[!] Wireless Report is not installed.${NC}"
        pause
        return
    fi

    echo -e "\n${RED}[!] WARNING: Removing Wireless Report...${NC}"
    printf " Are you sure? (y/n): "
    read confirm
    if [ "$confirm" = "y" ] || [ "$confirm" = "Y" ]; then
        [ -f "$CONF_FILE" ] && . "$CONF_FILE"

        # 2. REVERSE THE MOUNT (The "Un-insert" step)
        if mount | grep -q "menuTree.js"; then
            echo -e "${CYAN}[*] Detaching menu for cleaning...${NC}"
            # Force unmount so the WebUI stops looking at our modified file
            umount -l /www/require/modules/menuTree.js >/dev/null 2>&1
            
            # Remove the Wireless Report line from the source file
            sed -i '/tabName:[[:space:]]*"Wireless Report"/d' /tmp/menuTree.js
            
            # 3. COEXISTENCE CHECK: Did we leave other addons behind?
            if grep -q "tabName" /tmp/menuTree.js; then
                mount --bind /tmp/menuTree.js /www/require/modules/menuTree.js
            else
                echo -e "${CYAN}[*] Restoring menu to factory default.${NC}"
            fi
        fi

        if [ -n "$INSTALLED_PAGE" ]; then
            logger -t "Wireless Report" "Unmounting Wireless/Wireless Report as $INSTALLED_PAGE"
			umount -l "/www/user/$INSTALLED_PAGE" >/dev/null 2>&1
			rm -f "/www/user/$INSTALLED_PAGE" >/dev/null 2>&1
        fi

        sed -i "\|$MENU_SCRIPT|d" /jffs/scripts/services-start
        sed -i "/wireless_report/d" /jffs/scripts/service-event
        killall gen_report.sh >/dev/null 2>&1
        service restart_httpd >/dev/null 2>&1 || killall -HUP httpd >/dev/null 2>&1
        rm -rf "$INSTALL_DIR" 2>/dev/null
        rm -f /tmp/wireless.asp 2>/dev/null
        echo -e "${GREEN}[+] Uninstalled. Wireless Report tab is gone.${NC}"
    fi
    pause
}

pause() { printf "\nPress [Enter] to return..."; read discard; }

set_temp_unit() {
    clear
    echo -e "${CYAN}==================================================${NC}"
    echo -e "             TEMPERATURE UNIT SETTINGS                        "
    echo -e "${CYAN}==================================================${NC}"
    
    [ -f "$CONF_FILE" ] && . "$CONF_FILE"
    CURRENT_UNIT=${REPORT_UNIT:-F}
    
    echo -e " Current Setting: ${GREEN}$CURRENT_UNIT${NC}"
    echo -e ""
    echo -e "  (1)  Fahrenheit (°F) - Default"
    echo -e "  (2)  Celsius (°C)"
    echo -e "  (r)  Return to main menu"
    echo -e ""
    printf " Selection: "
    read t_choice

    case "$t_choice" in
        1) NEW_UNIT="F" ;;
        2) NEW_UNIT="C" ;;
        r|R) return ;;
        *) echo -e "${RED}Invalid selection.${NC}"; sleep 1; set_temp_unit; return ;;
    esac

    # Update the config file
    if [ ! -f "$CONF_FILE" ]; then
        echo "REPORT_UNIT=$NEW_UNIT" > "$CONF_FILE"
    else
        # Remove old entry if it exists and append new one
        sed -i '/REPORT_UNIT=/d' "$CONF_FILE"
        echo "REPORT_UNIT=$NEW_UNIT" >> "$CONF_FILE"
    fi

    echo -e "${GREEN}[+] Temperature unit set to $NEW_UNIT${NC}"
    
    # Trigger a report regeneration in the background to apply changes immediately
    [ -f "$REPORT_SCRIPT" ] && sh "$REPORT_SCRIPT" >/dev/null 2>&1 &
    
    pause
}
set_nicknames() {
    clear
    RAW_NODES=$(nvram get asus_device_list | sed 's/</\n/g' | grep '>2$')
    MAIN_IP=$(nvram get lan_ipaddr)
    MAIN_HW_MODEL=$(nvram get modelid)
    [ -z "$MAIN_HW_MODEL" ] && MAIN_HW_MODEL=$(nvram get productid)
    
    [ -f "$CONF_FILE" ] && . "$CONF_FILE"

    echo -e "${CYAN}==================================================${NC}"
    echo -e "              ROUTER NICKNAME SETUP               "
    echo -e "      (Press [Enter] to keep the current name)     "
    echo -e "${CYAN}==================================================${NC}"
    echo -e ""

    CURRENT_MAIN_NICK=${MAIN_NICK:-"$MAIN_HW_MODEL"}
    printf " Main $MAIN_IP [$CURRENT_MAIN_NICK]: "
    read input_main
    
    MAIN_NICK=${input_main:-"$CURRENT_MAIN_NICK"}
    sed -i '/^MAIN_NICK=/d' "$CONF_FILE"
    echo "MAIN_NICK=\"$MAIN_NICK\"" >> "$CONF_FILE"

    if [ -n "$RAW_NODES" ]; then
        VALID_NODES=$(echo "$RAW_NODES" | awk -F '>' '{print $2 "|" $3}' | sort -t . -k 4,4n)
        for node in $VALID_NODES; do
            ALIAS=$(echo "$node" | cut -d'|' -f1)
            IP=$(echo "$node" | cut -d'|' -f2)
            CLEAN_IP=$(echo "$IP" | tr '.' '_')
            
            # Fetch existing nickname or fallback to hardware model
            eval SAVED_NICK=\$NODE_NICK_$CLEAN_IP
            CURRENT_DISPLAY=${SAVED_NICK:-"$ALIAS"}
            
            # CLEAN PROMPT: Node 192.168.50.3 [RT-AX86U_Pro]:
            printf " Node $IP [$CURRENT_DISPLAY]: "
            read input_node
            
            NEW_NICK=${input_node:-"$CURRENT_DISPLAY"}
            sed -i "/^NODE_NICK_$CLEAN_IP=/d" "$CONF_FILE"
            echo "NODE_NICK_$CLEAN_IP=\"$NEW_NICK\"" >> "$CONF_FILE"
        done
    fi
    echo -e "\n${GREEN}[+] Nicknames updated successfully.${NC}"
}
while true; do
    clear; check_version; show_menu; read choice
    case "$choice" in
        1) do_install ;;
        2) do_uninstall ;;
        3) set_temp_unit ;;
        4) 
            if [ -f "$CONF_FILE" ]; then
                set_nicknames
                [ -f "$REPORT_SCRIPT" ] && sh "$REPORT_SCRIPT" >/dev/null 2>&1 &
                pause
            else
                echo -e "${RED}[!] Please install the report (Option 1) first.${NC}"
                sleep 2
            fi
            ;;
        e|E) clear; exit 0 ;;
        *) echo -e "${RED}Invalid selection.${NC}"; sleep 1 ;;
    esac
done