#!/usr/bin/env bash
set -euo pipefail
# POS: network download — aria2 RPC daemon + queue control (add/torrent/metalink, watch, limits)
# POS_SUBCMDS: start stop status add torrent metalink list info files peers pause resume remove purge move limit set watch restart retry replace
# POS_FLAGS: --dir --out --split --seed --force --upload --gid --tmux

source "$(dirname "$0")/../lib/common.sh" 2>/dev/null || source "$(dirname "$0")/common.sh"

command -v aria2c &>/dev/null || err "aria2c not found — install with: sudo apt install aria2"
command -v jq &>/dev/null || err "jq not found — install with: sudo apt install jq"
command -v curl &>/dev/null || err "curl not found — install with: sudo apt install curl"

# ── Config / seams (env overrides for tests) ───────────────────
RPC_PORT="${RPC_PORT:-6800}"
RPC_URL="http://127.0.0.1:$RPC_PORT/jsonrpc"
CONFIG_FILE="$CONFIG_DIR/download.env"
DOWNLOAD_DIR="${DOWNLOAD_DIR:-$HOME/Downloads}"
USER_SYSTEMD_DIR="${USER_SYSTEMD_DIR:-${XDG_CONFIG_HOME:-$HOME/.config}/systemd/user}"
SERVICE="pos-aria2.service"
HEALER_SERVICE="pos-aria2-retry.service"
HEALER_TIMER="pos-aria2-retry.timer"
RETRY_STATE="$CONFIG_DIR/download.retry"
NET_PROBE="${NET_PROBE:-timeout 3 bash -c '</dev/tcp/8.8.8.8/53>'}"
RETRY_INTERVAL="${RETRY_INTERVAL:-30}"
RETRY_VERIFY_SLEEP="${RETRY_VERIFY_SLEEP:-3}"

load_secret() {
    if [ -z "${RPC_SECRET:-}" ] && [ -f "$CONFIG_FILE" ]; then
        RPC_SECRET=$(grep -E '^RPC_SECRET=' "$CONFIG_FILE" 2>/dev/null | head -1 | cut -d= -f2-)
    fi
    RPC_SECRET="${RPC_SECRET:-}"
}
load_secret

# Shared awk helpers: hs = human speed, et = human eta (seconds).
ROW_AWK='
function hs(s){ if(s>=1073741824)return sprintf("%.1fG/s", s/1073741824); if(s>=1048576)return sprintf("%.1fM/s", s/1048576); if(s>=1024)return sprintf("%.1fK/s", s/1024); return sprintf("%dB/s", s) }
function et(s){ s=s+0; if(s<=0)return "—"; if(s>=86400)return sprintf("%dd%dh", int(s/86400), int(s%86400/3600)); if(s>=3600)return sprintf("%dh%dm", int(s/3600), int(s%3600/60)); if(s>=60)return sprintf("%dm%ds", int(s/60), s%60); return sprintf("%ds", s) }
'

usage() {
    cat <<EOF
Usage: pos network download <command> [args]

aria2 download daemon + queue control. Runs a persistent aria2c with JSON-RPC
(systemd user service on localhost:$RPC_PORT) and drives it via the RPC API.

Commands:
  start                     Start the daemon (installs the systemd user service, generates RPC secret)
  stop                      Stop the daemon and remove the service
  status                    Daemon health + global transfer stats
  add <url> [<url>...]      Enqueue HTTP/FTP downloads (auto-starts the daemon)
  torrent <file|magnet>...  Enqueue .torrent files or magnet links
  metalink <file|url>       Enqueue a .metalink file or URL
  list                      Show active / waiting / finished downloads
  info <gid>                Full status of one download
  files <gid>               Files of a download
  peers <gid>               Peers of a torrent download
  pause|resume [gid|all]    Pause / resume one or all downloads (default all)
  remove [gid|all]          Remove one or all (--force kills immediately)
  purge                     Clear finished/error download history
  move <gid> <pos>          Reorder the waiting queue
  limit [gid] <speed>       Speed limit, global or per-download (--upload = up; 0 = unlimited)
  set <key=value>...        Set global aria2 options (--gid <gid> = per-download)
  watch [gid]               Live progress, 2s refresh (Ctrl+C to detach)
  restart <gid>             Re-queue a finished/errored download (resumes partial files;
                            torrents re-add via magnet, HTTP via their original URLs)
  retry <gid|all>           Smart retry: waits out internet outages, then re-queues and
                            re-checks; reports real problems (dead source) instead of looping
                            (--once: single check without waiting — used by the healer timer)
  replace <gid> <url>       Give a dead download a fresh URL: same dir + file name, resumes
                            the partial file (status flags downloads that need this)

Options (add / torrent / metalink):
  --dir <path>    download directory (default: $DOWNLOAD_DIR)
  --out <name>    output file name (add only)
  --split <n>     connections per download (default: 16)
  --seed          torrent: keep seeding after download (torrent only)
  --tmux          show live progress in a tmux session named after the file
                  (attach with: tmux attach -t dl-<name>)

Options (retry):
  --interval <s>  poll seconds while waiting for internet (default: 30)
  --max-wait <s>  give up waiting after N seconds (default: 0 = wait forever)
  --once          single pass: no waiting for internet (healer timer mode)
  --quiet         suppress progress output (exit 0 even if retries failed)

Examples:
  pos network download add https://example.com/ubuntu.iso --tmux
  pos network download add https://example.com/a.bin https://example.com/b.bin --dir /mnt/hdd/dl
  pos network download torrent file.torrent --seed
  pos network download list
  pos network download limit 2M
  pos network download watch
  pos network download restart 2e9dffc4
  pos network download retry all
  pos network download replace 2e9dffc4 https://mirror.example.com/ubuntu.iso

Failed downloads are auto-retried by the 'retry healer' systemd timer while the
daemon runs; it arms on download start and disables itself when nothing is left.
Sources that fail with a real 404/410 are marked permanent — 'status' flags them
with a fresh-link hint and 'replace <gid> <url>' resumes them with a new URL.

Config: RPC secret in $CONFIG_FILE (chmod 600).
Env:   RPC_PORT, RPC_SECRET, DOWNLOAD_DIR, NET_PROBE override defaults (test seams).
EOF
    exit 0
}

# ── JSON helpers ────────────────────────────────────────────────
json_arr() { jq -nc '$ARGS.positional' --args "$@"; }
json_str() { jq -nc --arg s "$1" '$s'; }

opts_json() {  # key=value... → JSON object
    local kv k v o="{}"
    for kv in "$@"; do
        k="${kv%%=*}"; v="${kv#*=}"
        o=$(jq -nc --argjson o "$o" --arg k "$k" --arg v "$v" '$o + {($k): $v}')
    done
    printf '%s' "$o"
}

# ── RPC client ─────────────────────────────────────────────────
rpc() {  # rpc <method> [json-args...]
    local method="$1"; shift
    local params="" a
    for a in "$@"; do params="$params,$a"; done
    local body
    body=$(curl -fsS --noproxy '*' -m 60 -H 'Content-Type: application/json' \
        -d "{\"jsonrpc\":\"2.0\",\"id\":1,\"method\":\"$method\",\"params\":[\"token:${RPC_SECRET}\"${params}]}" \
        "$RPC_URL" 2>/dev/null) \
        || err "RPC failed — is the daemon running? (pos network download start)"
    local ecode
    ecode=$(printf '%s' "$body" | jq -r '.error.code // empty' 2>/dev/null || true)
    if [ -n "$ecode" ]; then
        err "aria2 RPC error $ecode: $(printf '%s' "$body" | jq -r '.error.message // "?"')"
    fi
    printf '%s' "$body"
}

# ── Daemon lifecycle (systemd user service) ────────────────────
daemon_active() { systemctl --user is-active --quiet "$SERVICE" 2>/dev/null; }

cmd_start() {
    if [ -z "$RPC_SECRET" ]; then
        RPC_SECRET=$(head -c 24 /dev/urandom | od -An -tx1 | tr -d ' \n')
        if [ "${DRY_RUN:-0}" -eq 1 ]; then
            log "(dry-run) write RPC secret to $CONFIG_FILE"
        else
            mkdir -p "$CONFIG_DIR"
            printf 'RPC_SECRET=%s\n' "$RPC_SECRET" > "$CONFIG_FILE"
            chmod 600 "$CONFIG_FILE"
            log "rpc secret written: $CONFIG_FILE"
        fi
    fi
    mkdir -p "$DOWNLOAD_DIR" 2>/dev/null || warn "cannot create download dir: $DOWNLOAD_DIR"
    if [ "${DRY_RUN:-0}" -eq 1 ]; then
        log "(dry-run) install unit $USER_SYSTEMD_DIR/$SERVICE"
        log "(dry-run) systemctl --user daemon-reload && enable --now $SERVICE"
    else
        mkdir -p "$USER_SYSTEMD_DIR"
        cat > "$USER_SYSTEMD_DIR/$SERVICE" <<EOF
[Unit]
Description=pos aria2 RPC daemon (linux-post-install)

[Service]
Type=simple
ExecStart=/usr/bin/aria2c --enable-rpc --rpc-listen-port=$RPC_PORT --rpc-secret=$RPC_SECRET --rpc-listen-all=false "--dir=$DOWNLOAD_DIR" --continue=true --max-connection-per-server=16 --split=16 --seed-time=0 --console-log-level=warn
Restart=on-failure
RestartSec=2
KillMode=control-group
TimeoutStopSec=5s

[Install]
WantedBy=default.target
EOF
        chmod 644 "$USER_SYSTEMD_DIR/$SERVICE"
        systemctl --user daemon-reload
        systemctl --user enable --now "$SERVICE"
        sleep 1
    fi
    log "aria2 daemon running on http://127.0.0.1:$RPC_PORT (dir: $DOWNLOAD_DIR)"
    if command -v loginctl >/dev/null 2>&1; then
        if ! loginctl show-user "$(id -un)" 2>/dev/null | grep -q '^Linger=yes'; then
            warn "enable linger so the daemon survives logout: sudo loginctl enable-linger $(id -un)"
        fi
    fi
}

cmd_stop() {
    if [ ! -f "$USER_SYSTEMD_DIR/$SERVICE" ]; then
        warn "no aria2 daemon service installed ($SERVICE)"
        exit 0
    fi
    if [ "${DRY_RUN:-0}" -eq 1 ]; then
        log "(dry-run) systemctl --user disable --now $SERVICE; remove unit"
    else
        systemctl --user disable --now "$SERVICE" 2>/dev/null || true
        rm -f "$USER_SYSTEMD_DIR/$SERVICE"
        systemctl --user daemon-reload
    fi
    log "aria2 daemon stopped and removed"
}

cmd_status() {
    if daemon_active; then echo "daemon:     running"; else echo "daemon:     not running"; fi
    if systemctl --user is-enabled "$SERVICE" >/dev/null 2>&1; then
        echo "autostart:  enabled (starts on login)"
    else
        echo "autostart:  disabled"
    fi
    echo "rpc:        http://127.0.0.1:$RPC_PORT"
    [ -f "$CONFIG_FILE" ] && echo "secret:     $CONFIG_FILE"
    echo "dir:        $DOWNLOAD_DIR"
    if daemon_active; then
        rpc aria2.getGlobalStat | jq -r '"active: \(.result.numActive)   waiting: \(.result.numWaiting)   stopped: \(.result.numStopped)   (history: \(.result.numStoppedTotal))"'
        echo "aria2:      $(rpc aria2.getVersion | jq -r '.result.version')"
        dead_source_advisory
    fi
}

dead_source_advisory() {
    # Stopped errored downloads whose source is marked permanently failing —
    # they need a fresh link via 'replace <gid> <url>'.
    local ids
    ids=$(grep . "$RETRY_STATE" 2>/dev/null || true)
    [ -n "$ids" ] || return 0
    rpc aria2.tellStopped 0 200 | jq -r --arg ids "$ids" '
        .result[] | select(.status == "error") |
        ((if .bittorrent then "bt:" + .bittorrent.infoHash else "url:" + (.files[0].uris[0].uri // .files[0].path // "?") end)) as $id |
        select(($ids | split("\n")) | index($id)) |
        [.gid, (if .bittorrent then (.bittorrent.info.name // "?") else (.files[0].path // "?" | split("/") | last) end)] | @tsv' \
        | while IFS=$'\t' read -r gid name; do
            printf '  needs fresh link: %s (%s) — pos network download replace %s <new-url>\n' "$name" "$gid" "$gid"
        done
}

ensure_daemon() {
    if ! daemon_active; then
        log "daemon not running — starting it"
        cmd_start
    fi
}

# ── Submit ─────────────────────────────────────────────────────
basename_url() {
    local u="$1" b
    b="${u##*/}"; b="${b%%\?*}"; b="${b%%\#*}"
    [ -n "$b" ] || b="download"
    printf '%s' "$b"
}

cmd_add() {
    local dir="$DOWNLOAD_DIR" out="" split="" tmux=0
    local urls=()
    while [ $# -gt 0 ]; do
        case "$1" in
            --dir)   [ $# -ge 2 ] || err "--dir requires a value"; dir="$2"; shift 2 ;;
            --dir=*) dir="${1#*=}"; shift ;;
            --out)   [ $# -ge 2 ] || err "--out requires a value"; out="$2"; shift 2 ;;
            --out=*) out="${1#*=}"; shift ;;
            --split) [ $# -ge 2 ] || err "--split requires a value"; split="$2"; shift 2 ;;
            --split=*) split="${1#*=}"; shift ;;
            --tmux)  tmux=1; shift ;;
            -h|--help) usage ;;
            -*) err "add: unknown option: $1" ;;
            *) urls+=("$1"); shift ;;
        esac
    done
    [ ${#urls[@]} -gt 0 ] || err "add: no URL given"
    [ "$tmux" -eq 1 ] && { command -v tmux &>/dev/null || err "tmux not found (install tmux) — needed for --tmux"; }
    dir="${dir/#\~/$HOME}"
    ensure_daemon
    local oa=()
    [ -n "$dir" ]   && oa+=("dir=$dir")
    [ -n "$out" ]   && oa+=("out=$out")
    [ -n "$split" ] && oa+=("split=$split")
    local resp gid
    resp=$(rpc aria2.addUri "$(json_arr "${urls[@]}")" "$(opts_json "${oa[@]}")")
    gid=$(printf '%s' "$resp" | jq -r '.result')
    log "added $gid: ${urls[*]}"
    ensure_healer
    if [ "$tmux" -eq 1 ]; then tmux_watch "$gid" "${out:-$(basename_url "${urls[0]}")}"; fi
}

cmd_torrent() {
    local dir="$DOWNLOAD_DIR" seed=0 tmux=0
    local items=()
    while [ $# -gt 0 ]; do
        case "$1" in
            --dir)   dir="${2:-$DOWNLOAD_DIR}"; shift 2 ;;
            --dir=*) dir="${1#*=}"; shift ;;
            --seed)  seed=1; shift ;;
            --tmux)  tmux=1; shift ;;
            -h|--help) usage ;;
            -*) err "torrent: unknown option: $1" ;;
            *) items+=("$1"); shift ;;
        esac
    done
    [ ${#items[@]} -gt 0 ] || err "torrent: need a .torrent file or magnet link"
    [ "$tmux" -eq 1 ] && { command -v tmux &>/dev/null || err "tmux not found (install tmux) — needed for --tmux"; }
    dir="${dir/#\~/$HOME}"
    ensure_daemon
    local item
    for item in "${items[@]}"; do
        local resp gid name
        if [[ "$item" =~ ^magnet: ]]; then
            resp=$(rpc aria2.addUri "$(json_arr "$item")" "$(opts_json "dir=$dir")")
            name="magnet"
        else
            [ -f "$item" ] || err "torrent: no such file: $item"
            name=$(basename "$item"); name="${name%.torrent}"
            resp=$(rpc aria2.addTorrent "$(json_str "$(base64 -w0 "$item")")" "[]" "$(opts_json "dir=$dir")")
        fi
        gid=$(printf '%s' "$resp" | jq -r '.result')
        if [ "$seed" -eq 1 ]; then
            rpc aria2.changeOption "$(json_str "$gid")" "$(opts_json "seed-time=14400 seed-ratio=0")" >/dev/null
            log "added torrent $gid: $item (seeding until stopped)"
        else
            log "added torrent $gid: $item"
        fi
        if [ "$tmux" -eq 1 ]; then tmux_watch "$gid" "$name"; fi
    done
    ensure_healer
}

cmd_metalink() {
    local dir="$DOWNLOAD_DIR" tmux=0 item=""
    while [ $# -gt 0 ]; do
        case "$1" in
            --dir)   dir="${2:-$DOWNLOAD_DIR}"; shift 2 ;;
            --dir=*) dir="${1#*=}"; shift ;;
            --tmux)  tmux=1; shift ;;
            -h|--help) usage ;;
            -*) err "metalink: unknown option: $1" ;;
            *) item="$1"; shift ;;
        esac
    done
    [ -n "$item" ] || err "metalink: need a .metalink file or URL"
    [ "$tmux" -eq 1 ] && { command -v tmux &>/dev/null || err "tmux not found (install tmux) — needed for --tmux"; }
    dir="${dir/#\~/$HOME}"
    ensure_daemon
    local src="$item" name
    if [[ "$item" =~ ^https?:// ]]; then
        src=$(mktemp)
        trap 'rm -f "$src"' EXIT
        curl -fsSL --noproxy '*' -m 120 -o "$src" "$item" || err "metalink: failed to fetch $item"
        name=$(basename_url "$item")
    else
        [ -f "$item" ] || err "metalink: no such file: $item"
        name=$(basename "$item")
    fi
    local resp gid
    resp=$(rpc aria2.addMetalink "$(json_str "$(base64 -w0 "$src")")" "{}")
    gid=$(printf '%s' "$resp" | jq -r '.result')
    if [[ "$item" =~ ^https?:// ]]; then
        trap - EXIT
        rm -f "$src"
    fi
    log "added metalink $gid: $item"
    ensure_healer
    if [ "$tmux" -eq 1 ]; then tmux_watch "$gid" "$name"; fi
}

# ── Restart / retry / healer ───────────────────────────────────
net_up() { bash -c "$NET_PROBE" >/dev/null 2>&1; }

resolve_gid() {
    # Accept a full 16-hex gid or the unique 8-char prefix shown by 'list'.
    local g="${1:-}" hits n
    [ -n "$g" ] || err "gid required (see 'pos network download list')"
    [[ "$g" =~ ^[0-9a-f]{16}$ ]] && { printf '%s' "$g"; return 0; }
    [[ "$g" =~ ^[0-9a-f]{1,15}$ ]] || err "invalid gid: $g (16 hex chars, or the short id from 'list')"
    hits=$( { rpc aria2.tellStopped 0 200; rpc aria2.tellActive; rpc aria2.tellWaiting 0 200; } \
        | jq -r --arg p "$g" '.result[]? | select(.gid | startswith($p)) | .gid' )
    hits=$(printf '%s\n' "$hits" | sed '/^$/d')
    n=$(printf '%s\n' "$hits" | sed '/^$/d' | wc -l | tr -d ' ')
    [ "$n" -eq 0 ] && err "no download matches gid '$g' (stopped history is lost when the daemon restarts)"
    [ "$n" -gt 1 ] && err "gid prefix '$g' is ambiguous — matches: $(printf '%s' "$hits" | paste -sd' ' -)"
    printf '%s\n' "$hits"
}

download_name() {  # download_name <tellStatus-json> → name of the file / torrent
    printf '%s' "$1" | jq -r '(.result.bittorrent.info.name // (.result.files[0].path // "")) | split("/") | last'
}

do_restart() {  # do_restart <gid> <dir> <seed 0|1> <split> → prints new gids
    local gid="$1" dir="$2" seed="$3" split="$4"
    local data st is_bt newgids=() uri fdir fname oa ng d magnet
    data=$(rpc aria2.tellStatus "$(json_str "$gid")")
    st=$(printf '%s' "$data" | jq -r '.result.status')
    case "$st" in
        complete|error|removed) ;;
        active) err "gid $gid is active — it's already downloading" ;;
        *) err "gid $gid has status '$st' — cannot restart" ;;
    esac
    is_bt=$(printf '%s' "$data" | jq -r 'if .result.bittorrent then 1 else 0 end')
    if [ "$is_bt" = "1" ]; then
        magnet=$(printf '%s' "$data" | jq -r '.result.bittorrent as $b | "magnet:?xt=urn:btih:\($b.infoHash)" + (($b.announceList // []) | flatten | map("&tr=" + @uri) | join(""))')
        d="${dir:-$(printf '%s' "$data" | jq -r '.result.dir')}"; d="${d/#\~/$HOME}"
        oa=()
        [ -n "$d" ] && oa+=("dir=$d")
        [ -n "$split" ] && oa+=("split=$split")
        ng=$(rpc aria2.addUri "$(json_arr "$magnet")" "$(opts_json "${oa[@]}")" | jq -r '.result')
        newgids+=("$ng")
        if [ "$seed" -eq 1 ]; then
            rpc aria2.changeOption "$(json_str "$ng")" "$(opts_json "seed-time=14400 seed-ratio=0")" >/dev/null
        fi
    else
        while IFS=$'\t' read -r uri fdir fname; do
            [ -n "$uri" ] || continue
            oa=()
            [ -n "$fdir" ] && oa+=("dir=$fdir")
            [ -n "$fname" ] && oa+=("out=$fname")
            [ -n "$split" ] && oa+=("split=$split")
            ng=$(rpc aria2.addUri "$(json_arr "$uri")" "$(opts_json "${oa[@]}")" | jq -r '.result')
            newgids+=("$ng")
        done < <(printf '%s' "$data" | jq -r '.result.files[] | [(.uris[0].uri // ""), (.path | split("/") | .[:-1] | join("/")), (.path | split("/") | last)] | @tsv')
    fi
    [ ${#newgids[@]} -gt 0 ] || err "gid $gid has no reusable URIs"
    printf '%s\n' "${newgids[@]}"
}

cmd_restart() {
    local gid="" dir="" split="" seed=0 tmux=0 newgids=() name=""
    while [ $# -gt 0 ]; do
        case "$1" in
            --dir)   dir="${2:-}"; shift 2 ;;
            --dir=*) dir="${1#*=}"; shift ;;
            --seed)  seed=1; shift ;;
            --split) split="${2:-}"; shift 2 ;;
            --split=*) split="${1#*=}"; shift ;;
            --tmux)  tmux=1; shift ;;
            -h|--help) usage ;;
            -*) err "restart: unknown option: $1" ;;
            *) gid="$1"; shift ;;
        esac
    done
    [ -n "$gid" ] || err "restart: gid required (see 'pos network download list')"
    [ "$tmux" -eq 1 ] && { command -v tmux &>/dev/null || err "tmux not found (install tmux) — needed for --tmux"; }
    dir="${dir/#\~/$HOME}"
    ensure_daemon
    gid=$(resolve_gid "$gid")
    mapfile -t newgids < <(do_restart "$gid" "$dir" "$seed" "$split")
    [ ${#newgids[@]} -gt 0 ] || err "restart failed"
    name=$(download_name "$(rpc aria2.tellStatus "$(json_str "$gid")")")
    [ -n "$name" ] || name="download"
    if [ "$seed" -eq 1 ]; then
        log "restarted $gid → ${newgids[*]}: $name (seeding until stopped)"
    else
        log "restarted $gid → ${newgids[*]}: $name"
    fi
    ensure_healer
    if [ "$tmux" -eq 1 ]; then tmux_watch "${newgids[0]}" "$name"; fi
}

cmd_replace() {  # replace <gid> <url> — fresh link for a dead download, same dir/file (resumes partial)
    local gid="" url="" dir="" split="" tmux=0 data st nfiles old_uri odir name newgid oa d
    while [ $# -gt 0 ]; do
        case "$1" in
            --dir)   dir="${2:-}"; shift 2 ;;
            --dir=*) dir="${1#*=}"; shift ;;
            --split) split="${2:-}"; shift 2 ;;
            --split=*) split="${1#*=}"; shift ;;
            --tmux)  tmux=1; shift ;;
            -h|--help) usage ;;
            -*) err "replace: unknown option: $1" ;;
            *) if [ -z "$gid" ]; then gid="$1"; else url="$1"; fi; shift ;;
        esac
    done
    [ -n "$gid" ] && [ -n "$url" ] || err "replace: usage: pos network download replace <gid> <new-url>"
    [ "$tmux" -eq 1 ] && { command -v tmux &>/dev/null || err "tmux not found (install tmux) — needed for --tmux"; }
    dir="${dir/#\~/$HOME}"
    ensure_daemon
    gid=$(resolve_gid "$gid")
    data=$(rpc aria2.tellStatus "$(json_str "$gid")")
    st=$(printf '%s' "$data" | jq -r '.result.status')
    case "$st" in
        complete|error|removed) ;;
        active) err "gid $gid is active — it's already downloading" ;;
        *) err "gid $gid has status '$st' — cannot replace" ;;
    esac
    if [ "$(printf '%s' "$data" | jq -r 'if .result.bittorrent then 1 else 0 end')" = "1" ]; then
        err "gid $gid is a torrent — a fresh URL can't replace it (use: restart $gid to re-add the same magnet)"
    fi
    nfiles=$(printf '%s' "$data" | jq -r '.result.files | length')
    [ "$nfiles" -eq 1 ] || err "gid $gid has $nfiles files — replace supports single-file downloads only"
    old_uri=$(printf '%s' "$data" | jq -r '.result.files[0].uris[0].uri // ""')
    name=$(download_name "$data"); [ -n "$name" ] || name="download"
    odir=$(printf '%s' "$data" | jq -r '.result.dir // ""')
    d="${dir:-$odir}"
    oa=()
    [ -n "$d" ] && oa+=("dir=$d")
    oa+=("out=$name")
    [ -n "$split" ] && oa+=("split=$split")
    newgid=$(rpc aria2.addUri "$(json_arr "$url")" "$(opts_json "${oa[@]}")" | jq -r '.result')
    log "replaced $gid → $newgid: $name — re-queued with fresh link (resumes partial file)"
    [ -n "$old_uri" ] && retry_unmark "url:$old_uri"
    retry_verify "$newgid" "url:$url" || err "replacement link also failed — see messages above"
    ensure_healer
    if [ "$tmux" -eq 1 ]; then tmux_watch "$newgid" "$name"; fi
}

retry_src_id() {  # stable identity for the permanent-failure list (survives new gids)
    local data
    data=$(rpc aria2.tellStatus "$(json_str "$1")")
    printf '%s' "$data" | jq -r 'if .result.bittorrent then "bt:" + .result.bittorrent.infoHash else "url:" + (.result.files[0].uris[0].uri // .result.files[0].path // "?") end'
}

retry_is_permanent() { grep -qxF "$1" "$RETRY_STATE" 2>/dev/null; }

retry_mark_permanent() {
    mkdir -p "$CONFIG_DIR"
    retry_is_permanent "$1" || printf '%s\n' "$1" >> "$RETRY_STATE"
}

retry_unmark() {  # retry_unmark <src-id> — drop one source from the permanent list
    [ -f "$RETRY_STATE" ] || return 0
    grep -vxF "$1" "$RETRY_STATE" > "$RETRY_STATE.tmp" || true
    mv "$RETRY_STATE.tmp" "$RETRY_STATE"
}

retry_verify() {  # retry_verify <newgid> <src-id> → 0 ok / 1 failed (marks real failures permanent)
    local g="$1" src="$2" attempt=0 st ec em data
    while [ "$attempt" -lt 5 ]; do
        attempt=$((attempt + 1))
        sleep "$RETRY_VERIFY_SLEEP"
        data=$(rpc aria2.tellStatus "$(json_str "$g")" 2>/dev/null || true)
        st=$(printf '%s' "$data" | jq -r '.result.status // "?"')
        case "$st" in
            active|waiting) [ "${quiet:-0}" -eq 1 ] || log "retry running — $g (${g:0:8})"; return 0 ;;
            complete) [ "${quiet:-0}" -eq 1 ] || log "retry finished instantly — $g already on disk"; return 0 ;;
            error|removed)
                ec=$(printf '%s' "$data" | jq -r '.result.errorCode // "?"')
                em=$(printf '%s' "$data" | jq -r '.result.errorMessage // "?"')
                if [ "$ec" = "3" ]; then
                    retry_mark_permanent "$src"
                    [ "${quiet:-0}" -eq 1 ] || err "real problem — source is gone (aria2 error 3: $em). Marked permanent; manual 'restart' overrides."
                    return 1
                fi
                [ "${quiet:-0}" -eq 1 ] || warn "retry attempt $attempt failed (aria2 error $ec: $em) — will re-check"
                ;;
        esac
    done
    [ "${quiet:-0}" -eq 1 ] || warn "retry of $g did not stabilize after 5 checks"
    return 1
}

cmd_retry() {
    local target="all" interval="$RETRY_INTERVAL" maxwait=0 once=0 quiet=0
    local rdir="" rsplit="" seed=0 tmux=0 rargs=() gids=() gid st failures=0 ng newgids src deadline=0
    while [ $# -gt 0 ]; do
        case "$1" in
            --interval)  interval="${2:-30}"; shift 2 ;;
            --interval=*) interval="${1#*=}"; shift ;;
            --max-wait)  maxwait="${2:-0}"; shift 2 ;;
            --max-wait=*) maxwait="${1#*=}"; shift ;;
            --once)  once=1; shift ;;
            --quiet) quiet=1; shift ;;
            --dir)   rdir="${2:-}"; shift 2 ;;
            --dir=*) rdir="${1#*=}"; shift ;;
            --seed)  seed=1; shift ;;
            --split) rsplit="${2:-}"; shift 2 ;;
            --split=*) rsplit="${1#*=}"; shift ;;
            --tmux)  tmux=1; shift ;;
            -h|--help) usage ;;
            -*) err "retry: unknown option: $1" ;;
            *) target="$1"; shift ;;
        esac
    done
    if ! daemon_active; then
        [ "$quiet" -eq 1 ] && return 0
        err "daemon not running — start it first (pos network download start)"
    fi
    rdir="${rdir/#\~/$HOME}"
    [ "$tmux" -eq 1 ] && { command -v tmux &>/dev/null || err "tmux not found (install tmux) — needed for --tmux"; }
    [ -n "$rdir" ] && rargs+=(--dir "$rdir")
    [ "$seed" -eq 1 ] && rargs+=(--seed)
    [ -n "$rsplit" ] && rargs+=(--split "$rsplit")
    [ "$tmux" -eq 1 ] && rargs+=(--tmux)
    if [ "$target" = "all" ]; then
        mapfile -t gids < <(rpc aria2.tellStopped 0 200 | jq -r '.result[] | select(.status == "error") | .gid')
    else
        gid=$(resolve_gid "$target")
        st=$(rpc aria2.tellStatus "$(json_str "$gid")" | jq -r '.result.status')
        [ "$st" = "error" ] || err "gid $target has status '$st' — nothing to retry"
        gids+=("$gid")
    fi
    if [ ${#gids[@]} -eq 0 ]; then
        [ "$quiet" -eq 1 ] || log "nothing to retry — queue clean"
        healer_done
        return 0
    fi
    [ "$maxwait" -gt 0 ] && deadline=$((SECONDS + maxwait))
    while ! net_up; do
        if [ "$once" -eq 1 ]; then
            [ "$quiet" -eq 1 ] || log "no internet — will retry on the next run"
            return 0
        fi
        [ "$quiet" -eq 1 ] || log "no internet — waiting for connectivity (checking every ${interval}s, Ctrl+C to abort)"
        [ "$maxwait" -gt 0 ] && [ "$SECONDS" -ge "$deadline" ] && err "timed out waiting for internet ($maxwait s)"
        sleep "$interval"
    done
    for gid in "${gids[@]}"; do
        src=$(retry_src_id "$gid")
        if retry_is_permanent "$src"; then
            [ "$quiet" -eq 1 ] || log "skip $gid — source marked permanently failing (override: pos network download restart $gid)"
            continue
        fi
        [ "$quiet" -eq 1 ] || log "retrying ${gid:0:8}…"
        newgids=$(do_restart "$gid" "$rdir" "$seed" "$rsplit" || true)
        [ -n "$newgids" ] || { failures=$((failures + 1)); continue; }
        ng=$(printf '%s\n' "$newgids" | head -1)
        retry_verify "$ng" "$src" || failures=$((failures + 1))
    done
    healer_done
    [ "$quiet" -eq 1 ] && return 0
    [ "$failures" -eq 0 ] || err "$failures retry(s) failed — see messages above"
}

install_healer_units() {
    mkdir -p "$USER_SYSTEMD_DIR"
    local runner
    if [ -x /usr/local/bin/pos-network-download ]; then
        runner=/usr/local/bin/pos-network-download
    else
        runner="$(cd "$(dirname "$0")/.." && pwd)/bin/pos-network-download"
        warn "using repo path $runner — re-run 'install.sh' so the healer survives a deleted repo"
    fi
    if [ "${DRY_RUN:-0}" -eq 1 ]; then
        log "(dry-run) install $HEALER_SERVICE + $HEALER_TIMER in $USER_SYSTEMD_DIR"
        return 0
    fi
    cat > "$USER_SYSTEMD_DIR/$HEALER_SERVICE" <<EOF
[Unit]
Description=pos aria2 retry healer (linux-post-install)

[Service]
Type=oneshot
Environment=PATH=/usr/local/bin:/usr/bin:/bin
ExecStart=$runner retry all --once --quiet
TimeoutStopSec=5s
EOF
    chmod 644 "$USER_SYSTEMD_DIR/$HEALER_SERVICE"
    cat > "$USER_SYSTEMD_DIR/$HEALER_TIMER" <<EOF
[Unit]
Description=pos aria2 retry healer timer (linux-post-install)

[Timer]
OnUnitActiveSec=2min
OnBootSec=2min
AccuracySec=30s
Persistent=true

[Install]
WantedBy=timers.target
EOF
    chmod 644 "$USER_SYSTEMD_DIR/$HEALER_TIMER"
    systemctl --user daemon-reload
}

ensure_healer() {
    # Arm the retry timer whenever a download starts; it heals failed downloads.
    if ! systemctl --user is-enabled "$HEALER_TIMER" >/dev/null 2>&1; then
        install_healer_units
        if [ "${DRY_RUN:-0}" -eq 1 ]; then
            log "(dry-run) systemctl --user enable --now $HEALER_TIMER"
        else
            systemctl --user enable --now "$HEALER_TIMER" >/dev/null 2>&1 \
                || warn "could not enable retry healer ($HEALER_TIMER)"
            log "retry healer armed — failed downloads auto-retry every 2 min"
        fi
    fi
}

healer_done() {
    # Disable the timer when nothing is left to heal (no active/waiting/errored).
    [ -f "$USER_SYSTEMD_DIR/$HEALER_TIMER" ] || return 0
    local n
    n=$(rpc aria2.tellActive | jq '.result | length')
    [ "$n" -gt 0 ] && return 0
    n=$(rpc aria2.tellWaiting 0 200 | jq '.result | length')
    [ "$n" -gt 0 ] && return 0
    n=$(rpc aria2.tellStopped 0 200 | jq '[.result[] | select(.status == "error")] | length')
    [ "$n" -gt 0 ] && return 0
    if [ "${DRY_RUN:-0}" -eq 1 ]; then
        log "(dry-run) systemctl --user disable --now $HEALER_TIMER"
    else
        systemctl --user disable --now "$HEALER_TIMER" >/dev/null 2>&1 || true
        log "retry healer disabled — no active or failed downloads left"
    fi
}

# ── Listing / inspection ───────────────────────────────────────
LIST_JQ='.result[] | [.gid[0:8], .status,
 (if ((.totalLength|tonumber) > 0) then (100*(.completedLength|tonumber)/(.totalLength|tonumber)) else 0 end),
 (.downloadSpeed|tonumber), (.uploadSpeed|tonumber),
 (.bittorrent.info.name // ((.files[0].path // "")|split("/")|last))] | @tsv'

fmt_list() {
    awk -F'\t' "$ROW_AWK"'{ printf "  %-8s %-9s %6.1f%% %9s %9s %s\n", $1, $2, $3, hs($4), hs($5), $6 }'
}

cmd_list() {
    local act wait stp n
    act=$(rpc aria2.tellActive)
    wait=$(rpc aria2.tellWaiting 0 100)
    stp=$(rpc aria2.tellStopped 0 50)
    n=$(printf '%s' "$act" | jq '.result | length')
    printf 'ACTIVE (%s)\n' "$n"
    printf '  %-8s %-9s %7s %9s %9s %s\n' GID STATUS PCT DL UL NAME
    printf '%s' "$act" | jq -r "$LIST_JQ" | fmt_list
    n=$(printf '%s' "$wait" | jq '.result | length')
    printf 'WAITING (%s)\n' "$n"
    printf '  %-8s %-9s %7s %9s %9s %s\n' GID STATUS PCT DL UL NAME
    printf '%s' "$wait" | jq -r "$LIST_JQ" | fmt_list
    n=$(printf '%s' "$stp" | jq '.result | length')
    printf 'STOPPED (last %s)\n' "$n"
    printf '  %-8s %-9s %7s %9s %9s %s\n' GID STATUS PCT DL UL NAME
    printf '%s' "$stp" | jq -r "$LIST_JQ" | fmt_list
}

cmd_info() {
    local gid="${1:-}"
    [ -n "$gid" ] || err "info: gid required (see 'pos network download list')"
    rpc aria2.tellStatus "$(json_str "$gid")" | jq -r '
      .result as $r |
      "gid:          \($r.gid)",
      "status:       \($r.status)",
      (if (($r.totalLength|tonumber) > 0) then "progress:     \(100*($r.completedLength|tonumber)/($r.totalLength|tonumber)|floor)%  \($r.completedLength)/\($r.totalLength) bytes" else "progress:     unknown (metadata still downloading)" end),
      "dl speed:     \($r.downloadSpeed) B/s",
      "up speed:     \($r.uploadSpeed) B/s",
      "connections:  \($r.connections)",
      (if ($r.numSeeders // 0 | tonumber) > 0 then "seeders:      \($r.numSeeders)" else empty end),
      (if ($r.errorCode | tonumber) > 0 then "error:        \($r.errorCode) — \($r.errorMessage // "?")" else empty end),
      (if $r.bittorrent then "torrent:      \($r.bittorrent.info.name // "?")" + "\n" + "info-hash:    \($r.infoHash)" else empty end),
      "dir:          \($r.dir)",
      "files:",
      ($r.files[] | "  [\(.index)] \(.path)  \(.completedLength)/\(.length) bytes")'
}

cmd_files() {
    local gid="${1:-}"
    [ -n "$gid" ] || err "files: gid required"
    rpc aria2.getFiles "$(json_str "$gid")" | jq -r '.result[] | [.index, .path, .length, .completedLength] | @tsv' \
        | awk -F'\t' '{ printf "  [%s] %s  %s/%s bytes\n", $1, $2, $4, $3 }'
}

cmd_peers() {
    local gid="${1:-}"
    [ -n "$gid" ] || err "peers: gid required"
    rpc aria2.getPeers "$(json_str "$gid")" | jq -r '.result[] | [.peerId[0:12], .ip, (.port|tostring), .downloadSpeed, .uploadSpeed, ((.progress|tonumber) * 100)] | @tsv' \
        | awk -F'\t' "$ROW_AWK"'{ printf "  %-12s %-16s %6s  dl %s  up %s  %5.1f%%\n", $1, $2, $3, hs($4), hs($5), $6 }'
}

# ── Queue control ──────────────────────────────────────────────
queue_cmd() {
    local single="$1" all="$2" verb="$3" target="${4:-all}"
    if [ "$target" = "all" ]; then
        rpc "$all" >/dev/null
    else
        rpc "$single" "$(json_str "$target")" >/dev/null
    fi
    log "$verb: $target"
}

cmd_pause()  { queue_cmd aria2.pause  aria2.pauseAll  "paused"  "$@"; }
cmd_resume() { queue_cmd aria2.unpause aria2.unpauseAll "resumed" "$@"; }

cmd_remove() {
    local force=0 target="${1:-all}"
    if [ "${1:-}" = "--force" ]; then force=1; target="${2:-all}"; fi
    if [ "$target" = "all" ]; then
        if [ "$force" -eq 1 ]; then rpc aria2.forceRemoveAll >/dev/null; else rpc aria2.removeAll >/dev/null; fi
    else
        if [ "$force" -eq 1 ]; then rpc aria2.forceRemove "$(json_str "$target")" >/dev/null; else rpc aria2.remove "$(json_str "$target")" >/dev/null; fi
    fi
    log "removed: $target"
}

cmd_purge() { rpc aria2.purgeDownloadResult >/dev/null; log "download history purged"; }

cmd_move() {
    local gid="${1:-}" pos="${2:-}"
    [ -n "$gid" ] && [ -n "$pos" ] || err "move: usage: pos network download move <gid> <position>"
    local newpos
    newpos=$(rpc aria2.changePosition "$(json_str "$gid")" "$(json_str "$pos")" "$(json_str "POS_SET")" | jq -r '.result')
    log "moved $gid to position $newpos"
}

# ── Tuning ─────────────────────────────────────────────────────
cmd_limit() {
    local upload=0 gid="" speed=""
    while [ $# -gt 0 ]; do
        case "$1" in
            --upload) upload=1; shift ;;
            -h|--help) usage ;;
            -*) err "limit: unknown option: $1" ;;
            *) if [ -z "$speed" ]; then speed="$1"; else gid="$speed"; speed="$1"; fi; shift ;;
        esac
    done
    [ -n "$speed" ] || err "limit: speed required (0 = unlimited; e.g. 2M)"
    local key="max-overall-download-limit"
    if [ "$upload" -eq 1 ]; then key="max-overall-upload-limit"; fi
    if [ -n "$gid" ]; then
        key="max-download-limit"; [ "$upload" -eq 1 ] && key="max-upload-limit"
        rpc aria2.changeOption "$(json_str "$gid")" "$(opts_json "$key=$speed")" >/dev/null
        log "limit: $gid $key=$speed"
    else
        rpc aria2.changeGlobalOption "$(opts_json "$key=$speed")" >/dev/null
        log "limit: global $key=$speed"
    fi
}

cmd_set() {
    local gid="" kvs=()
    while [ $# -gt 0 ]; do
        case "$1" in
            --gid)   gid="${2:-}"; shift 2 ;;
            --gid=*) gid="${1#*=}"; shift ;;
            -h|--help) usage ;;
            -*) err "set: unknown option: $1" ;;
            *) kvs+=("$1"); shift ;;
        esac
    done
    [ ${#kvs[@]} -gt 0 ] || err "set: need key=value (see: aria2c --help=#rpc-options)"
    local opts
    opts=$(opts_json "${kvs[@]}")
    if [ -n "$gid" ]; then
        rpc aria2.changeOption "$(json_str "$gid")" "$opts" >/dev/null
        log "set: gid $gid — ${kvs[*]}"
    else
        rpc aria2.changeGlobalOption "$opts" >/dev/null
        log "set: global — ${kvs[*]}"
    fi
}

# ── Live view ──────────────────────────────────────────────────
cmd_watch() {
    local gid="${1:-}"
    printf '\033[?25l'
    trap 'st=$?; printf "\033[?25h"; exit "$st"' INT TERM EXIT
    if [ -n "$gid" ]; then
        while :; do
            local raw row st ng
            raw=$(rpc aria2.tellStatus "$(json_str "$gid")")
            row=$(printf '%s' "$raw" | jq -r '
              .result as $r |
              [$r.status,
               (if (($r.totalLength|tonumber) > 0) then (100*($r.completedLength|tonumber)/($r.totalLength|tonumber)) else 0 end),
               $r.completedLength, $r.totalLength,
               ($r.downloadSpeed|tonumber), ($r.uploadSpeed|tonumber),
               ($r.connections|tonumber),
               (if (($r.downloadSpeed|tonumber) > 0) then (((($r.totalLength|tonumber) - ($r.completedLength|tonumber)) / ($r.downloadSpeed|tonumber))) else 0 end)]
              | @tsv' 2>/dev/null || true)
            st=$(printf '%s' "$row" | cut -f1)
            printf '\r\033[J'
            if [ -n "$row" ]; then
                printf '%s' "$row" | awk -F'\t' "$ROW_AWK"'{ printf "  %-9s %6.1f%%  %s / %s  dl %s  up %s  conn %s  eta %s", $1, $2, $3, $4, hs($5), hs($6), $7, et($8) }'
            fi
            case "$st" in
                complete) printf '\n'; exit 0 ;;
                error|removed)
                    if net_up; then
                        printf '\n'
                        err "download $gid failed while internet is up — real problem (diagnose: pos network download info $gid; retry: pos network download retry $gid)"
                    fi
                    printf '\033[J  network down — waiting for connectivity (auto-retry when back)\n'
                    while ! net_up; do sleep "$RETRY_INTERVAL"; done
                    printf '\r\033[J  network back — restarting download\n'
                    gid=$(resolve_gid "$gid")
                    ng=$(do_restart "$gid" "" 0 "" | head -1)
                    [ -n "$ng" ] || exit 1
                    name=$(download_name "$(rpc aria2.tellStatus "$(json_str "$gid")")")
                    [ -n "$name" ] || name="download"
                    log "auto-restarted $gid → $ng: $name"
                    gid="$ng"
                    ;;
            esac
            sleep 2
        done
    else
        while :; do
            local act g n
            act=$(rpc aria2.tellActive)
            g=$(rpc aria2.getGlobalStat)
            n=$(printf '%s' "$act" | jq '.result | length')
            printf '\r\033[J'
            if [ "$n" -eq 0 ]; then
                printf 'no active downloads\n'
                exit 0
            fi
            printf '%s' "$g" | jq -r '[.result.numActive, (.result.downloadSpeed|tonumber), (.result.uploadSpeed|tonumber)] | @tsv' | awk -F'\t' "$ROW_AWK"'{ printf "active %s   dl %s   up %s", $1, hs($2), hs($3) }'
            printf '  %-8s %-9s %7s %9s %9s %s\n' GID STATUS PCT DL UL NAME
            printf '%s' "$act" | jq -r "$LIST_JQ" | fmt_list
            sleep 2
        done
    fi
}

tmux_watch() {
    local gid="$1" name="$2"
    name=$(printf '%s' "$name" | tr -cs 'A-Za-z0-9._-' '-' | cut -c1-40)
    name="${name%-}"
    [ -n "$name" ] || name="download"
    local base sname n=2
    base="dl-$name"
    sname="$base"
    while tmux has-session -t "$sname" 2>/dev/null; do
        sname="${base}-${n}"; n=$((n+1))
    done
    local self
    self="$(cd "$(dirname "$0")" && pwd)/$(basename "$0")"
    if [ "${DRY_RUN:-0}" -eq 1 ]; then
        log "(dry-run) tmux new-session -d -s $sname -- $self watch $gid"
    else
        TERM="${TERM:-xterm-256color}" tmux new-session -d -s "$sname" "$self watch $gid"
    fi
    log "tmux session '$sname' started — attach: tmux attach -t '$sname'"
}

# ── Dispatch ───────────────────────────────────────────────────
overview() {
    cmd_status
    if daemon_active; then echo; cmd_list; fi
}

main() {
    local cmd="${1:-}"
    shift 2>/dev/null || true
    case "$cmd" in
        "")      overview ;;
        -h|--help) usage ;;
        start)   cmd_start ;;
        stop)    cmd_stop ;;
        status)  cmd_status ;;
        add)     cmd_add "$@" ;;
        torrent) cmd_torrent "$@" ;;
        metalink) cmd_metalink "$@" ;;
        list)    cmd_list ;;
        info|files|peers|pause|resume|remove|move|limit|set|watch|restart|retry|replace)
            "cmd_$cmd" "$@" ;;
        purge)   cmd_purge ;;
        *) err "unknown command: $cmd (see 'pos network download --help')" ;;
    esac
}

main "$@"
