#!/usr/bin/env bash
set -euo pipefail
source "$(dirname "$0")/../lib/common.sh"

usage() {
    cat <<EOF
Usage: bash apps/install.sh [OPTIONS] [app ...]

Install optional desktop applications.

Options:
  --all        Install all available apps without prompting
  -h, --help   Show this help message

Examples:
  bash apps/install.sh                  # interactive selection
  bash apps/install.sh --all            # install everything
  bash apps/install.sh brave vscode     # install specific apps
EOF
    exit 0
}

ALL=0
POSITIONAL=()

while [[ $# -gt 0 ]]; do
    case "$1" in
        --all) ALL=1; shift ;;
        -h|--help) usage ;;
        -*) err "Unknown option: $1" ;;
        *) POSITIONAL+=("$1"); shift ;;
    esac
done

APPS_DIR="$(dirname "$0")"

# ── Category display names ─────────────────────────────────────
declare -A CAT_NAMES=(
    [browsers]="Browsers"
    [development]="Development"
    [media]="Media"
    [networking]="Networking / VPN"
    [remote-access]="Remote Access"
    [system]="System / Virtualization"
    [utilities]="Utilities"
)

# ── Discover categories and apps ───────────────────────────────
CATEGORIES=()
declare -A CAT_APPS

for cat_dir in "$APPS_DIR"/*/; do
    [ -d "$cat_dir" ] || continue
    cat_name=$(basename "$cat_dir")
    # Skip non-app directories
    [[ "$cat_name" == "install" ]] && continue
    # Check if directory has any .sh files
    has_apps=0
    for f in "$cat_dir"*.sh; do
        [ -f "$f" ] && has_apps=1 && break
    done
    [ "$has_apps" -eq 0 ] && continue

    CATEGORIES+=("$cat_name")
    CAT_APPS["$cat_name"]=()
    for f in "$cat_dir"*.sh; do
        [ -f "$f" ] || continue
        app_name=$(basename "$f" .sh)
        CAT_APPS["$cat_name"]+=("$app_name")
    done
done

# ── Helper: find which category an app belongs to ──────────────
find_app_category() {
    local app="$1"
    for cat in "${CATEGORIES[@]}"; do
        for a in "${CAT_APPS[$cat]}"; do
            [ "$a" == "$app" ] && { echo "$cat"; return 0; }
        done
    done
    return 1
}

SELECTED=()

# ── Mode 1: specific apps requested on command line ────────────
if [ "${#POSITIONAL[@]}" -gt 0 ]; then
    for req in "${POSITIONAL[@]}"; do
        cat=$(find_app_category "$req" 2>/dev/null) && {
            SELECTED+=("$req")
        } || {
            warn "Unknown app: $req (skipping)"
        }
    done

# ── Mode 2: --all ─────────────────────────────────────────────
elif [ "$ALL" -eq 1 ]; then
    for cat in "${CATEGORIES[@]}"; do
        for app in "${CAT_APPS[$cat]}"; do
            SELECTED+=("$app")
        done
    done

# ── Mode 3: interactive TUI ───────────────────────────────────
else
    section "Optional Applications"
    echo "Select apps to install (y/n for each):"
    echo

    for cat in "${CATEGORIES[@]}"; do
        display="${CAT_NAMES[$cat]:-$cat}"
        echo "  $display"
        for app in "${CAT_APPS[$cat]}"; do
            read -rp "    Install ${app}? [y/N]: " yn
            if [[ "$yn" =~ ^[Yy] ]]; then
                SELECTED+=("$app")
            fi
        done
        echo
    done
fi

# ── Install selected apps ──────────────────────────────────────
[ "${#SELECTED[@]}" -eq 0 ] && { warn "No apps selected"; exit 0; }

echo
section "Installing ${SELECTED[*]}"

timer_start
count=1
total=${#SELECTED[@]}
for app in "${SELECTED[@]}"; do
    cat=$(find_app_category "$app")
    step "$count" "$total" "$app"
    bash "$APPS_DIR/$cat/$app.sh"
    count=$((count + 1))
    echo
done

echo
echo "${GREEN}════════════════════════════════════════════${RESET}"
echo "${GREEN}  Apps installed ($(timer_stop))${RESET}"
echo "${GREEN}════════════════════════════════════════════${RESET}"