#!/bin/bash

set -e # 出错立即停止

### 函数定义区 ###

# 1. 从SQLite数据库获取当前开放交易的币对列表
get_open_trades_pairs() {
  local db_path="$1"
  local pairs=()

  if [ ! -f "$db_path" ]; then
    echo "⚠️ 交易数据库文件 $db_path 不存在，开放交易对列表为空" >&2
    echo ""
    return
  fi

  local query_result=$(sqlite3 -header -csv "$db_path" "SELECT id, pair, open_date, amount, open_rate, max_rate, close_profit, stake_amount FROM trades WHERE is_open = 1 ORDER BY open_date DESC LIMIT 10;")
  pairs=($(echo "$query_result" | awk -F ',' 'NR > 1 {print $2}' | tr -d '"'))

  echo "从数据库获取到开放交易对: ${pairs[*]}" >&2
  echo "${pairs[@]}"
}

# 2. 从远程接口获取币对列表
get_remote_pairs() {
  local remote_url="$1"
  local pairs=()

  if [ -z "$remote_url" ]; then
    echo "⚠️ 未提供远程币对列表URL，远程币对列表为空" >&2
    echo ""
    return
  fi

  echo "正在从远程URL获取币对列表: $remote_url" >&2
  local pairs_json=$(curl -s "$remote_url")

  if [ $? -ne 0 ] || [ -z "$pairs_json" ]; then
    echo "⚠️ 远程币对列表获取失败，使用空列表" >&2
    echo ""
    return
  fi

  local parsed_pairs=$(echo "$pairs_json" | python3 -c "
import sys, json
data = json.load(sys.stdin)
pairs = [pair.replace('-', '/') for pair in data.get('pairlist', [])]
print(' '.join(pairs) if pairs else '')
")

  if [ -n "$parsed_pairs" ]; then
    pairs=($parsed_pairs)
    echo "从远程获取到币对: ${pairs[*]}" >&2
  else
    echo "⚠️ 远程币对列表解析为空" >&2
  fi

  echo "${pairs[@]}"
}

# 3. 合并并去重两个币对列表
merge_and_deduplicate_pairs() {
  local -a db_pairs=($1)
  local -a remote_pairs=($2)
  local -a merged=()

  merged=($(printf "%s\n" "${db_pairs[@]}" "${remote_pairs[@]}" | sort -u | tr '\n' ' '))

  echo "合并去重后的币对列表: ${merged[*]}" >&2
  echo "${merged[@]}"
}

# 4. 验证JSON文件格式是否有效
validate_json_file() {
  local json_path="$1"

  if ! command -v jq &>/dev/null; then
    echo "❌ 未安装jq工具，请先安装jq" >&2
    exit 1
  fi

  if [ ! -f "$json_path" ]; then
    echo "❌ JSON配置文件 $json_path 不存在" >&2
    exit 1
  fi

  if ! jq . "$json_path" >/dev/null 2>&1; then
    echo "❌ JSON文件格式错误：$json_path" >&2
    exit 1
  fi
}

# 5. 更新live.json中的pair_whitelist
update_live_json_pair_whitelist() {
  local json_path="$1"
  local -a pairlist=($2)

  validate_json_file "$json_path"

  local json_array=$(printf '%s\n' "${pairlist[@]}" | jq -R . | jq -s .)

  echo "正在更新 $json_path 中的exchange.pair_whitelist..." >&2
  jq --argjson pairs "$json_array" '.exchange.pair_whitelist = $pairs' "$json_path" >"$json_path.tmp" && mv "$json_path.tmp" "$json_path"

  if [ $? -eq 0 ]; then
    echo "✅ 成功更新 $json_path，新的pair_whitelist包含 ${#pairlist[@]} 个币对" >&2
  else
    echo "❌ 更新 $json_path 失败" >&2
    exit 1
  fi
}

# 6. 停止并移除指定名称的容器
remove_existing_container() {
  local container_name="$1"

  if [ -z "$container_name" ]; then
    echo "⚠️ 容器名称为空，跳过移除操作" >&2
    return
  fi

  # 检查容器是否存在
  if docker ps -a --format '{{.Names}}' | grep -q "^${container_name}$"; then
    echo "发现同名容器 $container_name，正在停止并移除..." >&2
    # 停止容器
    if ! docker stop "$container_name" >&2; then
      echo "⚠️ 停止容器 $container_name 失败，尝试强制移除..." >&2
    fi
    # 移除容器
    if docker rm "$container_name" >&2; then
      echo "✅ 已成功移除容器 $container_name" >&2
    else
      echo "❌ 移除容器 $container_name 失败，请手动处理" >&2
      exit 1
    fi
  else
    echo "未发现同名容器 $container_name，无需移除" >&2
  fi
}

### 主逻辑区 ###

# 检查 .env 文件
if [ ! -f ".env" ]; then
  echo "⚠️ 本地缺少 .env 文件，请创建并配置" >&2
  exit 1
fi

# 加载 .env 变量
export $(grep -v '^#' .env | xargs)

# 检查必要环境变量
if [ -z "$TEST_BRANCH" ] || [ -z "$DRYRUN_BRANCH" ]; then
  echo "⚠️ .env 文件缺少 TEST_BRANCH 或 DRYRUN_BRANCH 配置" >&2
  exit 1
fi

# 检查当前分支
current_branch=$(git rev-parse --abbrev-ref HEAD)
if [[ "$current_branch" != "$TEST_BRANCH" && "$current_branch" != "$DRYRUN_BRANCH" ]]; then
  echo "⚠️ 错误：当前分支 '$current_branch' 不符合配置要求" >&2
  exit 1
else
  echo "✅ 当前分支 '$current_branch' 符合配置要求" >&2
fi

# dryrun分支清理
if [[ "$current_branch" == *"dryrun"* ]]; then
  echo "当前为dryrun分支，执行git reset --hard清理工作区..." >&2
  git reset --hard || {
    echo "⚠️ git reset失败"
    exit 1
  }
  echo "✅ Git工作区已清理" >&2
fi

# 设置默认远程币对列表URL
DEFAULT_PAIR_REMOTE_URL="http://pairlist.xl.home/api/pairlist?mute=true&count=30"
PAIR_REMOTE_LIST_URL="$DEFAULT_PAIR_REMOTE_URL"

# 初始化策略配置（从 .env 加载）
STRATEGY_NAME=${STRATEGY_NAME:-TheForceV7}
CONFIG_FILE=${CONFIG_FILE:-basic.json}

# 解析命令行参数：支持策略名、配置文件、币对列表
while [[ $# -gt 0 ]]; do
  case "$1" in
  --strategy=*)
    STRATEGY_NAME="${1#*=}"
    shift
    ;;
  --strategy|-t)
    if [[ -n "$2" && "$2" != -* ]]; then
      STRATEGY_NAME="$2"
      shift 2
    else
      echo "错误：--strategy需要指定策略名" >&2
      exit 1
    fi
    ;;
  --config=*)
    CONFIG_FILE="${1#*=}"
    shift
    ;;
  --config|-c)
    if [[ -n "$2" && "$2" != -* ]]; then
      CONFIG_FILE="$2"
      shift 2
    else
      echo "错误：--config需要指定配置文件" >&2
      exit 1
    fi
    ;;
  --pairs=*)
    PAIRS_ARG="${1#*=}"
    shift
    ;;
  --pairs)
    if [[ -n "$2" && "$2" != -* ]]; then
      PAIRS_ARG="$2"
      shift 2
    else
      echo "错误：--pairs需要指定值" >&2
      exit 1
    fi
    ;;
  --pairRemoteList=*)
    PAIR_REMOTE_LIST_URL="${1#*=}"
    shift
    ;;
  --pairRemoteList)
    if [[ -n "$2" && "$2" != -* ]]; then
      PAIR_REMOTE_LIST_URL="$2"
      shift 2
    else
      echo "错误：--pairRemoteList需要指定值" >&2
      exit 1
    fi
    ;;
  *)
    shift
    ;;
  esac
done

# 处理策略名称，自动匹配配置文件
if [ -n "$STRATEGY_NAME" ]; then
  STRATEGY_CONFIG_LOWER="$(echo "$STRATEGY_NAME" | tr '[:upper:]' '[:lower:]').json"
  
  # 如果没有明确指定配置文件，先在 config_examples 中查找
  if [ -f "../config_examples/$STRATEGY_CONFIG_LOWER" ]; then
    CONFIG_FILE="$STRATEGY_CONFIG_LOWER"
    echo "✅ 自动匹配配置文件 (config_examples): $CONFIG_FILE" >&2
  elif [ -f "../freqtrade/templates/$STRATEGY_CONFIG_LOWER" ]; then
    # 其次在 templates 中查找（仅罢材材）
    CONFIG_FILE="$STRATEGY_CONFIG_LOWER"
    echo "✅ 自动匹配配置文件 (templates): $CONFIG_FILE" >&2
  fi
fi

PARAMS_NAME=$(echo "$STRATEGY_NAME" | tr '[:upper:]' '[:lower:]')

echo "✅ 使用策略: $STRATEGY_NAME" >&2
echo "✅ 使用配置: $CONFIG_FILE" >&2
echo "✅ dryrun分支: $DRYRUN_BRANCH" >&2
echo "✅ 远程币对列表URL: $PAIR_REMOTE_LIST_URL" >&2

### 核心：处理币对列表 ###

# 0. 判断是否使用 --pairs 传入的币对列表
if [ -n "$PAIRS_ARG" ]; then
  echo "✅ 使用命令行 --pairs 参数：$PAIRS_ARG" >&2
  # 更新配置文件中的 pair_whitelist
  if [ -f "../config_examples/$CONFIG_FILE" ]; then
    update_live_json_pair_whitelist "../config_examples/$CONFIG_FILE" "$PAIRS_ARG"
  else
    echo "⚠️ 找不到 ../config_examples/$CONFIG_FILE，跳过配置更新" >&2
  fi
else
  # 使用旧的逻辑：数据库 + 遥程 API
  echo "✅ 使用旧逻辑（数据库 + 遥程 API）" >&2
  
  # 1. 获取数据库币对（使用绝对路径）
  db_path="/home/ubuntu/freqtrade/user_data/tradesv3.sqlite"
  db_pairs=$(get_open_trades_pairs "$db_path")

  # 2. 获取遥程币对
  remote_pairs=$(get_remote_pairs "$PAIR_REMOTE_LIST_URL")

  # 3. 合并去重
  merged_pairs=$(merge_and_deduplicate_pairs "$db_pairs" "$remote_pairs")

  # 4. 更新配置文件
  if [ -f "../config_examples/$CONFIG_FILE" ]; then
    update_live_json_pair_whitelist "../config_examples/$CONFIG_FILE" "$merged_pairs"
  else
    echo "⚠️ 找不到 ../config_examples/$CONFIG_FILE，跳过配置更新" >&2
  fi
fi

### 启动容器 ###

GIT_COMMIT_SHORT=$(git rev-parse HEAD | cut -c 1-8)
CONTAINER_NAME="freqtrade-dryrun-${GIT_COMMIT_SHORT}"

echo "准备启动容器: $CONTAINER_NAME" >&2

# 移除已存在的同名容器
remove_existing_container "$CONTAINER_NAME"

# 启动新容器
docker-compose run -d --rm --name freqtrade-dryrun-${GIT_COMMIT_SHORT} -p 8080:8080 freqtrade trade \
  --logfile /freqtrade/user_data/logs/freqtrade.log \
  --db-url sqlite:////freqtrade/user_data/tradesv3.sqlite \
  --dry-run \
  --config /freqtrade/config_examples/$CONFIG_FILE \
  --strategy $STRATEGY_NAME \
  --fee 0.0008 \
  --strategy-path /freqtrade/templates

if [ $? -eq 0 ]; then
  echo "✅ 容器 $CONTAINER_NAME 启动完成" >&2
else
  echo "❌ 容器启动失败" >&2
  exit 1
fi