deploy.sh

#!/bin/bash
set -e

# Deep Research Agent - Main Deployment Orchestrator
# Routes to specific deployment scripts

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

# Functions
log_info() {
    echo -e "${GREEN}✓${NC} $1"
}

log_warn() {
    echo -e "${YELLOW}⚠${NC} $1"
}

log_error() {
    echo -e "${RED}✗${NC} $1"
}

log_step() {
    echo -e "${BLUE}▶${NC} $1"
}

# Display banner
display_banner() {
    echo "========================================"
    echo "  Deep Research Agent - Deployment"
    echo "========================================"
    echo ""
}

# Select AWS Region
select_region() {
    echo "Select AWS Region:"
    echo ""
    echo "  1) us-east-1      (US East - N. Virginia)"
    echo "  2) us-west-2      (US West - Oregon)"
    echo "  3) ap-northeast-1 (Asia Pacific - Tokyo)"
    echo "  4) ap-northeast-2 (Asia Pacific - Seoul)"
    echo "  5) ap-southeast-1 (Asia Pacific - Singapore)"
    echo "  6) eu-west-1      (Europe - Ireland)"
    echo "  7) eu-central-1   (Europe - Frankfurt)"
    echo "  8) Custom region"
    echo ""

    read -p "Select region (1-8) [default: 2]: " REGION_OPTION
    REGION_OPTION=${REGION_OPTION:-2}
    echo ""

    case $REGION_OPTION in
        1)
            AWS_REGION="us-east-1"
            ;;
        2)
            AWS_REGION="us-west-2"
            ;;
        3)
            AWS_REGION="ap-northeast-1"
            ;;
        4)
            AWS_REGION="ap-northeast-2"
            ;;
        5)
            AWS_REGION="ap-southeast-1"
            ;;
        6)
            AWS_REGION="eu-west-1"
            ;;
        7)
            AWS_REGION="eu-central-1"
            ;;
        8)
            read -p "Enter AWS region: " AWS_REGION
            if [ -z "$AWS_REGION" ]; then
                log_error "Region cannot be empty"
                exit 1
            fi
            ;;
        *)
            log_error "Invalid option. Using default region: us-west-2"
            AWS_REGION="us-west-2"
            ;;
    esac

    # Export region for deployment scripts
    export AWS_REGION
    export TF_VAR_aws_region="$AWS_REGION"

    log_info "Selected region: $AWS_REGION"
    echo ""
}

# Display menu
display_menu() {
    echo "What would you like to deploy?"
    echo ""
    echo "  1) Backend      (AgentCore Runtime + Infrastructure)"
    echo "  2) Frontend     (Cognito + Amplify/ECS + BFF)"
    echo "  3) Tools        (Gateway + Additional Services)"
    echo "  4) Full Stack   (Backend + Frontend)"
    echo "  5) Everything   (Backend + Frontend + Tools)"
    echo ""
    echo "  0) Exit"
    echo ""
}

# Check Docker
check_docker() {
    log_step "Checking Docker..."

    # Check if Docker is installed
    if ! command -v docker &> /dev/null; then
        log_error "Docker is not installed. Please install Docker first."
        echo "  Visit: https://docs.docker.com/get-docker/"
        exit 1
    fi

    # Check if Docker daemon is running
    if ! docker info &> /dev/null; then
        log_error "Docker daemon is not running. Please start Docker."
        echo "  On macOS: Open Docker Desktop"
        echo "  On Linux: sudo systemctl start docker"
        exit 1
    fi

    log_info "Docker is running"
    echo ""
}

# Check if scripts exist
check_scripts() {
    local missing=0

    if [ ! -f "./terraform/deploy-backend.sh" ]; then
        log_error "terraform/deploy-backend.sh not found"
        missing=1
    fi

    if [ ! -f "./terraform/deploy-frontend.sh" ]; then
        log_error "terraform/deploy-frontend.sh not found"
        missing=1
    fi

    if [ $missing -eq 1 ]; then
        log_error "Required deployment scripts are missing"
        exit 1
    fi

    # Make sure scripts are executable
    chmod +x ./terraform/deploy-backend.sh
    chmod +x ./terraform/deploy-frontend.sh
    chmod +x ./terraform/deploy-tools.sh
}

# Deploy backend
deploy_backend() {
    log_step "Deploying Backend..."
    echo ""
    ./terraform/deploy-backend.sh
}

# Deploy frontend
deploy_frontend() {
    log_step "Deploying Frontend..."
    echo ""
    ./terraform/deploy-frontend.sh
}

# Deploy tools
deploy_tools() {
    if [ -f "./terraform/deploy-tools.sh" ]; then
        log_step "Deploying Tools..."
        echo ""
        ./terraform/deploy-tools.sh
    else
        log_warn "terraform/deploy-tools.sh not yet implemented"
        echo ""
        echo "Coming soon: Gateway, Lambda Tools, etc."
    fi
}

# Main function
main() {
    display_banner
    check_docker
    check_scripts
    select_region
    display_menu

    read -p "Select option (0-5): " OPTION
    echo ""

    case $OPTION in
        1)
            echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
            echo "  Option 1: Backend Only"
            echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
            echo ""
            deploy_backend
            ;;
        2)
            echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
            echo "  Option 2: Frontend Only"
            echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
            echo ""
            deploy_frontend
            ;;
        3)
            echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
            echo "  Option 3: Tools Only"
            echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
            echo ""
            deploy_tools
            ;;
        4)
            echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
            echo "  Option 4: Full Stack (Backend + Frontend)"
            echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
            echo ""
            deploy_backend
            echo ""
            echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
            echo ""
            deploy_frontend
            ;;
        5)
            echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
            echo "  Option 5: Everything"
            echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
            echo ""
            deploy_backend
            echo ""
            echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
            echo ""
            deploy_frontend
            echo ""
            echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
            echo ""
            deploy_tools
            ;;
        0)
            log_info "Exiting..."
            exit 0
            ;;
        *)
            log_error "Invalid option. Please select 0-5."
            exit 1
            ;;
    esac

    echo ""
    echo "========================================"
    log_info "Deployment Complete!"
    echo "========================================"
}

# Run main
main