#!/bin/bash

set -e

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

echo -e "${BLUE}=== SeaweedFS PostgreSQL Test Setup ===${NC}"

# Function to wait for service
wait_for_service() {
    local service=$1
    local max_wait=$2
    local count=0
    
    echo -e "${YELLOW}Waiting for $service to be ready...${NC}"
    while [ $count -lt $max_wait ]; do
        if docker-compose ps $service | grep -q "healthy\|Up"; then
            echo -e "${GREEN}✓ $service is ready${NC}"
            return 0
        fi
        sleep 2
        count=$((count + 1))
        echo -n "."
    done
    
    echo -e "${RED}✗ Timeout waiting for $service${NC}"
    return 1
}

# Function to show logs
show_logs() {
    local service=$1
    echo -e "${BLUE}=== $service logs ===${NC}"
    docker-compose logs --tail=20 $service
    echo
}

# Parse command line arguments
case "$1" in
    "start")
        echo -e "${YELLOW}Starting SeaweedFS cluster and PostgreSQL server...${NC}"
        docker-compose up -d seaweedfs postgres-server
        
        wait_for_service "seaweedfs" 30
        wait_for_service "postgres-server" 15
        
        echo -e "${GREEN}✓ SeaweedFS and PostgreSQL server are running${NC}"
        echo
        echo "You can now:"
        echo "  • Run data producer: $0 produce"
        echo "  • Run test client: $0 test"
        echo "  • Connect with psql: $0 psql"
        echo "  • View logs: $0 logs [service]"
        echo "  • Stop services: $0 stop"
        ;;
        
    "produce")
        echo -e "${YELLOW}Creating MQ test data...${NC}"
        docker-compose up --build mq-producer
        
        if [ $? -eq 0 ]; then
            echo -e "${GREEN}✓ Test data created successfully${NC}"
            echo
            echo "You can now run: $0 test"
        else
            echo -e "${RED}✗ Data production failed${NC}"
            show_logs "mq-producer"
        fi
        ;;
        
    "test")
        echo -e "${YELLOW}Running PostgreSQL client tests...${NC}"
        docker-compose up --build postgres-client
        
        if [ $? -eq 0 ]; then
            echo -e "${GREEN}✓ Client tests completed${NC}"
        else
            echo -e "${RED}✗ Client tests failed${NC}"
            show_logs "postgres-client"
        fi
        ;;
        
    "psql")
        echo -e "${YELLOW}Connecting to PostgreSQL with psql...${NC}"
        docker-compose run --rm psql-cli psql -h postgres-server -p 5432 -U seaweedfs -d default
        ;;
        
    "logs")
        service=${2:-"seaweedfs"}
        show_logs "$service"
        ;;
        
    "status")
        echo -e "${BLUE}=== Service Status ===${NC}"
        docker-compose ps
        ;;
        
    "stop")
        echo -e "${YELLOW}Stopping all services...${NC}"
        docker-compose down
        echo -e "${GREEN}✓ All services stopped${NC}"
        ;;
        
    "clean")
        echo -e "${YELLOW}Cleaning up everything (including data)...${NC}"
        docker-compose down -v
        docker system prune -f
        echo -e "${GREEN}✓ Cleanup completed${NC}"
        ;;
        
    "all")
        echo -e "${YELLOW}Running complete test suite...${NC}"
        
        # Start services (wait_for_service ensures they're ready)
        $0 start
        
        # Create data (docker-compose up is synchronous)
        $0 produce
        
        # Run tests
        $0 test
        
        echo -e "${GREEN}✓ Complete test suite finished${NC}"
        ;;
        
    *)
        echo "Usage: $0 {start|produce|test|psql|logs|status|stop|clean|all}"
        echo
        echo "Commands:"
        echo "  start     - Start SeaweedFS and PostgreSQL server"
        echo "  produce   - Create MQ test data (run after start)"
        echo "  test      - Run PostgreSQL client tests (run after produce)"
        echo "  psql      - Connect with psql CLI"
        echo "  logs      - Show service logs (optionally specify service name)"
        echo "  status    - Show service status"
        echo "  stop      - Stop all services"
        echo "  clean     - Stop and remove all data"
        echo "  all       - Run complete test suite (start -> produce -> test)"
        echo
        echo "Example workflow:"
        echo "  $0 all                # Complete automated test"
        echo "  $0 start              # Manual step-by-step"
        echo "  $0 produce"
        echo "  $0 test"
        echo "  $0 psql               # Interactive testing"
        exit 1
        ;;
esac