#!/bin/sh
set -e

# Iris installer
# Usage: curl -sS https://raw.githubusercontent.com/versenilvis/iris/main/install.sh | sudo sh

REPO="versenilvis/iris"
BIN_DIR="${BIN_DIR:-/usr/local/bin}"

main() {
    echo "Installing iris..."

    arch=$(get_arch)
    echo "Detected architecture: ${arch}"

    tmp_dir=$(mktemp -d)
    trap 'rm -rf '"${tmp_dir}" EXIT
    cd "${tmp_dir}"

    download_url=$(get_download_url "${arch}")
    if [ -z "${download_url}" ]; then
        err "Could not find release for architecture: ${arch}"
    fi
    echo "Downloading: ${download_url}"

    if command -v curl >/dev/null 2>&1; then
        curl -sLO "${download_url}"
    elif command -v wget >/dev/null 2>&1; then
        wget -q "${download_url}"
    else
        err "curl or wget is required"
    fi

    archive=$(basename "${download_url}")
    case "${archive}" in
        *.tar.gz) tar -xzf "${archive}" ;;
        *.zip)
            if command -v unzip >/dev/null 2>&1; then
                unzip -q "${archive}"
            else
                err "unzip is required to extract ${archive}"
            fi
            ;;
        *) err "Unknown archive format: ${archive}" ;;
    esac

    bin=$(find . -name "iris" -type f | head -1)
    [ -z "$bin" ] && err "Binary not found in archive"

    mkdir -p "${BIN_DIR}"
    cp "$bin" "${BIN_DIR}/iris"
    chmod +x "${BIN_DIR}/iris"

    echo ""
    echo "Installed iris to ${BIN_DIR}/iris"
    echo ""

    if "${BIN_DIR}/iris" --version >/dev/null 2>&1; then
        echo "Installation verified."
    else
        echo "Warning: could not verify binary"
    fi

    echo ""
    echo "To complete setup, run:"
    echo "  iris setup"
}

get_arch() {
    os=$(uname -s | tr '[:upper:]' '[:lower:]')
    arch=$(uname -m)

    case "${os}" in
        linux)  os="linux" ;;
        darwin) os="darwin" ;;
        *) err "Unsupported OS: ${os}" ;;
    esac

    case "${arch}" in
        x86_64 | amd64)   arch="amd64" ;;
        aarch64 | arm64)  arch="arm64" ;;
        *) err "Unsupported architecture: ${arch}" ;;
    esac

    echo "${os}_${arch}"
}

get_download_url() {
    arch="$1"

    if command -v curl >/dev/null 2>&1; then
        releases=$(curl -sL \
            ${GITHUB_TOKEN:+-H "Authorization: Bearer ${GITHUB_TOKEN}"} \
            "https://api.github.com/repos/${REPO}/releases/latest")
    elif command -v wget >/dev/null 2>&1; then
        releases=$(wget -qO- \
            ${GITHUB_TOKEN:+--header "Authorization: Bearer ${GITHUB_TOKEN}"} \
            "https://api.github.com/repos/${REPO}/releases/latest")
    else
        err "curl or wget is required"
    fi

    if echo "${releases}" | grep -q "rate limit"; then
        err "GitHub API rate limited. Try again later or set GITHUB_TOKEN env variable"
    fi

    if echo "${releases}" | grep -q '"message"'; then
        msg=$(echo "${releases}" | grep '"message"' | head -1 | cut -d '"' -f 4)
        err "GitHub API error: ${msg}"
    fi

    url=$(echo "${releases}" | grep "browser_download_url" | grep "${arch}" | head -1 | cut -d '"' -f 4)
    echo "${url}"
}

err() {
    echo "Error: $1" >&2
    exit 1
}

main "$@"