#!/bin/bash

set -e

# Colors for output
GREEN='\033[0;32m'
BLUE='\033[0;34m'
YELLOW='\033[1;33m'
NC='\033[0m'

# ASCII Art
echo -e "${BLUE}"
cat << "EOF"
  ____              ______
 / __ \____  ____  / ____/___  _________ ___
/ / / / __ \/ __ \/ /_  / __ \/ ___/ __ `__ \
/ /_/ / /_/ / / / / __/ / /_/ / /  / / / / / /
\____/ .___/_/ /_/_/    \____/_/  /_/ /_/ /_/
    /_/
EOF
echo -e "${NC}"

# Default values
DEV_MODE=false
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_ROOT="$(dirname "$SCRIPT_DIR")"

# Parse command line arguments
while [[ "$#" -gt 0 ]]; do
    case $1 in
        --dev) DEV_MODE=true ;;
        *) echo "Unknown parameter: $1"; exit 1 ;;
    esac
    shift
done

cd "$PROJECT_ROOT"

is_truthy_env_value() {
    local value="$1"

    value="${value%%#*}"
    value="$(printf '%s' "$value" | sed -e 's/^[[:space:]]*//' -e 's/[[:space:]]*$//')"
    value="${value#\"}"
    value="${value%\"}"
    value="${value#\'}"
    value="${value%\'}"
    value="${value#(}"
    value="${value%)}"
    value="$(printf '%s' "$value" | tr '[:upper:]' '[:lower:]')"

    case "$value" in
        true|1|yes|on) return 0 ;;
        *) return 1 ;;
    esac
}

jwt_skip_validation_disabled_in_env() {
    local env_file="$1"
    local line value

    if [ ! -f "$env_file" ]; then
        return 1
    fi

    line="$(grep -E '^[[:space:]]*JWT_SKIP_IP_UA_VALIDATION[[:space:]]*=' "$env_file" | tail -n 1 || true)"
    if [ -z "$line" ]; then
        return 1
    fi

    value="${line#*=}"
    is_truthy_env_value "$value"
}

echo -e "${BLUE}Starting OpnForm Docker setup...${NC}"

# Run the environment setup script with --docker flag (only for production)
if [ "$DEV_MODE" = false ]; then
    echo -e "${GREEN}Setting up environment files...${NC}"
    bash "$SCRIPT_DIR/setup-env.sh" --docker

    if jwt_skip_validation_disabled_in_env "$PROJECT_ROOT/api/.env"; then
        echo -e "${YELLOW}Warning: JWT User Agent validation is disabled in api/.env.${NC}"
        echo -e "${YELLOW}Set JWT_SKIP_IP_UA_VALIDATION=false for production Docker deployments.${NC}"
        exit 1
    fi
else
    echo -e "${GREEN}Development mode - skipping .env generation (using docker-compose environment variables)${NC}"
fi

# Detect compose binary: prefer Docker Compose v2 plugin ("docker compose"), fallback to "docker-compose"
declare -a COMPOSE_CMD
if command -v docker >/dev/null 2>&1 && docker compose version >/dev/null 2>&1; then
  COMPOSE_CMD=(docker compose)
elif command -v docker-compose >/dev/null 2>&1; then
  COMPOSE_CMD=(docker-compose)
else
  echo -e "${YELLOW}Neither 'docker compose' nor 'docker-compose' found.${NC}"
  echo "Install Compose v2 (recommended):"
  echo "  sudo apt-get update && sudo apt-get install -y docker-compose-plugin"
  echo "Or install docker-compose classic."
  exit 1
fi

# Determine which compose file to use
if [ "$DEV_MODE" = true ]; then
    echo -e "${YELLOW}Development mode enabled - using minimal setup with docker-compose.dev.yml${NC}"
    COMPOSE_FILE="docker-compose.dev.yml"
else
    echo -e "${YELLOW}Production mode - using docker-compose.yml${NC}"
    COMPOSE_FILE="docker-compose.yml"
fi

# Build compose args as array
COMPOSE_ARGS=(-f "$COMPOSE_FILE")
if [ -f "docker-compose.override.yml" ]; then
    echo -e "${BLUE}Found docker-compose.override.yml - including local overrides${NC}"
    COMPOSE_ARGS+=(-f "docker-compose.override.yml")
fi

# Start Docker containers
echo -e "${GREEN}Starting Docker containers...${NC}"
"${COMPOSE_CMD[@]}" "${COMPOSE_ARGS[@]}" up -d

# Display access instructions
if [ "$DEV_MODE" = true ]; then
    echo -e "${BLUE}Development environment setup complete!${NC}"
    echo -e "${YELLOW}Please wait for the frontend to finish building (this may take a few minutes)${NC}"
    echo -e "${GREEN}Then visit: http://localhost:3000/setup${NC}"
else
    echo -e "${BLUE}Production environment setup complete!${NC}"
    echo -e "${YELLOW}Please wait a moment for all services to start${NC}"
    echo -e "${GREEN}Then visit: http://localhost${NC}"
fi