docker-build.sh

#!/bin/bash
# ============================================================
# Docker Build Script for City Service Application
# Usage: ./docker-build.sh [environment] [action]
# ============================================================

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

# Default values
ENVIRONMENT=${1:-dev}
ACTION=${2:-build}
PROJECT_NAME="city-service"

# Function to print colored output
print_status() {
    echo -e "${BLUE}[INFO]${NC} $1"
}

print_success() {
    echo -e "${GREEN}[SUCCESS]${NC} $1"
}

print_warning() {
    echo -e "${YELLOW}[WARNING]${NC} $1"
}

print_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

# Function to check prerequisites
check_prerequisites() {
    print_status "Checking prerequisites..."
    
    if ! command -v docker &> /dev/null; then
        print_error "Docker is not installed"
        exit 1
    fi
    
    if ! command -v docker-compose &> /dev/null; then
        print_error "Docker Compose is not installed"
        exit 1
    fi
    
    if [ ! -f ".env" ]; then
        print_warning ".env file not found, copying from env.example"
        cp env.example .env
        print_warning "Please update .env file with your configuration"
    fi
    
    print_success "Prerequisites check passed"
}

# Function to build images
build_images() {
    print_status "Building Docker images for $ENVIRONMENT environment..."
    
    case $ENVIRONMENT in
        "dev"|"development")
            docker-compose build --parallel
            ;;
        "prod"|"production")
            docker-compose -f docker-compose.yml -f docker-compose.prod.yml build --parallel
            ;;
        *)
            print_error "Unknown environment: $ENVIRONMENT"
            print_status "Available environments: dev, prod"
            exit 1
            ;;
    esac
    
    print_success "Images built successfully"
}

# Function to start services
start_services() {
    print_status "Starting services for $ENVIRONMENT environment..."
    
    case $ENVIRONMENT in
        "dev"|"development")
            docker-compose up -d
            ;;
        "prod"|"production")
            docker-compose -f docker-compose.yml -f docker-compose.prod.yml up -d
            ;;
    esac
    
    print_success "Services started successfully"
    print_status "Waiting for services to be ready..."
    sleep 30
    
    # Check service health
    check_health
}

# Function to stop services
stop_services() {
    print_status "Stopping services..."
    
    case $ENVIRONMENT in
        "dev"|"development")
            docker-compose down
            ;;
        "prod"|"production")
            docker-compose -f docker-compose.yml -f docker-compose.prod.yml down
            ;;
    esac
    
    print_success "Services stopped successfully"
}

# Function to check service health
check_health() {
    print_status "Checking service health..."
    
    # Check backend health
    if curl -f http://localhost:8080/api/health > /dev/null 2>&1; then
        print_success "Backend is healthy"
    else
        print_warning "Backend health check failed"
    fi
    
    # Check frontend health
    if curl -f http://localhost:3000 > /dev/null 2>&1; then
        print_success "Frontend is healthy"
    else
        print_warning "Frontend health check failed"
    fi
    
    # Show running containers
    print_status "Running containers:"
    docker-compose ps
}

# Function to show logs
show_logs() {
    print_status "Showing logs for $ENVIRONMENT environment..."
    
    case $ENVIRONMENT in
        "dev"|"development")
            docker-compose logs -f
            ;;
        "prod"|"production")
            docker-compose -f docker-compose.yml -f docker-compose.prod.yml logs -f
            ;;
    esac
}

# Function to clean up
cleanup() {
    print_status "Cleaning up Docker resources..."
    
    # Stop and remove containers
    docker-compose down --remove-orphans
    
    # Remove unused images
    docker image prune -f
    
    # Remove unused volumes (be careful with this in production)
    if [ "$ENVIRONMENT" = "dev" ]; then
        docker volume prune -f
    fi
    
    print_success "Cleanup completed"
}

# Function to show usage
show_usage() {
    echo "Usage: $0 [environment] [action]"
    echo ""
    echo "Environments:"
    echo "  dev, development  - Development environment"
    echo "  prod, production  - Production environment"
    echo ""
    echo "Actions:"
    echo "  build            - Build Docker images"
    echo "  start            - Start services"
    echo "  stop             - Stop services"
    echo "  restart          - Restart services"
    echo "  logs             - Show logs"
    echo "  health           - Check service health"
    echo "  cleanup          - Clean up Docker resources"
    echo "  help             - Show this help message"
    echo ""
    echo "Examples:"
    echo "  $0 dev build      # Build images for development"
    echo "  $0 prod start     # Start production services"
    echo "  $0 dev logs       # Show development logs"
}

# Main script logic
main() {
    print_status "City Service Docker Build Script"
    print_status "Environment: $ENVIRONMENT"
    print_status "Action: $ACTION"
    echo ""
    
    case $ACTION in
        "build")
            check_prerequisites
            build_images
            ;;
        "start")
            check_prerequisites
            build_images
            start_services
            ;;
        "stop")
            stop_services
            ;;
        "restart")
            stop_services
            sleep 5
            check_prerequisites
            build_images
            start_services
            ;;
        "logs")
            show_logs
            ;;
        "health")
            check_health
            ;;
        "cleanup")
            cleanup
            ;;
        "help"|"--help"|"-h")
            show_usage
            ;;
        *)
            print_error "Unknown action: $ACTION"
            show_usage
            exit 1
            ;;
    esac
}

# Run main function
main "$@"