#!/bin/sh
# consensor uninstaller for Claude Code. Reverses install.sh.
#
#   sh ~/.claude/consensor/uninstall.sh                    reverse consensor's changes
#   sh ~/.claude/consensor/uninstall.sh --restore-backups  put ~/.claude/settings.json and
#                                                          ~/.claude.json back byte-for-byte
#                                                          as they were before the first install
#
# Default behaviour:
#   - settings.json: if it has not changed since consensor last touched it, the
#     pre-install backup is restored byte-for-byte. If you have edited it since,
#     only the rules and status line consensor added are removed; your edits stay.
#   - ~/.claude.json: `claude mcp remove consensor --scope user`, only if consensor
#     added it. (This file holds live Claude Code state, so it is not rolled back
#     unless you pass --restore-backups.)
#   - Before anything changes, the current files are snapshotted to
#     ~/.claude/consensor/backups/<time>-pre-uninstall/. Backups are kept.
# Safe to run more than once.
# shellcheck shell=sh
set -eu
umask 077

CLAUDE_DIR="${HOME}/.claude"
DIR="${CLAUDE_DIR}/consensor"
STATE="${DIR}/state"
SETTINGS="${CLAUDE_DIR}/settings.json"
CLAUDE_JSON="${HOME}/.claude.json"

say() { printf '%s\n' "$*"; }
warn() { printf 'consensor: %s\n' "$*" >&2; }

restore_file() {
  # restore_file <backup dir> <live path>
  name=$(basename "$2")
  if [ -f "$1/${name}" ]; then
    cp -p "$1/${name}" "$2"
    say "  restored $2 from backup"
  elif [ -f "$1/${name}.absent" ]; then
    rm -f "$2"
    say "  removed $2 (it did not exist before consensor)"
  else
    warn "no backup of $2 found in $1; left unchanged"
  fi
}

run_merge() {
  case "$INTERP" in
    node) node "${DIR}/lib/merge.cjs" "$@" ;;
    python) python3 "${DIR}/lib/merge.py" "$@" ;;
    *) return 2 ;;
  esac
}

main() {
  RESTORE=0
  for arg in "$@"; do
    case "$arg" in
      --restore-backups) RESTORE=1 ;;
      -h|--help) sed -n '2,20p' "$0"; exit 0 ;;
      *) warn "unknown option: $arg"; exit 2 ;;
    esac
  done

  if [ ! -d "$STATE" ] && [ ! -f "${DIR}/manifest.json" ]; then
    say "consensor is not installed (nothing to undo)."
    exit 0
  fi

  INTERP=$(cat "${DIR}/interp" 2>/dev/null || printf 'none')
  ORIG=""
  if [ -f "${DIR}/backups/ORIGINAL" ]; then
    ORIG="${DIR}/backups/$(cat "${DIR}/backups/ORIGINAL")"
  fi

  # Snapshot what is there now, before changing anything.
  snap="${DIR}/backups/$(date -u +%Y%m%dT%H%M%SZ)-$$-pre-uninstall"
  mkdir -p "$snap"
  for f in "$SETTINGS" "$CLAUDE_JSON"; do
    if [ -f "$f" ]; then cp -p "$f" "$snap/"; else : > "$snap/$(basename "$f").absent"; fi
  done
  say "Snapshot of current files: ${snap}"

  # settings.json
  unchanged=0
  if [ -f "${STATE}/settings.post.cksum" ] && [ -f "$SETTINGS" ]; then
    if [ "$(cksum < "$SETTINGS")" = "$(cat "${STATE}/settings.post.cksum")" ]; then unchanged=1; fi
  fi
  if [ -n "$ORIG" ] && { [ "$RESTORE" -eq 1 ] || [ "$unchanged" -eq 1 ]; }; then
    restore_file "$ORIG" "$SETTINGS"
  elif [ -f "$SETTINGS" ]; then
    out=""
    if out=$(run_merge revert "$SETTINGS" "${DIR}/manifest.json"); then
      say "  removed consensor's rules and status line from ${SETTINGS}; your other settings were kept"
      if [ "$out" = "EMPTY" ] && [ -n "$ORIG" ] && [ -f "${ORIG}/settings.json.absent" ]; then
        rm -f "$SETTINGS"
      fi
    else
      warn "could not edit ${SETTINGS}; restore it from ${ORIG:-the backups folder} if needed"
    fi
  fi

  # ~/.claude.json (the connector)
  if [ "$RESTORE" -eq 1 ] && [ -n "$ORIG" ]; then
    restore_file "$ORIG" "$CLAUDE_JSON"
  elif [ -f "${STATE}/mcp_added" ]; then
    if command -v claude >/dev/null 2>&1; then
      if claude mcp remove consensor --scope user >/dev/null 2>&1; then
        say "  removed the consensor connector"
      else
        say "  the consensor connector was already gone"
      fi
    else
      warn "'claude' not found; remove the connector with: claude mcp remove consensor --scope user"
    fi
  else
    say "  the consensor connector was there before install; left in place"
  fi

  for cmd in csyntax ccents csync csensai; do
    if [ -f "${STATE}/command_${cmd}" ]; then
      rm -f "${CLAUDE_DIR}/commands/${cmd}.md" "${STATE}/command_${cmd}"
      say "  removed the /${cmd} command"
    fi
  done
  rm -f "${DIR}/statusline.sh" "${DIR}/install_id" "${DIR}/interp" "${DIR}/last_post" \
        "${DIR}/manifest.json" "${DIR}/backups/ORIGINAL" "${DIR}/lib/merge.cjs" "${DIR}/lib/merge.py"
  rm -f "${STATE}/mcp_added" "${STATE}/timing" "${STATE}/settings.post.cksum"
  rmdir "${DIR}/lib" "$STATE" 2>/dev/null || true
  rm -f "${DIR}/uninstall.sh"

  say ""
  say "consensor has been removed. Backups are kept in ${DIR}/backups/"
  say "Delete them whenever you like:  rm -r ~/.claude/consensor"
}

main "$@"