deploy.sh

#!/bin/bash

# Vidya Chat Agent Deployment Script
set -e

echo "🚀 Starting Vidya Chat Agent Deployment..."

# Check if GEMINI_API_KEY is set
if [ -z "$GEMINI_API_KEY" ]; then
    echo "❌ Error: GEMINI_API_KEY environment variable is not set"
    echo "Please set your Gemini API key: export GEMINI_API_KEY='your_api_key_here'"
    exit 1
fi

# Check if Docker is running
if ! docker info > /dev/null 2>&1; then
    echo "❌ Error: Docker is not running"
    echo "Please start Docker and try again"
    exit 1
fi

# Check if docker-compose is available
if ! command -v docker-compose &> /dev/null; then
    echo "❌ Error: docker-compose is not installed"
    echo "Please install docker-compose and try again"
    exit 1
fi

# Function to deploy development environment
deploy_dev() {
    echo "🔧 Deploying Development Environment..."
    
    # Build and start services
    docker-compose up -d --build
    
    # Wait for services to be healthy
    echo "⏳ Waiting for services to be ready..."
    sleep 30
    
    # Run database migrations
    echo "🗄️ Running database migrations..."
    docker-compose exec backend uv run alembic upgrade head
    
    echo "✅ Development environment deployed successfully!"
    echo "🌐 API available at: http://localhost:8000"
    echo "📚 API docs at: http://localhost:8000/docs"
    echo "🗄️ pgAdmin at: http://localhost:5050 (admin@vidya.com / admin123)"
}

# Function to deploy production environment
deploy_prod() {
    echo "🏭 Deploying Production Environment..."
    
    # Check for required environment variables
    if [ -z "$POSTGRES_PASSWORD" ]; then
        echo "⚠️ Warning: POSTGRES_PASSWORD not set, using default"
        export POSTGRES_PASSWORD="secure_password_here"
    fi
    
    # Build and start services
    docker-compose -f docker-compose.prod.yml up -d --build
    
    # Wait for services to be healthy
    echo "⏳ Waiting for services to be ready..."
    sleep 30
    
    # Run database migrations
    echo "🗄️ Running database migrations..."
    docker-compose -f docker-compose.prod.yml exec backend uv run alembic upgrade head
    
    echo "✅ Production environment deployed successfully!"
    echo "🌐 API available at: http://localhost"
    echo "📚 API docs at: http://localhost/docs"
}

# Function to stop services
stop_services() {
    echo "🛑 Stopping services..."
    
    if [ "$1" = "prod" ]; then
        docker-compose -f docker-compose.prod.yml down
    else
        docker-compose down
    fi
    
    echo "✅ Services stopped"
}

# Function to show logs
show_logs() {
    if [ "$1" = "prod" ]; then
        docker-compose -f docker-compose.prod.yml logs -f
    else
        docker-compose logs -f
    fi
}

# Function to show status
show_status() {
    if [ "$1" = "prod" ]; then
        docker-compose -f docker-compose.prod.yml ps
    else
        docker-compose ps
    fi
}

# Main script logic
case "${1:-dev}" in
    "dev")
        deploy_dev
        ;;
    "prod")
        deploy_prod
        ;;
    "stop")
        stop_services "${2:-dev}"
        ;;
    "logs")
        show_logs "${2:-dev}"
        ;;
    "status")
        show_status "${2:-dev}"
        ;;
    "help"|"-h"|"--help")
        echo "Vidya Chat Agent Deployment Script"
        echo ""
        echo "Usage: $0 [COMMAND] [ENVIRONMENT]"
        echo ""
        echo "Commands:"
        echo "  dev     Deploy development environment (default)"
        echo "  prod    Deploy production environment"
        echo "  stop    Stop services"
        echo "  logs    Show service logs"
        echo "  status  Show service status"
        echo "  help    Show this help message"
        echo ""
        echo "Environments:"
        echo "  dev     Development (default)"
        echo "  prod    Production"
        echo ""
        echo "Examples:"
        echo "  $0 dev          # Deploy development environment"
        echo "  $0 prod         # Deploy production environment"
        echo "  $0 stop dev     # Stop development services"
        echo "  $0 logs prod    # Show production logs"
        echo ""
        echo "Required Environment Variables:"
        echo "  GEMINI_API_KEY  Your Google Gemini API key"
        echo ""
        echo "Optional Environment Variables:"
        echo "  POSTGRES_PASSWORD  Database password (default: secure_password_here)"
        ;;
    *)
        echo "❌ Unknown command: $1"
        echo "Use '$0 help' for usage information"
        exit 1
        ;;
esac