#!/usr/bin/env bash
# ==============================================================================
# ani-sync Universal Auto-Installer (Linux & macOS)
# Automatically installs ani-sync, FZF fuzzy search, MPV, yt-dlp, and Python dependencies
# ==============================================================================

set -e

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
CYAN='\033[0;36m'
YELLOW='\033[1;33m'
MAGENTA='\033[0;35m'
BOLD='\033[1m'
DIM='\033[2m'
NC='\033[0m' # No Color

echo -e "${CYAN}${BOLD}"
echo "  ============================================"
echo "          ani-sync Universal Installer        "
echo "     Stream Anime & Auto-Sync Watch Progress  "
echo "  ============================================"
echo -e "${NC}"

# Detect installation target directory
if [ "$EUID" -eq 0 ]; then
    INSTALL_DIR="/usr/local/bin"
    SHARE_DIR="/usr/local/share/ani-sync"
    SUDO_CMD=""
else
    INSTALL_DIR="${HOME}/.local/bin"
    SHARE_DIR="${HOME}/.local/share/ani-sync"
    mkdir -p "$INSTALL_DIR"
    if command -v sudo &>/dev/null; then
        SUDO_CMD="sudo"
    else
        SUDO_CMD=""
    fi
fi

mkdir -p "$SHARE_DIR"

# ------------------------------------------------------------------------------
# 1. Package Manager Detection & System Dependencies (fzf, mpv, yt-dlp, curl, etc.)
# ------------------------------------------------------------------------------
echo -e "${CYAN}[1/4] Checking and installing system dependencies (fzf, mpv, yt-dlp, curl)...${NC}"

install_system_packages() {
    if command -v apt-get &>/dev/null; then
        echo -e "  ${DIM}Detected Debian/Ubuntu (APT) package manager...${NC}"
        $SUDO_CMD apt-get update -y -qq || true
        $SUDO_CMD apt-get install -y -qq fzf mpv yt-dlp curl git python3 python3-pip python3-requests python3-tqdm 2>/dev/null || \
        $SUDO_CMD apt-get install -y -qq fzf mpv curl git python3 python3-pip 2>/dev/null || true
    elif command -v pacman &>/dev/null; then
        echo -e "  ${DIM}Detected Arch Linux (pacman) package manager...${NC}"
        $SUDO_CMD pacman -Sy --noconfirm --needed fzf mpv yt-dlp curl git python python-pip python-requests python-tqdm 2>/dev/null || true
    elif command -v dnf &>/dev/null; then
        echo -e "  ${DIM}Detected Fedora/RHEL (DNF) package manager...${NC}"
        $SUDO_CMD dnf install -y -q fzf mpv yt-dlp curl git python3 python3-pip python3-requests python3-tqdm 2>/dev/null || true
    elif command -v zypper &>/dev/null; then
        echo -e "  ${DIM}Detected openSUSE (zypper) package manager...${NC}"
        $SUDO_CMD zypper --non-interactive in fzf mpv yt-dlp curl git python3 python3-pip python3-requests python3-tqdm 2>/dev/null || true
    elif command -v apk &>/dev/null; then
        echo -e "  ${DIM}Detected Alpine Linux (apk) package manager...${NC}"
        $SUDO_CMD apk add --no-cache fzf mpv yt-dlp curl git python3 py3-pip py3-requests py3-tqdm 2>/dev/null || true
    elif command -v brew &>/dev/null; then
        echo -e "  ${DIM}Detected macOS (Homebrew) package manager...${NC}"
        brew install fzf mpv yt-dlp curl git python3 2>/dev/null || true
    elif command -v xbps-install &>/dev/null; then
        echo -e "  ${DIM}Detected Void Linux (XBPS) package manager...${NC}"
        $SUDO_CMD xbps-install -Sy fzf mpv yt-dlp curl git python3 python3-pip python3-requests python3-tqdm 2>/dev/null || true
    elif command -v pkg &>/dev/null; then
        echo -e "  ${DIM}Detected Termux/FreeBSD (pkg) package manager...${NC}"
        pkg install -y fzf mpv yt-dlp curl git python 2>/dev/null || true
    fi
}

install_system_packages || true

# ------------------------------------------------------------------------------
# 2. Standalone FZF Binary Auto-Downloader Fallback
# ------------------------------------------------------------------------------
if ! command -v fzf &>/dev/null && [ ! -f "$INSTALL_DIR/fzf" ]; then
    echo -e "${YELLOW}  Installing standalone FZF binary from GitHub releases...${NC}"
    ARCH="$(uname -m)"
    OS="$(uname -s | tr '[:upper:]' '[:lower:]')"
    FZF_ARCH="amd64"

    case "$ARCH" in
        x86_64|amd64) FZF_ARCH="amd64" ;;
        aarch64|arm64|armv8*) FZF_ARCH="arm64" ;;
        armv7*) FZF_ARCH="armv7" ;;
        armv6*) FZF_ARCH="armv6" ;;
        i386|i686) FZF_ARCH="386" ;;
        *) FZF_ARCH="amd64" ;;
    esac

    FZF_VERSION="0.60.3"
    FZF_URL="https://github.com/junegunn/fzf/releases/download/v${FZF_VERSION}/fzf-${FZF_VERSION}-${OS}_${FZF_ARCH}.tar.gz"
    
    TMP_FZF_DIR="$(mktemp -d)"
    if curl -fsSL "$FZF_URL" -o "$TMP_FZF_DIR/fzf.tar.gz" 2>/dev/null; then
        tar -xzf "$TMP_FZF_DIR/fzf.tar.gz" -C "$TMP_FZF_DIR"
        if [ -f "$TMP_FZF_DIR/fzf" ]; then
            cp "$TMP_FZF_DIR/fzf" "$INSTALL_DIR/fzf"
            chmod +x "$INSTALL_DIR/fzf"
            echo -e "  ${GREEN}✓ Standalone FZF installed to $INSTALL_DIR/fzf${NC}"
        fi
    fi
    rm -rf "$TMP_FZF_DIR"
fi

# Check Python 3
if ! command -v python3 &>/dev/null; then
    echo -e "${RED}Error: python3 is not installed.${NC} Please install Python 3 and rerun this script."
    exit 1
fi

# ------------------------------------------------------------------------------
# 3. Python Dependencies (requests, tqdm, yt-dlp)
# ------------------------------------------------------------------------------
echo -e "${CYAN}[2/4] Checking and installing Python packages (requests, tqdm, yt-dlp)...${NC}"
python3 -m pip install --upgrade --quiet requests tqdm yt-dlp ani-cli 2>/dev/null || \
python3 -m pip install --break-system-packages --upgrade --quiet requests tqdm yt-dlp ani-cli 2>/dev/null || \
python3 -m pip install --user --break-system-packages --quiet requests tqdm yt-dlp ani-cli 2>/dev/null || \
python3 -m pip install --user --quiet requests tqdm yt-dlp ani-cli 2>/dev/null || {
    echo -e "${YELLOW}Warning: Could not run pip automatically. Python system packages will be used.${NC}"
}

# ------------------------------------------------------------------------------
# 4. Install ani-sync Script & Wrapper
# ------------------------------------------------------------------------------
echo -e "${CYAN}[3/4] Installing ani-sync to ${INSTALL_DIR}...${NC}"

# If running directly from git cloned repo
if [ -f "ani_sync.py" ]; then
    cp ani_sync.py "$SHARE_DIR/ani_sync.py"
    if [ -d "assets" ]; then
        cp -r assets "$SHARE_DIR/"
    fi
else
    # Downloading directly from raw GitHub if run via curl / standalone
    echo -e "  Downloading latest ani-sync script from GitHub..."
    curl -fsSL https://raw.githubusercontent.com/idrisharis12/ani-sync/main/ani_sync.py -o "$SHARE_DIR/ani_sync.py"
fi

# Create launcher wrapper in INSTALL_DIR
cat << 'LAUNCHER' > "$INSTALL_DIR/ani-sync"
#!/usr/bin/env bash
export PATH="${HOME}/.local/bin:/usr/local/bin:$PATH"
SHARE_DIR_USER="${HOME}/.local/share/ani-sync"
SHARE_DIR_SYS="/usr/local/share/ani-sync"

if [ -f "$SHARE_DIR_SYS/ani_sync.py" ]; then
    exec python3 "$SHARE_DIR_SYS/ani_sync.py" "$@"
elif [ -f "$SHARE_DIR_USER/ani_sync.py" ]; then
    exec python3 "$SHARE_DIR_USER/ani_sync.py" "$@"
else
    echo "Error: ani_sync.py not found." >&2
    exit 1
fi
LAUNCHER

chmod +x "$SHARE_DIR/ani_sync.py"
chmod +x "$INSTALL_DIR/ani-sync"

# ------------------------------------------------------------------------------
# 5. Environment & PATH Verification
# ------------------------------------------------------------------------------
echo -e "${CYAN}[4/4] Finalizing setup & PATH configuration...${NC}"

# Ensure ~/.local/bin is in PATH for non-root users
if [ "$EUID" -ne 0 ]; then
    case ":$PATH:" in
        *":$INSTALL_DIR:"*) ;;
        *)
            for rc in "$HOME/.bashrc" "$HOME/.zshrc" "$HOME/.profile"; do
                if [ -f "$rc" ] && ! grep -q 'export PATH="$HOME/.local/bin:$PATH"' "$rc"; then
                    echo 'export PATH="$HOME/.local/bin:$PATH"' >> "$rc"
                fi
            done
            export PATH="$INSTALL_DIR:$PATH"
            ;;
    esac
fi

# Set up automatic APT update hook if running as root on Debian/Ubuntu
if [ "$EUID" -eq 0 ] && [ -d "/etc/apt/apt.conf.d" ]; then
    cat << 'APTHOOK' > /etc/apt/apt.conf.d/99ani-sync-updater
APT::Update::Post-Invoke-Success { "if command -v ani-sync >/dev/null 2>&1; then ani-sync update --quiet || true; fi"; };
APTHOOK
    chmod 644 /etc/apt/apt.conf.d/99ani-sync-updater
    echo -e "  ${GREEN}✓ Registered APT auto-update hook (/etc/apt/apt.conf.d/99ani-sync-updater)${NC}"
fi

# ------------------------------------------------------------------------------
# Summary
# ------------------------------------------------------------------------------
echo -e "\n${GREEN}${BOLD}============================================================${NC}"
echo -e "${GREEN}${BOLD}           ✓ Successfully installed ani-sync!               ${NC}"
echo -e "${GREEN}${BOLD}============================================================${NC}"
echo -e "  ${CYAN}• FZF Search:${NC}      $([ "$(command -v fzf 2>/dev/null || [ -f "$INSTALL_DIR/fzf" ] && echo "yes")" ] && echo -e "${GREEN}Enabled (Interactive Fuzzy Finder)${NC}" || echo -e "${YELLOW}Numbered Menu Fallback${NC}")"
echo -e "  ${CYAN}• Media Player:${NC}    $([ "$(command -v mpv 2>/dev/null)" ] && echo -e "${GREEN}mpv detected${NC}" || echo -e "${YELLOW}mpv recommended (sudo apt install mpv / pacman -S mpv)${NC}")"
echo -e "  ${CYAN}• Turbo Engine:${NC}    $([ "$(command -v yt-dlp 2>/dev/null)" ] && echo -e "${GREEN}yt-dlp ready${NC}" || echo -e "${YELLOW}yt-dlp installed via python${NC}")"
echo -e "\n${BOLD}Quick Start:${NC}"
echo -e "  ${CYAN}ani-sync${NC}                     Interactive anime search & launcher"
echo -e "  ${CYAN}ani-sync \"frieren\"${NC}           Search and play specific anime"
echo -e "  ${CYAN}ani-sync -c${NC}                  Continue watching next episode"
echo -e "  ${CYAN}ani-sync history${NC}             Browse watch history with interactive FZF"
echo -e "  ${CYAN}ani-sync auth${NC}                Connect MyAnimeList / AniList / Kitsu"
echo -e "  ${CYAN}ani-sync doctor${NC}              Verify dependencies and system health"
echo ""