#!/bin/bash
set -e

# luxctl installer
# Usage: curl -fsSL https://raw.githubusercontent.com/thearyanahmed/luxctl/master/install.sh | bash
#
# Downloads a pre-built binary when available (Linux x86_64/aarch64, macOS aarch64).
# Falls back to `cargo install` for unsupported platforms (e.g. macOS Intel).
#
# Environment variables:
#   LUXCTL_VERSION   - pin a specific version (e.g. "v0.8.2"), default: latest
#   INSTALL_DIR      - where to put the binary, default: /usr/local/bin

REPO="thearyanahmed/luxctl"
VERSION="${LUXCTL_VERSION:-}"
INSTALL_DIR="${INSTALL_DIR:-/usr/local/bin}"

info() {
    printf "\033[0;34m==>\033[0m %s\n" "$1"
}

success() {
    printf "\033[0;32m==>\033[0m %s\n" "$1"
}

error() {
    printf "\033[0;31merror:\033[0m %s\n" "$1" >&2
    exit 1
}

check_cmd() {
    command -v "$1" >/dev/null 2>&1
}

detect_platform() {
    OS="$(uname -s)"
    ARCH="$(uname -m)"

    case "$OS" in
        Linux)          PLATFORM="linux" ;;
        Darwin)         PLATFORM="macos" ;;
        MINGW*|MSYS*|CYGWIN*)  PLATFORM="windows" ;;
        *)              PLATFORM="unknown" ;;
    esac

    case "$ARCH" in
        x86_64|amd64)   ARCH="x86_64" ;;
        aarch64|arm64)  ARCH="aarch64" ;;
        *)              ARCH="unknown" ;;
    esac
}

# resolve the asset filename for this platform, empty string if no binary available
resolve_asset() {
    case "${PLATFORM}-${ARCH}" in
        linux-x86_64)    ASSET="luxctl-linux-x86_64.tar.gz" ;;
        linux-aarch64)   ASSET="luxctl-linux-aarch64.tar.gz" ;;
        macos-aarch64)   ASSET="luxctl-macos-aarch64.tar.gz" ;;
        macos-x86_64)    ASSET="luxctl-macos-x86_64.tar.gz" ;;
        windows-x86_64)  ASSET="luxctl-windows-x86_64.exe.zip" ;;
        *)               ASSET="" ;;
    esac
}

fetch_latest_version() {
    if check_cmd curl; then
        VERSION=$(curl -fsSL "https://api.github.com/repos/${REPO}/releases/latest" \
            | grep '"tag_name"' | head -1 | cut -d '"' -f 4)
    elif check_cmd wget; then
        VERSION=$(wget -qO- "https://api.github.com/repos/${REPO}/releases/latest" \
            | grep '"tag_name"' | head -1 | cut -d '"' -f 4)
    else
        error "curl or wget is required"
    fi

    if [ -z "$VERSION" ]; then
        error "could not determine latest version from GitHub"
    fi
}

download_binary() {
    local url="https://github.com/${REPO}/releases/download/${VERSION}/${ASSET}"
    local tmpdir
    tmpdir="$(mktemp -d)"
    trap 'rm -rf "$tmpdir"' EXIT

    info "Downloading luxctl ${VERSION} (${PLATFORM}/${ARCH})..."

    if check_cmd curl; then
        curl -fsSL "$url" -o "${tmpdir}/${ASSET}"
    elif check_cmd wget; then
        wget -qO "${tmpdir}/${ASSET}" "$url"
    fi

    # windows assets are .zip with an .exe inside
    if [ "$PLATFORM" = "windows" ]; then
        if check_cmd unzip; then
            unzip -q "${tmpdir}/${ASSET}" -d "$tmpdir"
        elif check_cmd 7z; then
            7z x "${tmpdir}/${ASSET}" -o"$tmpdir" -y >/dev/null
        else
            error "unzip or 7z is required to extract the Windows binary"
        fi

        local exe_name="${ASSET%.zip}"
        if [ -f "${tmpdir}/${exe_name}" ]; then
            mv "${tmpdir}/${exe_name}" "${tmpdir}/luxctl.exe"
        fi

        local bin="luxctl.exe"
        local dest="${INSTALL_DIR}/luxctl.exe"

        if [ -w "$INSTALL_DIR" ]; then
            mv "${tmpdir}/${bin}" "$dest"
        else
            info "Writing to ${INSTALL_DIR} requires elevated permissions..."
            mv "${tmpdir}/${bin}" "$dest"
        fi

        return
    fi

    # unix: .tar.gz
    tar -xzf "${tmpdir}/${ASSET}" -C "$tmpdir"

    # binary inside the tarball is named after the asset (e.g. luxctl-macos-aarch64)
    local bin_name="${ASSET%.tar.gz}"
    if [ -f "${tmpdir}/${bin_name}" ]; then
        mv "${tmpdir}/${bin_name}" "${tmpdir}/luxctl"
    elif [ ! -f "${tmpdir}/luxctl" ]; then
        error "archive did not contain a 'luxctl' or '${bin_name}' binary"
    fi

    chmod +x "${tmpdir}/luxctl"

    # install to INSTALL_DIR, use sudo if needed
    if [ -w "$INSTALL_DIR" ]; then
        mv "${tmpdir}/luxctl" "${INSTALL_DIR}/luxctl"
    else
        info "Writing to ${INSTALL_DIR} requires elevated permissions..."
        sudo mv "${tmpdir}/luxctl" "${INSTALL_DIR}/luxctl"
    fi
}

install_via_cargo() {
    error "No pre-built binary available for ${PLATFORM}/${ARCH}. Please open an issue: https://github.com/${REPO}/issues"
    # cargo install builds without the production client secret, so API calls will fail with HTTP 403.

    if ! check_cmd cargo; then
        info "Rust not found. Installing via rustup..."
        curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y --no-modify-path

        CARGO_BIN="${CARGO_HOME:-$HOME/.cargo}/bin"
        if [ -f "$CARGO_BIN/../env" ]; then
            . "$CARGO_BIN/../env"
        fi
        if ! check_cmd cargo; then
            export PATH="$CARGO_BIN:$PATH"
        fi
    fi

    if ! check_cmd cargo; then
        error "cargo not found after installation. Please install Rust manually: https://rustup.rs"
    fi

    local cargo_version="${VERSION#v}"  # strip leading v for crates.io
    if [ -n "$cargo_version" ]; then
        cargo install luxctl --version "$cargo_version"
    else
        cargo install luxctl
    fi
}

main() {
    echo ""
    echo "  luxctl - projectlighthouse.io"
    echo ""

    detect_platform

    # default install dir for windows is user-writable
    if [ "$PLATFORM" = "windows" ] && [ "$INSTALL_DIR" = "/usr/local/bin" ]; then
        INSTALL_DIR="${USERPROFILE:-$HOME}/bin"
        mkdir -p "$INSTALL_DIR"
    fi

    resolve_asset

    if [ -z "$VERSION" ]; then
        fetch_latest_version
    fi

    if [ -n "$ASSET" ]; then
        download_binary
    else
        install_via_cargo
    fi

    # verify
    local bin_path="${INSTALL_DIR}/luxctl"
    if [ "$PLATFORM" = "windows" ]; then
        bin_path="${INSTALL_DIR}/luxctl.exe"
    fi

    if check_cmd luxctl; then
        INSTALLED_VERSION=$(luxctl --version 2>/dev/null | grep -oE '[0-9]+\.[0-9]+\.[0-9]+' | head -1 || echo "unknown")
        success "luxctl ${INSTALLED_VERSION} installed successfully!"
    elif [ -f "$bin_path" ]; then
        success "luxctl installed to ${bin_path}"
        echo ""
        echo "Make sure ${INSTALL_DIR} is in your PATH."
    else
        error "installation failed"
    fi

    echo ""
    echo "Get started:"
    echo "  luxctl auth --token <YOUR_TOKEN>"
    echo "  luxctl doctor"
    echo ""
}

main