#!/bin/bash

# 02-dev-tools-setup.sh
# Installs various development tools and utilities.

echo "--- Starting Development Tools Setup ---"

# Setup NVIM
echo "Setting up Neovim..."
_nvim_asset_name() {
    case "$(uname -m)" in
        x86_64)  echo "nvim-linux-x86_64" ;;
        aarch64) echo "nvim-linux-arm64" ;;
        *) echo "Unsupported arch for Neovim: $(uname -m)" >&2; exit 1 ;;
    esac
}

NVIM_LATEST="$(curl -fsSL https://api.github.com/repos/neovim/neovim/releases/latest | grep '"tag_name"' | cut -d'"' -f4)"
NVIM_INSTALLED="$(nvim --version 2>/dev/null | head -n1 | awk '{print $2}')"

if [ "$NVIM_INSTALLED" = "$NVIM_LATEST" ]; then
    echo "Neovim ${NVIM_LATEST} already up-to-date."
else
    NVIM_ASSET="$(_nvim_asset_name)"
    echo "Installing Neovim ${NVIM_LATEST} (${NVIM_ASSET})..."
    TEMP_NVIM_DIR=$(mktemp -d)
    curl -fsSLo "${TEMP_NVIM_DIR}/${NVIM_ASSET}.tar.gz" \
        "https://github.com/neovim/neovim/releases/latest/download/${NVIM_ASSET}.tar.gz"
    sudo rm -rf /opt/nvim
    sudo tar -C /opt -xzf "${TEMP_NVIM_DIR}/${NVIM_ASSET}.tar.gz"
    sudo mv "/opt/${NVIM_ASSET}" /opt/nvim 2>/dev/null || true
    rm -rf "${TEMP_NVIM_DIR}"
    sudo ln -sf /opt/nvim/bin/nvim /usr/local/bin/nvim
    echo "Neovim ${NVIM_LATEST} installed to /opt/nvim, symlinked to /usr/local/bin/nvim"
fi

# pynvim
echo "Installing pynvim for Neovim Python support..."
if /usr/bin/python3 -m pip show pynvim &>/dev/null; then
    echo "pynvim already installed."
else
    /usr/bin/python3 -m pip install --user pynvim
    echo "pynvim installed for the current user."
fi

# Install Nerd Font (Hack)
echo "Installing Hack Nerd Font..."
NERDFONT_VERSION="v3.2.1" # Or use "latest" if API allows, else check manually
NERDFONT_NAME="Hack"
FONT_DIR="$HOME/.local/share/fonts"
mkdir -p "$FONT_DIR"

# Check if a Hack Nerd Font is already installed to avoid re-downloading
if fc-list | grep -qi "Hack Nerd Font"; then
    echo "Hack Nerd Font already installed."
else
    echo "Downloading and installing Hack Nerd Font..."
    TEMP_FONT_DIR=$(mktemp -d)
    wget -qO "${TEMP_FONT_DIR}/${NERDFONT_NAME}.zip" "https://github.com/ryanoasis/nerd-fonts/releases/download/${NERDFONT_VERSION}/${NERDFONT_NAME}.zip"
    if [ $? -eq 0 ]; then
        unzip -q "${TEMP_FONT_DIR}/${NERDFONT_NAME}.zip" -d "${TEMP_FONT_DIR}/${NERDFONT_NAME}NerdFont"
        # Copy only .ttf or .otf files
        find "${TEMP_FONT_DIR}/${NERDFONT_NAME}NerdFont" \( -name "*.ttf" -o -name "*.otf" \) -exec cp {} "$FONT_DIR/" \;
        echo "Updating font cache..."
        fc-cache -f -v
        echo "Hack Nerd Font installed."
    else
        echo "ERROR: Failed to download Hack Nerd Font."
    fi
    rm -rf "${TEMP_FONT_DIR}" # Clean up
fi

OS="$(uname | tr '[:upper:]' '[:lower:]')"
ARCH_RAW="$(uname -m)"
case "${ARCH_RAW}" in
	x86_64) ARCH=amd64 ;;
	aarch64|arm64) ARCH=arm64 ;;
	*) echo "Unsupported arch: ${ARCH_RAW}" >&2; exit 1 ;;
esac

_add_to_rc() {
  local rcfile="$1" snippet="$2"
  grep -qxF "${snippet}" "${rcfile}" 2>/dev/null || {
    echo "# added by 02-dev-tools-setup.sh" >> "${rcfile}"
    echo "${snippet}" >> "${rcfile}"
    echo "  → updated ${rcfile}"
  }
}

if ! command -v go &>/dev/null; then
  echo "➤ Fetching latest Go version..."
  LATEST_GO_FULL="$(curl -fsSL https://go.dev/VERSION?m=text)"  
  # e.g. "go1.21.4"
  LATEST_GO="${LATEST_GO_FULL#go}"                    
  TAR="go${LATEST_GO}.${OS}-${ARCH}.tar.gz"          
  URL="https://go.dev/dl/${TAR}"

  echo "   latest is ${LATEST_GO_FULL}, downloading ${TAR}..."
  TMPDIR="$(mktemp -d)"
  curl -fsSL "${URL}" -o "${TMPDIR}/${TAR}"
  sudo rm -rf /usr/local/go
  sudo tar -C /usr/local -xzf "${TMPDIR}/${TAR}"
  rm -rf "${TMPDIR}"
  echo "Installed Go ${LATEST_GO_FULL} to /usr/local/go"

  # ensure PATH in your RCs
  SNIPPET='export PATH=$PATH:/usr/local/go/bin'
  for rc in "${HOME}/.bashrc" "${HOME}/.zshrc"; do
    [ -f "${rc}" ] && _add_to_rc "${rc}" "${SNIPPET}"
  done
else
  INSTALLED_FULL="$(go version | awk '{print $3}')" # e.g. "go1.21.4"
  LATEST_GO_FULL="$(curl -fsSL https://go.dev/VERSION?m=text)"
  if [ "${INSTALLED_FULL}" = "${LATEST_GO_FULL}" ]; then
    echo "Go ${INSTALLED_FULL} already up-to-date"
  else
    echo "Go ${INSTALLED_FULL} installed but ${LATEST_GO_FULL} is available."
    echo "    rerun this script to upgrade or uninstall manually."
  fi
fi

echo

if ! command -v oh-my-posh &>/dev/null; then
  echo "➤ Installing Oh-My-Posh latest release..."
  # GitHub /releases/latest/download will 302 redirect to actual asset
  BIN_URL="https://github.com/JanDeDobbeleer/oh-my-posh/releases/latest/download/posh-${OS}-${ARCH}"

  mkdir -p "${HOME}/.local/bin"
  TMPBIN="$(mktemp)"
  curl -fsSL -L "${BIN_URL}" -o "${TMPBIN}"
  install -m755 "${TMPBIN}" "${HOME}/.local/bin/oh-my-posh"
  rm -f "${TMPBIN}"
  echo "Installed oh-my-posh to ~/.local/bin/oh-my-posh"

  echo "Theme is managed via dotfiles repo at terminal/posh-theme.omp.json"
  echo "Run 04-config-symlinks.sh to link it to ~/.config/oh-my-posh/theme.omp.json"

else
  echo "✔ Oh-My-Posh already installed: $(oh-my-posh --version | head -n1)"
fi

# Install Rust Up
echo "Installing Rust via rustup..."
if command -v rustc &>/dev/null; then
    echo "Rust (rustc) already installed."
else
    # The -y flag automates the installation, --no-modify-path prevents it from altering .profile/.bashrc directly
    # You'll need to source "$HOME/.cargo/env" or add it to your shell's config manually
    curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y --no-modify-path
    echo "Rust installed via rustup. Source \"\$HOME/.cargo/env\" or add it to your shell config."
    echo "For the current session, you can run: source \"\$HOME/.cargo/env\""
fi

echo "--- Development Tools Setup Finished ---"