#!/usr/bin/env bash
# activate_self_tdd.sh — Timmy Time dev environment bootstrapper
#
# Usage:
#   bash activate_self_tdd.sh              # standard (Ollama) setup
#   bash activate_self_tdd.sh --big-brain  # install AirLLM extra too
#
# What it does:
#   1. Creates a Python venv (or reuses an existing one)
#   2. Installs Timmy Time (+ dev deps, optionally bigbrain)
#   3. Runs the full test suite — aborts if anything fails
#   4. Launches the self-TDD watchdog in the background
#   5. Starts the dashboard
#
# Everything stays local. No cloud. Sats are sovereignty, boss.

set -euo pipefail

REPO_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
VENV_DIR="$REPO_DIR/.venv"
BIG_BRAIN=0

for arg in "$@"; do
  case $arg in
    --big-brain) BIG_BRAIN=1 ;;
    *) echo "Unknown argument: $arg"; exit 1 ;;
  esac
done

echo "==> Timmy Time — sovereign AI agent bootstrapper"
echo "    Working directory: $REPO_DIR"

# ── 1. Virtual environment ────────────────────────────────────────────────────
if [[ ! -d "$VENV_DIR" ]]; then
  echo "==> Creating virtual environment..."
  python3 -m venv "$VENV_DIR"
fi

# shellcheck disable=SC1091
source "$VENV_DIR/bin/activate"
echo "==> Virtual environment active: $VENV_DIR"

# ── 2. Install dependencies ───────────────────────────────────────────────────
if [[ $BIG_BRAIN -eq 1 ]]; then
  echo "==> Installing with bigbrain extra (AirLLM)..."
  pip install --quiet -e "$REPO_DIR[dev,bigbrain]"
  # On Apple Silicon, also install the MLX backend.
  if [[ "$(uname -m)" == "arm64" && "$(uname -s)" == "Darwin" ]]; then
    echo "==> Apple Silicon detected — installing AirLLM MLX backend..."
    pip install --quiet "airllm[mlx]"
  fi
else
  echo "==> Installing standard dependencies..."
  pip install --quiet -e "$REPO_DIR[dev]"
fi

# ── 3. Run tests ──────────────────────────────────────────────────────────────
echo "==> Running test suite..."
python -m pytest "$REPO_DIR/tests/" -q --tb=short
echo "==> All tests passed."

# ── 4. Self-TDD watchdog (background) ────────────────────────────────────────
echo "==> Starting self-TDD watchdog (60s interval) in background..."
self-tdd watch --interval 60 &
WATCHDOG_PID=$!
echo "    Watchdog PID: $WATCHDOG_PID"
echo "    Kill with: kill $WATCHDOG_PID"

# ── 5. Dashboard ─────────────────────────────────────────────────────────────
echo ""
echo "==> Starting Timmy Time dashboard at http://localhost:8000"
echo "    Ctrl-C stops the dashboard (watchdog continues until you kill it)"
echo ""
uvicorn dashboard.app:app --reload --host 0.0.0.0 --port 8000