#!/bin/bash
#
# MyBeacon Backend Startup Script
#

set -e

# Change to backend directory
cd "$(dirname "$0")"

# Colors
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

# Default values
MODE="${1:-dev}"
HOST="${HOST:-0.0.0.0}"
PORT="${PORT:-8000}"
LOG_LEVEL="${LOG_LEVEL:-info}"

echo -e "${GREEN}=== MyBeacon Backend Startup ===${NC}"
echo -e "Mode: ${YELLOW}${MODE}${NC}"
echo -e "Host: ${YELLOW}${HOST}${NC}"
echo -e "Port: ${YELLOW}${PORT}${NC}"
echo ""

# Check if .env exists
if [ ! -f ".env" ]; then
    echo -e "${RED}Error: .env file not found!${NC}"
    echo "Please create .env file from .env.example"
    exit 1
fi

# Check if poetry is installed
if ! command -v poetry &> /dev/null; then
    echo -e "${RED}Error: poetry not found!${NC}"
    echo "Install poetry: curl -sSL https://install.python-poetry.org | python3 -"
    exit 1
fi

# Install dependencies if needed
if [ ! -d ".venv" ] && [ ! -d "$(poetry env info -p 2>/dev/null)" ]; then
    echo -e "${YELLOW}Installing dependencies...${NC}"
    poetry install
fi

# Function to stop backend
cleanup() {
    echo -e "\n${YELLOW}Stopping backend...${NC}"
    pkill -f "uvicorn app.main:app" 2>/dev/null || true
    exit 0
}

trap cleanup SIGINT SIGTERM

# Start backend based on mode
case "$MODE" in
    dev)
        echo -e "${GREEN}Starting in DEVELOPMENT mode...${NC}"
        echo -e "${YELLOW}Logs: /tmp/backend_dev.log${NC}"
        echo ""
        poetry run python3 -m uvicorn app.main:app \
            --host "$HOST" \
            --port "$PORT" \
            --reload \
            --log-level "$LOG_LEVEL" \
            2>&1 | tee /tmp/backend_dev.log
        ;;

    prod)
        echo -e "${GREEN}Starting in PRODUCTION mode...${NC}"
        echo -e "${YELLOW}Logs: /var/log/mybeacon/backend.log${NC}"
        echo ""

        # Create log directory
        sudo mkdir -p /var/log/mybeacon
        sudo chown $USER:$USER /var/log/mybeacon

        poetry run python3 -m uvicorn app.main:app \
            --host "$HOST" \
            --port "$PORT" \
            --workers 4 \
            --log-level "$LOG_LEVEL" \
            --access-log \
            2>&1 | tee -a /var/log/mybeacon/backend.log
        ;;

    background)
        echo -e "${GREEN}Starting in BACKGROUND mode...${NC}"
        echo -e "${YELLOW}Logs: /tmp/backend_live.log${NC}"
        echo ""

        # Kill existing backend
        pkill -f "uvicorn app.main:app" 2>/dev/null || true
        sleep 1

        # Start in background
        poetry run python3 -m uvicorn app.main:app \
            --host "$HOST" \
            --port "$PORT" \
            --log-level "$LOG_LEVEL" \
            2>&1 < /dev/null | tee /tmp/backend_live.log &

        PID=$!
        echo -e "${GREEN}Backend started with PID: ${PID}${NC}"
        echo -e "Check logs: ${YELLOW}tail -f /tmp/backend_live.log${NC}"
        echo -e "Stop: ${YELLOW}pkill -f 'uvicorn app.main:app'${NC}"
        ;;

    stop)
        echo -e "${YELLOW}Stopping backend...${NC}"
        pkill -f "uvicorn app.main:app" 2>/dev/null && echo -e "${GREEN}Backend stopped${NC}" || echo -e "${YELLOW}Backend not running${NC}"
        ;;

    restart)
        echo -e "${YELLOW}Restarting backend...${NC}"
        pkill -f "uvicorn app.main:app" 2>/dev/null || true
        sleep 2
        exec "$0" background
        ;;

    status)
        if pgrep -f "uvicorn app.main:app" > /dev/null; then
            echo -e "${GREEN}Backend is running${NC}"
            ps aux | grep "uvicorn app.main:app" | grep -v grep
        else
            echo -e "${YELLOW}Backend is not running${NC}"
        fi
        ;;

    *)
        echo "Usage: $0 {dev|prod|background|stop|restart|status}"
        echo ""
        echo "Modes:"
        echo "  dev        - Development mode with auto-reload"
        echo "  prod       - Production mode with 4 workers"
        echo "  background - Run in background"
        echo "  stop       - Stop backend"
        echo "  restart    - Restart backend"
        echo "  status     - Check backend status"
        echo ""
        echo "Environment variables:"
        echo "  HOST       - Bind host (default: 0.0.0.0)"
        echo "  PORT       - Bind port (default: 8000)"
        echo "  LOG_LEVEL  - Log level (default: info)"
        echo ""
        echo "Examples:"
        echo "  $0 dev                    # Development mode"
        echo "  $0 background             # Background mode"
        echo "  PORT=8001 $0 dev          # Custom port"
        exit 1
        ;;
esac