#!/bin/bash
set -euo pipefail

# niteshift-setup.sh - Setup script for niteshiftdev/umami
# This script sets up the Umami analytics application with all required dependencies

# Logging setup
LOG_FILE="${NITESHIFT_LOG_FILE:-/dev/stdout}"

log() {
  echo "[$(date '+%Y-%m-%d %H:%M:%S')] $*" | tee -a "$LOG_FILE" >/dev/null
  echo "$*"
}

log_error() {
  echo "[$(date '+%Y-%m-%d %H:%M:%S')] ERROR: $*" | tee -a "$LOG_FILE" >&2
  echo "ERROR: $*" >&2
}

DB_SETUP_SENTINEL="${DB_SETUP_SENTINEL:-$(pwd)/.niteshift-db-setup-complete}"

# Pre-flight checks
log "Starting niteshift setup for umami..."

USE_PREBAKED_SETUP=0
if [[ "${UMAMI_PREBAKED:-0}" == "1" ]]; then
  if command -v umami-prebake-sync >/dev/null 2>&1; then
    log "Prebaked Modal image detected, syncing cached artifacts..."
    if umami-prebake-sync "$(pwd)"; then
      USE_PREBAKED_SETUP=1
      log "✓ Prebaked dependencies and artifacts restored"
    else
      log_error "Prebaked sync failed, falling back to full setup"
    fi
  else
    log_error "UMAMI_PREBAKED=1 but umami-prebake-sync is missing; falling back to full setup"
  fi
fi

# 1. Check DATABASE_URL is set
if [ -z "${DATABASE_URL:-}" ]; then
  log_error "DATABASE_URL environment variable is not set"
  log_error "This script assumes DATABASE_URL is already configured"
  exit 1
fi
log "✓ DATABASE_URL is set"

# 2. Check Node.js version
if ! command -v node &> /dev/null; then
  log_error "Node.js is not installed"
  exit 1
fi

NODE_VERSION=$(node -v | sed 's/v//')
NODE_MAJOR=$(echo "$NODE_VERSION" | cut -d. -f1)
if [ "$NODE_MAJOR" -lt 18 ]; then
  log_error "Node.js 18.18+ is required, found version $NODE_VERSION"
  exit 1
fi
log "✓ Node.js $NODE_VERSION is installed"

# 3. Enable corepack for pnpm
if ! command -v pnpm &> /dev/null; then
  log "Enabling corepack for pnpm..."
  corepack enable
  if ! command -v pnpm &> /dev/null; then
    log_error "Failed to enable pnpm via corepack"
    exit 1
  fi
fi
log "✓ pnpm is available"

# 4. Install dependencies
if [[ "$USE_PREBAKED_SETUP" -eq 0 ]]; then
  log "Installing dependencies with pnpm..."
  if ! pnpm install --package-import-method=copy --frozen-lockfile; then
    log_error "Failed to install dependencies"
    exit 1
  fi
  log "✓ Dependencies installed"
else
  log "Linking dependencies from prebaked pnpm store..."
  if [ -d node_modules ]; then
    log "node_modules already present, skipping pnpm install for prebaked setup"
  elif pnpm install --package-import-method=copy --frozen-lockfile --offline; then
    log "✓ Dependencies installed (offline)"
  else
    log_error "Offline install failed, falling back to full pnpm install"
    if ! pnpm install --package-import-method=copy --frozen-lockfile; then
      log_error "Failed to install dependencies even after fallback"
      exit 1
    fi
    log "✓ Dependencies installed via fallback"
  fi
fi

# 5. Build only what's needed for dev (skip production Next.js build)
# For dev mode we need:
# - check-env: validates environment variables
# - build-db: generates Prisma client and builds database
# - check-db: verifies database connection and applies migrations
# - build-tracker: bundles tracker script (needed for tracking functionality)
# - build-geo: processes geographic data (needed for geo features)
# We skip build-app (production Next.js build) since dev mode compiles on-the-fly
if [[ "$USE_PREBAKED_SETUP" -eq 0 ]]; then
  log "Validating environment..."
  if ! pnpm run check-env; then
    log_error "Environment validation failed"
    exit 1
  fi
  log "✓ Environment validated"
else
  log "Skipping check-env (prebaked fast path)"
fi

PREBAKED_DB_SETUP_DONE=0
if [[ "$USE_PREBAKED_SETUP" -eq 1 ]] && [ -f "$DB_SETUP_SENTINEL" ]; then
  PREBAKED_DB_SETUP_DONE=1
fi

if [[ "$USE_PREBAKED_SETUP" -eq 0 ]]; then
  log "Building database client..."
  if ! pnpm run build-db; then
    log_error "Database client build failed"
    exit 1
  fi
  log "✓ Database client built"
else
  if [[ "$PREBAKED_DB_SETUP_DONE" -eq 1 ]]; then
    log "Skipping build-db (prebaked Prisma client detected; db setup already done)"
  else
    log "Building database client (prebaked image, first run)..."
    if ! pnpm run build-db; then
      log_error "Database client build failed"
      exit 1
    fi
    log "✓ Database client built"
  fi
fi

if [[ "$USE_PREBAKED_SETUP" -eq 0 ]]; then
  log "Checking database and applying migrations..."
  if ! pnpm run check-db; then
    log_error "Database check failed"
    exit 1
  fi
  log "✓ Database migrations applied"
else
  if [[ "$PREBAKED_DB_SETUP_DONE" -eq 1 ]]; then
    log "Skipping check-db (prebaked fast path; db setup already done)"
  else
    log "Checking database and applying migrations (prebaked image, first run)..."
    if ! pnpm run check-db; then
      log_error "Database check failed"
      exit 1
    fi
    log "✓ Database migrations applied"
    if ! touch "$DB_SETUP_SENTINEL"; then
      log_error "Failed to write db setup sentinel at $DB_SETUP_SENTINEL"
      exit 1
    fi
    PREBAKED_DB_SETUP_DONE=1
  fi
fi

if [[ "$USE_PREBAKED_SETUP" -eq 0 ]]; then
  log "Building tracker script..."
  if ! pnpm run build-tracker; then
    log_error "Tracker build failed"
    exit 1
  fi
  log "✓ Tracker script built"
else
  log "Skipping tracker build (prebaked bundle detected)"
fi

if [[ "$USE_PREBAKED_SETUP" -eq 0 ]]; then
  log "Building geo database..."
  if ! pnpm run build-geo; then
    log_error "Geo build failed"
    exit 1
  fi
  log "✓ Geo database built"
else
  log "Skipping geo build (prebaked GeoLite database detected)"
fi

# 6. Start the dev server in the background
log "Starting development server on port 3001 (with hot reload)..."
# Log dev server output to NITESHIFT_LOG_FILE (or stdout if not set)
pnpm run dev >> "$LOG_FILE" 2>&1 &
SERVER_PID=$!
log "✓ Dev server started with PID $SERVER_PID"

# 7. Wait for dev server to be ready
log "Waiting for dev server to be ready (this may take 30-60 seconds on first run)..."
MAX_ATTEMPTS=60
ATTEMPT=0
while [ $ATTEMPT -lt $MAX_ATTEMPTS ]; do
  if curl -s -o /dev/null -w "%{http_code}" http://localhost:3001/api/heartbeat 2>/dev/null | grep -q "200"; then
    log "✓ Dev server is ready and responding"
    break
  fi
  ATTEMPT=$((ATTEMPT + 1))
  if [ $ATTEMPT -eq $MAX_ATTEMPTS ]; then
    log_error "Dev server failed to start within 60 seconds"
    log_error "Check the log file for details: $LOG_FILE"
    kill $SERVER_PID 2>/dev/null || true
    exit 1
  fi
  sleep 1
done

# 8. Warm up the main application routes
log "Warming up main application routes..."
if curl -s -o /dev/null --max-time 30 http://localhost:3001/ 2>/dev/null; then
  log "✓ Main routes pre-compiled"
else
  log "Warning: Route warm-up timed out or failed (non-critical)"
fi

log ""
log "================================================================"
log "Setup complete! Umami dev server is running on http://localhost:3001"
log "================================================================"
log ""
log "Default credentials:"
log "  Username: admin"
log "  Password: umami"
log ""
log "Server PID: $SERVER_PID"
if [ "$LOG_FILE" != "/dev/stdout" ]; then
  log "Server logs: $LOG_FILE"
fi
log ""
log "Hot reload is enabled - changes will be reflected automatically"
log ""
log "To stop the server: kill $SERVER_PID"
log ""