#!/bin/sh
# bee-boot-status — boot progress display on tty1.
# Shows live service status until all bee services are done or failed,
# then exits so getty can show the login prompt.

CRITICAL="bee-preflight bee-nvidia bee-audit"
ALL="bee-sshsetup ssh bee-network bee-nvidia bee-preflight bee-audit bee-web"

svc_state() { systemctl is-active "$1.service" 2>/dev/null || echo "inactive"; }

svc_icon() {
    case "$(svc_state "$1")" in
        active)       printf '\033[32m[  OK  ]\033[0m' ;;
        failed)       printf '\033[31m[ FAIL ]\033[0m' ;;
        activating)   printf '\033[33m[  ..  ]\033[0m' ;;
        deactivating) printf '\033[33m[ stop ]\033[0m' ;;
        inactive)     printf '\033[90m[      ]\033[0m' ;;
        *)            printf '\033[90m[  ?   ]\033[0m' ;;
    esac
}

svc_detail() {
    local svc="$1" state
    state="$(svc_state "$svc")"
    case "$state" in
        failed)
            local res
            res="$(systemctl show -p Result "$svc.service" 2>/dev/null | cut -d= -f2)"
            [ -n "$res" ] && [ "$res" != "success" ] && printf '  \033[31m(%s)\033[0m' "$res"
            ;;
        activating)
            local line
            line="$(journalctl -u "$svc.service" -n 1 --no-pager --output=cat 2>/dev/null | cut -c1-55)"
            [ -n "$line" ] && printf '  \033[90m%s\033[0m' "$line"
            ;;
    esac
}

all_critical_done() {
    for svc in $CRITICAL; do
        case "$(svc_state "$svc")" in
            active|failed|inactive) ;;
            *) return 1 ;;
        esac
    done
    return 0
}

while true; do
    # move to top-left and clear screen
    printf '\033[H\033[2J'

    printf '\n'
    printf '  \033[33m███████╗ █████╗ ███████╗██╗   ██╗      ██████╗ ███████╗███████╗\033[0m\n'
    printf '  \033[33m██╔════╝██╔══██╗██╔════╝╚██╗ ██╔╝      ██╔══██╗██╔════╝██╔════╝\033[0m\n'
    printf '  \033[33m█████╗  ███████║███████╗ ╚████╔╝ █████╗██████╔╝█████╗  █████╗\033[0m\n'
    printf '  \033[33m██╔══╝  ██╔══██║╚════██║  ╚██╔╝  ╚════╝██╔══██╗██╔══╝  ██╔══╝\033[0m\n'
    printf '  \033[33m███████╗██║  ██║███████║   ██║         ██████╔╝███████╗███████╗\033[0m\n'
    printf '  \033[33m╚══════╝╚═╝  ╚═╝╚══════╝   ╚═╝         ╚═════╝ ╚══════╝╚══════╝\033[0m\n'
    printf '  Hardware Audit LiveCD\n'
    printf '\n'

    for svc in $ALL; do
        printf '  %s  %-20s%s\n' "$(svc_icon "$svc")" "$svc" "$(svc_detail "$svc")"
    done
    printf '\n'

    # Network
    ips="$(ip -4 addr show scope global 2>/dev/null | awk '/inet /{printf "  %-16s %s\n", $NF, $2}')"
    if [ -n "$ips" ]; then
        printf '  \033[1mNetwork:\033[0m\n'
        printf '%s\n' "$ips"
        printf '\n'
    fi

    if all_critical_done; then
        printf '  \033[1;32mSystem ready.\033[0m  Audit is running in the background.\n'
        first_ip="$(ip -4 addr show scope global 2>/dev/null | awk '/inet /{print $2}' | cut -d/ -f1 | head -1)"
        if [ -n "$first_ip" ]; then
            printf '  Web UI: \033[1mhttp://%s/\033[0m\n' "$first_ip"
        fi
        printf '\n'
        sleep 3
        break
    fi

    printf '  \033[90mStarting up...\033[0m\n'
    sleep 3
done
