#!/bin/bash

# 배포 스크립트
# Auto Stock Trading System Docker 배포

set -e

echo "====================================="
echo "Auto Stock Trading System 배포 시작"
echo "====================================="

# 색상 정의
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
RED='\033[0;31m'
NC='\033[0m' # No Color

# 환경 변수 파일 확인
if [ ! -f .env ]; then
    echo -e "${RED}오류: .env 파일이 없습니다.${NC}"
    echo "먼저 .env.example을 복사하여 .env 파일을 생성하고 설정을 완료해주세요."
    echo "cp .env.example .env"
    exit 1
fi

echo -e "${YELLOW}환경 변수 로드 중...${NC}"
source .env

# 필수 환경 변수 확인 (KIS_MODE 기반)
KIS_MODE="${KIS_MODE:-virtual}"
if [ "$KIS_MODE" = "real" ]; then
    REQUIRED_VARS=("KIS_REAL_APP_KEY" "KIS_REAL_APP_SECRET" "KIS_REAL_ACCOUNT_NO")
else
    REQUIRED_VARS=("KIS_VIRTUAL_APP_KEY" "KIS_VIRTUAL_APP_SECRET" "KIS_VIRTUAL_ACCOUNT_NO")
fi

for var in "${REQUIRED_VARS[@]}"; do
    if [ -z "${!var}" ]; then
        echo -e "${RED}오류: $var 환경 변수가 설정되지 않았습니다.${NC}"
        exit 1
    fi
done

echo -e "${GREEN}✓ 환경 변수 확인 완료${NC}"

# Docker 및 Docker Compose 확인
if ! command -v docker &> /dev/null; then
    echo -e "${RED}오류: Docker가 설치되어 있지 않습니다.${NC}"
    exit 1
fi

if ! command -v docker-compose &> /dev/null && ! docker compose version &> /dev/null; then
    echo -e "${RED}오류: Docker Compose가 설치되어 있지 않습니다.${NC}"
    exit 1
fi

echo -e "${GREEN}✓ Docker 확인 완료${NC}"

# 기존 컨테이너 중지 및 제거
echo -e "${YELLOW}기존 컨테이너 중지 중...${NC}"
docker-compose down 2>/dev/null || docker compose down 2>/dev/null || true

# Docker 이미지 빌드
echo -e "${YELLOW}Docker 이미지 빌드 중...${NC}"
if docker compose version &> /dev/null; then
    docker compose build
else
    docker-compose build
fi

echo -e "${GREEN}✓ 이미지 빌드 완료${NC}"

# 컨테이너 시작
echo -e "${YELLOW}컨테이너 시작 중...${NC}"
if docker compose version &> /dev/null; then
    docker compose up -d
else
    docker-compose up -d
fi

# 서비스 시작 대기
echo -e "${YELLOW}서비스 시작 대기 중...${NC}"
sleep 10

# 서비스 상태 확인
echo ""
echo "====================================="
echo "서비스 상태 확인"
echo "====================================="

services=("postgres" "redis" "rabbitmq" "mcp-server" "backend" "frontend")
all_healthy=true

for service in "${services[@]}"; do
    container_name="autostock-$service"
    if docker ps | grep -q "$container_name"; then
        echo -e "${GREEN}✓ $service: 실행 중${NC}"
    else
        echo -e "${RED}✗ $service: 실행 실패${NC}"
        all_healthy=false
    fi
done

echo ""
echo "====================================="
echo "서비스 URL"
echo "====================================="
echo -e "Frontend:  ${GREEN}http://localhost:3000${NC}"
echo -e "Backend:   ${GREEN}http://localhost:8000${NC}"
echo -e "MCP Server:${GREEN}http://localhost:8001${NC}"
echo -e "RabbitMQ:  ${GREEN}http://localhost:15672${NC} (guest/guest)"
echo ""

if [ "$all_healthy" = true ]; then
    echo -e "${GREEN}✓ 배포 완료!${NC}"
    echo ""
    echo "로그 확인: ./scripts/logs.sh"
    echo "중지: ./scripts/stop.sh"
    exit 0
else
    echo -e "${RED}✗ 일부 서비스 시작 실패${NC}"
    echo "로그 확인: docker-compose logs"
    exit 1
fi