#!/bin/bash
# Dynasty Revolution MOTD Script
[ -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
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'

IPV4_ADDR=$(hostname -I | awk '{print $1}')
IPV6_ADDR=$(ip -6 addr show scope global 2>/dev/null | awk '/inet6/ {print $2}' | head -n 1)
[ -z "$IPV6_ADDR" ] && IPV6_ADDR="N/A"

HOSTNAME=$(hostname)
MAC_ADDR=$(ip link show | awk '/ether/ {print $2}' | head -n 1)

NB_IP=$(ip addr show wt0 2>/dev/null | grep "inet " | awk '{print $2}' | cut -d/ -f1)
if [ -z "$NB_IP" ]; then
    NB_OUT="${RED}OFFLINE [Not Provisioned]${NC}"
else
    NB_VER=$(netbird version 2>/dev/null)
    NB_OUT="${GOLD}${NB_IP} Version ${NB_VER}${NC}"
fi

UPTIME_SEC=$(cat /proc/uptime | awk '{print int($1)}')
UP_DAYS=$((UPTIME_SEC / 86400))
UP_HOURS=$(((UPTIME_SEC % 86400) / 3600))
UP_MINS=$(((UPTIME_SEC % 3600) / 60))
if [ $UP_DAYS -gt 0 ]; then
    UPTIME="Alive for ${UP_DAYS} days, ${UP_HOURS} hours"
elif [ $UP_HOURS -gt 0 ]; then
    UPTIME="Alive for ${UP_HOURS} hours, ${UP_MINS} minutes"
else
    UPTIME="Alive for ${UP_MINS} minutes"
fi

LAST_INFO=$(last -i -a -F -n 2 | grep -v "still logged" | head -n 1)
LAST_USER=$(echo $LAST_INFO | awk '{print $1}')
LAST_SRC=$(echo $LAST_INFO | awk '{print $NF}')
# Catch local Proxmox console logins that don't generate an IP
[[ "$LAST_SRC" == \(* ]] && LAST_SRC="Local Console"
[[ "$LAST_SRC" == "0.0.0.0" ]] && LAST_SRC="Local Console"

LAST_RAW_DATE=$(echo $LAST_INFO | awk '{print $3, $4, $5, $6, $7}')
LAST_DATE=$(LC_TIME=en_US.UTF-8 date -d "$LAST_RAW_DATE" +'%m-%d-%Y %I:%M:%S %p %Z' 2>/dev/null)
[ -z "$LAST_USER" ] || [ "$LAST_USER" = "wtmp" ] && LAST_USER="System" && LAST_SRC="Local" && LAST_DATE="Initial Boot"

SERVER_TIME=$(LC_TIME=en_US.UTF-8 date +'%m-%d-%Y %I:%M:%S %p %Z')

CPU_LOAD=$(vmstat 1 2 | tail -1 | awk '{print 100 - $15}')
CPU_CORES="$(nproc) Cores"
MEM_USAGE=$(free | grep Mem | awk '{print $3/$2 * 100.0}' | cut -d. -f1)
MEM_DETAILS=$(free -h | awk '/Mem:/ {print $3"/"$2}' | sed -e 's/i//g' -e 's/M/MB/g' -e 's/G/GB/g')
DISK_USAGE=$(df -h / | grep / | awk '{print $5}' | sed 's/%//' | head -n 1)
DISK_DETAILS=$(df -h / | awk 'NR==2 {print $3"/"$2}' | sed -e 's/i//g' -e 's/M/MB/g' -e 's/G/GB/g')

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} .... ${GOLD}$UPTIME${NC}"
echo -e " ${BLUE}Internal IP:${NC}       ${GOLD}IPv4: $IPV4_ADDR     IPv6: $IPV6_ADDR${NC}"
echo -e " ${BLUE}NetBird IP:${NC}        $NB_OUT"
echo -e " ${BLUE}Internet Status:${NC}   $STATUS"
echo -e " ${BLUE}System Time:${NC}       ${GOLD}$SERVER_TIME${NC}"
echo -e " ${BLUE}Last Access:${NC}       ${GOLD}$LAST_USER${NC} on ${GOLD}$LAST_SRC${NC} @ ${GOLD}$LAST_DATE${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) $CPU_CORES    "
echo -ne " ${BLUE}RAM Usage:${NC}    $(get_color $MEM_USAGE) $MEM_DETAILS    "
echo -e  " ${BLUE}Disk Usage:${NC}   $(get_color $DISK_USAGE) $DISK_DETAILS"
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 [$MAC_ADDR]${NC}"
echo ""
INNER_EOF

chmod +x /etc/update-motd.d/05-dynr-welcome
echo -e "\\033[38;5;178mDynasty Revolution MOTD Installed Successfully.\\033[0m"