#!/usr/bin/env bash
# AFFiNE ebuild updater for nbdy_overlay
#
# Version detection strategy:
#   GitHub releases API → latest stable release tag (vX.Y.Z format)
#   SRC_URI: https://github.com/toeverything/AFFiNE/releases/download/v${PV}/affine-${PV}-stable-linux-x64.appimage

set -euo pipefail

# Cleanup on failure: remove any newly created ebuild
CLEANUP_FILES=()
cleanup_on_exit() {
    local exit_code=$?
    if (( exit_code != 0 )) && (( ${#CLEANUP_FILES[@]} > 0 )); then
        for f in "${CLEANUP_FILES[@]}"; do
            [[ -f "$f" ]] && rm -v "$f"
        done
    fi
}
trap cleanup_on_exit EXIT

OVERLAY_DIR="$(dirname "$(dirname "$(realpath "$0")")")"
EBUILD_DIR="${OVERLAY_DIR}/app-office/affine-bin"
PKG_NAME="affine-bin"
GITHUB_REPO="toeverything/AFFiNE"

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

log_info()    { echo -e "${BLUE}[INFO]${NC} $1" >&2; }
log_success() { echo -e "${GREEN}[OK]${NC} $1" >&2; }
log_warn()    { echo -e "${YELLOW}[WARN]${NC} $1" >&2; }
log_error()   { echo -e "${RED}[ERROR]${NC} $1" >&2; }

get_current_version() {
    local ebuild
    ebuild=$(ls "${EBUILD_DIR}"/*.ebuild 2>/dev/null | sort -V | tail -1)
    if [[ -z "$ebuild" ]]; then
        echo ""
        return
    fi
    basename "$ebuild" .ebuild | sed "s/${PKG_NAME}-//"
}

get_latest_version() {
    log_info "Fetching latest release from GitHub..."

    local curl_opts=(-sfL --max-time 30)
    [[ -n "${GITHUB_TOKEN:-}" ]] && curl_opts+=(-H "Authorization: Bearer ${GITHUB_TOKEN}")

    local json
    json=$(curl "${curl_opts[@]}" \
        "https://api.github.com/repos/${GITHUB_REPO}/releases/latest" 2>/dev/null) || {
        log_error "Failed to fetch latest release from GitHub"
        return 1
    }

    local tag_name
    tag_name=$(echo "$json" | grep -oP '"tag_name"\s*:\s*"\Kv[0-9]+\.[0-9]+\.[0-9]+' | head -1)

    if [[ -z "$tag_name" ]]; then
        log_error "Could not parse version from GitHub release"
        return 1
    fi

    # Strip leading 'v'
    local version="${tag_name#v}"

    # Verify the AppImage asset exists
    local asset_url="https://github.com/${GITHUB_REPO}/releases/download/${tag_name}/affine-${version}-stable-linux-x64.appimage"
    local http_code
    http_code=$(curl -sfIL -o /dev/null -w '%{http_code}' "$asset_url" 2>/dev/null)

    if [[ "$http_code" != "200" && "$http_code" != "302" ]]; then
        log_warn "AppImage not found for ${version} (HTTP ${http_code}), checking previous releases..."

        # Fallback: iterate recent releases
        local releases
        releases=$(curl "${curl_opts[@]}" \
            "https://api.github.com/repos/${GITHUB_REPO}/releases?per_page=10" 2>/dev/null) || return 1

        local tag
        while IFS= read -r tag; do
            [[ -z "$tag" ]] && continue
            local v="${tag#v}"
            asset_url="https://github.com/${GITHUB_REPO}/releases/download/${tag}/affine-${v}-stable-linux-x64.appimage"
            http_code=$(curl -sfIL -o /dev/null -w '%{http_code}' "$asset_url" 2>/dev/null)
            if [[ "$http_code" == "200" || "$http_code" == "302" ]]; then
                log_success "Found AppImage for ${v}"
                echo "$v"
                return 0
            fi
        done < <(echo "$releases" | grep -oP '"tag_name"\s*:\s*"\Kv[0-9]+\.[0-9]+\.[0-9]+')

        log_error "No AppImage found in recent releases"
        return 1
    fi

    log_success "Latest version: ${version} (AppImage verified)"
    echo "$version"
}

create_ebuild() {
    local new_version="$1"
    local old_ebuild new_ebuild

    old_ebuild=$(ls "${EBUILD_DIR}"/*.ebuild 2>/dev/null | sort -V | tail -1)
    new_ebuild="${EBUILD_DIR}/${PKG_NAME}-${new_version}.ebuild"

    if [[ -z "$old_ebuild" ]]; then
        log_error "No existing ebuild found to copy"
        return 1
    fi

    if [[ -f "$new_ebuild" ]]; then
        log_warn "Ebuild already exists: $new_ebuild"
        return 0
    fi

    cp "$old_ebuild" "$new_ebuild"
    CLEANUP_FILES+=("$new_ebuild")
    log_success "Created: ${PKG_NAME}-${new_version}.ebuild"
}

generate_manifest() {
    log_info "Generating Manifest..."

    local ebuild
    ebuild=$(ls "${EBUILD_DIR}"/*.ebuild 2>/dev/null | sort -V | tail -1)

    if [[ -z "$ebuild" ]]; then
        log_error "No ebuild found"
        return 1
    fi

    local pv distfile src_url
    pv=$(basename "$ebuild" .ebuild | sed "s/${PKG_NAME}-//")
    distfile="affine-${pv}-stable-linux-x64.appimage"
    src_url="https://github.com/${GITHUB_REPO}/releases/download/v${pv}/${distfile}"

    log_info "Downloading ${distfile} for checksum (~172 MB)..."
    local tmpfile
    tmpfile=$(mktemp)

    if ! curl -sfL -o "$tmpfile" "$src_url"; then
        log_error "Failed to download ${src_url}"
        rm -f "$tmpfile"
        return 1
    fi

    local filesize blake2b sha512
    filesize=$(wc -c < "$tmpfile")
    blake2b=$(b2sum "$tmpfile" | awk '{print $1}')
    sha512=$(sha512sum "$tmpfile" | awk '{print $1}')
    rm -f "$tmpfile"

    echo "DIST ${distfile} ${filesize} BLAKE2B ${blake2b} SHA512 ${sha512}" > "${EBUILD_DIR}/Manifest"
    log_success "Manifest generated for ${distfile} (${filesize} bytes)"
}

cleanup_old_versions() {
    local keep="${1:-1}"
    local ebuilds

    mapfile -t ebuilds < <(ls "${EBUILD_DIR}"/*.ebuild 2>/dev/null | sort -V)
    local count=${#ebuilds[@]}

    if (( count <= keep )); then
        log_info "No cleanup needed (${count} versions, keeping ${keep})"
        return
    fi

    local to_remove=$((count - keep))
    log_info "Removing ${to_remove} old version(s)..."

    for ((i = 0; i < to_remove; i++)); do
        rm -v "${ebuilds[$i]}"
    done

    log_success "Cleanup complete"
}

git_commit() {
    local version="$1"

    cd "${OVERLAY_DIR}"
    git add "${EBUILD_DIR}/"

    if git diff --cached --quiet; then
        log_info "No changes to commit"
        return 0
    fi

    local msg="app-office/affine-bin: bump to ${version}"
    git commit -m "$msg"
    log_success "Committed: $msg"
}

main() {
    local dry_run=false
    local force=false
    local no_commit=false
    local keep_versions=1

    while [[ $# -gt 0 ]]; do
        case $1 in
            -n|--dry-run)  dry_run=true;        shift ;;
            -f|--force)    force=true;           shift ;;
            --no-commit)   no_commit=true;       shift ;;
            --keep)        keep_versions="$2";   shift 2 ;;
            -h|--help)
                echo "Usage: $0 [OPTIONS]"
                echo "  -n, --dry-run    Preview changes without writing anything"
                echo "  -f, --force      Force update even if version appears current"
                echo "  --no-commit      Skip git commit step"
                echo "  --keep N         Keep N ebuild versions (default: 1)"
                exit 0
                ;;
            *) log_error "Unknown option: $1"; exit 1 ;;
        esac
    done

    echo "============================================"
    echo "  AFFiNE Ebuild Updater"
    echo "============================================"
    echo ""

    local current_version latest_version
    current_version=$(get_current_version)
    latest_version=$(get_latest_version) || exit 1

    echo ""
    echo "Current: ${current_version:-<none>}"
    echo "Latest:  ${latest_version}"
    echo ""

    if [[ "$current_version" == "$latest_version" ]] && [[ "$force" != "true" ]]; then
        log_success "Already up to date!"
        exit 0
    fi

    if [[ "$dry_run" == "true" ]]; then
        log_info "[DRY-RUN] Would update to ${latest_version}"
        exit 0
    fi

    create_ebuild "$latest_version"
    generate_manifest
    cleanup_old_versions "$keep_versions"

    if [[ "$no_commit" != "true" ]]; then
        git_commit "$latest_version"
    fi

    log_success "Update complete: ${latest_version}"
}

main "$@"