#!/bin/bash
set -euo pipefail

for arg in "$@"; do
  # allows: ./install.sh macos   (sets $macos=1)
  declare "${arg}=1"
done

if [[ -n "${macos:-}" ]]; then
  linux=""
else
  linux=1
fi

install_neovim() {
  if command -v nvim >/dev/null 2>&1; then
    return 0
  fi

  if [[ -z "${linux:-}" ]]; then
    # macOS
    if command -v brew >/dev/null 2>&1; then
      brew install neovim
      return 0
    fi
    echo "Neovim not found and Homebrew is not installed. Install Homebrew or install Neovim manually." >&2
    return 1
  fi

  # Linux: curl official tarball, install under /opt, symlink to /usr/local/bin (no AppImage)
  if ! command -v curl >/dev/null 2>&1; then
    echo "curl is required to install Neovim on Linux." >&2
    return 1
  fi
  if ! command -v tar >/dev/null 2>&1; then
    echo "tar is required to install Neovim on Linux." >&2
    return 1
  fi

  arch="$(uname -m)"
  case "$arch" in
    x86_64) nvim_arch="linux-x86_64" ;;
    aarch64|arm64) nvim_arch="linux-arm64" ;;
    *)
      echo "Unsupported architecture for Neovim tarball install: $arch" >&2
      return 1
      ;;
  esac

  tmpdir="$(mktemp -d)"
  trap 'rm -rf "$tmpdir"' EXIT

  url="https://github.com/neovim/neovim/releases/latest/download/nvim-${nvim_arch}.tar.gz"
  curl -fsSL "$url" -o "$tmpdir/nvim.tar.gz"
  tar -xzf "$tmpdir/nvim.tar.gz" -C "$tmpdir"

  src_dir="$tmpdir/nvim-${nvim_arch}"

  # Install to /opt/nvim (replace if present), then symlink binary
  sudo rm -rf /opt/nvim
  sudo mv "$src_dir" /opt/nvim
  sudo ln -sf /opt/nvim/bin/nvim /usr/local/bin/nvim
}

mkdir -p "$HOME/.config"

# install neovim if missing
install_neovim

# install neovim config
if [ ! -d "$HOME/.config/nvim" ]; then
  ln -s "$HOME/dotfiles/nvim" "$HOME/.config/nvim" || :
fi

# install starship config
ln -s "$HOME/dotfiles/starship.toml" "$HOME/.config/starship.toml" || :

if [ ! -n "${linux:-}" ]; then
  # macos-only configs
  ln -s "$HOME/dotfiles/aerospace.toml" "$HOME/.aerospace.toml" || :

  if [ ! -d "$HOME/.config/sketchybar" ]; then
    ln -s "$HOME/dotfiles/sketchybar" "$HOME/.config/sketchybar" || :
  fi
  if [ ! -d "$HOME/.config/borders" ]; then
    ln -s "$HOME/dotfiles/borders" "$HOME/.config/borders" || :
  fi
  if [ ! -d "$HOME/.config/neovide" ]; then
    ln -s "$HOME/dotfiles/neovide" "$HOME/.config/neovide" || :
  fi
  if [ ! -d "$HOME/.config/ghostty" ]; then
    ln -s "$HOME/dotfiles/ghostty" "$HOME/.config/ghostty" || :
  fi
else
  # linux-only configs
  if [ ! -d "$HOME/.config/sway" ]; then
    ln -s "$HOME/dotfiles/sway" "$HOME/.config/sway" || :
  fi
  if [ ! -d "$HOME/.config/waybar" ]; then
    ln -s "$HOME/dotfiles/waybar" "$HOME/.config/waybar" || :
  fi
  if [ ! -d "$HOME/.config/rofi" ]; then
    ln -s "$HOME/dotfiles/rofi" "$HOME/.config/rofi" || :
  fi
  if [ ! -d "$HOME/.config/dunst" ]; then
    ln -s "$HOME/dotfiles/dunst" "$HOME/.config/dunst" || :
  fi
fi