#!/usr/bin/env bash
# crypto_monitor 一键环境部署（Linux / macOS / Git Bash）
#
# 用法:
#   bash deploy/setup_env.sh
#   bash deploy/setup_env.sh --only binance,gate_bot
#   bash deploy/setup_env.sh --skip-pm2
#   bash deploy/setup_env.sh --recreate-venv
#   bash deploy/setup_env.sh --install-system-deps   # root + apt 时安装 python*-venv
#
set -e
set -u
# 避免 Windows CRLF 导致 set -euo pipefail 一行报错；pipefail 仅 bash 支持
if [ -n "${BASH_VERSION:-}" ]; then
  set -o pipefail
fi

DEPLOY_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
REPO_ROOT="$(cd "${DEPLOY_DIR}/.." && pwd)"
REQ_FILE="${REPO_ROOT}/requirements.txt"
HUB_REQ="${REPO_ROOT}/manual_trading_hub/requirements.txt"

ONLY="all"
SKIP_PM2=0
SKIP_ENV_COPY=0
RECREATE_VENV=0
INSTALL_APT_DEPS=0
PY=""

usage() {
  sed -n '2,12p' "$0" | sed 's/^# \?//'
  exit "${1:-0}"
}

while [[ $# -gt 0 ]]; do
  case "$1" in
    --only) ONLY="${2:-all}"; shift 2 ;;
    --skip-pm2) SKIP_PM2=1; shift ;;
    --skip-env-copy) SKIP_ENV_COPY=1; shift ;;
    --recreate-venv) RECREATE_VENV=1; shift ;;
    --install-system-deps) INSTALL_APT_DEPS=1; shift ;;
    -h|--help) usage 0 ;;
    *) echo "未知参数: $1" >&2; usage 1 ;;
  esac
done

step() { echo ""; echo "==> $*"; }

should_include() {
  local key="$1"
  if [[ "${ONLY}" == "all" ]]; then
    return 0
  fi
  local item
  IFS=',' read -ra PARTS <<< "${ONLY}"
  for item in "${PARTS[@]}"; do
    item="$(echo "${item}" | tr '[:upper:]' '[:lower:]' | xargs)"
    [[ "${item}" == "${key}" ]] && return 0
  done
  return 1
}

find_python() {
  if command -v python3 >/dev/null 2>&1; then
    echo python3
    return
  fi
  if command -v python >/dev/null 2>&1; then
    echo python
    return
  fi
  echo "未找到 python3/python，请先安装 Python 3.10+" >&2
  exit 1
}

check_python_version() {
  local py="$1"
  local ver
  ver="$("${py}" -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")')"
  local major minor
  major="${ver%%.*}"
  minor="${ver#*.}"
  if [[ "${major}" -lt 3 ]] || [[ "${major}" -eq 3 && "${minor}" -lt 10 ]]; then
    echo "需要 Python 3.10+，当前: ${ver}" >&2
    exit 1
  fi
  echo "Python: $("${py}" --version 2>&1)"
}

python_minor_version() {
  local py="$1"
  "${py}" -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")'
}

check_venv_available() {
  local py="$1"
  local tmp
  tmp="$(mktemp -d 2>/dev/null || mktemp -d -t cmvenv)"
  if "${py}" -m venv "${tmp}" >/dev/null 2>&1 && [[ -x "${tmp}/bin/python" ]]; then
    rm -rf "${tmp}"
    return 0
  fi
  rm -rf "${tmp}" 2>/dev/null || true
  return 1
}

install_debian_venv_packages() {
  local py="$1"
  local ver
  ver="$(python_minor_version "${py}")"
  if ! command -v apt-get >/dev/null 2>&1; then
    echo "  未检测到 apt-get，请手动安装 python${ver}-venv 与 python3-pip" >&2
    return 1
  fi
  if [[ "$(id -u)" -ne 0 ]]; then
    echo "  需要 root 安装系统包，请执行:" >&2
    echo "    sudo apt update && sudo apt install -y python${ver}-venv python3-pip curl" >&2
    echo "  或: sudo bash deploy/setup_env.sh --install-system-deps" >&2
    return 1
  fi
  step "安装系统依赖 (python${ver}-venv) ..."
  export DEBIAN_FRONTEND=noninteractive
  apt-get update -qq
  if ! apt-get install -y "python${ver}-venv" python3-pip curl ca-certificates; then
    apt-get install -y python3-venv python3-pip curl ca-certificates
  fi
}

ensure_venv_prereqs() {
  local py="$1"
  if check_venv_available "${py}"; then
    return 0
  fi
  echo "  当前 Python 无法创建 venv（缺少 ensurepip，常见于未安装 python*-venv）" >&2
  if [[ "${INSTALL_APT_DEPS}" -eq 1 ]] || [[ "$(id -u)" -eq 0 ]]; then
    install_debian_venv_packages "${py}" || exit 1
    if check_venv_available "${py}"; then
      return 0
    fi
  fi
  local ver
  ver="$(python_minor_version "${py}")"
  echo "请安装后重试:" >&2
  echo "  apt update && apt install -y python${ver}-venv python3-pip" >&2
  echo "  bash deploy/setup_env.sh" >&2
  exit 1
}

create_project_venv() {
  local py="$1"
  if [[ "${RECREATE_VENV}" -eq 1 && -d .venv ]]; then
    echo "  删除旧 venv ..."
    rm -rf .venv
  fi
  if [[ -d .venv && ! -x .venv/bin/python ]]; then
    echo "  清理未完成的 venv ..."
    rm -rf .venv
  fi
  if [[ -x .venv/bin/python ]]; then
    return 0
  fi
  echo "  创建 venv ..."
  if ! "${py}" -m venv .venv; then
    rm -rf .venv 2>/dev/null || true
    echo "  venv 创建失败" >&2
    exit 1
  fi
}

setup_monitor() {
  local dir_name="$1"
  local proj="${REPO_ROOT}/${dir_name}"
  if [[ ! -d "${proj}" ]]; then
    echo "  跳过（目录不存在）: ${dir_name}"
    return
  fi
  step "${dir_name}"
  cd "${proj}"
  create_project_venv "${PY}"
  echo "  升级 pip ..."
  .venv/bin/python -m pip install -U pip setuptools wheel -q
  echo "  安装依赖 ..."
  .venv/bin/pip install -r "${REQ_FILE}" -q
  if [[ "${SKIP_ENV_COPY}" -eq 0 ]]; then
    if [[ -f .env.example && ! -f .env ]]; then
      cp -n .env.example .env 2>/dev/null || cp .env.example .env
      echo "  已复制 .env.example -> .env"
    elif [[ -f .env ]]; then
      echo "  保留已有 .env"
    else
      echo "  无 .env.example，请手动配置 .env"
    fi
  fi
  mkdir -p static/images/order_charts
  echo "  完成: ${proj}/.venv/bin/python"
}

setup_hub() {
  local proj="${REPO_ROOT}/manual_trading_hub"
  if [[ ! -d "${proj}" ]]; then
    echo "  跳过 hub（目录不存在）"
    return
  fi
  step "manual_trading_hub"
  cd "${proj}"
  create_project_venv "${PY}"
  .venv/bin/python -m pip install -U pip setuptools wheel -q
  if [[ -f "${HUB_REQ}" ]]; then
    .venv/bin/pip install -r "${HUB_REQ}" -q
  fi
  if [[ "${SKIP_ENV_COPY}" -eq 0 && -f .env.example && ! -f .env ]]; then
    cp -n .env.example .env 2>/dev/null || cp .env.example .env
    echo "  已复制 .env.example -> .env"
  fi
  echo "  完成: ${proj}/.venv/bin/python"
}

install_pm2() {
  if [[ "${SKIP_PM2}" -eq 1 ]]; then
    return
  fi
  step "PM2（可选）"
  if ! command -v node >/dev/null 2>&1; then
    echo "  未检测到 Node.js，跳过。安装后执行: npm install -g pm2"
    return
  fi
  if command -v pm2 >/dev/null 2>&1; then
    echo "  PM2 已安装: $(pm2 -v)"
    return
  fi
  echo "  正在安装 pm2 ..."
  npm install -g pm2
  echo "  各子目录: pm2 start ecosystem.config.cjs"
}

echo "crypto_monitor 环境部署"
echo "仓库根目录: ${REPO_ROOT}"

[[ -f "${REQ_FILE}" ]] || { echo "缺少 ${REQ_FILE}" >&2; exit 1; }

PY="$(find_python)"
check_python_version "${PY}"
ensure_venv_prereqs "${PY}"

should_include binance && setup_monitor crypto_monitor_binance
should_include gate && setup_monitor crypto_monitor_gate
should_include gate_bot && setup_monitor crypto_monitor_gate_bot
should_include okx && setup_monitor crypto_monitor_okx
should_include hub && setup_hub

install_pm2

echo ""
echo "部署完成。下一步："
echo "  1. 编辑各子目录 .env"
echo "  2. 启动示例: cd crypto_monitor_binance && source .venv/bin/activate && python app.py"
echo "  3. Windows 可用: powershell -File deploy/setup_env.ps1"
echo ""