#!/usr/bin/env bash
# Kowloon installer
# Usage: curl -fsSL https://install.kwln.social | bash
# Or run directly: bash install.sh

set -euo pipefail

SETUP_IMAGE="ghcr.io/jzellis/kowloon-setup:latest"
APP_IMAGE="ghcr.io/jzellis/kowloon:latest"
INSTALL_DIR="${KOWLOON_DIR:-$HOME/kowloon}"
SETUP_PORT=2999

# ── Colors ────────────────────────────────────────────────────────────────────
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BOLD='\033[1m'
RESET='\033[0m'

info()    { echo -e "${BOLD}==> $*${RESET}"; }
success() { echo -e "${GREEN}==> $*${RESET}"; }
warn()    { echo -e "${YELLOW}==> WARNING: $*${RESET}"; }
error()   { echo -e "${RED}==> ERROR: $*${RESET}" >&2; exit 1; }

# ── Check Docker ──────────────────────────────────────────────────────────────
info "Checking requirements..."

if ! command -v docker &>/dev/null; then
  warn "Docker is not installed."
  echo ""
  echo "  Install Docker from: https://docs.docker.com/get-docker/"
  echo "  Then re-run this installer."
  echo ""
  exit 1
fi

if ! docker info &>/dev/null; then
  error "Docker is installed but not running. Start Docker and try again."
fi

DOCKER_VERSION=$(docker version --format '{{.Server.Version}}' 2>/dev/null || echo "unknown")
info "Docker $DOCKER_VERSION found."

# Compose v2 check
if ! docker compose version &>/dev/null; then
  error "Docker Compose v2 is required (docker compose, not docker-compose).\nInstall it from: https://docs.docker.com/compose/install/"
fi

# ── Working directory ─────────────────────────────────────────────────────────
info "Setting up install directory: $INSTALL_DIR"
mkdir -p "$INSTALL_DIR"
cd "$INSTALL_DIR"

# ── Get server IP ─────────────────────────────────────────────────────────────
SERVER_IP=""
# Try common sources in order
for iface_ip in \
  "$(curl -sf --max-time 3 https://api.ipify.org 2>/dev/null)" \
  "$(curl -sf --max-time 3 https://icanhazip.com 2>/dev/null)" \
  "$(ip route get 1.1.1.1 2>/dev/null | awk '{for(i=1;i<=NF;i++) if($i=="src") {print $(i+1); exit}}')"; do
  if [[ -n "$iface_ip" ]]; then
    SERVER_IP="$iface_ip"
    break
  fi
done

echo ""
echo -e "${BOLD}Your server's public IP address is: ${GREEN}${SERVER_IP:-unknown}${RESET}"
echo "Make sure your domain's A record points to this IP before continuing."
echo ""

# ── Pull setup image ──────────────────────────────────────────────────────────
info "Pulling Kowloon setup image..."
docker pull "$SETUP_IMAGE"

# ── Run setup container ───────────────────────────────────────────────────────
info "Starting installation wizard..."
echo ""
echo -e "  Open this URL in your browser: ${BOLD}http://${SERVER_IP:-<your-server-ip>}:${SETUP_PORT}${RESET}"
echo ""
echo "  (If running locally, use: http://localhost:${SETUP_PORT})"
echo ""
echo "  The installer will guide you through the rest of the setup."
echo "  This terminal will wait until you complete the form..."
echo ""

# Run in foreground — blocks until setup server calls process.exit(0).
# The /config volume is mounted to our INSTALL_DIR so generated files land here.
docker run --rm \
  --name kowloon-setup \
  -p "${SETUP_PORT}:${SETUP_PORT}" \
  -v "${INSTALL_DIR}:/config" \
  "$SETUP_IMAGE"

# ── Verify config was written ─────────────────────────────────────────────────
if [[ ! -f "$INSTALL_DIR/.env" ]]; then
  error "Setup did not complete — .env was not created. Run the installer again."
fi

if [[ ! -f "$INSTALL_DIR/docker-compose.yml" ]]; then
  error "Setup did not complete — docker-compose.yml was not created. Run the installer again."
fi

success "Configuration written to $INSTALL_DIR"

# ── Pull app images ───────────────────────────────────────────────────────────
info "Pulling Kowloon app images (this may take a minute)..."
docker compose pull

# ── Start the stack ───────────────────────────────────────────────────────────
info "Starting Kowloon..."
docker compose up -d

# ── Wait for health ───────────────────────────────────────────────────────────
DOMAIN=$(grep '^DOMAIN=' "$INSTALL_DIR/.env" | cut -d= -f2 | tr -d '[:space:]')
info "Waiting for Kowloon to become available..."

MAX_WAIT=120
WAITED=0
HEALTH_URL="https://${DOMAIN}/health"

while true; do
  STATUS=$(curl -sf --max-time 5 "$HEALTH_URL" 2>/dev/null | python3 -c "import sys,json; d=json.load(sys.stdin); print('ok' if d.get('ok') else 'wait')" 2>/dev/null || echo "wait")
  if [[ "$STATUS" == "ok" ]]; then
    break
  fi
  if [[ $WAITED -ge $MAX_WAIT ]]; then
    warn "Kowloon is not responding at $HEALTH_URL after ${MAX_WAIT}s."
    warn "It may still be starting. Check status with: docker compose logs -f"
    break
  fi
  sleep 5
  WAITED=$((WAITED + 5))
  echo -n "."
done
echo ""

# ── Done ──────────────────────────────────────────────────────────────────────
echo ""
echo -e "${GREEN}${BOLD}============================================================${RESET}"
echo -e "${GREEN}${BOLD}  Kowloon is installed!${RESET}"
echo -e "${GREEN}${BOLD}============================================================${RESET}"
echo ""
echo -e "  Your server: ${BOLD}https://${DOMAIN}${RESET}"
echo ""
echo "  Useful commands (run from $INSTALL_DIR):"
echo "    docker compose logs -f       # follow logs"
echo "    docker compose restart       # restart all services"
echo "    docker compose down          # stop"
echo "    docker compose pull && docker compose up -d  # upgrade"
echo ""