#!/bin/sh
set -eu

REPO="fabro-sh/fabro"

# Colors (only when stderr is a terminal)
if [ -t 2 ]; then
  RED='\033[0;31m'
  GREEN='\033[0;32m'
  DIM='\033[2m'
  BOLD='\033[1m'
  BOLD_CYAN='\033[1;36m'
  RESET='\033[0m'
else
  RED=''
  GREEN=''
  DIM=''
  BOLD=''
  BOLD_CYAN=''
  RESET=''
fi

info()    { printf "  %b\n" "$1" >&2; }
step()    { printf "  ${BOLD}%b${RESET}\n" "$1" >&2; }
dim()     { printf "  ${DIM}%b${RESET}\n" "$1" >&2; }
success() { printf "  ${GREEN}✔${RESET} %b\n" "$1" >&2; }
error()   { printf "  ${RED}✗ %b${RESET}\n" "$1" >&2; exit 1; }

# --- Header ---
printf "\n  ⚒️  ${BOLD}Fabro Install${RESET}\n\n" >&2

# --- Require gh CLI ---
if ! command -v gh >/dev/null 2>&1; then
  error "gh CLI is required but not installed. Install it from ${BOLD_CYAN}https://cli.github.com${RESET}"
fi

# --- Detect platform ---
OS="$(uname -s)"
ARCH="$(uname -m)"

case "$OS" in
  Darwin)
    # Detect Rosetta translation
    if [ "$ARCH" = "x86_64" ]; then
      if sysctl -n sysctl.proc_translated 2>/dev/null | grep -q 1; then
        ARCH="arm64"
      fi
    fi
    case "$ARCH" in
      arm64) TARGET="aarch64-apple-darwin" ;;
      *)     error "Unsupported macOS architecture: $ARCH. Supported: Apple Silicon (arm64)" ;;
    esac
    ;;
  Linux)
    case "$ARCH" in
      x86_64)  TARGET="x86_64-unknown-linux-gnu" ;;
      aarch64) TARGET="aarch64-unknown-linux-gnu" ;;
      *)       error "Unsupported Linux architecture: $ARCH. Supported: x86_64, aarch64" ;;
    esac
    if ldd --version 2>&1 | grep -qi musl; then
      TARGET="${TARGET%-gnu}-musl"
    fi
    ;;
  *)
    error "Unsupported OS: $OS. Supported platforms: macOS (Apple Silicon), Linux (x86_64, aarch64)"
    ;;
esac

ASSET="fabro-${TARGET}.tar.gz"
TMPDIR="$(mktemp -d)"
trap 'rm -rf "$TMPDIR"' EXIT

TAG="$(gh api "repos/${REPO}/releases/latest" --jq '.tag_name')"
if [ -z "$TAG" ]; then
  error "Could not resolve the latest stable release tag"
fi

dim "Downloading fabro for ${TARGET}..."
gh release download "$TAG" --repo "$REPO" --pattern "$ASSET" --dir "$TMPDIR" --clobber

dim "Extracting..."
tar xzf "${TMPDIR}/${ASSET}" -C "$TMPDIR"

# --- Install binary ---
INSTALL_DIR="${FABRO_INSTALL_DIR:-$HOME/.fabro/bin}"
mkdir -p "$INSTALL_DIR"
mv "${TMPDIR}/fabro-${TARGET}/fabro" "${INSTALL_DIR}/fabro"

chmod +x "${INSTALL_DIR}/fabro"

# --- Verify ---
VERSION="$("${INSTALL_DIR}/fabro" --version 2>/dev/null || true)"
if [ -z "$VERSION" ]; then
  error "Installation failed: could not run fabro --version"
fi

tildify() {
  if [ "${1#"$HOME"/}" != "$1" ]; then
    echo "~/${1#"$HOME"/}"
  else
    echo "$1"
  fi
}

success "Installed ${VERSION} to ${BOLD_CYAN}$(tildify "${INSTALL_DIR}/fabro")${RESET}"

# --- Ensure install dir is on PATH ---
if command -v fabro >/dev/null 2>&1; then
  dim "fabro is already on \$PATH, skipping shell config"
else
  tilde_bin_dir=$(tildify "$INSTALL_DIR")
  echo "" >&2

  if [ -t 2 ] && [ -e /dev/tty ]; then
    case $(basename "${SHELL:-sh}") in
    zsh)
      : "${ZDOTDIR:="$HOME"}"
      shell_config="${ZDOTDIR%/}/.zshrc"
      {
        printf '\n# fabro\n'
        echo "export PATH=\"$INSTALL_DIR:\$PATH\""
      } >>"$shell_config"
      info "Added ${BOLD_CYAN}${tilde_bin_dir}${RESET} to \$PATH in ${BOLD_CYAN}$(tildify "$shell_config")${RESET}"
      ;;
    bash)
      shell_config="$HOME/.bashrc"
      if [ -f "$HOME/.bash_profile" ]; then
        shell_config="$HOME/.bash_profile"
      fi
      {
        printf '\n# fabro\n'
        echo "export PATH=\"$INSTALL_DIR:\$PATH\""
      } >>"$shell_config"
      info "Added ${BOLD_CYAN}${tilde_bin_dir}${RESET} to \$PATH in ${BOLD_CYAN}$(tildify "$shell_config")${RESET}"
      ;;
    fish)
      fish_config="$HOME/.config/fish/config.fish"
      mkdir -p "$(dirname "$fish_config")"
      {
        printf '\n# fabro\n'
        echo "fish_add_path $INSTALL_DIR"
      } >>"$fish_config"
      info "Added ${BOLD_CYAN}${tilde_bin_dir}${RESET} to \$PATH in ${BOLD_CYAN}$(tildify "$fish_config")${RESET}"
      ;;
    *)
      info "Add ${BOLD_CYAN}${tilde_bin_dir}${RESET} to your PATH:"
      echo "" >&2
      info "  ${BOLD}export PATH=\"${INSTALL_DIR}:\$PATH\"${RESET}"
      ;;
    esac
  else
    info "Add ${BOLD_CYAN}${tilde_bin_dir}${RESET} to your PATH:"
    echo "" >&2
    info "  ${BOLD}export PATH=\"${INSTALL_DIR}:\$PATH\"${RESET}"
  fi

  export PATH="${INSTALL_DIR}:$PATH"
fi
echo "" >&2

# --- Prompt to run setup wizard ---
if [ -t 2 ] && [ -e /dev/tty ]; then
  printf "  ${BOLD}Run ${BOLD_CYAN}fabro install${RESET}${BOLD} now to complete setup? [Y/n]${RESET} " >&2
  read -r answer </dev/tty
  case "$answer" in
    [nN]*) dim "Skipping. Run ${BOLD_CYAN}fabro install${RESET}${DIM} whenever you're ready." ;;
    *)     echo "" >&2; exec "${INSTALL_DIR}/fabro" install ;;
  esac
else
  info "Run ${BOLD_CYAN}fabro install${RESET} to complete setup."
fi
