#!/bin/bash
set -euo pipefail

# Constants for color formatting
RESET="\033[0m"
declare -A COLORS=(
    ["bright_white"]="\033[97m" ["black"]="\033[30m" ["red"]="\033[31m"
    ["green"]="\033[32m" ["yellow"]="\033[33m" ["blue"]="\033[34m"
    ["magenta"]="\033[35m" ["cyan"]="\033[36m" ["white"]="\033[37m"
    ["bg_black"]="\033[40m" ["bg_red"]="\033[41m" ["bg_green"]="\033[42m"
    ["bg_yellow"]="\033[43m" ["bg_blue"]="\033[44m" ["bg_magenta"]="\033[45m"
    ["bg_cyan"]="\033[46m" ["bg_white"]="\033[47m" ["bg_bright_black"]="\033[100m"
    ["bg_bright_red"]="\033[101m" ["bg_bright_green"]="\033[102m"
    ["bg_bright_yellow"]="\033[103m" ["bg_bright_blue"]="\033[104m"
    ["bg_bright_magenta"]="\033[105m" ["bg_bright_cyan"]="\033[106m"
    ["bg_bright_white"]="\033[107m"
)

# Helper functions
color_text() { echo -e "${COLORS[$1]}$2${RESET}"; }

center_text() {
    local text="$1" cols=$(tput cols)
    while IFS= read -r line; do
        local padding=$(( (cols - ${#line}) / 2 ))
        printf "%*s%s\n" "$padding" " " "$line"
    done <<< "$text"
}

# System information functions
display_system_info() {
    local os=$(awk -F= '/^PRETTY_NAME/ {gsub(/"/, "", $2); print $2}' /etc/os-release)
    local hostname="$(hostname)"
    local model="$(awk '!(NR%2){print$1,p}{p=$0}' /sys/devices/virtual/dmi/id/board_{name,vendor})"
    local kernel_name="$(uname -s)"
    local kernel_ver="$(uname -r)"
    local uptime="$(uptime -p)"
    center_text "$(color_text "bg_green" "Operating System: $os")"
    center_text "$(color_text "bg_bright_green" "Host: $hostname")"
    center_text "$(color_text "bg_bright_blue" "Model: $model")"
    center_text "$(color_text "bg_yellow" "Kernel: $kernel_name $kernel_ver")"
    center_text "$(color_text "bg_cyan" "Uptime: $uptime")"
}

display_memory_usage() {
    local mem=$(free -h --si | awk '/^Mem/ {print $3 "/" $2}')
    local mem_pct=$(free --si | awk '/^Mem/ {printf("%.2f%%", $3/$2 * 100)}')
    local swap=$(free -h --si | awk '/^Swap/ {print $3 "/" $2}')
    local swap_pct=$(free --si | awk '/^Swap/ {printf("%.2f%%", $3/$2 * 100)}')
    center_text "$(color_text "bg_bright_cyan" "Memory: $mem ($mem_pct)")"
    center_text "$(color_text "bg_bright_yellow" "Swap: $swap ($swap_pct)")"
}

display_cpu_info() {
    local cpu_model=$(awk -F: '/^model name/ {print $2; exit}' /proc/cpuinfo | sed 's/^[ \t]*//')
    local cpu_cores=$(grep -c ^processor /proc/cpuinfo)
    local load_avg=$(awk '{print $1, $2, $3}' /proc/loadavg)
    center_text "$(color_text "bg_bright_magenta" "CPU Model: $cpu_model")"
    center_text "$(color_text "bg_blue" "CPU Cores: $cpu_cores")"
    center_text "$(color_text "bg_red" "Load Average: $load_avg")"
}

display_temperature_info() {
    local cpu_temp=$(sensors | awk '/^Package id 0:/ {print $4}')
    [ -n "$cpu_temp" ] && center_text "$(color_text "bg_red" "CPU Temperature: $cpu_temp")"
}

display_disk_usage() {
    center_text "$(color_text "bg_yellow" "Disk I/O Statistics")"
    iostat | while IFS= read -r line; do center_text "$line"; done

    center_text "$(color_text "bg_bright_black" "Mounted Drives")"
    local drives=$(df -h | awk 'NR==1 {next} {printf "%-20s %6s %6s %6s\n", $1, $2, $3, $5}')
    while IFS= read -r line; do center_text "$line"; done <<< "$drives"
}

display_battery_info() {
    local battery_info=$(upower -i $(upower -e | grep BAT) | grep -E "state|to full|percentage")
    center_text "$(color_text "bg_bright_black" "Battery Info")"
    while IFS= read -r line; do center_text "$line"; done <<< "$battery_info"
}

display_network_info() {
    local ip_address=$(hostname -i | awk '{print $1}')
    local interfaces=$(ip -o link show | awk -F': ' '{print $2}')
    local gateway=$(ip route | awk '/default/ {print $3}')
    local dns=$(awk '/nameserver/ {print $2}' /etc/resolv.conf | tr '\n' ' ')
    center_text "$(color_text "bg_cyan" "IP Address: $ip_address")"
    center_text "$(color_text "bg_cyan" "Gateway: $gateway")"
    center_text "$(color_text "bg_cyan" "DNS: $dns")"
    center_text "$(color_text "bg_magenta" "Active Network Interfaces: $interfaces")"
}

display_gpu(){
    local gpu_info=$(lspci | grep VGA | cut -d ':' -f 3 | cut -d '[' -f 1,2 | sed 's/^ *//')
    center_text "$(color_text "bg_yellow" "GPU INFO: $gpu_info")"
}

# Menu function
show_menu() {
    clear
    center_text "$(color_text "bg_blue" "System Information Menu")"
    echo
    center_text "$(color_text "green" "1. Display System Information")"
    center_text "$(color_text "green" "2. Display Memory Usage")"
    center_text "$(color_text "green" "3. Display CPU Information")"
    center_text "$(color_text "green" "4. Display Temperature Information")"
    center_text "$(color_text "green" "5. Display GPU Information")"
    center_text "$(color_text "green" "6. Display Disk Usage")"
    center_text "$(color_text "green" "7. Display Battery Information")"
    center_text "$(color_text "green" "8. Display Network Information")"
    center_text "$(color_text "red"   "9. Display All Information")" # Changed 'bright_white' to 'red' for emphasis
    echo
    center_text "$(color_text "red"   "0. Exit")"
    echo
    center_text "$(color_text "yellow" "Enter your choice: ")"
}

# Main function to display all information
display_info() {
    display_system_info
    display_memory_usage
    display_cpu_info
    display_temperature_info
    display_gpu
    display_disk_usage
    display_battery_info
    display_network_info
}

# Menu loop
while true; do
    show_menu
    read -r choice

    case "$choice" in
        1) display_system_info ;;
        2) display_memory_usage ;;
        3) display_cpu_info ;;
        4) display_temperature_info ;;
        5) display_gpu ;;
        6) display_disk_usage ;;
        7) display_battery_info ;;
        8) display_network_info ;;
        9) display_info ;;
        0) exit 0 ;;
        *) center_text "$(color_text "red" "Invalid choice. Please try again.")"
           sleep 1 ;;
    esac

    # Pause after displaying info (optional)
    if [[ "$choice" != 0 ]]; then
        center_text "$(color_text "yellow" "Press Enter to continue...")"
        read -r
    fi
done