dotfiles/install.sh

#!/usr/bin/env bash
# install.sh: Install dependencies and stow all packages.
# Run this after cloning on a new machine.

set -euo pipefail

DOTFILES_DIR="$(cd "$(dirname "$0")" && pwd)"

# ---------------------------------------------------------------------------
# Detect Linux distribution
# ---------------------------------------------------------------------------

detect_distro() {
  if [[ -f /etc/os-release ]]; then
    . /etc/os-release
    echo "$ID"
  else
    echo "unknown"
  fi
}

DISTRO=$(detect_distro)

# ---------------------------------------------------------------------------
# Dependencies
# ---------------------------------------------------------------------------

case "$DISTRO" in
  endeavouros)
    # EndeavourOS is Arch-based, use pacman
    PACMAN_PACKAGES=(
      stow
      # prompt
      starship
      # sway ecosystem
      swayidle
      # bar & notifications
      mako
      libnotify        # notify-send
      # terminal & launcher
      fuzzel
      # audio & display
      wireplumber      # wpctl
      pipewire-pulse   # pactl
      brightnessctl
      alsa-utils       # speaker-test
      # screen recording
      wf-recorder
      slurp
      zenity
      ffmpeg
      # misc
      jq
      firefox
    )

    echo "Installing packages for EndeavourOS..."
    sudo pacman -S --noconfirm "${PACMAN_PACKAGES[@]}"
    ;;
  
  fedora)
    DNF_PACKAGES=(
      stow
      # prompt
      starship
      # sway ecosystem
      swayidle
      # bar & notifications
      mako
      libnotify        # notify-send
      # terminal & launcher
      fuzzel
      # audio & display
      wireplumber      # wpctl
      pipewire-utils   # pactl
      brightnessctl
      alsa-utils       # speaker-test
      # screen recording
      wf-recorder
      slurp
      zenity
      ffmpeg           # needs RPM Fusion: https://rpmfusion.org
      # misc
      jq
      firefox
    )
    
    echo "Installing packages for Fedora..."
    sudo dnf install -y "${DNF_PACKAGES[@]}"
    ;;
  
  arch)
    PACMAN_PACKAGES=(
      stow
      # prompt
      starship
      # sway ecosystem
      swayidle
      # bar & notifications
      mako
      libnotify        # notify-send
      # terminal & launcher
      fuzzel
      # audio & display
      wireplumber      # wpctl
      pipewire-pulse   # pactl
      brightnessctl
      alsa-utils       # speaker-test
      # screen recording
      wf-recorder
      slurp
      zenity
      ffmpeg
      # misc
      jq
      firefox
    )
    
    echo "Installing packages for Arch Linux..."
    sudo pacman -S --noconfirm "${PACMAN_PACKAGES[@]}"
    ;;
  
  *)
    echo "ERROR: Unsupported distribution: $DISTRO"
    echo "Supported distributions: fedora, arch, endeavouros"
    exit 1
    ;;
esac

# ---------------------------------------------------------------------------
# Packages not in Fedora repos — install manually if missing
# ---------------------------------------------------------------------------

warn_missing() {
  local cmd="$1" msg="$2"
  if ! command -v "$cmd" &>/dev/null; then
    echo "WARNING: '$cmd' not found — $msg"
  fi
}

warn_missing autotiling  "install via pip: pip install --user autotiling"
warn_missing eww         "install from https://github.com/elkowar/eww/releases"
warn_missing wezterm     "install from https://wezfurlong.org/wezterm/install/linux.html"

# ---------------------------------------------------------------------------
# NPM packages
# ---------------------------------------------------------------------------

echo ""
echo "Installing npm packages..."
npm install -g @mariozechner/pi-coding-agent

# ---------------------------------------------------------------------------
# Fonts
# ---------------------------------------------------------------------------

FONT_DIR="$HOME/.local/share/fonts/JetBrainsMono"
if [[ ! -d "$FONT_DIR" ]]; then
  echo ""
  echo "Installing JetBrains Mono Nerd Font..."
  TMP="$(mktemp -d)"
  curl -fLo "$TMP/JetBrainsMono.zip" \
    "https://github.com/ryanoasis/nerd-fonts/releases/latest/download/JetBrainsMono.zip"
  mkdir -p "$FONT_DIR"
  unzip -q "$TMP/JetBrainsMono.zip" -d "$FONT_DIR"
  rm -rf "$TMP"
  fc-cache -f "$FONT_DIR"
  echo "Font installed."
else
  echo "JetBrains Mono Nerd Font already installed, skipping."
fi

# ---------------------------------------------------------------------------
# Configure shell
# ---------------------------------------------------------------------------

echo ""
echo "Configuring zsh..."

# Ensure .zshrc exists
if [[ ! -f "$HOME/.zshrc" ]]; then
  touch "$HOME/.zshrc"
fi

# Ensure starship is initialized in .zshrc
if ! grep -q 'eval "$(starship init zsh)"' "$HOME/.zshrc"; then
  echo 'eval "$(starship init zsh)"' >> "$HOME/.zshrc"
  echo "  Added starship initialization to ~/.zshrc"
else
  echo "  Starship already configured in ~/.zshrc"
fi

# ---------------------------------------------------------------------------
# Stow all packages
# ---------------------------------------------------------------------------

echo ""
echo "Stowing dotfiles..."

# Check for conflicts before stowing
CONFLICTS=()
for PACKAGE in "$DOTFILES_DIR"/*/; do
  PACKAGE="$(basename "$PACKAGE")"
  
  # Skip hidden directories and .git
  [[ "$PACKAGE" == .* ]] && continue
  
  # Check if package has files that would conflict
  while IFS= read -r -d '' dotfile; do
    # Get the target path this would create
    TARGET_PATH="${HOME}/${dotfile#$DOTFILES_DIR/$PACKAGE/}"
    
    # If target exists and is NOT a symlink to the correct location, it's a conflict
    if [[ -e "$TARGET_PATH" ]] && [[ ! -L "$TARGET_PATH" ]]; then
      CONFLICTS+=("$PACKAGE (file: $TARGET_PATH)")
    fi
  done < <(find "$DOTFILES_DIR/$PACKAGE" -type f -print0 2>/dev/null || true)
done

if [[ ${#CONFLICTS[@]} -gt 0 ]]; then
  echo "WARNING: Found conflicts with existing configs:"
  for conflict in "${CONFLICTS[@]}"; do
    echo "  ✗ $conflict"
  done
  echo ""
fi

# Stow all packages
for PACKAGE in "$DOTFILES_DIR"/*/; do
  PACKAGE="$(basename "$PACKAGE")"
  
  # Skip hidden directories
  [[ "$PACKAGE" == .* ]] && continue
  
  echo "  $PACKAGE"
  stow --dir="$DOTFILES_DIR" --target="$HOME" --restow "$PACKAGE"
done

echo "Done."