#!/usr/bin/env bash

# Copyright (c) 2021-2025 community-scripts
# Author: MickLesk
# License: MIT
# https://github.com/community-scripts/ProxmoxVE/raw/main/LICENSE

set -eEuo pipefail
BL=$(echo "\033[36m")
RD=$(echo "\033[01;31m")
GN=$(echo "\033[1;92m")
CL=$(echo "\033[m")

function header_info {
    clear
    cat <<"EOF"
    ____                                      _    ____________     __           ____                                      _    ________
   / __ \_________  _  ______ ___  ____  _  _| |  / / ____/ __ \   / /_____     / __ \_________  _  ______ ___  ____  _  _| |  / / ____/
  / /_/ / ___/ __ \| |/_/ __ `__ \/ __ \| |/_/ | / / __/ / / / /  / __/ __ \   / /_/ / ___/ __ \| |/_/ __ `__ \/ __ \| |/_/ | / / __/
 / ____/ /  / /_/ />  </ / / / / / /_/ />  < | |/ / /___/ /_/ /  / /_/ /_/ /  / ____/ /  / /_/ />  </ / / / / / /_/ />  < | |/ / /___
/_/   /_/   \____/_/|_/_/ /_/ /_/\____/_/|_| |___/_____/_____/   \__/\____/  /_/   /_/   \____/_/|_/_/ /_/ /_/\____/_/|_| |___/_____/
EOF
}

function update_container() {
    container=$1
    os=$(pct config "$container" | awk '/^ostype/ {print $2}')

    if [[ "$os" == "ubuntu" || "$os" == "debian" || "$os" == "alpine" ]]; then
        echo -e "${BL}[Info]${GN} Checking /usr/bin/update in ${BL}$container${CL} (OS: ${GN}$os${CL})"

        if pct exec "$container" -- [ -e /usr/bin/update ]; then
            pct exec "$container" -- bash -c "sed -i 's/ProxmoxVED/ProxmoxVE/g' /usr/bin/update"

            if pct exec "$container" -- grep -q "ProxmoxVE" /usr/bin/update; then
                echo -e "${GN}[Success]${CL} /usr/bin/update updated in ${BL}$container${CL}.\n"
            else
                echo -e "${RD}[Error]${CL} /usr/bin/update in ${BL}$container${CL} could not be updated properly.\n"
            fi
        else
            echo -e "${RD}[Error]${CL} /usr/bin/update not found in container ${BL}$container${CL}.\n"
        fi
    fi
}

function update_motd() {
    container=$1
    os=$(pct config "$container" | awk '/^ostype/ {print $2}')

    echo -e "${BL}[Debug]${GN} Processing container: ${BL}$container${CL} (OS: ${GN}$os${CL})"

    if [[ "$os" == "ubuntu" || "$os" == "debian" ]]; then
        echo -e "${BL}[Debug]${GN} Updating Debian/Ubuntu MOTD in ${BL}$container${CL}"

        pct exec "$container" -- bash -c "
          PROFILE_FILE='/etc/profile.d/00_motd.sh'
          mkdir -p /etc/profile.d/
          echo 'echo -e \"\"' > \"\$PROFILE_FILE\"
          echo 'echo -e \"🌐 Provided by: community-scripts ORG | GitHub: https://github.com/community-scripts/ProxmoxVE\"' >> \"\$PROFILE_FILE\"
          echo 'echo -e \"🖥️ OS: \$(grep ^NAME /etc/os-release | cut -d= -f2 | tr -d '\"') - Version: \$(grep ^VERSION_ID /etc/os-release | cut -d= -f2 | tr -d '\"')\"' >> \"\$PROFILE_FILE\"
          echo 'echo -e \"🏠 Hostname: \$(hostname)\"' >> \"\$PROFILE_FILE\"
          echo 'echo -e \"💡 IP Address: \$(hostname -I | awk '\''{print \$1}'\'')\"' >> \"\$PROFILE_FILE\"
          chmod -x /etc/update-motd.d/*
        "

        echo -e "${GN}[Debug] Finished Debian/Ubuntu MOTD update for ${BL}$container${CL}"

    elif [[ "$os" == "alpine" ]]; then
        echo -e "${BL}[Debug]${GN} Updating Alpine MOTD in ${BL}$container${CL}"

        pct exec "$container" -- /bin/sh -c '
          echo "[Debug] Alpine: Start updating MOTD" > /tmp/motd_debug.log
          echo "export TERM=\"xterm-256color\"" >> /root/.bashrc
          echo "[Debug] Alpine: Set TERM variable" >> /tmp/motd_debug.log

          mkdir -p /etc/profile.d/
          echo "[Debug] Alpine: Created /etc/profile.d/" >> /tmp/motd_debug.log

          IP=$(ip -4 addr show eth0 | awk "/inet / {print \$2}" | cut -d/ -f1 | head -n 1)
          echo "[Debug] Alpine: Fetched IP: $IP" >> /tmp/motd_debug.log

          PROFILE_FILE="/etc/profile.d/00_lxc-details.sh"
          echo "[Debug] Alpine: Writing to profile file" >> /tmp/motd_debug.log

          echo "echo -e \"\"" > \"$PROFILE_FILE\"
          echo "echo -e \" LXC Container\"" >> \"$PROFILE_FILE\"
          echo "echo -e \" 🌐 Provided by: community-scripts ORG | GitHub: https://github.com/community-scripts/ProxmoxVE\"" >> \"$PROFILE_FILE\"
          echo "echo -e \"🖥️ OS: $(grep ^NAME /etc/os-release | cut -d= -f2 | tr -d '\"') - Version: $(grep ^VERSION_ID /etc/os-release | cut -d= -f2 | tr -d '\"')\"" >> \"$PROFILE_FILE\"
          echo "echo -e \"🏠 Hostname: $(hostname)\"" >> \"$PROFILE_FILE\"
          echo "echo -e \"💡 IP Address: $IP\"" >> \"$PROFILE_FILE\"
          echo "[Debug] Alpine: Finished writing MOTD" >> /tmp/motd_debug.log
        '

        echo -e "${GN}[Debug] Finished Alpine MOTD update for ${BL}$container${CL}"
    fi
}

function remove_dev_tag() {
    container=$1
    current_tags=$(pct config "$container" | awk '/^tags/ {print $2}')

    if [[ "$current_tags" == *"dev"* ]]; then
        new_tags=$(echo "$current_tags" | sed 's/,*dev,*//g' | sed 's/^,//' | sed 's/,$//')

        if [[ -z "$new_tags" ]]; then
            pct set "$container" -delete tags
        else
            pct set "$container" -tags "$new_tags"
        fi

        echo -e "${GN}[Success]${CL} 'dev' tag removed from ${BL}$container${CL}.\n"
    fi
}

header_info
echo "Searching for containers with 'dev' tag..."
for container in $(pct list | awk '{if(NR>1) print $1}'); do
    tags=$(pct config "$container" | awk '/^tags/ {print $2}')
    if [[ "$tags" == *"dev"* ]]; then
        update_container "$container"
        update_motd "$container"
        remove_dev_tag "$container"
    fi
done

header_info
echo -e "${GN}The process is complete.${CL}\n"