install.sh

#!/usr/bin/env bash

set -e

# Colors and styling via gum
BOLD="\033[1m"
RESET="\033[0m"

# ─────────────────────────────────────────────────────────────────────────────
# Helpers
# ─────────────────────────────────────────────────────────────────────────────

ensure_homebrew() {
  if ! command -v brew &>/dev/null; then
    echo "Installing Homebrew..."
    /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
    eval "$(/opt/homebrew/bin/brew shellenv)"
    echo "Homebrew installed."
  else
    echo "Homebrew already installed."
  fi
}

ensure_gum() {
  if ! command -v gum &>/dev/null; then
    echo "Installing gum..."
    brew install gum
  fi
}

# ─────────────────────────────────────────────────────────────────────────────
# Installation steps
# ─────────────────────────────────────────────────────────────────────────────

step_dotfiles() {
  gum style --border normal --padding "0 1" --foreground 4 "Copying dotfiles to $HOME"

  local files=(
    ".aliases"
    ".exports"
    ".gitignore"
    ".zshrc"
  )

  local dirs=(
    ".config/ghostty"
    ".config/nvim"
    ".config/zellij"
    ".config/nchat"
    ".gnupg"
  )

  for file in "${files[@]}"; do
    if [[ -f "$file" ]]; then
      cp "$file" "$HOME/$file"
      gum style --foreground 2 "  copied $file"
    fi
  done

  for dir in "${dirs[@]}"; do
    if [[ -d "$dir" ]]; then
      mkdir -p "$HOME/$dir"
      cp -r "$dir"/* "$HOME/$dir/" 2>/dev/null || true
      gum style --foreground 2 "  copied $dir/"
    fi
  done
}

step_git_config() {
  gum style --border normal --padding "0 1" --foreground 4 "Git configuration"

  # Copy .gitconfig base template if it exists
  if [[ -f ".gitconfig" ]]; then
    cp ".gitconfig" "$HOME/.gitconfig"
    gum style --foreground 2 "  copied .gitconfig"
  fi

  local existing_name existing_email
  existing_name=$(git config --global user.name 2>/dev/null || true)
  existing_email=$(git config --global user.email 2>/dev/null || true)

  local git_name="$existing_name"
  local git_email="$existing_email"

  if [[ -n "$existing_name" && -n "$existing_email" ]]; then
    gum style --foreground 3 "  Current: $existing_name <$existing_email>"
    if gum confirm "Change git user/email?"; then
      git_name=$(gum input --placeholder "Your Git user.name" --prompt "Name: " --value "$existing_name")
      git_email=$(gum input --placeholder "Your Git user.email" --prompt "Email: " --value "$existing_email")
    fi
  else
    git_name=$(gum input --placeholder "Your Git user.name" --prompt "Name: ")
    git_email=$(gum input --placeholder "Your Git user.email" --prompt "Email: ")
  fi

  git config --global user.name "$git_name"
  git config --global user.email "$git_email"

  if gum confirm "Set up GPG signing for commits?"; then
    if ! command -v gpg &>/dev/null; then
      gum style --foreground 3 "  gpg not found. Installing via Homebrew..."
      brew install gnupg
    fi

    local gpg_key
    gpg_key=$(gpg --list-secret-keys --keyid-format=long 2>/dev/null \
      | grep -E 'sec\s+rsa[0-9]+/' | sed 's|.*/||' | awk '{print $1}' | head -1 || true)

    if [[ -z "$gpg_key" ]]; then
      gum style --foreground 3 "  No GPG key found. Generating a new one..."

      local key_name key_email
      key_name=$(gum input --placeholder "Full name for GPG key" --prompt "Name: " --value "$git_name")
      key_email=$(gum input --placeholder "Email for GPG key" --prompt "Email: " --value "$git_email")

      gpg --batch --gen-key <<GPGEOF
%no-protection
Key-Type: RSA
Key-Length: 4096
Subkey-Type: RSA
Subkey-Length: 4096
Name-Real: $key_name
Name-Email: $key_email
Expire-Date: 0
%commit
GPGEOF

      gpg_key=$(gpg --list-secret-keys --keyid-format=long 2>/dev/null \
        | grep -E 'sec\s+rsa4096/' | sed 's|.*/||' | awk '{print $1}' | head -1)
      gum style --foreground 2 "  GPG key $gpg_key generated."

      # Offer to add a second email UID
      if gum confirm "Add another email to this GPG key?"; then
        local extra_email
        extra_email=$(gum input --placeholder "Additional email" --prompt "Email: ")
        local fingerprint
        fingerprint=$(gpg --list-secret-keys --keyid-format=long 2>/dev/null \
          | grep -A1 "^sec" | tail -1 | tr -d ' ')
        gpg --quick-add-uid "$fingerprint" "$key_name <$extra_email>"
        gum style --foreground 2 "  Added $extra_email to GPG key."
      fi

      # Copy public key to clipboard for GitHub
      gpg --armor --export "$gpg_key" | pbcopy
      gum style --foreground 3 "  Public key copied to clipboard — add it to GitHub (Settings > SSH and GPG keys)."
      gum confirm "Press enter once you've added the key to GitHub" --affirmative "Done" --negative ""
    else
      gum style --foreground 2 "  Found existing GPG key: $gpg_key"
    fi

    git config --global user.signingkey "$gpg_key"
    git config --global commit.gpgsign true
    gum style --foreground 2 "  GPG signing enabled with key $gpg_key"
  fi

  gum style --foreground 2 "  Git configured as $git_name <$git_email>"
}

step_brew_cli() {
  gum style --border normal --padding "0 1" --foreground 4 "Installing CLI tools via Homebrew"
  gum spin --spinner dot --title "Installing CLI tools..." -- bash brew.sh
  gum style --foreground 2 "  CLI tools installed."
}

step_brew_cask() {
  gum style --border normal --padding "0 1" --foreground 4 "Installing GUI apps via Homebrew Cask"
  gum spin --spinner dot --title "Installing GUI apps..." -- bash cask.sh
  gum style --foreground 2 "  GUI apps installed."
}

step_oh_my_zsh() {
  gum style --border normal --padding "0 1" --foreground 4 "Installing Oh My Zsh"

  if [[ -d "$HOME/.oh-my-zsh" ]]; then
    gum style --foreground 3 "  Oh My Zsh already installed, skipping."
  else
    gum spin --spinner dot --title "Installing Oh My Zsh..." -- \
      sh -c "$(curl -fsSL https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh)" "" --unattended
    gum style --foreground 2 "  Oh My Zsh installed."
  fi

  # Spaceship prompt
  local spaceship_dir="${ZSH_CUSTOM:-$HOME/.oh-my-zsh/custom}/themes/spaceship-prompt"
  if [[ -d "$spaceship_dir" ]]; then
    gum style --foreground 3 "  Spaceship prompt already installed, skipping."
  else
    gum spin --spinner dot --title "Installing Spaceship prompt..." -- \
      git clone https://github.com/spaceship-prompt/spaceship-prompt.git "$spaceship_dir" --depth=1
    ln -sf "$spaceship_dir/spaceship.zsh-theme" "${ZSH_CUSTOM:-$HOME/.oh-my-zsh/custom}/themes/spaceship.zsh-theme"
    gum style --foreground 2 "  Spaceship prompt installed."
  fi
}

step_languages() {
  gum style --border normal --padding "0 1" --foreground 4 "Installing programming languages via asdf"

  if ! command -v asdf &>/dev/null; then
    gum spin --spinner dot --title "Installing asdf..." -- brew install asdf
    gum style --foreground 2 "  asdf installed."
  fi

  local languages
  languages=$(gum choose --no-limit --selected "golang,nodejs,elixir,rust" \
    "golang" \
    "nodejs" \
    "elixir" \
    "rust")

  while IFS= read -r lang; do
    [[ -z "$lang" ]] && continue

    if asdf plugin list 2>/dev/null | grep -q "^${lang}$"; then
      gum style --foreground 3 "  $lang plugin already added."
    else
      gum spin --spinner dot --title "Adding asdf plugin $lang..." -- \
        asdf plugin add "$lang"
      gum style --foreground 2 "  $lang plugin added."
    fi

    gum spin --spinner dot --title "Installing latest $lang..." -- \
      asdf install "$lang" latest
    asdf set --home "$lang" latest
    gum style --foreground 2 "  $lang latest installed and set as global default."
  done <<< "$languages"
}

step_claude_code() {
  gum style --border normal --padding "0 1" --foreground 4 "Configuring Claude Code"

  if ! command -v claude &>/dev/null; then
    gum style --foreground 3 "  Claude Code not found. Installing..."
    gum spin --spinner dot --title "Installing Claude Code..." -- \
      bash -c 'curl -fsSL https://claude.ai/install.sh | bash'
  fi

  mkdir -p "$HOME/.claude"

  if [[ -f "$HOME/.claude/settings.json" ]]; then
    # Merge hooks into existing settings using a temp file
    local tmp
    tmp=$(mktemp)
    # Use python3 (ships with macOS) to merge hooks from dotfiles into existing settings
    python3 -c "
import json, sys
with open('$HOME/.claude/settings.json') as f: existing = json.load(f)
with open('claude-code/settings.json') as f: dotfiles = json.load(f)
existing['hooks'] = dotfiles.get('hooks', {})
with open('$tmp', 'w') as f: json.dump(existing, f, indent=2, ensure_ascii=False)
print('merged')
" && mv "$tmp" "$HOME/.claude/settings.json"
    gum style --foreground 2 "  Merged notification hooks into existing settings."
  else
    cp claude-code/settings.json "$HOME/.claude/settings.json"
    gum style --foreground 2 "  Copied Claude Code settings."
  fi
}

step_macos() {
  gum style --border normal --padding "0 1" --foreground 4 "Applying macOS preferences"

  if gum confirm "This will change macOS system preferences. Continue?"; then
    bash osx.sh
    gum style --foreground 2 "  macOS preferences applied."
  else
    gum style --foreground 3 "  Skipped macOS preferences."
  fi
}

# ─────────────────────────────────────────────────────────────────────────────
# Main
# ─────────────────────────────────────────────────────────────────────────────

main() {
  ensure_homebrew
  ensure_gum

  gum style \
    --border double \
    --align center \
    --padding "1 3" \
    --foreground 4 \
    --bold \
    "Dotfiles Installer" \
    "github.com/vitorleal/dotfiles"

  echo ""

  local steps
  steps=$(gum choose --no-limit --selected "Dotfiles,Git Config,CLI Tools,GUI Apps,Oh My Zsh,Languages (asdf),Claude Code,macOS Preferences" \
    "Dotfiles" \
    "Git Config" \
    "CLI Tools" \
    "GUI Apps" \
    "Oh My Zsh" \
    "Languages (asdf)" \
    "Claude Code" \
    "macOS Preferences")

  echo ""

  while IFS= read -r step <&3; do
    case "$step" in
      "Dotfiles")           step_dotfiles ;;
      "Git Config")         step_git_config ;;
      "CLI Tools")          step_brew_cli ;;
      "GUI Apps")           step_brew_cask ;;
      "Oh My Zsh")          step_oh_my_zsh ;;
      "Languages (asdf)")   step_languages ;;
      "Claude Code")        step_claude_code ;;
      "macOS Preferences")  step_macos ;;
    esac
    echo ""
  done 3<<< "$steps"

  # Offer to re-run setup scripts
  if gum confirm "Re-run any setup scripts (brew, cask, macOS)?"; then
    local rerun
    rerun=$(gum choose --no-limit \
      "CLI Tools (brew.sh)" \
      "GUI Apps (cask.sh)" \
      "macOS Preferences (osx.sh)")

    while IFS= read -r choice <&4; do
      case "$choice" in
        "CLI Tools (brew.sh)")          step_brew_cli ;;
        "GUI Apps (cask.sh)")           step_brew_cask ;;
        "macOS Preferences (osx.sh)")   step_macos ;;
      esac
    done 4<<< "$rerun"
  fi

  gum style \
    --border double \
    --align center \
    --padding "1 3" \
    --foreground 2 \
    --bold \
    "All done!" \
    "Restart your terminal to apply changes."
}

main "$@"