install.sh

#!/bin/bash

set -e

# Constants
DEFAULT_INSTALL_DIR="$HOME/.privateer/bin"
PVTR_REPO="privateerproj/privateer"
LATEST_RELEASE_URL="https://api.github.com/repos/${PVTR_REPO}/releases/latest"

# Detect OS (darwin = macOS, linux = Linux, msys or cygwin = Windows)
OS=""
case "$(uname -s)" in
    Darwin)
        OS="darwin"
        ;;
    Linux)
        OS="linux"
        ;;
    CYGWIN*|MSYS*|MINGW*)
        OS="windows"
        ;;
    *)
        echo "Unsupported Environment: $(uname -s)"
        exit 1
        ;;
esac

# Detect Architecture (x86_64 = amd64, arm64 = arm64, i386/i686 = 386)
ARCH=""
case "$(uname -m)" in
    x86_64)
        ARCH="x86_64"
        ;;
    i386)
        ARCH="i386"
        ;;
    arm64)
        ARCH="arm64"
        ;;
    *)
        echo "Unsupported Architecture: $(uname -m)"
        exit 1
        ;;
esac

download_latest_release() {
    local install_dir="$1"
    local install_file="$install_dir/pvtr"

    # Ensure the directory exists
    mkdir -p "$install_dir"

    # Build the grep pattern based on OS
    local pattern
    if [[ "$OS" == "darwin" ]]; then
        pattern="browser_download_url.*${OS}.*"
    else
        pattern="browser_download_url.*${OS}.*${ARCH}.*"
    fi

    # Fetch the download URL for the latest release
    local url
    url=$(curl -s ${LATEST_RELEASE_URL} | grep -i "$pattern" | cut -d '"' -f 4)

    if [[ -z "$url" ]]; then
        echo "Failed to fetch the download URL for the latest release."
        exit 1
    fi

    echo "Downloading from: $url"

    # Download the binary to the specified install directory
    curl -L -0 "$url" | tar xvf - -C "$install_dir"

    if [[ $? -ne 0 ]]; then
        echo "Failed to download the binary."
        exit 1
    fi

    # Ensure the binary is executable
    chmod +x "$install_file"

    echo "Downloaded binary to $install_file"
}

update_path() {
    local install_dir="$1"

    # Check if the install directory is already in PATH
    if [[ ":$PATH:" != *":$install_dir:"* ]]; then
        echo "$install_dir is not in the PATH."

        # Detect current shell
        current_shell=$(basename "$SHELL")

        case "$current_shell" in
            bash)
                config_file="$HOME/.bash_profile"
                ;;
            zsh)
                config_file="$HOME/.zshrc"
                ;;
            fish)
                config_file="$HOME/.config/fish/config.fish"
                ;;
            *)
                echo "Unsupported shell: $current_shell. You may need to manually add $install_dir to your PATH."
                return
                ;;
        esac

        # Check if the path is already added to the config file
        if ! grep -q "$install_dir" "$config_file"; then
            echo "export PATH=\"$install_dir:\$PATH\"" >> "$config_file"
            echo "$install_dir added to $config_file"
            source $config_file
        else
            echo "$install_dir is already in $config_file."
        fi
    else
        echo "$install_dir is already in the PATH."
    fi
}

# Main logic
main() {
    local install_dir="$DEFAULT_INSTALL_DIR"

    # Handle CLI arguments for installation path override
    while getopts "p:" opt; do
        case $opt in
            p)
                install_dir="$OPTARG"
                ;;
            *)
                echo "Usage: $0 [-p install_path]"
                exit 1
                ;;
        esac
    done

    mkdir -p "$install_dir"

    # Download the latest release
    download_latest_release "$install_dir"

    # Ensure the binary is accessible via PATH
    update_path "$install_dir"

    echo "pvtr installation complete!"
}

main "$@"