#!/bin/bash
# Agent Orchestrator setup script
# Validates prerequisites, installs dependencies, builds packages, and links the CLI globally

set -e  # Exit on error

REPO_ROOT="$(cd "$(dirname "$0")/.." && pwd)"

echo "Agent Orchestrator Setup"
echo ""

# ─── Hard requirements (exit 1 if missing) ────────────────────────────────────

# Node.js >= 20
if ! command -v node &> /dev/null; then
  echo "ERROR: Node.js is not installed."
  echo "  Install Node.js 20+: https://nodejs.org/en/download"
  exit 1
fi

NODE_MAJOR=$(node -e "process.stdout.write(String(process.versions.node.split('.')[0]))")
if [ "$NODE_MAJOR" -lt 20 ]; then
  echo "ERROR: Node.js $NODE_MAJOR.x detected, but 20+ is required."
  echo "  Install Node.js 20+: https://nodejs.org/en/download"
  exit 1
fi
echo "[ok] Node.js $(node --version)"

# git >= 2.25 (required for worktree support)
if ! command -v git &> /dev/null; then
  echo "ERROR: git is not installed."
  echo "  Install git 2.25+: https://git-scm.com/downloads"
  exit 1
fi

GIT_VERSION=$(git --version | grep -oE '[0-9]+\.[0-9]+' | head -1)
GIT_MAJOR=$(echo "$GIT_VERSION" | cut -d. -f1)
GIT_MINOR=$(echo "$GIT_VERSION" | cut -d. -f2)
if [ "$GIT_MAJOR" -lt 2 ] || { [ "$GIT_MAJOR" -eq 2 ] && [ "$GIT_MINOR" -lt 25 ]; }; then
  echo "ERROR: git $GIT_VERSION detected, but 2.25+ is required (worktree support)."
  echo "  Upgrade git: https://git-scm.com/downloads"
  exit 1
fi
echo "[ok] git $GIT_VERSION"

# ─── Soft requirements (warn + offer interactive fix) ─────────────────────────

# Detect interactive terminal for optional prompts (skip in CI/Docker)
INTERACTIVE=false
if [ -t 0 ]; then
  INTERACTIVE=true
fi

# tmux
if ! command -v tmux &> /dev/null; then
  echo ""
  echo "WARNING: tmux is not installed (default runtime requires it)."
  if [ "$INTERACTIVE" = true ] && command -v brew &> /dev/null; then
    read -r -p "  Install tmux via Homebrew? [Y/n] " response
    response=${response:-Y}
    if [[ "$response" =~ ^[Yy]$ ]]; then
      brew install tmux
      echo "[ok] tmux installed"
    else
      echo "  Skipping. Install later: brew install tmux"
    fi
  else
    echo "  Install tmux: https://github.com/tmux/tmux/wiki/Installing"
  fi
else
  echo "[ok] tmux $(tmux -V | grep -oE '[0-9]+\.[0-9a-z]+')"
fi

# gh CLI authentication
if ! command -v gh &> /dev/null; then
  echo ""
  echo "WARNING: GitHub CLI (gh) is not installed."
  echo "  Install: https://cli.github.com/"
else
  if ! gh auth status &> /dev/null; then
    echo ""
    echo "WARNING: GitHub CLI is not authenticated."
    if [ "$INTERACTIVE" = true ]; then
      read -r -p "  Run 'gh auth login' now? [Y/n] " response
      response=${response:-Y}
      if [[ "$response" =~ ^[Yy]$ ]]; then
        gh auth login
      else
        echo "  Skipping. Authenticate later: gh auth login"
      fi
    else
      echo "  Authenticate later: gh auth login"
    fi
  else
    echo "[ok] gh authenticated"
  fi
fi

# claude CLI
if ! command -v claude &> /dev/null; then
  echo ""
  echo "WARNING: Claude CLI is not installed (required for claude-code agent)."
  echo "  Install: npm install -g @anthropic-ai/claude-code"
fi

echo ""

# ─── Install pnpm ────────────────────────────────────────────────────────────

if command -v pnpm &> /dev/null; then
  echo "[ok] pnpm $(pnpm --version) (already installed)"
else
  echo "Installing pnpm via corepack..."
  if corepack enable && corepack prepare --activate 2>/dev/null; then
    echo "[ok] pnpm $(pnpm --version)"
  else
    echo "  corepack failed (likely permissions), falling back to npm install..."
    npm install -g pnpm
    echo "[ok] pnpm $(pnpm --version)"
  fi
fi

# ─── Install, build, link ────────────────────────────────────────────────────

echo ""
echo "Installing dependencies..."
pnpm install

echo ""
echo "Cleaning stale build artifacts..."
rm -rf packages/web/.next

echo ""
echo "Building all packages..."
pnpm build

echo ""
echo "Linking CLI globally..."
cd packages/ao
npm_link_error="$(mktemp)"
if npm link --force 2>"$npm_link_error"; then
  rm -f "$npm_link_error"
elif [ "$INTERACTIVE" = true ]; then
  rm -f "$npm_link_error"
  echo "  Launcher refresh failed. Retrying with sudo..."
  if ! sudo npm link --force; then
    echo "ERROR: sudo npm link --force failed. Inspect npm output above." >&2
    exit 1
  fi
else
  cat "$npm_link_error" >&2
  rm -f "$npm_link_error"
  echo "ERROR: Launcher refresh failed. Run manually: cd packages/ao && sudo npm link --force"
  exit 1
fi
cd "$REPO_ROOT"

# ─── Verify ao is in PATH ────────────────────────────────────────────────────

echo ""
if command -v ao &> /dev/null; then
  echo "[ok] 'ao' command is available in PATH"
else
  NPM_BIN="$(npm bin -g 2>/dev/null || npm config get prefix)/bin"
  echo "WARNING: 'ao' is not in your PATH."
  echo "  Add this to your shell profile (~/.zshrc or ~/.bashrc):"
  echo ""
  echo "    export PATH=\"$NPM_BIN:\$PATH\""
  echo ""
  echo "  Then restart your terminal or run: source ~/.zshrc"
fi

# ─── Done ─────────────────────────────────────────────────────────────────────

echo ""
echo "Setup complete!"
echo ""
echo "What's next:"
echo ""
echo "  Navigate to your project directory and start:"
echo ""
echo "    cd ~/your-project"
echo "    ao start            # auto-detects, creates config, launches dashboard"
echo ""
echo "  Want to add more projects later?"
echo ""
echo "    ao start ~/path/to/another-repo"
echo ""