#!/bin/bash
set -e

################################################################
# Command line args (any --key value sets a config override)
################################################################

_CLI_OVERRIDES=$(mktemp)

while [[ $# -gt 0 ]]; do
  case $1 in
    --config) CONFIG_FILE="$2"; shift 2 ;;
    --name) echo "sprite_name: $2" >> "$_CLI_OVERRIDES"; shift 2 ;;
    --repo) echo "repo: $2" >> "$_CLI_OVERRIDES"; shift 2 ;;
    --*) echo "${1#--}: $2" >> "$_CLI_OVERRIDES"; shift 2 ;;
    *) echo "Unknown option: $1"; exit 1 ;;
  esac
done

################################################################
# Load config
################################################################

DIR="$(cd "$(dirname "$0")" 2>/dev/null && pwd)"
BASE_URL="https://dissonantp.github.io/sprite-environment"

# Parse a value from config YAML (flat key: value only)
# Resolution: CLI overrides > --config file > local config.yaml > remote config.yaml
cfg() {
  local key="$1" default="$2" val=""
  # Check CLI overrides first
  val=$(grep "^${key}:" "$_CLI_OVERRIDES" 2>/dev/null | tail -1 | sed 's/^[^:]*: *//' | sed 's/ *$//')
  # Then config file
  if [ -z "$val" ]; then
    local file="${CONFIG_FILE:-}"
    if [ -z "$file" ] && [ -f "$DIR/config.yaml" ]; then
      file="$DIR/config.yaml"
    fi
    if [ -z "$file" ]; then
      if [ -z "$_REMOTE_CONFIG" ]; then
        _REMOTE_CONFIG=$(mktemp)
        curl -sL "$BASE_URL/config.yaml" -o "$_REMOTE_CONFIG"
      fi
      file="$_REMOTE_CONFIG"
    fi
    val=$(grep "^${key}:" "$file" 2>/dev/null | head -1 | sed 's/^[^:]*: *//' | sed 's/ *$//')
  fi
  # Expand $HOME in values
  val=$(eval echo "$val")
  if [ -z "$val" ]; then echo "$default"; else echo "$val"; fi
}

################################################################
# Resolve sprite name
################################################################

export SPRITE_NAME=$(cfg sprite_name "")

if [ -z "$SPRITE_NAME" ]; then
  echo "Usage: setup.sh --name <sprite-name> [--config path] [--key value ...]"
  exit 1
fi

if [[ ! "$SPRITE_NAME" =~ ^[a-z0-9]([a-z0-9-]*[a-z0-9])?$ ]]; then
  echo "Error: sprite name must be lowercase alphanumeric with hyphens (e.g. my-sprite)"
  exit 1
fi

################################################################
# Create Sprite (skip if already exists)
################################################################

if sprite exec -s $SPRITE_NAME true > /dev/null 2>&1; then
  echo "==> Sprite '$SPRITE_NAME' already exists, updating"
else
  echo "==> Creating sprite: $SPRITE_NAME"
  sprite create -skip-console $SPRITE_NAME
fi

################################################################
# Script runner (local or remote)
################################################################

if [ -f "$DIR/scripts/install_docker.sh" ]; then
  run_script() { SPRITE_NAME=$SPRITE_NAME bash "$DIR/$1"; }
else
  run_script() {
    local tmp=$(mktemp)
    curl -sL "$BASE_URL/$1" -o "$tmp"
    SPRITE_NAME=$SPRITE_NAME bash "$tmp"
    rm -f "$tmp"
  }
fi

################################################################
# Run scripts
################################################################

# INSTALL GH CLI (first, so docker can use gh token for ghcr.io)
if [ "$(cfg install_gh true)" = "true" ]; then
  echo "==> Installing GitHub CLI"
  export GH_SSH_KEY=$(cfg gh_ssh_key "$HOME/.ssh/id_ed25519_dissonantP")
  run_script "scripts/install_gh.sh"
else
  echo "==> Skipping GitHub CLI"
fi

# INSTALL OPENSSH
export INSTALL_OPENSSH=$(cfg install_openssh false)
if [ "$INSTALL_OPENSSH" = "true" ]; then
  echo "==> Installing OpenSSH"
  run_script "scripts/install_openssh.sh"
else
  echo "==> Skipping OpenSSH"
fi

# INSTALL DOCKER
if [ "$(cfg install_docker true)" = "true" ]; then
  echo "==> Installing Docker"
  export DOCKER_GHCR_LOGIN=$(cfg docker_ghcr_login true)
  export DOCKER_GHCR_USER=$(cfg docker_ghcr_user dissonantP)
  run_script "scripts/install_docker.sh"
else
  echo "==> Skipping Docker"
fi

# INSTALL YARN
if [ "$(cfg install_yarn true)" = "true" ]; then
  echo "==> Installing Yarn"
  run_script "scripts/install_yarn.sh"
else
  echo "==> Skipping Yarn"
fi

# INSTALL CODEX
if [ "$(cfg install_codex true)" = "true" ]; then
  echo "==> Installing Codex"
  export CODEX_AUTH_FILE=$(cfg codex_auth_file "$HOME/.codex/auth.json")
  run_script "scripts/install_codex.sh"
else
  echo "==> Skipping Codex"
fi

# INSTALL PLAYWRIGHT MCP
if [ "$(cfg install_playwright_mcp true)" = "true" ]; then
  echo "==> Installing Playwright MCP"
  run_script "scripts/install_playwright_mcp.sh"
else
  echo "==> Skipping Playwright MCP"
fi

# CLONE REPO
REPO=$(cfg repo "")
if [ -n "$REPO" ]; then
  echo "==> Cloning repo: $REPO"
  sprite exec -s $SPRITE_NAME gh repo clone "$REPO"
fi

# CHEATSHEET
echo "==> Installing cheatsheet"
run_script "scripts/install_cheatsheet.sh"

# VALIDATE
echo "==> Validating"
run_script "scripts/validate.sh"

# ALL DONE!
echo "==> Done"