#!/bin/sh
set -e

# --- Detect download tool ---
if command -v curl >/dev/null 2>&1; then
    FETCH="curl"
elif command -v wget >/dev/null 2>&1; then
    FETCH="wget"
else
    echo "Error: either 'curl' or 'wget' is required but neither is installed." >&2
    echo "Install one with your package manager, e.g.: apt install -y curl" >&2
    exit 1
fi

for cmd in sed grep uname chmod mkdir; do
    if ! command -v "$cmd" >/dev/null 2>&1; then
        echo "Error: '$cmd' is required but not installed." >&2
        echo "Install it with your package manager, e.g.: apt install -y $cmd" >&2
        exit 1
    fi
done

fetch() {
    if [ "$FETCH" = "curl" ]; then
        curl -fsSL "$1"
    else
        wget -qO- "$1"
    fi
}

fetch_to_file() {
    if [ "$FETCH" = "curl" ]; then
        curl -fSL -o "$2" "$1"
    else
        wget -q -O "$2" "$1"
    fi
}

# The directory to install the binary to. Can be overridden by setting the DIR environment variable.
DIR="${DIR:-"$HOME/.local/bin"}"

# --- Auto-detect OS and Architecture ---
OS=$(uname -s)
case "$OS" in
    Linux*)   PLATFORM="linux" ;;
    Darwin*)  PLATFORM="darwin" ;;
    *)        echo "Error: Unsupported OS '$OS'. Haloy supports Linux and macOS." >&2; exit 1 ;;
esac

ARCH=$(uname -m)
case "$ARCH" in
    x86_64)   ARCH="amd64" ;;
    arm64|aarch64) ARCH="arm64" ;;
    *)        echo "Error: Unsupported architecture '$ARCH'. Haloy supports amd64 (x86_64) and arm64." >&2; exit 1 ;;
esac

# --- Fetch the latest version from GitHub ---
echo "Finding the latest version of Haloy..."
GITHUB_API_URL="https://api.github.com/repos/haloydev/haloy/releases"
GITHUB_RESPONSE=$(fetch "$GITHUB_API_URL" 2>/dev/null || true)

# Check if the response indicates no releases
if echo "$GITHUB_RESPONSE" | grep -q '"message": "Not Found"'; then
    echo "Error: No releases found for Haloy. Please check https://github.com/haloydev/haloy/releases" >&2
    exit 1
fi

GITHUB_LATEST_VERSION=$(echo "$GITHUB_RESPONSE" | grep -o '"tag_name": *"[^"]*"' | head -1 | sed 's/.*"\([^"]*\)"$/\1/')

if [ -z "$GITHUB_LATEST_VERSION" ]; then
    echo "Error: Could not determine the latest Haloy version from GitHub." >&2
    exit 1
fi

# --- Download and Install ---
BINARY_NAME="haloy-${PLATFORM}-${ARCH}"
DOWNLOAD_URL="https://github.com/haloydev/haloy/releases/download/${GITHUB_LATEST_VERSION}/${BINARY_NAME}"
INSTALL_PATH="$DIR/haloy"

# Create the installation directory if it doesn't exist
mkdir -p "$DIR"

echo "Downloading Haloy ${GITHUB_LATEST_VERSION} for ${PLATFORM}/${ARCH}..."
fetch_to_file "$DOWNLOAD_URL" "$INSTALL_PATH"
chmod +x "$INSTALL_PATH"

echo ""
echo "Haloy client has been installed to '$INSTALL_PATH'"
echo ""

# --- Check if DIR is in PATH ---
# Use case statement for POSIX-compliant substring check
case ":$PATH:" in
    *":$DIR:"*)
        echo "You can now run 'haloy' from anywhere!"
        ;;
    *)
        echo "Warning: '$DIR' is not in your PATH."
        echo ""
        echo "Add the following line to your shell profile (~/.bashrc, ~/.zshrc, or equivalent):"
        echo ""
        echo "    export PATH=\"\$HOME/.local/bin:\$PATH\""
        echo ""
        echo "Then restart your shell or run: source ~/.bashrc (or ~/.zshrc)"
        ;;
esac

# --- Install shell completions ---
SHELL_NAME="$(basename "$SHELL" 2>/dev/null || true)"

install_bash_completions() {
    if [ "$(id -u)" = "0" ]; then
        COMP_DIR="/etc/bash_completion.d"
    else
        COMP_DIR="$HOME/.local/share/bash-completion/completions"
    fi
    mkdir -p "$COMP_DIR"
    "$INSTALL_PATH" completion bash > "$COMP_DIR/haloy" 2>/dev/null
    echo "Bash completions installed to $COMP_DIR/haloy"
}

install_zsh_completions() {
    COMP_DIR="$HOME/.local/share/zsh/site-functions"
    mkdir -p "$COMP_DIR"
    "$INSTALL_PATH" completion zsh > "$COMP_DIR/_haloy" 2>/dev/null
    echo "Zsh completions installed to $COMP_DIR/_haloy"
    echo "Note: You may need to add the following to your ~/.zshrc if completions don't work:"
    echo "    fpath=(~/.local/share/zsh/site-functions \$fpath)"
    echo "    autoload -U compinit && compinit"
}

install_fish_completions() {
    COMP_DIR="$HOME/.config/fish/completions"
    mkdir -p "$COMP_DIR"
    "$INSTALL_PATH" completion fish > "$COMP_DIR/haloy.fish" 2>/dev/null
    echo "Fish completions installed to $COMP_DIR/haloy.fish"
}

echo ""
case "$SHELL_NAME" in
    bash)
        install_bash_completions
        ;;
    zsh)
        install_zsh_completions
        ;;
    fish)
        install_fish_completions
        ;;
    *)
        echo "Shell completions were not installed (unrecognized shell: $SHELL_NAME)."
        echo "Run 'haloy completion --help' to install them manually."
        ;;
esac