#!/bin/bash
# install hrp with one shell command
# bash -c "$(curl -ksSL https://httprunner.com/script/install.sh)"

set -e

function echoError() {
    echo -e "\033[31m✘ $1\033[0m" # red
}
export -f echoError

function echoInfo() {
    echo -e "\033[32m✔ $1\033[0m" # green
}
export -f echoInfo

function echoWarn() {
    echo -e "\033[33m! $1\033[0m" # yellow
}
export -f echoError

github_api_url="https://api.github.com/repos/httprunner/httprunner/releases/latest"

function get_latest_version() {
    # get latest release version from GitHub API
    version=$(curl -s $github_api_url | grep '"tag_name":' | sed -E 's/.*"([^"]+)".*/\1/')
    echo "$version"
}

function get_os() {
    os=$(uname -s)
    echo "$os" | tr '[:upper:]' '[:lower:]'
}

function get_arch() {
    arch=$(uname -m)
    if [ "$arch" == "x86_64" ]; then
        arch="amd64"
    fi
    echo "$arch"
}

function main() {
    echoInfo "Detect target hrp package..."
    version=$(get_latest_version)
    if [[ $version != v* ]]; then
        echo "get hrp latest version failed:"
        echo "$version"
        exit 1
    fi
    echo "Latest version: $version"

    os=$(get_os)
    echo "Current OS: $os"
    if [[ $os == mingw* ]]; then
        echoWarn "Current OS is MinGW, try to use windows package"
        os="windows"
    fi

    arch=$(get_arch)
    echo "Current ARCH: $arch"
    pkg_suffix=".tar.gz"
    pkg="hrp-$version-$os-$arch$pkg_suffix"
    echo "Download package: $pkg"

    download_url=$(curl -s $github_api_url | grep "browser_download_url.*$pkg" | cut -d '"' -f 4)
    echo "Download url: $download_url"
    echo

    echoInfo "Downloading..."
    echo "$ curl -kL $download_url -o $pkg"
    curl -kL $download_url -o "$pkg"
    echo

    # for windows, only extract package to current directory
    if [[ $os == windows ]]; then # windows
        # extract to current directory
        echoInfo "Extracting..."
        echo "$ unzip -o $pkg -d ."
        unzip -o $pkg -d .

        echo "$ hrp.exe -v"
        hrp.exe -v
        echo "$ hrp.exe -h"
        hrp.exe -h
        exit 0
    fi

    # for linux or darwin, install hrp to /usr/local/bin
    # extract to temp directory
    echoInfo "Created temp dir..."
    echo "$ mktemp -d -t hrp.XXXX"
    tmp_dir=$(mktemp -d -t hrp.XXXX)
    echo "$tmp_dir"
    echo "$ mv $pkg $tmp_dir && cd $tmp_dir"
    mv $pkg $tmp_dir
    cd "$tmp_dir"
    echo

    echoInfo "Extracting..."
    echo "$ tar -xzf $pkg"
    tar -xzf "$pkg"

    echo "$ ls -lh"
    ls -lh
    echo

    echoInfo "Installing..."
    if hrp -v > /dev/null && [ $(command -v hrp) != "./hrp" ]; then
        echoWarn "$(hrp -v) exists, remove first !!!"
        echo "$ rm -rf $(command -v hrp)"
        rm -rf "$(command -v hrp)"
    fi

    echo "$ chmod +x hrp && mv hrp /usr/local/bin/"
    chmod +x hrp
    mv hrp /usr/local/bin/
    echo

    echoInfo "Check installation..."
    echo "$ command -v hrp"
    command -v hrp
    echo "$ hrp -v"
    hrp -v
    echo "$ hrp -h"
    hrp -h
    echo

    if [[ -f $HOME/.hrp/venv/bin/pip3 ]]; then
        echoInfo "Upgrade httprunner..."
        echo "$ $HOME/.hrp/venv/bin/pip3 install --upgrade httprunner==$version --index-url https://pypi.org/simple"
        $HOME/.hrp/venv/bin/pip3 install --upgrade httprunner==$version --index-url https://pypi.org/simple
    fi
}

main