#!/bin/bash

# Auto Stock Trading System - 전체 종료 스크립트

echo "================================================"
echo "  Auto Stock Trading System Stopping..."
echo "================================================"

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

# 프로젝트 루트 디렉토리
PROJECT_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$PROJECT_ROOT"

echo ""
echo "🛑 Stopping all services..."
echo ""

# PID 파일에서 프로세스 종료
stop_service() {
    local service_name=$1
    local pid_file="$PROJECT_ROOT/logs/${service_name}.pid"
    
    if [ -f "$pid_file" ]; then
        local pid=$(cat "$pid_file")
        if ps -p $pid > /dev/null 2>&1; then
            echo "  └─ Stopping $service_name (PID: $pid)..."
            kill $pid
            sleep 2
            
            # 강제 종료가 필요한 경우
            if ps -p $pid > /dev/null 2>&1; then
                echo "     ${YELLOW}Force killing $service_name...${NC}"
                kill -9 $pid
            fi
            
            echo "     ${GREEN}✓ $service_name stopped${NC}"
        else
            echo "  └─ $service_name is not running"
        fi
        rm -f "$pid_file"
    else
        echo "  └─ No PID file found for $service_name"
    fi
}

# 각 서비스 종료
stop_service "backend"
stop_service "mcp-server"
stop_service "frontend"

echo ""

# 포트로 실행 중인 프로세스 강제 종료
echo "🔍 Checking for remaining processes on ports..."
echo ""

# Backend (8000)
if lsof -ti:8000 > /dev/null 2>&1; then
    echo "  └─ Killing process on port 8000..."
    kill -9 $(lsof -ti:8000) 2>/dev/null
    echo "     ${GREEN}✓ Port 8000 cleared${NC}"
fi

# MCP Server (8001)
if lsof -ti:8001 > /dev/null 2>&1; then
    echo "  └─ Killing process on port 8001..."
    kill -9 $(lsof -ti:8001) 2>/dev/null
    echo "     ${GREEN}✓ Port 8001 cleared${NC}"
fi

# Frontend (3000)
if lsof -ti:3000 > /dev/null 2>&1; then
    echo "  └─ Killing process on port 3000..."
    kill -9 $(lsof -ti:3000) 2>/dev/null
    echo "     ${GREEN}✓ Port 3000 cleared${NC}"
fi

# Python 프로세스 정리
echo ""
echo "🧹 Cleaning up Python processes..."
pkill -f "uvicorn.*8000" 2>/dev/null && echo "  ${GREEN}✓ Backend processes cleaned${NC}"
pkill -f "main.py" 2>/dev/null && echo "  ${GREEN}✓ MCP Server processes cleaned${NC}"

# Node 프로세스 정리
echo ""
echo "🧹 Cleaning up Node processes..."
pkill -f "vite.*3000" 2>/dev/null && echo "  ${GREEN}✓ Frontend processes cleaned${NC}"

echo ""
echo "================================================"
echo "  ${GREEN}All Services Stopped Successfully!${NC}"
echo "================================================"
echo ""
echo "🚀 To start all services again, run: ./start-all.sh"
echo ""