#!/usr/bin/env bash

# archwiki-tui - Automated Installer
# ----------------------------------
# This script performs a full zero-labor installation of archwiki-tui.
# It automatically detects your distro and installs all missing dependencies.

set -euo pipefail

REPO_URL="https://github.com/Harshil-Anuwadia/archwiki-tui.git"
APP_NAME="archwiki"
INSTALL_DIR="/usr/local/bin"

# Colors
RED='\033[0;31m'
GREEN='\033[0;32m'
BLUE='\033[0;34m'
BOLD='\033[1m'
NC='\033[0m'

info() { echo -e "${BLUE}[INFO]${NC} $1"; }
success() { echo -e "${GREEN}[SUCCESS]${NC} $1"; }
error() { echo -e "${RED}[ERROR]${NC} $1" >&2; exit 1; }

banner() {
    echo -e "${BLUE}${BOLD}"
    echo "  ▄▀█ █▀█ █▀▀ █░█ █   █ █ █▄▀ █"
    echo "  █▀█ █▀▄ █▄▄ █▀█ █▄█▄█ █ █░█ █"
    echo -e "${NC}"
    echo -e "  The definitive Arch Wiki terminal browser.\n"
}

# Check for root/sudo
if [ "$EUID" -ne 0 ]; then
    error "Please run this installer with sudo: curl ... | sudo bash"
fi

banner

# Detect Distro and install base tools
info "Detecting system environment..."
if [ -f /etc/os-release ]; then
    . /etc/os-release
    DISTRO=$ID
else
    DISTRO="unknown"
fi

install_pkg() {
    info "Installing missing dependency: $1..."
    case "$DISTRO" in
        arch|manjaro) pacman -S --needed --noconfirm "$@" ;;
        ubuntu|debian|kali|pop|linuxmint|raspbian) 
            apt-get update -qq
            apt-get install -y -qq "$@" 
            ;;
        fedora|rhel|centos|rocky|almalinux) dnf install -y -q "$@" ;;
        suse|opensuse*) zypper install -y "$@" ;;
        *) info "Unsupported distro ($DISTRO) for auto-install. Please install '$1' manually." ;;
    esac
}

# Ensure core build tools are present
info "Verifying system dependencies..."
for cmd in git make gcc curl tar; do
    if ! command -v "$cmd" &> /dev/null; then
        pkg=$cmd
        if [ "$cmd" == "gcc" ]; then
            case "$DISTRO" in
                ubuntu|debian|kali|pop|linuxmint) pkg="build-essential" ;;
                fedora|rhel|centos) pkg="gcc gcc-c++" ;;
            esac
        fi
        install_pkg "$pkg"
    fi
done

# Handle Go dependency
GO_PATH="/usr/local/go/bin/go"
if ! command -v go &> /dev/null && [ ! -f "$GO_PATH" ]; then
    GO_VER="1.25.0"
    info "Go not found. Installing Go ${GO_VER}..."
    GO_ARCH="amd64"
    if [[ "$(uname -m)" == "aarch64" ]]; then GO_ARCH="arm64"; fi
    
    GO_TMP="/tmp/go${GO_VER}.linux-${GO_ARCH}.tar.gz"
    curl -fL "https://go.dev/dl/go${GO_VER}.linux-${GO_ARCH}.tar.gz" -o "$GO_TMP"
    
    rm -rf /usr/local/go
    tar -C /usr/local -xzf "$GO_TMP"
    rm "$GO_TMP"
    success "Go installed to /usr/local/go."
fi

# Ensure we use the right Go binary
if [ -f "$GO_PATH" ]; then
    export PATH="/usr/local/go/bin:$PATH"
fi

# Build from source
TMP_DIR=$(mktemp -d)
info "Cloning latest source code..."
git clone --depth 1 "$REPO_URL" "$TMP_DIR" &> /dev/null || error "Failed to clone repository."

cd "$TMP_DIR"

info "Building $APP_NAME..."
# We run make with an explicit GO variable to ensure it uses the one we just installed
if ! make build GO="$(command -v go)"; then
    echo "--------------------------------------------------"
    echo "BUILD FAILED. Diagnostic Information:"
    echo "Go Version: $(go version || echo 'Not Found')"
    echo "Make Version: $(make --version | head -n 1 || echo 'Not Found')"
    echo "PATH: $PATH"
    echo "--------------------------------------------------"
    error "Build failed. See logs above for details."
fi

# Final installation
info "Deploying binary to $INSTALL_DIR/$APP_NAME..."
cp "bin/$APP_NAME" "$INSTALL_DIR/"
chmod +x "$INSTALL_DIR/$APP_NAME"

# Cleanup
info "Cleaning up..."
rm -rf "$TMP_DIR"

echo ""
success "Installation complete! Run '$APP_NAME' to start."
info "Tip: Run '$APP_NAME <query>' for a direct search."