#!/bin/bash

set -e

# Source the utilities file
source "$(dirname "$0")/utils.sh"

# Set the compose command explicitly to use docker compose subcommand
COMPOSE_CMD="docker compose"
log_info "Using $COMPOSE_CMD as compose command for update application"

# Navigate to the directory where this script is located
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
# Project root directory (one level up from scripts)
PROJECT_ROOT="$( cd "$SCRIPT_DIR/.." &> /dev/null && pwd )"
# Path to the 05_run_services.sh script (Corrected from original update.sh which had 04)
RUN_SERVICES_SCRIPT="$SCRIPT_DIR/05_run_services.sh"
# Compose files (Not strictly needed here unless used directly, but good for context)
# MAIN_COMPOSE_FILE="$PROJECT_ROOT/docker-compose.yml"
# SUPABASE_COMPOSE_FILE="$PROJECT_ROOT/supabase/docker/docker-compose.yml"
ENV_FILE="$PROJECT_ROOT/.env"

# Check if run services script exists
if [ ! -f "$RUN_SERVICES_SCRIPT" ]; then
    log_error "$RUN_SERVICES_SCRIPT not found."
    exit 1
fi

cd "$PROJECT_ROOT"

# Stop all services
log_info "Stopping all services..."
$COMPOSE_CMD down || { 
  log_warning "Failed to stop containers with 'docker compose down'. Continuing with update anyway..."; 
}

# Pull latest versions of all containers
log_info "Pulling latest versions of all containers..."
$COMPOSE_CMD pull || { log_error "Failed to pull Docker images. Check network connection and Docker Hub status."; exit 1; }

# Ask user about n8n import and modify .env file
if [ -f "$ENV_FILE" ]; then
    N8N_WORKFLOWS_IMPORTED_EVER=$(grep "^N8N_WORKFLOWS_IMPORTED_EVER=" "$ENV_FILE" | cut -d'=' -f2 | tr -d '"' || echo "false")

    if [[ "$N8N_WORKFLOWS_IMPORTED_EVER" == "true" ]]; then
        log_info "n8n workflows have been imported previously. Skipping import prompt."
        # Use a temporary file for sed portability
        sed 's/^RUN_N8N_IMPORT=.*/RUN_N8N_IMPORT=false/' "$ENV_FILE" > "${ENV_FILE}.tmp" && mv "${ENV_FILE}.tmp" "$ENV_FILE" || {
            log_error "Failed to set RUN_N8N_IMPORT=false in $ENV_FILE. Check permissions."
            rm -f "${ENV_FILE}.tmp"
        }
    else
        echo ""
        read -p "Import n8n workflow? (y/n) : " import_choice
        case "$import_choice" in
            [yY] | [yY][eE][sS] )
                # Use a temporary file for sed portability
                sed 's/^RUN_N8N_IMPORT=.*/RUN_N8N_IMPORT=true/' "$ENV_FILE" > "${ENV_FILE}.tmp" && mv "${ENV_FILE}.tmp" "$ENV_FILE" || {
                    log_error "Failed to set RUN_N8N_IMPORT=true in $ENV_FILE. Check permissions."
                    rm -f "${ENV_FILE}.tmp"
                }
                # Update N8N_WORKFLOWS_IMPORTED_EVER to true
                if grep -q "^N8N_WORKFLOWS_IMPORTED_EVER=" "$ENV_FILE"; then
                    sed 's/^N8N_WORKFLOWS_IMPORTED_EVER=.*/N8N_WORKFLOWS_IMPORTED_EVER=true/' "$ENV_FILE" > "${ENV_FILE}.tmp" && mv "${ENV_FILE}.tmp" "$ENV_FILE" || {
                        log_error "Failed to set N8N_WORKFLOWS_IMPORTED_EVER=true in $ENV_FILE. Check permissions."
                        rm -f "${ENV_FILE}.tmp"
                    }
                else
                    echo "N8N_WORKFLOWS_IMPORTED_EVER=true" >> "$ENV_FILE"
                fi
                ;;
            * )
                # Use a temporary file for sed portability
                sed 's/^RUN_N8N_IMPORT=.*/RUN_N8N_IMPORT=false/' "$ENV_FILE" > "${ENV_FILE}.tmp" && mv "${ENV_FILE}.tmp" "$ENV_FILE" || {
                    log_error "Failed to set RUN_N8N_IMPORT=false in $ENV_FILE. Check permissions."
                    rm -f "${ENV_FILE}.tmp"
                }
                # N8N_WORKFLOWS_IMPORTED_EVER remains false (or its current state if already in file)
                ;;
        esac
    fi

    # Ask user about n8n worker count
    if grep -q "^N8N_WORKER_COUNT=" "$ENV_FILE"; then
        CURRENT_WORKER_COUNT=$(grep "^N8N_WORKER_COUNT=" "$ENV_FILE" | cut -d'=' -f2 | tr -d '"')
        echo ""
        read -p "Enter new n8n worker count (leave empty to keep current: $CURRENT_WORKER_COUNT): " new_worker_count_raw

        if [[ -n "$new_worker_count_raw" ]]; then
            # Validate input: must be a positive integer
            if [[ "$new_worker_count_raw" =~ ^[1-9][0-9]*$ ]]; then
                NEW_WORKER_COUNT="$new_worker_count_raw"
                log_info "Updating n8n worker count to $NEW_WORKER_COUNT in $ENV_FILE..."
                # Use a temporary file for sed portability (-i needs backup suffix on macOS without -e)
                sed "s/^N8N_WORKER_COUNT=.*/N8N_WORKER_COUNT=\"$NEW_WORKER_COUNT\"/" "$ENV_FILE" > "${ENV_FILE}.tmp" && mv "${ENV_FILE}.tmp" "$ENV_FILE" || {
                    log_error "Failed to update N8N_WORKER_COUNT in $ENV_FILE. Check permissions."
                    rm -f "${ENV_FILE}.tmp" # Clean up temp file on failure
                }
            else
                log_warning "Invalid input '$new_worker_count_raw'. Worker count must be a positive integer. Keeping current value ($CURRENT_WORKER_COUNT)."
            fi
        else
            log_info "Keeping current n8n worker count ($CURRENT_WORKER_COUNT)."
        fi
    else
        # This case might occur if .env exists but N8N_WORKER_COUNT was manually removed.
        # 03_generate_secrets.sh should ensure it exists on initial setup.
        log_warning "N8N_WORKER_COUNT line not found in $ENV_FILE. Cannot update worker count during this update."
        # Optionally, prompt user to add it if needed:
        # echo ""
        # read -p "N8N_WORKER_COUNT line not found. Add it now? (Enter number, or leave empty to skip): " add_worker_count
        # if [[ "$add_worker_count" =~ ^[1-9][0-9]*$ ]]; then
        #     echo "N8N_WORKER_COUNT="$add_worker_count"" >> "$ENV_FILE"
        #     log_info "Added N8N_WORKER_COUNT=$add_worker_count to $ENV_FILE."
    fi
else
    log_warning "$ENV_FILE not found. Cannot configure RUN_N8N_IMPORT or N8N_WORKER_COUNT."
fi

# Start services using the 05_run_services.sh script
log_info "Running Services..."
bash "$RUN_SERVICES_SCRIPT" || { log_error "Failed to start services. Check logs for details."; exit 1; }

log_success "Update application completed successfully!"

exit 0