#!/usr/bin/env bash
# icoextract ebuild updater for nbdy_overlay
# Source: PyPI (the ebuild inherits 'pypi' eclass and uses pypi_sdist_url)
# Asset: icoextract-${PV}.tar.gz from PyPI sdist

set -euo pipefail

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}/dev-python/icoextract"
PKG_NAME="icoextract"
PYPI_NAME="icoextract"

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"; }
log_success() { echo -e "${GREEN}[OK]${NC} $1"; }
log_warn() { echo -e "${YELLOW}[WARN]${NC} $1"; }
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)
    [[ -z "$ebuild" ]] && { echo ""; return; }
    basename "$ebuild" .ebuild | sed "s/${PKG_NAME}-//"
}

get_latest_version() {
    # PyPI JSON API: https://pypi.org/pypi/<name>/json -> .info.version
    local json
    json=$(curl -sfL "https://pypi.org/pypi/${PYPI_NAME}/json" 2>/dev/null) || {
        log_error "Failed to fetch PyPI metadata for ${PYPI_NAME}"
        return 1
    }

    local version
    version=$(echo "$json" | grep -oP '"version"\s*:\s*"\K[^"]+' | head -1 || echo "")

    if [[ -z "$version" ]]; then
        log_error "Could not parse version from PyPI JSON"
        return 1
    fi

    echo "$version"
}

# Get the sdist URL for a specific version from PyPI metadata
get_sdist_url() {
    local version="$1"
    local json
    json=$(curl -sfL "https://pypi.org/pypi/${PYPI_NAME}/${version}/json" 2>/dev/null) || {
        log_error "Failed to fetch PyPI release metadata for ${PYPI_NAME}-${version}"
        return 1
    }

    # Find the sdist (packagetype = sdist), capture its URL
    # Greedy regex on a flat JSON file is fine here.
    local url
    url=$(echo "$json" \
        | python3 -c '
import json, sys
try:
    data = json.load(sys.stdin)
except Exception as e:
    sys.exit(1)
for f in data.get("urls", []):
    if f.get("packagetype") == "sdist":
        print(f.get("url",""))
        break
' 2>/dev/null || echo "")

    if [[ -z "$url" ]]; then
        log_error "No sdist URL found for ${PYPI_NAME}-${version}"
        return 1
    fi

    echo "$url"
}

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 src_url
    pv=$(basename "$ebuild" .ebuild | sed "s/${PKG_NAME}-//")

    src_url=$(get_sdist_url "$pv") || return 1
    local distfile
    distfile=$(basename "$src_url")

    log_info "Downloading ${distfile} for checksum..."
    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="dev-python/icoextract: 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"
                echo "  -f, --force      Force update"
                echo "  --no-commit      Don't git commit"
                echo "  --keep N         Keep N old versions (default: 1)"
                exit 0
                ;;
            *) log_error "Unknown option: $1"; exit 1 ;;
        esac
    done

    echo "============================================"
    echo "  icoextract Ebuild Updater"
    echo "============================================"
    echo ""

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

    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 "$@"