setup

#!/usr/bin/env bash
# heygen-stack setup — first-time installation for Claude Code and OpenClaw users.
# Idempotent: safe to run multiple times.
set -euo pipefail

# ─── Colors (degrade gracefully) ─────────────────────────────
if [ -t 1 ] && command -v tput &>/dev/null && [ "$(tput colors 2>/dev/null || echo 0)" -ge 8 ]; then
  GREEN=$(tput setaf 2) YELLOW=$(tput setaf 3) RED=$(tput setaf 1) BOLD=$(tput bold) RESET=$(tput sgr0)
else
  GREEN="" YELLOW="" RED="" BOLD="" RESET=""
fi

ok()   { echo "${GREEN}✓${RESET} $1"; }
warn() { echo "${YELLOW}!${RESET} $1"; }
err()  { echo "${RED}✗${RESET} $1"; }

# ─── Locate repo root ────────────────────────────────────────
STACK_DIR="$(cd "$(dirname "$0")" && pwd)"
if [ ! -f "$STACK_DIR/SKILL.md" ]; then
  err "Cannot find SKILL.md — run this script from the heygen-stack repo root."
  exit 1
fi

# ─── Parse flags ────────────────────────────────────────────
HOST_OVERRIDE=""
while [ $# -gt 0 ]; do
  case "$1" in
    --host)
      shift
      if [ $# -eq 0 ]; then
        err "--host requires a value: claude or openclaw"
        exit 1
      fi
      case "$1" in
        claude|claude-code) HOST_OVERRIDE="claude-code" ;;
        openclaw)           HOST_OVERRIDE="openclaw" ;;
        *)
          err "Unknown host agent: $1 (expected: claude or openclaw)"
          exit 1
          ;;
      esac
      shift
      ;;
    --host=*)
      val="${1#--host=}"
      case "$val" in
        claude|claude-code) HOST_OVERRIDE="claude-code" ;;
        openclaw)           HOST_OVERRIDE="openclaw" ;;
        *)
          err "Unknown host agent: $val (expected: claude or openclaw)"
          exit 1
          ;;
      esac
      shift
      ;;
    *)
      warn "Unknown flag: $1"
      shift
      ;;
  esac
done

# ─── Detect host agent ───────────────────────────────────────
# Priority: 1) --host flag, 2) script path, 3) directory existence fallback
AGENT="unknown"
SKILLS_DIR=""

if [ -n "$HOST_OVERRIDE" ]; then
  # Explicit --host flag takes highest priority
  AGENT="$HOST_OVERRIDE"
elif [[ "$STACK_DIR" == *"/.openclaw/workspace/skills/"* ]]; then
  AGENT="openclaw"
elif [[ "$STACK_DIR" == *"/.claude/skills/"* ]]; then
  AGENT="claude-code"
elif [ -d "$HOME/.openclaw" ] && [ -d "$HOME/.claude" ]; then
  # Both installed — path was ambiguous, warn and ask for --host
  warn "Both Claude Code and OpenClaw detected. Cannot auto-detect host agent."
  warn "Re-run with: ./setup --host claude  OR  ./setup --host openclaw"
elif [ -d "$HOME/.claude" ]; then
  AGENT="claude-code"
elif [ -d "$HOME/.openclaw" ]; then
  AGENT="openclaw"
fi

# Set skills directory based on detected agent
case "$AGENT" in
  claude-code) SKILLS_DIR="$HOME/.claude/skills" ;;
  openclaw)    SKILLS_DIR="$HOME/.openclaw/workspace/skills" ;;
esac

echo ""
echo "${BOLD}HeyGen Stack Setup${RESET}"
echo "─────────────────────────────"
if [ "$AGENT" = "unknown" ]; then
  warn "Could not detect Claude Code or OpenClaw. See INSTALL.md for manual setup."
else
  ok "Detected host agent: ${BOLD}${AGENT}${RESET}"
fi

# ─── Create skill symlinks ───────────────────────────────────
SKILLS=(heygen-avatar heygen-video)

if [ -n "$SKILLS_DIR" ]; then
  STACK_SKILL_DIR="$SKILLS_DIR/heygen-stack"
  mkdir -p "$STACK_SKILL_DIR"
  for skill in "${SKILLS[@]}"; do
    src="$STACK_DIR/$skill/SKILL.md"
    [ ! -f "$src" ] && { warn "Skipping $skill — SKILL.md not found"; continue; }
    mkdir -p "$STACK_SKILL_DIR/$skill" "$SKILLS_DIR/$skill"
    ln -sf "$src" "$STACK_SKILL_DIR/$skill/SKILL.md"
    ln -sf "$src" "$SKILLS_DIR/$skill/SKILL.md"
  done
  ln -sf "$STACK_DIR/SKILL.md" "$STACK_SKILL_DIR/SKILL.md"
  [ -d "$STACK_DIR/references" ] && ln -sfn "$STACK_DIR/references" "$STACK_SKILL_DIR/references"
  ok "Skill symlinks created in ${SKILLS_DIR}"
  for skill in "${SKILLS[@]}"; do echo "    ${skill}/SKILL.md"; done
fi

# ─── Persistent config ──────────────────────────────────────
HEYGEN_CONFIG_DIR="$HOME/.heygen"
HEYGEN_CONFIG="$HEYGEN_CONFIG_DIR/config"

# Load key from config file if env var is not set
if [ -z "${HEYGEN_API_KEY:-}" ] && [ -f "$HEYGEN_CONFIG" ]; then
  # shellcheck source=/dev/null
  source "$HEYGEN_CONFIG"
fi

# ─── Check HEYGEN_API_KEY ────────────────────────────────────
echo ""
if [ -z "${HEYGEN_API_KEY:-}" ]; then
  echo "  Get your key: ${BOLD}https://app.heygen.com/settings/api${RESET}"
  echo ""
  if [ -t 0 ]; then
    printf "  Enter your HeyGen API key: "
    read -r HEYGEN_API_KEY
    if [ -z "$HEYGEN_API_KEY" ]; then
      warn "No key entered. You can set it later:"
      echo "    export HEYGEN_API_KEY=<your-key>"
      echo "    Or re-run ./setup"
    fi
  else
    warn "HEYGEN_API_KEY is not set (non-interactive shell)."
    echo "  Set it manually:  export HEYGEN_API_KEY=<your-key>"
    echo "  Or re-run ./setup in an interactive terminal."
  fi
fi

if [ -n "${HEYGEN_API_KEY:-}" ]; then
  echo -n "Validating API key... "
  HTTP_CODE=$(curl -s -o /dev/null -w "%{http_code}" "https://api.heygen.com/v3/users/me" \
    -H "X-Api-Key: $HEYGEN_API_KEY" 2>/dev/null || echo "000")
  if [ "$HTTP_CODE" = "200" ]; then
    ok "API key is valid."
    # Persist to ~/.heygen/config
    mkdir -p "$HEYGEN_CONFIG_DIR"
    echo "HEYGEN_API_KEY=$HEYGEN_API_KEY" > "$HEYGEN_CONFIG"
    chmod 600 "$HEYGEN_CONFIG"
    ok "Key saved to ${BOLD}~/.heygen/config${RESET} (persists across sessions)."
  else
    err "API key validation failed (HTTP $HTTP_CODE)."
    echo "  Check your key: https://app.heygen.com/settings/api"
  fi
fi

# ─── Summary ─────────────────────────────────────────────────
echo ""
echo "${BOLD}Installed skills:${RESET}"
for skill in "${SKILLS[@]}"; do
  echo "  - $skill"
done

echo ""
echo "${BOLD}Next steps:${RESET}"
if [ -z "${HEYGEN_API_KEY:-}" ]; then
  echo "  1. Re-run ./setup and enter your API key, or:"
  echo "     export HEYGEN_API_KEY=<your-key>"
  echo "  2. Ask your agent:  \"Create my HeyGen avatar\""
else
  echo "  Ask your agent: \"Create my HeyGen avatar\""
fi
echo ""