#!/bin/bash
# roborev installer
# Usage: curl -fsSL https://raw.githubusercontent.com/roborev-dev/roborev/main/scripts/install.sh | bash

set -e

REPO="roborev-dev/roborev"
BINARY_NAME="roborev"

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

info() { echo -e "${GREEN}$1${NC}"; }
warn() { echo -e "${YELLOW}$1${NC}"; }
error() { echo -e "${RED}$1${NC}" >&2; exit 1; }

# Detect OS
detect_os() {
    case "$(uname -s)" in
        Darwin) echo "darwin" ;;
        Linux) echo "linux" ;;
        MINGW*|MSYS*|CYGWIN*) echo "windows" ;;
        *) error "Unsupported OS: $(uname -s)" ;;
    esac
}

# Detect architecture
detect_arch() {
    case "$(uname -m)" in
        x86_64|amd64) echo "amd64" ;;
        aarch64|arm64) echo "arm64" ;;
        armv7*|armhf) echo "arm" ;;
        *) error "Unsupported architecture: $(uname -m)" ;;
    esac
}

# Find install directory
find_install_dir() {
    if [ -w "/usr/local/bin" ]; then
        echo "/usr/local/bin"
    elif [ -w "$HOME/.local/bin" ]; then
        mkdir -p "$HOME/.local/bin"
        echo "$HOME/.local/bin"
    else
        mkdir -p "$HOME/.local/bin"
        echo "$HOME/.local/bin"
    fi
}

# Download with curl or wget
download() {
    local url="$1"
    local output="$2"
    if command -v curl &>/dev/null; then
        curl -fsSL "$url" -o "$output"
    elif command -v wget &>/dev/null; then
        wget -q "$url" -O "$output"
    else
        error "Neither curl nor wget found"
    fi
}

# Get latest release version
get_latest_version() {
    local url="https://api.github.com/repos/${REPO}/releases/latest"
    if command -v curl &>/dev/null; then
        curl -fsSL "$url" | grep '"tag_name"' | head -1 | cut -d'"' -f4
    elif command -v wget &>/dev/null; then
        wget -qO- "$url" | grep '"tag_name"' | head -1 | cut -d'"' -f4
    fi
}

# Install from GitHub releases
install_from_release() {
    local os="$1"
    local arch="$2"
    local install_dir="$3"

    info "Fetching latest release..."
    local version=$(get_latest_version)

    if [ -z "$version" ]; then
        return 1
    fi

    info "Found version: $version"

    local platform="${os}_${arch}"
    local filename="roborev_${version#v}_${platform}.tar.gz"
    local url="https://github.com/${REPO}/releases/download/${version}/${filename}"

    local tmpdir=$(mktemp -d)
    trap "rm -rf $tmpdir" EXIT

    info "Downloading ${filename}..."
    if ! download "$url" "$tmpdir/release.tar.gz"; then
        return 1
    fi

    info "Extracting..."
    tar -xzf "$tmpdir/release.tar.gz" -C "$tmpdir"

    # Install binary
    if [ -f "$tmpdir/roborev" ]; then
        if [ -w "$install_dir" ]; then
            mv "$tmpdir/roborev" "$install_dir/"
        else
            sudo mv "$tmpdir/roborev" "$install_dir/"
        fi
        chmod +x "$install_dir/roborev"
    fi

    # macOS code signing
    if [ "$os" = "darwin" ] && [ -f "$install_dir/roborev" ]; then
        codesign -s - "$install_dir/roborev" 2>/dev/null || true
    fi

    return 0
}

# Install using go install
install_from_go() {
    local install_dir="$1"

    if ! command -v go &>/dev/null; then
        return 1
    fi

    info "Installing via 'go install'..."
    go install "github.com/${REPO}/cmd/roborev@latest"

    return 0
}

# Main
main() {
    info "Installing roborev..."
    echo

    local os=$(detect_os)
    local arch=$(detect_arch)
    local install_dir=$(find_install_dir)

    info "Platform: ${os}/${arch}"
    info "Install directory: ${install_dir}"
    echo

    # Try release first, then go install
    if install_from_release "$os" "$arch" "$install_dir"; then
        info "Installed from GitHub release"
    elif install_from_go "$install_dir"; then
        info "Installed via go install"
        install_dir="$(go env GOPATH)/bin"
    else
        error "Installation failed. Install Go from https://go.dev and try again."
    fi

    echo
    info "Installation complete!"
    echo

    # Check PATH
    if ! echo "$PATH" | grep -q "$install_dir"; then
        warn "Add this to your shell profile:"
        echo "  export PATH=\"\$PATH:$install_dir\""
        echo
    fi

    echo "Get started:"
    echo "  cd your-repo"
    echo "  roborev init"
    echo
    echo "For AI agent skills (Claude Code, Codex):"
    echo "  roborev skills install"
}

main "$@"