#!/usr/bin/env bash
# ThinRPA 企业级 RPA 平台 一键安装加载器（由 SN 许可服务渲染）
#
# 本脚本只做两件事：定位/下载"核心安装包"（内含 TUI 向导），然后透传全部
# 参数 exec 包内向导。环境检测、冲突处理、依赖安装全部由向导完成。
#
# 用法：
#   curl -fsSL 'https://sn.thinkalike.com.cn/install.sh?product=THINRPA' | bash -s -- --sn <SN码>
# 常用选项（完整选项由向导 --help 给出）：
#   --sn <SN>      许可序列号（缺省自动申请 90 天试用）
#   --host <name>  nginx 对外主机名/IP
#   --yes          非交互模式（全部按推荐项执行）
#   --scope <s>    full|standard|minimal
#   --tmp <dir>    指定下载/解压的临时目录（默认自动挑一块够大的盘）
#   --dry-run      仅环境检测
#
# 离线采集模式：把 fetch-all.sh 采集到的 thinrpa-core.tar.gz 放在本
# 脚本同目录，将直接复用（不联网下载）。
set -euo pipefail

SN_BASE="https://sn.thinkalike.com.cn"
CORE_URL="$SN_BASE/download/thinrpa-core.tar.gz"
SN=""
ARGS=()

while [ $# -gt 0 ]; do
  case "$1" in
    --sn) SN="${2:-}"; shift 2 ;;
    --tmp) TMP_OVERRIDE="${2:-}"; shift 2 ;;
    --sn-base) SN_BASE="${2:-}"; CORE_URL="$SN_BASE/download/thinrpa-core.tar.gz"; ARGS+=(--sn-base "$SN_BASE"); shift 2 ;;
    *) ARGS+=("$1"); shift ;;
  esac
done

[ "$(id -u)" -eq 0 ] || { echo "run as root"; exit 1; }

# ---- 定位核心包：同目录携带 > 采集目录/当前目录 > 已下载缓存 > 在线下载（断点续传） ----
PKG=""
SRC="${BASH_SOURCE[0]:-}"
if [ -n "$SRC" ]; then SELF_DIR="$(dirname "$SRC")"; else SELF_DIR="."; fi
# 说明书与 README 给了两条都能走通的路径（在 <code>-files 里直接跑、或先解压 core 再
# 从 <code>-core/ 里调 ../<code>-fullinstall/install.sh），加载器必须都找得到包；
# 否则离线交付会静默退化成重新公网下载数 GB（客户现场往往没网）。
for d in "$SELF_DIR" "$PWD" "$SELF_DIR/.." "$PWD/.."; do
  if [ -f "$d/thinrpa-core.tar.gz" ]; then
    PKG="$d/thinrpa-core.tar.gz"
    echo "[loader] 使用随包携带的核心包（离线采集模式）：$PKG"
    break
  fi
done

CORE_NAME="thinrpa-core.tar.gz"
# SN 是 FastAPI，HEAD 返 405，只能用 Range 探针从 Content-Range 里拿总长。
probe_size() {
  curl -sS --connect-timeout 15 --max-time 30 -r 0-0 -D - -o /dev/null "$1" 2>/dev/null \
    | tr -d '\r' | sed -n 's/^[Cc]ontent-[Rr]ange: bytes [0-9]*-[0-9]*\/\([0-9][0-9]*\)$/\1/p' | head -n1
}
avail_kb() { df -P -k "$1" 2>/dev/null | awk 'NR==2{print $4}'; }

# 落地目录不能写死 /tmp：/tmp 常常是 tmpfs 内存盘（实测一台 3.5G 只剩 1.1G），
# 1.4GB 的核心包下到一半就 ENOSPC，客户只看到一句“下载失败”。
SIZE_RAW="$(probe_size "$CORE_URL")"
SIZE="${SIZE_RAW:-0}"
[ "$SIZE" -ge 0 ] 2>/dev/null || SIZE=0
if [ "$SIZE" -eq 0 ]; then
  SIZE=$((2 * 1024 * 1024 * 1024))   # 探不到就按 2GB 保守估，不拿“无上限”当默认
  echo "[loader] NOTE: 未探到核心包大小（离线镜像/代理屏蔽了 Content-Range），按 2GB 预估空间"
fi
# 下载一份 + 解压一份（实测核心包压缩 1.39GB / 解压后 1.23GB ≈ 0.9 份）+ 256MB 余量
NEED_KB=$(( SIZE / 1024 * 2 + 262144 ))
NEED_GB=$(( (NEED_KB + 1048575) / 1048576 ))
WORK=""
for d in "${TMP_OVERRIDE:-}" /var/tmp /tmp "$SELF_DIR"; do
  [ -n "$d" ] || continue
  [ -d "$d" ] && [ -w "$d" ] || continue
  KB="$(avail_kb "$d")"
  [ -n "$KB" ] || continue
  if [ "$KB" -ge "$NEED_KB" ]; then WORK="$d"; break; fi
done
if [ -z "$WORK" ]; then
  echo "[loader] ERROR: 找不到可写且剩余空间≥${NEED_GB}GB 的临时目录（依次试过 --tmp / /var/tmp / /tmp / 当前目录）。"
  echo "         用 --tmp <目录> 指一块数据盘，或先在联网机跑 fetch-all.sh 采集后离线安装。"
  exit 1
fi
echo "[loader] 临时目录: $WORK（核心包 $((SIZE / 1048576))MB，需约 ${NEED_GB}GB）"

# 复用缓存必须确认文件已完整：半截包直接交给 tar 只会报一看不懂的“invalid gzip”。
if [ -z "$PKG" ]; then
  for d in "$WORK" /tmp; do
    [ -f "$d/$CORE_NAME" ] || continue
    HAVE="$(stat -c %s "$d/$CORE_NAME" 2>/dev/null || echo 0)"
    if [ -z "$SIZE_RAW" ] || [ "$HAVE" = "$SIZE" ]; then
      PKG="$d/$CORE_NAME"
      echo "[loader] 复用已下载的核心包：$PKG"
      break
    fi
    echo "[loader] $d/$CORE_NAME 大小 $HAVE 与服务端 $SIZE 不符（上次中断），续传补齐"
  done
fi
if [ -z "$PKG" ]; then
  echo "[loader] 下载核心包: $CORE_URL"
  curl -fL --retry 3 -C - "$CORE_URL" -o "$WORK/$CORE_NAME" \
    || { echo "[loader] ERROR: 核心包下载失败（重跑本命令可断点续传；离线环境请先用 fetch-all.sh 采集，或把 $CORE_NAME 放到本脚本同目录）"; exit 1; }
  PKG="$WORK/$CORE_NAME"
fi

# ---- 解压并移交向导 ----
STAGE="$(mktemp -d "$WORK/thinrpa-core.XXXXXX")"
trap 'rm -rf "$STAGE"' EXIT
tar -xzf "$PKG" -C "$STAGE"
sed -i 's/\r$//' "$STAGE/install.sh" 2>/dev/null || true
[ -f "$STAGE/install.sh" ] || { echo "[loader] ERROR: 核心包内缺少 install.sh 向导"; exit 1; }

# 采集模式的加选包（浏览器/ollama）若在同目录，一并带入向导可见位置
if [ -n "$SRC" ]; then
  for f in "$(dirname "$SRC")"/thinrpa-browser-*.tar.gz \
           "$(dirname "$SRC")"/thinrpa-ollama-*.tar.gz; do
    [ -f "$f" ] && cp -n "$f" "$STAGE/" 2>/dev/null || true
  done
fi

echo "[loader] 移交安装向导…"
SN_ARGS=()
[ -n "$SN" ] && SN_ARGS=(--sn "$SN")
exec bash "$STAGE/install.sh" "${SN_ARGS[@]}" "${ARGS[@]}"