#!/usr/bin/env bash
# bootstrap-remote.sh — runs ON the R720, invoked over SSH by
# bootstrap-local.sh. Idempotent ; resumable via PHASE env var.
#
# Inputs (from SSH-passed env vars) :
#   FORGEJO_REGISTRATION_TOKEN    short-lived token to register runner
#   FORGEJO_API_URL               default: https://forgejo.talas.group
#
# Each phase logs to /var/log/talas-bootstrap.log AND emits structured
# >>>PHASE:<name>:<status><<< markers on stdout for the local script.

# lib.sh is concatenated upstream by bootstrap-local before this file is
# piped to bash. When run standalone, source it manually.
if ! declare -F info >/dev/null 2>&1; then
    SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
    # shellcheck source=lib.sh
    . "$SCRIPT_DIR/lib.sh"
fi
trap_errors

# Persistent log on R720 — useful when the SSH stream gets cut off.
exec > >(tee -a /var/log/talas-bootstrap.log) 2>&1

: "${FORGEJO_API_URL:=https://forgejo.talas.group}"

# ============================================================================
# Phase R1 — Incus profiles
# ============================================================================
remote_phase_1_profiles() {
    section "R1 — Incus profiles (veza-app, veza-data, veza-net)"
    _current_phase=r1_profiles
    phase r1_profiles START

    if skip_if_done r1_profiles "incus profiles"; then
        phase r1_profiles DONE; return 0
    fi

    for p in veza-app veza-data veza-net; do
        if incus profile show "$p" >/dev/null 2>&1; then
            ok "profile $p already exists"
        else
            incus profile create "$p"
            ok "profile $p created (empty — operator may add limits later)"
        fi
    done

    # If there's an existing veza-net network, add it to veza-net profile
    # so containers using that profile pick it up by default. Otherwise
    # leave the profile empty (caller passes --network on launch).
    if incus network show veza-net >/dev/null 2>&1; then
        if ! incus profile device show veza-net 2>/dev/null | grep -q '^eth0:'; then
            incus profile device add veza-net eth0 nic \
                network=veza-net \
                name=eth0 >/dev/null
            ok "veza-net profile : eth0 → network veza-net"
        else
            ok "veza-net profile : eth0 device already configured"
        fi
    else
        warn "incus network 'veza-net' not found — containers will need explicit --network on launch"
    fi

    mark_done r1_profiles
    phase r1_profiles DONE
}

# ============================================================================
# Phase R2 — mount Incus socket into forgejo-runner container
# ============================================================================
remote_phase_2_runner_socket() {
    section "R2 — mount /var/lib/incus/unix.socket into forgejo-runner"
    _current_phase=r2_runner_socket
    phase r2_runner_socket START

    if skip_if_done r2_runner_socket "runner socket mount"; then
        phase r2_runner_socket DONE; return 0
    fi

    if ! incus info forgejo-runner >/dev/null 2>&1; then
        die "container 'forgejo-runner' not found ; expected at the IP shown in the design"
    fi

    if incus config device show forgejo-runner 2>/dev/null | grep -q '^incus-socket:'; then
        ok "incus-socket device already attached"
    else
        info "attaching unix socket as a disk device"
        incus config device add forgejo-runner incus-socket disk \
            source=/var/lib/incus/unix.socket \
            path=/var/lib/incus/unix.socket >/dev/null
        ok "device added"
    fi

    if [[ "$(incus config get forgejo-runner security.nesting)" != "true" ]]; then
        info "enabling security.nesting"
        incus config set forgejo-runner security.nesting=true
        ok "nesting=true ; restart required"
        info "restarting forgejo-runner container"
        incus restart forgejo-runner
        sleep 3
    fi

    info "ensuring incus client is installed inside the runner"
    if ! incus exec forgejo-runner -- command -v incus >/dev/null 2>&1; then
        incus exec forgejo-runner -- apt-get update -qq
        incus exec forgejo-runner -- apt-get install -y incus-client >/dev/null
        ok "incus-client installed in runner"
    else
        ok "incus-client already in runner"
    fi

    info "smoke-test : runner can incus list"
    if ! incus exec forgejo-runner -- incus list >/dev/null 2>&1; then
        die "runner cannot reach Incus socket — verify nesting + permissions"
    fi
    ok "runner has Incus access"

    mark_done r2_runner_socket
    phase r2_runner_socket DONE
}

# ============================================================================
# Phase R3 — runner label = 'incus'
# ============================================================================
remote_phase_3_runner_labels() {
    section "R3 — forgejo-runner labelled 'incus,self-hosted'"
    _current_phase=r3_runner_labels
    phase r3_runner_labels START

    if skip_if_done r3_runner_labels "runner labels"; then
        phase r3_runner_labels DONE; return 0
    fi

    require_env FORGEJO_REGISTRATION_TOKEN \
        "set on the SSH command-line by bootstrap-local.sh"

    # Find the runner config inside the container. Path varies by install
    # method ; act_runner default is /etc/forgejo-runner/.runner.
    local runner_cfg
    runner_cfg=$(incus exec forgejo-runner -- bash -c '
        for f in /etc/forgejo-runner/.runner /var/lib/forgejo-runner/.runner /opt/forgejo-runner/.runner; do
            [[ -f "$f" ]] && echo "$f" && exit 0
        done
        exit 1
    ' 2>/dev/null) || true

    local labels=""
    if [[ -n "$runner_cfg" ]]; then
        labels=$(incus exec forgejo-runner -- jq -r '.labels[]?' "$runner_cfg" 2>/dev/null \
                || incus exec forgejo-runner -- grep -oE '"labels":\[[^]]+' "$runner_cfg" 2>/dev/null \
                || echo "")
    fi

    if echo "$labels" | grep -qw incus; then
        ok "runner already has 'incus' label"
        mark_done r3_runner_labels
        phase r3_runner_labels DONE
        return 0
    fi

    info "re-registering runner with labels incus,self-hosted"

    # Stop systemd unit, wipe old registration, re-register, start.
    incus exec forgejo-runner -- systemctl stop forgejo-runner.service 2>/dev/null \
        || incus exec forgejo-runner -- systemctl stop act_runner.service 2>/dev/null \
        || warn "no systemd unit to stop ; will skip"

    [[ -n "$runner_cfg" ]] && incus exec forgejo-runner -- rm -f "$runner_cfg"

    # Detect runner binary name
    local runner_bin
    runner_bin=$(incus exec forgejo-runner -- bash -c '
        for b in forgejo-runner act_runner; do
            command -v "$b" >/dev/null 2>&1 && echo "$b" && exit 0
        done
        exit 1
    ' 2>/dev/null) || die "no forgejo-runner / act_runner binary found in container"

    incus exec forgejo-runner -- "$runner_bin" register \
        --no-interactive \
        --instance "$FORGEJO_API_URL" \
        --token "$FORGEJO_REGISTRATION_TOKEN" \
        --name "r720-incus" \
        --labels "incus,self-hosted"

    incus exec forgejo-runner -- systemctl start "$runner_bin.service" \
        || incus exec forgejo-runner -- systemctl start forgejo-runner.service

    ok "runner re-registered with incus label"

    mark_done r3_runner_labels
    phase r3_runner_labels DONE
}

# ============================================================================
# Phase R4 — sanity, summary
# ============================================================================
remote_phase_4_sanity() {
    section "R4 — sanity check"
    _current_phase=r4_sanity
    phase r4_sanity START

    info "incus profiles :"
    incus profile list -f csv | grep -E '^veza-' | awk -F, '{print "  " $1}'

    info "forgejo-runner status :"
    incus exec forgejo-runner -- systemctl is-active forgejo-runner.service 2>/dev/null \
        || incus exec forgejo-runner -- systemctl is-active act_runner.service 2>/dev/null \
        || warn "no active runner service — verify manually"

    info "forgejo container reachable from runner :"
    if incus exec forgejo-runner -- curl -sSf -o /dev/null --max-time 5 \
            "$FORGEJO_API_URL" 2>/dev/null \
       || incus exec forgejo-runner -- curl -sSf -ko /dev/null --max-time 5 \
            https://10.0.20.105:3000/ 2>/dev/null \
       || incus exec forgejo-runner -- curl -sSf -o /dev/null --max-time 5 \
            http://10.0.20.105:3000/ 2>/dev/null; then
        ok "runner can reach Forgejo"
    else
        warn "runner cannot reach Forgejo — check WireGuard / DNS / firewall"
    fi

    mark_done r4_sanity
    phase r4_sanity DONE
}

main() {
    local start=${PHASE:-1}
    info "remote bootstrap starting at phase $start (log: /var/log/talas-bootstrap.log)"

    [[ $start -le 1 ]] && remote_phase_1_profiles
    [[ $start -le 2 ]] && remote_phase_2_runner_socket
    [[ $start -le 3 ]] && remote_phase_3_runner_labels
    [[ $start -le 4 ]] && remote_phase_4_sanity

    ok "remote bootstrap done"
}

main "$@"