#!/bin/bash
# Dynasty Revolution MOTD Installer
[ -f /etc/profile.d/00_lxc-details.sh ] && rm -f /etc/profile.d/00_lxc-details.sh
[ -f /etc/motd ] && echo "" > /etc/motd
mkdir -p /etc/update-motd.d/
chmod -x /etc/update-motd.d/* 2>/dev/null

cat << 'INNER_EOF' > /etc/update-motd.d/05-dynr-welcome
#!/bin/bash
# Dynasty Revolution Official MOTD
BLUE='\033[38;5;45m'; PURPLE='\033[38;5;127m'; GOLD='\033[38;5;178m'; HOTPINK='\033[38;5;201m'
RED='\033[38;5;196m'; GREEN='\033[38;5;82m'; YELLOW='\033[38;5;226m'; NC='\033[0m'

# Stats
IP_ADDR=$(hostname -I | awk '{print $1}')
HOSTNAME=$(hostname)
NB_IP=$(ip addr show wt0 2>/dev/null | grep "inet " | awk '{print $2}' | cut -d/ -f1)
[ -z "$NB_IP" ] && NB_IP="${RED}OFFLINE${NC}"
LAST_LOG=$(last -i -n 2 | awk 'NR==2 {print $3}')
[ -z "$LAST_LOG" ] && LAST_LOG="None"

CPU_LOAD=$(vmstat 1 2 | tail -1 | awk '{print 100 - $15}')
MEM_USAGE=$(free | grep Mem | awk '{print $3/$2 * 100.0}' | cut -d. -f1)
DISK_USAGE=$(df -h / | grep / | awk '{print $5}' | sed 's/%//' | head -n 1)

get_color() {
    if [ "$1" -ge 90 ]; then echo -e "${RED}$1%${NC}";
    elif [ "$1" -ge 70 ]; then echo -e "${YELLOW}$1%${NC}";
    else echo -e "${GREEN}$1%${NC}"; fi
}

ping -c 1 1.1.1.1 &>/dev/null && STATUS="${GREEN}ONLINE${NC}" || STATUS="${RED}OFFLINE${NC}"

echo -e "${PURPLE}"
echo "  _____ __     __ _   _ _____           _____ ______ ________      ________ _____   _____ "
echo " |  __ \\\ \\   / /| \\ | |  __ \\         / ____|  ____|  __  \\ \\    / /  ____|  __ \\ / ____|"
echo " | |  | |\\ \\_/ / |  \\| | |__) |        | (___ | |__  | |__) \\ \\  / /| |__  | |__)| (___  "
echo " | |  | | \\   /  | .   |  _  /          \\___ \\|  __| |  _  / \\ \\/ / |  __| |  _  / \\___ \\ "
echo " | |__| |  | |   | |\\  | | \\ \\          ____) | |____| | \\ \\  \\  /  | |____| | \\ \\ ____) |"
echo " |_____/   |_|   |_| \\_|_|  \\_\\        |_____/|______|_|  \\_\\  \\/   |______|_|  \\_|_____/ "
echo -e "          ${GOLD}D Y N A S T Y   R E V O L U T I O N${NC}"
echo ""
echo -e " ${BLUE}Welcome to LXC:${NC}    ${HOTPINK}$HOSTNAME${NC}"
echo -e " ${BLUE}Internal IP:${NC}       ${GOLD}$IP_ADDR${NC}"
echo -e " ${BLUE}NetBird IP:${NC}        ${GOLD}$NB_IP${NC}"
echo -e " ${BLUE}Internet Status:${NC}   $STATUS"
echo -e " ${BLUE}Last Login From:${NC}   ${GOLD}$LAST_LOG${NC}"
echo -e " ${BLUE}Support:${NC}           support@dynastyrevolution.com"
echo -e " ${BLUE}Abuse/Report:${NC}      report@dynastyrevolution.com"
echo -e "${GOLD}--------------------------------------------------------------------------------${NC}"

echo -ne " ${BLUE}CPU Usage:${NC}    $(get_color $CPU_LOAD)    "
echo -ne " ${BLUE}RAM Usage:${NC}    $(get_color $MEM_USAGE)    "
echo -e  " ${BLUE}Disk Usage:${NC}   $(get_color $DISK_USAGE)"
echo -e "${GOLD}--------------------------------------------------------------------------------${NC}"

echo -e "${RED}⚠️  WARNING: Unauthorized access is prohibited. All CLI commands are logged.${NC}"
echo -e "${RED}DYNASTY REVOLUTION INTERNAL SYSTEMS SYSTEM ID: $HOSTNAME${NC}"
echo ""
INNER_EOF

chmod +x /etc/update-motd.d/05-dynr-welcome
echo "Dynasty Revolution MOTD Installed Successfully."