setup

#!/usr/bin/env bash
# kstack setup — install/uninstall/verify skills for Kiro CLI
# Usage: ./setup          (install)
#        ./setup remove   (uninstall)
#        ./setup verify   (check installation)
#        ./setup help     (show usage)
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
SKILLS_DIR="$HOME/.kiro/skills"
SKILLS=(plan-product plan-eng code-review ship qa retro)
VERSION=$(cat "$SCRIPT_DIR/VERSION" 2>/dev/null || echo "unknown")

# --- Help ---
if [[ "${1:-}" == "help" || "${1:-}" == "--help" || "${1:-}" == "-h" ]]; then
  echo "kstack v${VERSION} — cognitive mode skills for Kiro CLI"
  echo ""
  echo "Usage:"
  echo "  ./setup          Install skills (safe to re-run)"
  echo "  ./setup verify   Check installation health"
  echo "  ./setup remove   Uninstall skills (keeps repo)"
  echo "  ./setup help     Show this message"
  echo ""
  echo "Supported: macOS, Linux, WSL"
  echo "Requires: Kiro CLI (https://kiro.dev)"
  exit 0
fi

# --- Verify ---
if [[ "${1:-}" == "verify" ]]; then
  echo "Verifying kstack v${VERSION}..."
  ERRORS=0

  # Check kiro-cli
  if command -v kiro-cli &>/dev/null; then
    KIRO_VER=$(kiro-cli --version 2>/dev/null || echo "unknown")
    echo "  ✅ kiro-cli: $KIRO_VER"
  else
    echo "  ❌ kiro-cli not found in PATH"
    ERRORS=$((ERRORS + 1))
  fi

  # Check skills dir
  if [[ -d "$SKILLS_DIR" ]]; then
    echo "  ✅ skills dir: $SKILLS_DIR"
  else
    echo "  ❌ skills dir missing: $SKILLS_DIR"
    ERRORS=$((ERRORS + 1))
  fi

  # Check each skill
  for skill in "${SKILLS[@]}"; do
    dest="$SKILLS_DIR/$skill"
    if [[ -L "$dest" ]]; then
      target=$(readlink -f "$dest" 2>/dev/null || readlink "$dest")
      if [[ -f "$target/SKILL.md" ]]; then
        echo "  ✅ $skill → $target"
      else
        echo "  ❌ $skill symlink exists but SKILL.md missing at $target"
        ERRORS=$((ERRORS + 1))
      fi
    elif [[ -d "$dest" && -f "$dest/SKILL.md" ]]; then
      echo "  ⚠️  $skill is a directory (not symlink) — works but won't auto-update"
    else
      echo "  ❌ $skill not installed"
      ERRORS=$((ERRORS + 1))
    fi
  done

  echo ""
  if [[ $ERRORS -eq 0 ]]; then
    echo "All checks passed. ✅"
  else
    echo "$ERRORS issue(s) found. Run ./setup to fix."
  fi
  exit $ERRORS
fi

# --- Uninstall ---
if [[ "${1:-}" == "remove" ]]; then
  echo "Uninstalling kstack v${VERSION}..."
  REMOVED=0
  for skill in "${SKILLS[@]}"; do
    dest="$SKILLS_DIR/$skill"
    if [[ -L "$dest" ]]; then
      rm -f "$dest"
      echo "  ✅ removed $skill"
      REMOVED=$((REMOVED + 1))
    else
      echo "  ⏭️  $skill not a symlink, skipping"
    fi
  done
  echo ""
  if [[ $REMOVED -eq 0 ]]; then
    echo "Nothing to remove — kstack was not installed."
  else
    echo "kstack uninstalled ($REMOVED skills removed). Repo at $SCRIPT_DIR left intact."
    echo "To fully remove: rm -rf $SCRIPT_DIR"
  fi
  exit 0
fi

# --- Invalid argument ---
if [[ -n "${1:-}" ]]; then
  echo "Unknown command: $1"
  echo "Run ./setup help for usage."
  exit 1
fi

# --- Install ---

# Pre-flight: check kiro-cli
if ! command -v kiro-cli &>/dev/null; then
  echo "⚠️  kiro-cli not found in PATH."
  echo "   Install it first: https://kiro.dev"
  echo "   Continuing anyway (skills will be ready when kiro-cli is installed)..."
  echo ""
fi

echo "Installing kstack v${VERSION}..."
mkdir -p "$SKILLS_DIR"

INSTALLED=0
SKIPPED=0
for skill in "${SKILLS[@]}"; do
  src="$SCRIPT_DIR/$skill"
  dest="$SKILLS_DIR/$skill"

  if [[ ! -d "$src" ]]; then
    echo "  ⚠️  $skill not found in repo"
    continue
  fi

  # Already correctly linked — skip
  if [[ -L "$dest" ]]; then
    existing=$(readlink -f "$dest" 2>/dev/null || readlink "$dest")
    expected=$(cd "$src" && pwd)
    if [[ "$existing" == "$expected" ]]; then
      echo "  ✅ $skill (already installed)"
      SKIPPED=$((SKIPPED + 1))
      continue
    fi
    rm -f "$dest"
  fi

  # Handle existing non-symlink directory
  if [[ -e "$dest" && ! -L "$dest" ]]; then
    echo "  ⚠️  $skill exists as directory, backing up to ${dest}.bak"
    mv "$dest" "${dest}.bak"
  fi

  ln -sf "$src" "$dest"
  echo "  ✅ $skill"
  INSTALLED=$((INSTALLED + 1))
done

echo ""
if [[ $SKIPPED -eq ${#SKILLS[@]} ]]; then
  echo "kstack v${VERSION}: already installed, all ${#SKILLS[@]} skills up to date."
else
  echo "kstack v${VERSION}: ${INSTALLED} installed, ${SKIPPED} already up to date."
fi
echo ""
echo "Usage — tell Kiro any of these:"
echo "  • plan product    — rethink the problem, find the 10-star version"
echo "  • plan eng        — architecture, diagrams, failure modes, test matrix"
echo "  • code review     — paranoid review: what blows up in prod?"
echo "  • ship            — sync main → test → push → PR"
echo "  • qa              — diff-aware testing with health score"
echo "  • retro           — data-driven retrospective"
echo ""
echo "Run ./setup verify to confirm everything is working."