#!/usr/bin/env bash
set -euo pipefail

# ── Parse --no-color BEFORE sourcing common.sh ──────────────────
NO_COLOR=0
for arg in "$@"; do
    [ "$arg" = "--no-color" ] && NO_COLOR=1
done
if [ "$NO_COLOR" -eq 1 ]; then
    export TERM=dumb
    unset CYAN GREEN YELLOW RED BLUE BOLD RESET
fi

source "$(dirname "$0")/lib/common.sh"
source "$(dirname "$0")/lib/flags.sh"

# Return the current install.sh version (e.g. "0.0c174").
# Derived from git commit count. Empty when .git is absent.
# Override with INSTALL_VERSION_OVERRIDE for testing.
install_version() {
    if [ "${INSTALL_VERSION_OVERRIDE+x}" ]; then
        printf '%s' "$INSTALL_VERSION_OVERRIDE"
        return
    fi
    local repo_dir count
    repo_dir="$(cd "$(dirname "$0")" && pwd)"
    count="$(git -C "$repo_dir" rev-list --count HEAD 2>/dev/null)" || { printf ''; return; }
    printf '0.0c%s' "$count"
}

# Exported so child phases (preinstall.sh, postinstall.sh) inherit it —
# otherwise '--dry-run' silently executes them for real.
export DRY_RUN=0
RUN_APPS=0
RUN_FEATURES=0
FORCE=0
SKIP_PHASES=""
STEPS_SPEC=""

# Expand range syntax in a --steps spec ("1-3" → "1,2,3"; also "1,3-4,6").
normalize_steps_spec() {
    local spec="$1" out="" part from to p
    local oldIFS="$IFS"
    IFS=','
    for part in $spec; do
        if [[ "$part" =~ ^([0-9]+)-([0-9]+)$ ]]; then
            from="${BASH_REMATCH[1]}"
            to="${BASH_REMATCH[2]}"
            if [ "$to" -lt "$from" ]; then
                err "Invalid --steps range '$part' (end < start)"
            fi
            for ((p = from; p <= to; p++)); do
                out="${out:+$out,}$p"
            done
        elif [[ "$part" =~ ^[0-9]+$ ]]; then
            out="${out:+$out,}$part"
        else
            err "Invalid --steps value '$part' (expected 1-4, comma list, or N-M range)"
        fi
    done
    IFS="$oldIFS"
    printf '%s' "$out"
}

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

Bootstrap a fresh Debian/Ubuntu install.

Options:
  --apps            Run interactive app picker after core install
  --full            Core install + all optional apps (non-interactive)
  --feature         Install features/ scripts (prompts before overwriting)
  --dry-run         Show what would be done without executing
  --force           Re-install even if the version matches
  --skip <phase>    Skip a phase (repeatable):
                      preinstall, scripts, postinstall, scalepoint, apps
  --steps <spec>    Run only specific phases. Format: 1,3,4 or 1-3
                    (1=preinstall, 2=scripts, 3=postinstall, 4=scalepoint)
  --no-color        Disable colored output
  -h, --help        Show this help message

Uninstall optional apps later with:
  ./apps/install.sh --uninstall            # interactive uninstall selection
  ./apps/install.sh --uninstall --all      # uninstall everything
  ./apps/install.sh --uninstall <app>...   # uninstall specific apps
EOF
    exit 0
}

while [[ $# -gt 0 ]]; do
    case "$1" in
        --apps) RUN_APPS=1; shift ;;
        --full) RUN_APPS=2; shift ;;
        --feature) RUN_FEATURES=1; shift ;;
        --dry-run) DRY_RUN=1; shift ;;
        --force) FORCE=1; shift ;;
        --skip)
            [ -z "${2:-}" ] && err "Missing value for --skip"
            SKIP_PHASES="${SKIP_PHASES:+$SKIP_PHASES,}$2"
            shift 2
            ;;
        --steps)
            [ -z "${2:-}" ] && err "Missing value for --steps"
            STEPS_SPEC="$(normalize_steps_spec "$2")"
            shift 2
            ;;
        --no-color) shift ;;
        -h|--help) usage ;;
        *) err "Unknown option: $1" ;;
    esac
done

# ── Version gate ────────────────────────────────────────────────
CURRENT_VERSION="$(install_version)"
if [ "${FORCE:-0}" -ne 1 ] && [ -n "$CURRENT_VERSION" ]; then
    INSTALLED_VERSION="$(flag_value installed_version 2>/dev/null)" || true
    if [ -n "$INSTALLED_VERSION" ]; then
        current_num="${CURRENT_VERSION#0.0c}"
        installed_num="${INSTALLED_VERSION#0.0c}"
        if [ "$current_num" -eq "$installed_num" ] 2>/dev/null; then
            if [ "${DRY_RUN:-0}" -eq 1 ]; then
                log "(dry-run) Would skip install: already at version $CURRENT_VERSION"
            else
                log "Already installed ($CURRENT_VERSION). Use --force to re-install."
            fi
            exit 0
        fi
    fi
fi

# ── Phase runner ────────────────────────────────────────────────
# Phase names → numbers:  preinstall=1 scripts=2 postinstall=3 scalepoint=4
should_run() {
    local phase_num="$1"
    local phase_name="$2"

    # --skip takes precedence
    if [[ ",$SKIP_PHASES," == *",$phase_name,"* ]]; then
        return 1
    fi

    # --steps restricts to listed phases only
    if [ -n "$STEPS_SPEC" ]; then
        if [[ ",$STEPS_SPEC," != *",$phase_num,"* ]]; then
            return 1
        fi
    fi

    return 0
}

section "Linux_post_install Bootstrap"
timer_start

# ── Phase 1: preinstall ────────────────────────────────────────
if should_run 1 preinstall; then
    step 1 4 "Installing system packages"
    if [ -f preinstall.sh ]; then
        # preinstall.sh owns 'apt update' — no duplicate update here.
        bash preinstall.sh
    fi
fi

# ── Phase 2: install wrappers ──────────────────────────────────
if should_run 2 scripts; then
    step 2 4 "Installing wrapper scripts"
    run sudo mkdir -p /usr/local/bin
    count=0
    for f in bin/*; do
        [ -f "$f" ] || continue
        run sudo install -m 755 "$f" /usr/local/bin/
        count=$((count + 1))
    done
    lib_count=0
    lib_names=""
    for lf in common.sh flags.sh notify.sh entertainment-lib.sh scheduler-lib.sh config-ui.sh user-timers-lib.sh entertainment-plugin-lib.sh usb-lib.sh share-lib.sh menu-lib.sh registry.sh yt-lib.sh bank-lib.sh; do
        run sudo install -m 644 "lib/$lf" "/usr/local/bin/$lf"
        lib_count=$((lib_count + 1))
        lib_names+="$lf "
    done
    log "libs -> /usr/local/bin (644): ${lib_names% }"

    # ── AI provider adapters ────────────────────────────────────
    # Installed into /usr/local/bin/ai-providers/ for pos-ai.
    ap_count=0
    ap_names=""
    run sudo mkdir -p /usr/local/bin/ai-providers
    for apf in lib/ai-providers/*.sh; do
        [ -f "$apf" ] || continue
        run sudo install -m 644 "$apf" "/usr/local/bin/ai-providers/"
        ap_count=$((ap_count + 1))
        ap_names+="$(basename "$apf") "
    done
    [ "$ap_count" -gt 0 ] && log "ai-providers -> /usr/local/bin/ai-providers (644): ${ap_names% }"

    # ── Entertainment plugins ────────────────────────────────
    # Installed into /usr/local/bin so the repo can be deleted afterwards.
    pcount=0
    pnames=""
    for f in entertainment/*.sh; do
        [ -f "$f" ] || continue
        run sudo install -m 755 "$f" /usr/local/bin/
        pcount=$((pcount + 1))
        pnames+="$(basename "$f") "
    done
    ok "$pcount entertainment plugins -> /usr/local/bin: ${pnames% }"

    # ── Precompiled architecture binaries ─────────────────────
    # Manually-compiled binaries (not available on the internet),
    # copied straight into /usr/local/bin for the matching arch.
    # Drop an arm64_bin/ folder later — it is picked up automatically.
    case "$(uname -m)" in
        x86_64)          prebuilt_dir="x64_bin" ;;
        aarch64|arm64)   prebuilt_dir="arm64_bin" ;;
        *)               prebuilt_dir="" ;;
    esac
    if [ -n "$prebuilt_dir" ] && [ -d "$prebuilt_dir" ]; then
        bcount=0
        bnames=""
        for f in "$prebuilt_dir"/*; do
            [ -f "$f" ] || continue
            run sudo install -m 755 "$f" /usr/local/bin/
            bcount=$((bcount + 1))
            bnames+="$(basename "$f") "
        done
        ok "$bcount $prebuilt_dir binaries -> /usr/local/bin: ${bnames% }"
    fi

    # ── Optional features ──────────────────────────────────────
    if [ "$RUN_FEATURES" -eq 1 ]; then
        fcount=0
        fnames=""
        for f in features/*; do
            [ -f "$f" ] || continue
            name=$(basename "$f")
            flag_name="${name%.sh}"
            dest="/usr/local/bin/$name"
            if [ -e "$dest" ]; then
                if confirm "Overwrite existing $dest?" n; then
                    run sudo install -m 755 "$f" "$dest"
                    log "feature overwritten: $name"
                else
                    log "Keeping existing $dest"
                fi
            else
                run sudo install -m 755 "$f" "$dest"
                log "feature installed: $name"
            fi
            flag_set "$flag_name"
            log "feature flag set: $flag_name"
            fcount=$((fcount + 1))
            fnames+="${name%.sh} "
        done
        ok "$fcount features installed: ${fnames% }"
    fi
    ok "$count scripts + $lib_count libs -> /usr/local/bin"
fi

# ── Phase 3: postinstall ───────────────────────────────────────
if should_run 3 postinstall; then
    step 3 4 "Post-install configuration"
    if [ -f postinstall.sh ]; then
        bash postinstall.sh
    fi
fi

# ── Phase 4: ScaleTail templates ────────────────────────────────
if should_run 4 scalepoint; then
    step 4 4 "Cloning ScaleTail templates"
    scale_dest="/usr/local/share/linux_post_install/scale-tail"
    if [ ! -d "$scale_dest" ]; then
        spawn "Cloning ScaleTail" sudo git clone --depth 1 \
            https://github.com/tailscale-dev/ScaleTail.git "$scale_dest"
    else
        log "ScaleTail already cloned"
    fi
fi

echo
echo "${GREEN}════════════════════════════════════════════${RESET}"
echo "${GREEN}  Bootstrap complete ($(timer_stop))${RESET}"
echo "${GREEN}════════════════════════════════════════════${RESET}"

# ── Record installed version ────────────────────────────────────
if [ "${DRY_RUN:-0}" -ne 1 ] && [ -n "${CURRENT_VERSION:-}" ]; then
    flag_set installed_version "$CURRENT_VERSION"
fi

# ── Optional apps ──────────────────────────────────────────────
if [ "$RUN_APPS" -eq 1 ]; then
    echo
    bash apps/install.sh
elif [ "$RUN_APPS" -eq 2 ]; then
    echo
    bash apps/install.sh --all
elif should_run 5 apps && [ -f apps/install.sh ]; then
    # --skip apps disables app phase even if --apps/--full is not used
    true
fi