#!/bin/bash
  
  # One-click startup script for SearchEngine
  # This script starts everything you need
  

  # Don't exit on error - we want to continue even if some checks fail
  # set -e

  
  cd "$(dirname "$0")"
  
  GREEN='\033[0;32m'
  YELLOW='\033[1;33m'
  RED='\033[0;31m'
  NC='\033[0m'
  
  echo -e "${GREEN}========================================${NC}"
  echo -e "${GREEN}SearchEngine一键启动脚本${NC}"
  echo -e "${GREEN}========================================${NC}"
  
  # Step 1: Setup environment
  echo -e "\n${YELLOW}Step 1/4: 设置环境${NC}"
  ./setup.sh
  
  # Step 2: Check if data is already ingested
  echo -e "\n${YELLOW}Step 2/4: 检查数据${NC}"
  source /home/tw/miniconda3/etc/profile.d/conda.sh
  conda activate searchengine
  
  # Check if index exists
  INDEX_EXISTS=$(python -c "
  from config.env_config import get_es_config
  from utils.es_client import ESClient
  from config import ConfigLoader
  
  es_config = get_es_config()
  es_client = ESClient(hosts=[es_config['host']], username=es_config.get('username'), password=es_config.get('password'))
  
  config_loader = ConfigLoader('config/schema')
  config = config_loader.load_customer_config('customer1')
  
  if es_client.index_exists(config.es_index_name):
      doc_count = es_client.count(config.es_index_name)
      print(f'{doc_count}')
  else:
      print('0')
  " 2>/dev/null || echo "0")
  
  if [ "$INDEX_EXISTS" = "0" ]; then
      echo -e "${YELLOW}索引不存在，开始导入数据...${NC}"
      echo -e "${YELLOW}注意: 首次导入会下载模型文件，可能需要10-30分钟${NC}"
      echo -e "${YELLOW}导入1000条数据进行快速测试（跳过embedding以加快速度）${NC}"
      ./scripts/ingest.sh 1000 true
  else
      echo -e "${GREEN}数据已存在，包含 $INDEX_EXISTS 条文档${NC}"
  fi
  
  # Step 3: Start backend in background
  echo -e "\n${YELLOW}Step 3/4: 启动后端服务${NC}"
  echo -e "${YELLOW}后端服务将在后台运行...${NC}"
  nohup ./scripts/start_backend.sh > logs/backend.log 2>&1 &
  BACKEND_PID=$!
  echo $BACKEND_PID > logs/backend.pid
  echo -e "${GREEN}后端服务已启动 (PID: $BACKEND_PID)${NC}"
  echo -e "${GREEN}日志文件: logs/backend.log${NC}"
  
  # Wait for backend to start
  echo -e "${YELLOW}等待后端服务启动...${NC}"

  MAX_RETRIES=12
  RETRY_COUNT=0
  BACKEND_READY=false
  
  while [ $RETRY_COUNT -lt $MAX_RETRIES ]; do
      sleep 2
      if curl -s http://localhost:6002/ > /dev/null 2>&1; then
          BACKEND_READY=true
          break
      fi
      RETRY_COUNT=$((RETRY_COUNT + 1))
      echo -e "${YELLOW}  等待中... ($RETRY_COUNT/$MAX_RETRIES)${NC}"
  done

  
  # Check if backend is running

  if [ "$BACKEND_READY" = true ]; then

      echo -e "${GREEN}✓ 后端服务运行正常${NC}"

      # Try health check, but don't fail if it's not ready yet
      if curl -s http://localhost:6002/admin/health > /dev/null 2>&1; then
          echo -e "${GREEN}✓ 健康检查通过${NC}"
      else
          echo -e "${YELLOW}⚠ 健康检查未通过，但服务已启动${NC}"
      fi

  else
      echo -e "${RED}✗ 后端服务启动失败，请检查日志: logs/backend.log${NC}"

      echo -e "${YELLOW}提示: 后端服务可能需要更多时间启动，或者检查端口是否被占用${NC}"
      # Don't exit - let user decide whether to continue

  fi
  
  # Step 4: Start frontend
  echo -e "\n${YELLOW}Step 4/4: 启动前端服务${NC}"
  echo -e "${GREEN}========================================${NC}"
  echo -e "${GREEN}所有服务启动完成!${NC}"
  echo -e "${GREEN}========================================${NC}"
  echo ""
  echo -e "访问地址:"

  echo -e "  ${GREEN}前端界面: http://localhost:6003${NC}"
  echo -e "  ${GREEN}后端API: http://localhost:6002${NC}"
  echo -e "  ${GREEN}API文档: http://localhost:6002/docs${NC}"

  echo ""
  echo -e "日志文件:"
  echo -e "  后端: logs/backend.log"
  echo ""
  echo -e "停止服务:"
  echo -e "  后端: kill \$(cat logs/backend.pid)"
  echo -e "  前端: Ctrl+C"
  echo ""
  echo -e "${YELLOW}正在启动前端服务...${NC}"
  echo -e "${YELLOW}按 Ctrl+C 停止前端服务${NC}"
  echo ""
  
  ./scripts/start_frontend.sh