#!/bin/bash
# 安装 lazygit（brew / apt / GitHub Release 二进制）

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
# shellcheck disable=SC1091
source "$SCRIPT_DIR/common/help.sh"

usage() {
    cat <<'EOF'
用法: lazygit-install

按顺序尝试安装 lazygit:
  1. Homebrew (macOS/Linux)
  2. apt (Debian/Ubuntu)
  3. 从 GitHub Release 下载到 ~/.local/bin

若已安装则打印版本并退出。

选项:
  -h, --help  显示此帮助
EOF
}

dotfile_help_requested "${1:-}" && dotfile_show_help

set -e

GREEN='\033[0;32m'
YELLOW='\033[1;33m'
RED='\033[0;31m'
NC='\033[0m'

if command -v lazygit &>/dev/null; then
    echo -e "${GREEN}lazygit 已安装:${NC}"
    lazygit --version
    exit 0
fi

install_via_brew() {
    if ! command -v brew &>/dev/null; then
        return 1
    fi
    echo -e "${YELLOW}使用 Homebrew 安装 lazygit...${NC}"
    brew install lazygit
}

install_via_apt() {
    if ! command -v apt-get &>/dev/null; then
        return 1
    fi
    echo -e "${YELLOW}使用 apt 安装 lazygit...${NC}"
    sudo apt-get update
    sudo apt-get install -y lazygit
}

install_via_release() {
    local ver
    ver=$(curl -fsSL "https://api.github.com/repos/jesseduffield/lazygit/releases/latest" |
        grep '"tag_name":' |
        sed -E 's/.*"v([^"]+)".*/\1/')
    if [[ -z "$ver" ]]; then
        echo -e "${RED}无法解析 lazygit 最新版本号${NC}" >&2
        return 1
    fi

    local os_raw arch_raw asset url
    os_raw=$(uname -s)
    arch_raw=$(uname -m)

    case "${os_raw}-${arch_raw}" in
    Darwin-arm64) asset="lazygit_${ver}_Darwin_arm64.tar.gz" ;;
    Darwin-x86_64) asset="lazygit_${ver}_Darwin_x86_64.tar.gz" ;;
    Linux-x86_64) asset="lazygit_${ver}_Linux_x86_64.tar.gz" ;;
    Linux-aarch64 | Linux-arm64) asset="lazygit_${ver}_Linux_arm64.tar.gz" ;;
    *)
        echo -e "${RED}当前平台不支持自动下载二进制: ${os_raw} ${arch_raw}${NC}" >&2
        return 1
        ;;
    esac

    url="https://github.com/jesseduffield/lazygit/releases/download/v${ver}/${asset}"
    local tmpdir dest_bin
    tmpdir=$(mktemp -d)
    dest_bin="${HOME}/.local/bin"

    echo -e "${YELLOW}下载 lazygit v${ver} (${asset})...${NC}"
    curl -fsSL -o "${tmpdir}/${asset}" "${url}"
    tar -xzf "${tmpdir}/${asset}" -C "${tmpdir}"

    mkdir -p "${dest_bin}"
    mv -f "${tmpdir}/lazygit" "${dest_bin}/lazygit"
    chmod +x "${dest_bin}/lazygit"
    rm -rf "${tmpdir}"

    echo -e "${GREEN}已安装到 ${dest_bin}/lazygit${NC}"
    if [[ ":${PATH}:" != *":${dest_bin}:"* ]]; then
        echo -e "${YELLOW}提示: 请将 ${dest_bin} 加入 PATH${NC}"
    fi
}

main() {
    if install_via_brew; then
        :
    elif install_via_apt; then
        :
    elif install_via_release; then
        :
    else
        echo -e "${RED}lazygit 安装失败${NC}" >&2
        exit 1
    fi

    echo -e "${GREEN}lazygit 安装完成:${NC}"
    command -v lazygit
    lazygit --version
}

main