#!/bin/sh
# 观澜·如东样板 v2 — Linux/macOS 离线安装 (支持 银河麒麟 V10 / 统信 UOS 20 / Ubuntu / CentOS / macOS)
# 用法: sh install.sh            (自动选 Python; 无合适系统 Python 时用包内便携运行时)
#       GUANLAN_PY=/path/python3 sh install.sh   (指定 Python, 多版本机器用)
set -e
# 选项 (用户令 2026-09-17): --force 跳过"已安装?"的确认; --service 装成 systemd 服务(需 root)
FORCE=0; DO_SERVICE=0
for arg in "$@"; do
  case "$arg" in
    --force|-f) FORCE=1 ;;
    --service|-s) DO_SERVICE=1 ;;
  esac
done
# 同理: 全局 PYTHONPATH 会让 pip 误判依赖已满足而不装进 .venv (见 install.ps1 注释)
unset PYTHONPATH
cd "$(dirname "$0")"; ROOT=$(pwd); export PYTHONUTF8=1 LC_ALL=${LC_ALL:-C.UTF-8}
say(){ printf '%s\n' "$*"; }

ARCH=$(uname -m); case "$ARCH" in x86_64|amd64) A=x86_64;; aarch64|arm64) A=aarch64;; *) A=$ARCH;; esac
OS=$(uname -s)
DISTRO=$( [ -f /etc/os-release ] && . /etc/os-release && echo "${PRETTY_NAME:-$NAME}" || echo "$OS" )
say "== 观澜 v2 安装 · $DISTRO · $ARCH"

# 0) 安装前检查: 是否已安装及版本 (用户令 2026-09-17)
say "== 0/7 安装前检查: 是否已安装及版本差异"
PKG_VER=$(sed -n "s/^VERSION *= *'\([^']*\)'.*/\1/p" src/version.py 2>/dev/null | head -1)
[ -n "$PKG_VER" ] || PKG_VER=$(sed -n 's/.*"version": *"\([^"]*\)".*/\1/p' dist-manifest.json 2>/dev/null | head -1)
PREV_VER=""; PREV_AT=""; PREV_MODE=""
if [ -f install-info.json ]; then
  PREV_VER=$("$ROOT/.venv/bin/python" -c "import json;print(json.load(open('install-info.json', encoding='utf-8')).get('version',''))" 2>/dev/null \
            || sed -n 's/.*"version": *"\([^"]*\).*/\1/p' install-info.json | head -1)
  PREV_AT=$(sed -n 's/.*"installed_at": *"\([^"]*\).*/\1/p' install-info.json | head -1)
  PREV_MODE=$(sed -n 's/.*"mode": *"\([^"]*\).*/\1/p' install-info.json | head -1)
fi
if [ -z "$PREV_VER" ] && [ ! -f install-info.json ]; then
  say "   未检测到已安装的观澜（将全新安装 v$PKG_VER）"
else
  say "   已安装: v${PREV_VER:-（无版本记录）}（装于 ${PREV_AT:-?}; 运行方式 ${PREV_MODE:-?}）"
  if [ "$PREV_VER" = "$PKG_VER" ]; then
    say "   [i] 版本相同（v$PKG_VER）: 重装会重建 .venv、覆盖程序文件; data/ 与 outputs/ 不受影响"
  elif [ -z "$PREV_VER" ]; then
    say "   [!] 旧安装没有版本记录 —— 本包 v$PKG_VER 将按'重装/升级'处理"
  else
    say "   [!] 版本差异: v$PREV_VER -> v$PKG_VER"
    case "$(printf '%s\n%s\n' "$PREV_VER" "$PKG_VER" | sort -V | tail -1)" in
      "$PREV_VER") say "   [X] 已装版本比本包更新 —— 用旧包覆盖新装通常不是你想要的, 请确认";;
    esac
  fi
  if [ "$FORCE" = "1" ]; then say "   (--force: 跳过询问, 直接继续)"
  elif [ ! -t 0 ]; then say "   (非交互 stdin: 自动继续安装)"
  else
    printf "   是否重新安装/覆盖? 输入 y 继续, 其它键退出: "
    read -r ANS
    case "$ANS" in y|Y|yes|YES) ;; *) say "   已取消（未改任何文件）"; exit 0;; esac
  fi
fi


# 1) 选 Python: 显式指定 > 系统 3.11/3.12 > 包内便携运行时 (麒麟/统信自带多为 3.7/3.9, 跑不了本项目依赖)
PY=""
if [ -n "$GUANLAN_PY" ]; then PY="$GUANLAN_PY"
else
  for c in python3.12 python3.13 python3.11 python3; do
    p=$(command -v $c 2>/dev/null) || continue
    "$p" -c 'import sys; sys.exit(0 if sys.version_info[:2] >= (3,11) else 1)' 2>/dev/null && { PY="$p"; break; }
  done
fi
if [ -z "$PY" ]; then
  RT=$(ls vendor/python/cpython-3.12*-$A-*linux-gnu-install_only*.tar.gz 2>/dev/null | head -1)
  [ "$OS" = Darwin ] && RT=""
  if [ -n "$RT" ]; then
    say "   系统无 Python 3.11+, 用包内便携运行时: $(basename "$RT")"
    rm -rf runtime && mkdir -p runtime && tar -xzf "$RT" -C runtime
    PY="$ROOT/runtime/python/bin/python3"
  else
    say "   ✗ 未找到 Python 3.11+ , 包内也没有本架构 ($A) 的便携运行时。"
    say "     办法一: 装系统 Python 3.11+ ; 办法二: GUANLAN_PY=/路径/python3 sh install.sh"
    exit 2
  fi
fi
say "   Python: $PY ($("$PY" -V 2>&1))"

# 2) 虚拟环境 + 离线依赖 (轮子在 wheels/linux_<arch> 或 wheels/macos_arm64)
"$PY" -m venv .venv 2>/dev/null || "$PY" -m venv --without-pip .venv
VPY=$ROOT/.venv/bin/python
"$VPY" -m ensurepip --upgrade >/dev/null 2>&1 || true
"$VPY" -m pip install --upgrade pip --quiet 2>/dev/null || true
case "$OS" in Darwin) W=wheels/macos_arm64;; *) W=wheels/linux_$A;; esac
if [ -d "$W" ] && [ "$(ls -A "$W" 2>/dev/null | wc -l)" -gt 0 ]; then
  say "== 离线安装依赖 ($W)"; "$VPY" -m pip install --no-index --find-links "$W" -r requirements.txt --quiet
else
  say "== 包内无本平台轮子 ($W), 改为联网安装"; "$VPY" -m pip install -r requirements.txt --quiet
fi

# 3) 写配置 (python 指向本 venv)
"$VPY" - <<'PYEOF'
import json, pathlib, sys
p = pathlib.Path('configs/serve.json')   # portability-allow: 安装脚本已 cd 到安装根
c = json.loads(p.read_text(encoding='utf-8'))
# 写**相对**路径 (相对安装根): 整个目录拷到别的电脑/别的盘后仍然有效 (绝对路径会在换机后失效)
exe = pathlib.Path(sys.executable).resolve(); root = pathlib.Path('.').resolve()
try:
    c['python'] = exe.relative_to(root).as_posix()
except ValueError:
    c['python'] = exe.as_posix()
p.write_text(json.dumps(c, ensure_ascii=False, indent=2) + '\n', encoding='utf-8')
print('   configs/serve.json -> ' + c['python'] + ' (相对安装根)')
PYEOF

# 4) Ollama (可选; 只影响问答与本地审核)
if command -v ollama >/dev/null 2>&1; then say "== Ollama 已安装: $(ollama -v 2>&1 | head -1)"
elif [ -f vendor/ollama/ollama-linux-$( [ "$A" = x86_64 ] && echo amd64 || echo arm64 ).tar.zst ]; then
  say "== 检测到包内 Ollama 离线包; 安装需 root:"
  say "   sudo tar -I zstd -C /usr -xf vendor/ollama/ollama-linux-*.tar.zst && sudo useradd -r -s /bin/false -m -d /usr/share/ollama ollama 2>/dev/null; ollama serve &"
  say "   (麒麟/统信若无 zstd: sudo yum install zstd 或 sudo apt install zstd)"
else say "== 未装 Ollama: 问答与本地审核不可用, 其余页面正常 (装法见 说明书 §5)"; fi

say "== 自检"; "$VPY" guanlan.py check || true
say ""
# 服务化 (可选; 用户令 2026-09-17): systemd 单元 guanlan.service, 需 root
if [ "$DO_SERVICE" = "1" ]; then
  say "== 服务化 (systemd, 需 root)"
  "$VPY" scripts/service_ctl.py install || say "   [!] 服务注册没成功 —— 不影响使用: 仍可用 guanlan.py serve 手工启动"
else
  say "== 服务化 (未指定 --service, 跳过)"
  say "   装成开机自启: sudo $VPY scripts/service_ctl.py install   (先看计划加 --dry-run)"
fi

# 安装记录 (版本单一真源 src/version.py): 安装检查与 guanlan.py check 都读它
MODE=manual; [ "$DO_SERVICE" = "1" ] && MODE=systemd
"$VPY" -c "import sys,pathlib; sys.path.insert(0,str(pathlib.Path('.').resolve())); from src import version as V; print('   安装记录: ' + str(V.write_installed(pathlib.Path('.').resolve(), python=sys.executable, mode='$MODE')))"

say ""
say "安装完成. 服务: $VPY scripts/service_ctl.py status|start|stop"
say "         手工: $VPY guanlan.py serve    停止: $VPY guanlan.py stop    入口: http://127.0.0.1:28084/"