#!/usr/bin/env bash
# awesome-claude-output-styles installer
#
# Install one style (and make it active):
#   curl -fsSL https://raw.githubusercontent.com/smixs/awesome-claude-output-styles/main/install.sh | bash -s -- eli15
#
# Install everything (nothing is activated, pick later via /config):
#   curl -fsSL https://raw.githubusercontent.com/smixs/awesome-claude-output-styles/main/install.sh | bash -s -- --all
#
# List available styles:
#   curl -fsSL https://raw.githubusercontent.com/smixs/awesome-claude-output-styles/main/install.sh | bash -s -- --list
#
# Add --enforce to also install a UserPromptSubmit hook that re-reminds Claude
# of the active style every turn (Claude Code only does this for built-in
# styles; custom styles fade in long sessions without it):
#   ... | bash -s -- eli15 --enforce
set -euo pipefail

CLAUDE_DIR="${CLAUDE_DIR:-$HOME/.claude}"
RAW="${RAW:-https://raw.githubusercontent.com/smixs/awesome-claude-output-styles/main}"

STYLES=(
  wait-what plain-english eli15 analogy-engine feynman thing-explainer ladder
  executive smart-brevity coach
  caveman adhd no-slop no-ai-slop unslop
  street gen-z sportscaster yoda bedtime-story
)

usage() {
  echo "Usage: install.sh <style> [<style>...] [--enforce] | --all [--enforce] | --list"
  echo "Styles: ${STYLES[*]} style-maker style-command"
  echo "--enforce: install a per-turn reminder hook so the style never fades"
  echo "style-command: install /style, a picker for switching styles"
  echo "--body <style>: print the style body to stdout, for other agents"
}

fetch() { # fetch <remote-path> <local-path>
  if [ -f "$SELF_DIR/$1" ]; then
    mkdir -p "$(dirname "$2")"
    cp "$SELF_DIR/$1" "$2"
  else
    curl -fsSL "$RAW/$1" -o "$2"
  fi
}

install_style() { # install_style <name>
  mkdir -p "$CLAUDE_DIR/output-styles"
  fetch "output-styles/$1.md" "$CLAUDE_DIR/output-styles/$1.md"
  echo "installed: output-styles/$1.md"
}

install_style_maker() {
  mkdir -p "$CLAUDE_DIR/skills/style-maker"
  fetch "skills/style-maker/SKILL.md" "$CLAUDE_DIR/skills/style-maker/SKILL.md"
  echo "installed: skills/style-maker (run it by asking Claude to \"make my output style\")"
}

# Print a style body (the file minus its YAML frontmatter) to stdout, for agents
# that read plain instruction files instead of Claude Code output styles.
print_body() { # print_body <name>
  local tmp
  tmp="$(mktemp)"
  fetch "output-styles/$1.md" "$tmp"
  awk 'body {print} /^---$/ {n++; if (n == 2) body = 1}' "$tmp"
  rm -f "$tmp"
}

install_slash_command() {
  mkdir -p "$CLAUDE_DIR/commands"
  fetch "commands/style.md" "$CLAUDE_DIR/commands/style.md"
  echo "installed: commands/style.md (switch styles with /style)"
}

install_enforce_hook() {
  mkdir -p "$CLAUDE_DIR/hooks"
  fetch "hooks/style-reminder.sh" "$CLAUDE_DIR/hooks/style-reminder.sh"
  chmod +x "$CLAUDE_DIR/hooks/style-reminder.sh"
  local settings="$CLAUDE_DIR/settings.json"
  if command -v python3 >/dev/null 2>&1; then
    python3 - "$settings" "$CLAUDE_DIR/hooks/style-reminder.sh" <<'PY'
import json, os, sys
path, cmd = sys.argv[1], sys.argv[2]
data = {}
if os.path.exists(path):
    with open(path) as f:
        data = json.load(f)
hooks = data.setdefault("hooks", {}).setdefault("UserPromptSubmit", [])
entry = {"hooks": [{"type": "command", "command": cmd}]}
if not any(h.get("command") == cmd
           for group in hooks for h in group.get("hooks", [])):
    hooks.append(entry)
with open(path, "w") as f:
    json.dump(data, f, indent=2, ensure_ascii=False)
    f.write("\n")
PY
    echo "installed: hooks/style-reminder.sh (per-turn style reminder, registered in $settings)"
  else
    echo "python3 not found. Register the hook manually in $settings:"
    echo '  {"hooks":{"UserPromptSubmit":[{"hooks":[{"type":"command","command":"'"$CLAUDE_DIR"'/hooks/style-reminder.sh"}]}]}}'
  fi
}

activate() { # activate <style-file>
  local file="$CLAUDE_DIR/output-styles/$1.md"
  local style_name
  style_name=$(sed -n 's/^name:[[:space:]]*//p' "$file" | head -1)
  [ -n "$style_name" ] || { echo "warning: no name in frontmatter, skipping activation"; return; }
  local settings="$CLAUDE_DIR/settings.json"
  if command -v python3 >/dev/null 2>&1; then
    python3 - "$settings" "$style_name" <<'PY'
import json, os, sys
path, name = sys.argv[1], sys.argv[2]
data = {}
if os.path.exists(path):
    with open(path) as f:
        data = json.load(f)
data["outputStyle"] = name
with open(path, "w") as f:
    json.dump(data, f, indent=2, ensure_ascii=False)
    f.write("\n")
PY
    echo "active style: $style_name (written to $settings)"
    echo "Restart Claude Code (or /clear) to apply."
  else
    echo "python3 not found. Activate manually: /config -> Output style -> $style_name"
  fi
}

# When run from a local checkout, install from local files instead of curl.
SELF_DIR="$(cd "$(dirname "${BASH_SOURCE[0]:-/dev/null}")" 2>/dev/null && pwd || echo /nonexistent)"
[ -f "$SELF_DIR/output-styles/${STYLES[0]}.md" ] || SELF_DIR=/nonexistent

[ $# -ge 1 ] || { usage; exit 1; }

ENFORCE=0
args=()
for a in "$@"; do
  if [ "$a" = "--enforce" ]; then ENFORCE=1; else args+=("$a"); fi
done
set -- ${args[@]+"${args[@]}"}
[ $# -ge 1 ] || { usage; exit 1; }

case "$1" in
  --list)
    printf '%s\n' "${STYLES[@]}" style-maker style-command
    exit 0
    ;;
  --body)
    [ $# -eq 2 ] || { echo "Usage: install.sh --body <style>"; exit 1; }
    print_body "$2"
    exit 0
    ;;
  --all)
    for s in "${STYLES[@]}"; do install_style "$s"; done
    install_style_maker
    install_slash_command
    [ "$ENFORCE" = 1 ] && install_enforce_hook
    echo
    echo "All styles installed. Pick one with /style, or /config -> Output style (takes effect after restart or /clear)."
    exit 0
    ;;
  -h|--help)
    usage
    exit 0
    ;;
esac

requested=("$@")
for s in "${requested[@]}"; do
  if [ "$s" = "style-maker" ]; then
    install_style_maker
    continue
  fi
  if [ "$s" = "style-command" ]; then
    install_slash_command
    continue
  fi
  found=0
  for known in "${STYLES[@]}"; do
    [ "$s" = "$known" ] && found=1 && break
  done
  [ "$found" = 1 ] || { echo "unknown style: $s"; usage; exit 1; }
  install_style "$s"
done

[ "$ENFORCE" = 1 ] && install_enforce_hook

# Exactly one style requested -> make it the active style.
if [ ${#requested[@]} -eq 1 ] && [ "${requested[0]}" != "style-maker" ] \
   && [ "${requested[0]}" != "style-command" ]; then
  activate "${requested[0]}"
else
  echo
  echo "Pick a style: /config -> Output style (takes effect after restart or /clear)."
fi