local_dev.sh

#!/bin/bash
# =============================================================================
# Local Development Script for Content Generation Accelerator
# =============================================================================
#
# This script sets up and runs the application locally for development.
#
# Usage:
#   ./local_dev.sh              # Start both backend and frontend
#   ./local_dev.sh backend      # Start only the backend
#   ./local_dev.sh frontend     # Start only the frontend
#   ./local_dev.sh setup        # Set up virtual environment and install dependencies
#   ./local_dev.sh env          # Generate .env file from Azure resources
#
# Prerequisites:
#   - Python 3.11+
#   - Node.js 18+
#   - Azure CLI (for fetching environment variables)
#   - Azure Developer CLI (azd) - optional, for env generation
#
# =============================================================================

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

# Get the script directory
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
SRC_DIR="$PROJECT_ROOT/src"
BACKEND_DIR="$SRC_DIR/backend"
FRONTEND_DIR="$SRC_DIR/App"

# Default ports
BACKEND_PORT=${BACKEND_PORT:-5000}
FRONTEND_PORT=${FRONTEND_PORT:-3000}

# =============================================================================
# Helper Functions
# =============================================================================

print_header() {
    echo -e "\n${BLUE}============================================${NC}"
    echo -e "${BLUE}$1${NC}"
    echo -e "${BLUE}============================================${NC}\n"
}

print_success() {
    echo -e "${GREEN}✓ $1${NC}"
}

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

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

print_info() {
    echo -e "${BLUE}→ $1${NC}"
}

check_command() {
    if ! command -v "$1" &> /dev/null; then
        print_error "$1 is not installed. Please install it first."
        return 1
    fi
    return 0
}

# =============================================================================
# Azure Authentication & Role Functions
# =============================================================================

ensure_azure_login() {
    if ! command -v az &> /dev/null; then
        print_error "Azure CLI is not installed. Please install it first."
        exit 1
    fi
    
    if az account show --query id -o tsv &> /dev/null; then
        local account_name
        account_name=$(az account show --query name -o tsv)
        print_success "Logged in to Azure: $account_name"
    else
        print_info "Not logged in. Running az login..."
        az login --use-device-code --output none
        if ! az account show --query id -o tsv &> /dev/null; then
            print_error "Azure login failed."
            exit 1
        fi
        print_success "Azure login successful."
    fi
}

ensure_azure_ai_user_role() {
    print_info "Checking Azure AI User role..."
    
    local existing_project_id=""
    local foundry_resource_id=""
    if [ -f ".env" ]; then
        existing_project_id=$(grep "^AZURE_EXISTING_AIPROJECT_RESOURCE_ID=" .env | cut -d'=' -f2- | tr -d '"' | tr -d "'" || echo "")
        foundry_resource_id=$(grep "^AI_FOUNDRY_RESOURCE_ID=" .env | cut -d'=' -f2- | tr -d '"' | tr -d "'" || echo "")
    fi
    
    local scope=""
    if [ -n "$existing_project_id" ]; then
        scope="$existing_project_id"
    elif [ -n "$foundry_resource_id" ]; then
        scope="$foundry_resource_id"
    else
        print_error "Neither AZURE_EXISTING_AIPROJECT_RESOURCE_ID nor AI_FOUNDRY_RESOURCE_ID found in .env"
        exit 1
    fi
    
    local signed_user_id
    signed_user_id=$(az ad signed-in-user show --query id -o tsv 2>/dev/null)
    if [ -z "$signed_user_id" ]; then
        print_error "Could not get signed-in user ID."
        exit 1
    fi
    
    local role_id="53ca6127-db72-4b80-b1b0-d745d6d5456d"
    local existing
    existing=$(MSYS_NO_PATHCONV=1 az role assignment list --assignee "$signed_user_id" --role "$role_id" --scope "$scope" --query "[0].id" -o tsv 2>/dev/null)
    
    if [ -n "$existing" ]; then
        print_success "Azure AI User role already assigned."
    else
        print_info "Assigning Azure AI User role..."
        if ! MSYS_NO_PATHCONV=1 az role assignment create --assignee "$signed_user_id" --role "$role_id" --scope "$scope" --output none 2>/dev/null; then
            print_error "Failed to assign Azure AI User role."
            exit 1
        fi
        print_success "Azure AI User role assigned."
    fi
}

ensure_cosmosdb_role() {
    print_info "Checking Cosmos DB Data Contributor role..."
    
    local cosmos_account=""
    local resource_group=""
    if [ -f ".env" ]; then
        cosmos_account=$(grep "^COSMOSDB_ACCOUNT_NAME=" .env | cut -d'=' -f2- | tr -d '"' | tr -d "'" || echo "")
        resource_group=$(grep "^RESOURCE_GROUP_NAME=" .env | cut -d'=' -f2- | tr -d '"' | tr -d "'" || echo "")
    fi
    
    if [ -z "$cosmos_account" ] || [ -z "$resource_group" ]; then
        print_error "COSMOSDB_ACCOUNT_NAME or RESOURCE_GROUP_NAME not found in .env"
        exit 1
    fi
    
    local signed_user_id
    signed_user_id=$(az ad signed-in-user show --query id -o tsv 2>/dev/null)
    if [ -z "$signed_user_id" ]; then
        print_error "Could not get signed-in user ID."
        exit 1
    fi
    
    local role_def_id="00000000-0000-0000-0000-000000000002"
    
    # Check if role already assigned
    local existing
    existing=$(az cosmosdb sql role assignment list --resource-group "$resource_group" --account-name "$cosmos_account" --query "[?principalId=='$signed_user_id' && contains(roleDefinitionId, '$role_def_id')].id | [0]" -o tsv 2>/dev/null)
    
    if [ -n "$existing" ]; then
        print_success "Cosmos DB role already assigned."
    else
        print_info "Assigning Cosmos DB role..."
        MSYS_NO_PATHCONV=1 az cosmosdb sql role assignment create --resource-group "$resource_group" --account-name "$cosmos_account" --role-definition-id "$role_def_id" --principal-id "$signed_user_id" --scope "/" --output none 2>/dev/null
        if [ $? -ne 0 ]; then
            print_error "Failed to assign Cosmos DB role."
            exit 1
        fi
        print_success "Cosmos DB role assigned."
    fi
}

ensure_storage_role() {
    print_info "Checking Storage Blob Data Contributor role..."
    
    local storage_account=""
    local resource_group=""
    if [ -f ".env" ]; then
        storage_account=$(grep "^AZURE_BLOB_ACCOUNT_NAME=" .env | cut -d'=' -f2- | tr -d '"' | tr -d "'" || echo "")
        resource_group=$(grep "^RESOURCE_GROUP_NAME=" .env | cut -d'=' -f2- | tr -d '"' | tr -d "'" || echo "")
    fi
    
    if [ -z "$storage_account" ] || [ -z "$resource_group" ]; then
        print_error "AZURE_BLOB_ACCOUNT_NAME or RESOURCE_GROUP_NAME not found in .env"
        exit 1
    fi
    
    local signed_user_id
    signed_user_id=$(az ad signed-in-user show --query id -o tsv 2>/dev/null)
    if [ -z "$signed_user_id" ]; then
        print_error "Could not get signed-in user ID."
        exit 1
    fi
    
    # Get storage account resource ID
    local storage_resource_id
    storage_resource_id=$(az storage account show --name "$storage_account" --resource-group "$resource_group" --query id -o tsv 2>/dev/null)
    if [ -z "$storage_resource_id" ]; then
        print_error "Could not get storage account resource ID."
        exit 1
    fi
    
    local role_name="Storage Blob Data Contributor"
    local existing
    existing=$(MSYS_NO_PATHCONV=1 az role assignment list --assignee "$signed_user_id" --role "$role_name" --scope "$storage_resource_id" --query "[0].id" -o tsv 2>/dev/null)
    
    if [ -n "$existing" ]; then
        print_success "Storage Blob Data Contributor role already assigned."
    else
        print_info "Assigning Storage Blob Data Contributor role..."
        if ! MSYS_NO_PATHCONV=1 az role assignment create --assignee "$signed_user_id" --role "$role_name" --scope "$storage_resource_id" --output none 2>/dev/null; then
            print_error "Failed to assign Storage Blob Data Contributor role."
            exit 1
        fi
        print_success "Storage Blob Data Contributor role assigned."
    fi
}

# =============================================================================
# Setup Function
# =============================================================================

setup() {
    print_header "Setting Up Local Development Environment"
    
    cd "$PROJECT_ROOT"
    
    # Check Python
    print_info "Checking Python installation..."
    if ! check_command python3; then
        exit 1
    fi
    python_version=$(python3 --version 2>&1 | cut -d' ' -f2)
    print_success "Python $python_version found"
    
    # Check Node.js
    print_info "Checking Node.js installation..."
    if ! check_command node; then
        exit 1
    fi
    node_version=$(node --version)
    print_success "Node.js $node_version found"
    
    # Create virtual environment
    print_info "Creating Python virtual environment..."
    if [ ! -d ".venv" ]; then
        python3 -m venv .venv
        print_success "Virtual environment created"
    else
        print_warning "Virtual environment already exists"
    fi
    
    # Activate virtual environment
    source .venv/bin/activate
    
    # Install Python dependencies
    print_info "Installing Python dependencies..."
    pip install --upgrade pip > /dev/null
    pip install -r "$BACKEND_DIR/requirements.txt" > /dev/null 2>&1
    print_success "Python dependencies installed"
    
    # Install frontend dependencies
    print_info "Installing frontend dependencies..."
    cd "$FRONTEND_DIR"
    npm install > /dev/null 2>&1
    print_success "Frontend dependencies installed"
    
    cd "$PROJECT_ROOT"
    
    # Check for .env file
    if [ ! -f ".env" ]; then
        print_warning ".env file not found"
        if [ -f ".env.sample" ]; then
            print_info "Copying .env.sample to .env..."
            cp .env.sample .env
            print_warning "Please update .env with your Azure resource values"
        fi
    else
        print_success ".env file found"
    fi
    
    print_header "Setup Complete!"
    echo -e "To start development, run: ${GREEN}./scripts/local_dev.sh${NC}"
    echo -e "Or start individually:"
    echo -e "  Backend:  ${GREEN}./scripts/local_dev.sh backend${NC}"
    echo -e "  Frontend: ${GREEN}./scripts/local_dev.sh frontend${NC}"
}

# =============================================================================
# Environment Generation Function
# =============================================================================

generate_env() {
    print_header "Generating Environment Variables from Azure"
    
    cd "$PROJECT_ROOT"
    
    ensure_azure_login
    
    # If using azd
    if command -v azd &> /dev/null && [ -f "azure.yaml" ]; then
        print_info "Azure Developer CLI detected. Generating .env from azd..."
        azd env get-values > .env.azd 2>/dev/null || true
        
        if [ -s ".env.azd" ]; then
            print_success "Environment variables exported to .env.azd"
            print_info "Merging with .env..."
            
            # Merge with existing .env
            if [ -f ".env" ]; then
                # Backup existing
                cp .env .env.backup
                # Append new values (avoiding duplicates)
                while IFS= read -r line; do
                    key=$(echo "$line" | cut -d'=' -f1)
                    if ! grep -q "^$key=" .env 2>/dev/null; then
                        echo "$line" >> .env
                    fi
                done < .env.azd
            else
                mv .env.azd .env
            fi
            rm -f .env.azd
            print_success "Environment variables merged"
        fi
    else
        print_warning "Azure Developer CLI not found or no azure.yaml"
        print_info "Please manually update .env with your Azure resource values"
        
        if [ ! -f ".env" ] && [ -f ".env.sample" ]; then
            cp .env.sample .env
            print_info "Created .env from .env.sample"
        fi
    fi
    
    print_success "Environment setup complete"
    print_warning "Review .env and ensure all required values are set"
}

# =============================================================================
# Backend Start Function
# =============================================================================

start_backend() {
    print_header "Starting Backend Server"
    
    cd "$PROJECT_ROOT"
    
    # Check for .env
    if [ ! -f ".env" ]; then
        print_error ".env file not found. Run: ./scripts/local_dev.sh setup"
        exit 1
    fi
    
    # Ensure Azure roles
    ensure_azure_login
    ensure_azure_ai_user_role
    ensure_cosmosdb_role
    ensure_storage_role
    
    # Activate virtual environment
    if [ -d ".venv" ]; then
        source .venv/bin/activate
        print_success "Virtual environment activated"
    else
        print_error "Virtual environment not found. Run: ./scripts/local_dev.sh setup"
        exit 1
    fi
    
    # Set environment variables
    export PYTHONPATH="$BACKEND_DIR"
    export DOTENV_PATH="$PROJECT_ROOT/.env"
    
    # Load .env file
    set -a
    source "$PROJECT_ROOT/.env"
    set +a
    
    print_info "Starting Quart backend on port $BACKEND_PORT..."
    print_info "API will be available at: http://localhost:$BACKEND_PORT"
    print_info "Health check: http://localhost:$BACKEND_PORT/api/health"
    echo ""
    
    cd "$BACKEND_DIR"
    
    # Use hypercorn for async support (same as production)
    if command -v hypercorn &> /dev/null; then
        hypercorn app:app --bind "0.0.0.0:$BACKEND_PORT" --reload
    else
        # Fallback to quart run
        python -m quart --app app:app run --host 0.0.0.0 --port "$BACKEND_PORT" --reload
    fi
}

# =============================================================================
# Frontend Start Function
# =============================================================================

start_frontend() {
    print_header "Starting Frontend Development Server"
    
    cd "$FRONTEND_DIR"
    
    # Check if node_modules exists
    if [ ! -d "node_modules" ]; then
        print_error "Node modules not found. Run: ./scripts/local_dev.sh setup"
        exit 1
    fi
    
    print_info "Starting Vite dev server on port $FRONTEND_PORT..."
    print_info "Frontend will be available at: http://localhost:$FRONTEND_PORT"
    print_info "API requests will proxy to: http://localhost:$BACKEND_PORT"
    echo ""
    
    npm run dev
}

# =============================================================================
# Start Both (using background processes)
# =============================================================================

start_all() {
    print_header "Starting Full Development Environment"
    
    cd "$PROJECT_ROOT"
    
    # Handle .env: generate if missing, or prompt to refresh if exists
    if [ ! -f ".env" ]; then
        print_warning ".env file not found. Generating from Azure resources..."
        generate_env
        # Fallback to .env.sample if generation didn't produce .env
        if [ ! -f ".env" ] && [ -f ".env.sample" ]; then
            print_warning "Env generation did not create .env. Copying .env.sample as fallback..."
            cp .env.sample .env
            print_info "Created .env from .env.sample. Update the file with your Azure resource values."
        fi
        if [ ! -f ".env" ]; then
            print_error "Failed to create .env. Run './scripts/local_dev.sh env' or create .env manually from .env.sample."
            exit 1
        fi
    else
        print_success "Found existing .env file"
        read -p "Do you want to overwrite it with fresh values from Azure deployment? (y/N): " overwrite
        if [ "$overwrite" = "y" ] || [ "$overwrite" = "Y" ]; then
            print_info "Regenerating .env from Azure resources..."
            rm -f .env
            generate_env
            print_success "Environment variables refreshed."
        else
            print_info "Preserving existing .env. Using local configuration."
        fi
    fi
    
    # Auto-run setup if prerequisites are missing
    local needs_setup=false
    if [ ! -d ".venv" ]; then
        print_warning "Virtual environment not found. Running setup..."
        needs_setup=true
    fi
    if [ ! -d "$FRONTEND_DIR/node_modules" ]; then
        print_warning "Frontend dependencies not found. Running setup..."
        needs_setup=true
    fi
    if [ "$needs_setup" = true ]; then
        setup
    fi
    
    # Ensure Azure authentication and role assignments
    ensure_azure_login
    ensure_azure_ai_user_role
    ensure_cosmosdb_role
    ensure_storage_role
    
    print_info "Starting backend and frontend in parallel..."
    print_info ""
    print_info "Services:"
    print_info "  Backend API:  http://localhost:$BACKEND_PORT"
    print_info "  Frontend:     http://localhost:$FRONTEND_PORT"
    print_info ""
    print_info "Press Ctrl+C to stop all services"
    echo ""
    
    # Trap Ctrl+C to kill all background jobs
    trap 'echo ""; print_info "Stopping all services..."; kill $(jobs -p) 2>/dev/null; exit 0' INT TERM
    
    # Start backend in background
    (
        cd "$PROJECT_ROOT"
        source .venv/bin/activate
        export PYTHONPATH="$BACKEND_DIR"
        export DOTENV_PATH="$PROJECT_ROOT/.env"
        set -a
        source "$PROJECT_ROOT/.env"
        set +a
        cd "$BACKEND_DIR"
        
        if command -v hypercorn &> /dev/null; then
            hypercorn app:app --bind "0.0.0.0:$BACKEND_PORT" --reload 2>&1 | sed 's/^/[Backend] /'
        else
            python -m quart --app app:app run --host 0.0.0.0 --port "$BACKEND_PORT" --reload 2>&1 | sed 's/^/[Backend] /'
        fi
    ) &
    
    # Give backend a moment to start
    sleep 2
    
    # Start frontend in background
    (
        cd "$FRONTEND_DIR"
        npm run dev 2>&1 | sed 's/^/[Frontend] /'
    ) &
    
    # Wait for all background jobs
    wait
}

# =============================================================================
# Build Function
# =============================================================================

build() {
    print_header "Building for Production"
    
    cd "$FRONTEND_DIR"
    
    print_info "Building frontend..."
    npm run build
    
    print_success "Build complete!"
    print_info "Static files are in: $SRC_DIR/static"
}

# =============================================================================
# Clean Function
# =============================================================================

clean() {
    print_header "Cleaning Development Environment"
    
    cd "$PROJECT_ROOT"
    
    print_info "Removing Python cache..."
    find . -type d -name "__pycache__" -exec rm -rf {} + 2>/dev/null || true
    find . -type f -name "*.pyc" -delete 2>/dev/null || true
    
    print_info "Removing node_modules..."
    rm -rf "$FRONTEND_DIR/node_modules" 2>/dev/null || true
    
    print_info "Removing build artifacts..."
    rm -rf "$SRC_DIR/static" 2>/dev/null || true
    
    print_success "Clean complete!"
}

# =============================================================================
# Main Script
# =============================================================================

case "${1:-}" in
    setup)
        setup
        ;;
    env)
        generate_env
        ;;
    backend)
        start_backend
        ;;
    frontend)
        start_frontend
        ;;
    build)
        build
        ;;
    clean)
        clean
        ;;
    ""|all)
        start_all
        ;;
    help|*)
        echo "Content Generation Accelerator - Local Development"
        echo ""
        echo "Usage: $0 [command]"
        echo ""
        echo "Commands:"
        echo "  setup     Set up virtual environment and install dependencies"
        echo "  env       Generate .env file from Azure resources"
        echo "  backend   Start only the backend server"
        echo "  frontend  Start only the frontend dev server"
        echo "  all       Start both backend and frontend (default)"
        echo "  build     Build frontend for production"
        echo "  clean     Remove cache and build artifacts"
        echo "  help      Show this help message"
        echo ""
        echo "Environment Variables:"
        echo "  BACKEND_PORT   Backend port (default: 5000)"
        echo "  FRONTEND_PORT  Frontend port (default: 3000)"
        ;;
esac