#!/bin/bash

usrPath="/usr/local/bin"

checkRootPermit() {
    [[ $EUID -ne 0 ]] && echo "请使用sudo/root权限运行本脚本" && exit 1
}

checkSystemArch() {
    arch=$(uname -m)
    if [[ $arch == "x86_64" ]]; then
        archParam="amd64"
    fi

    if [[ $arch == "aarch64" ]]; then
        archParam="arm64"
    fi

    if [[ $arch == "arm64" ]]; then
        archParam="arm64"
    fi

    if [[ $archParam == "" ]]; then
        echo "未知的系统架构，请联系作者"
        exit 1
    fi
}

checkSystemDistribution() {
    case "$OSTYPE" in
    darwin*)
        osDistribution="darwin"
        downPath="/var/tmp/nexttrace"
        ;;
    linux*)
        osDistribution="linux"
        downPath="/var/tmp/nexttrace"
        ;;
    *)
        echo "unknown: $OSTYPE"
        exit 1
        ;;
    esac
}

getLocation() {
    echo "正在获取地理位置信息..."
    countryCode=$(curl -s "http://ip-api.com/line/?fields=countryCode")
}

installWgetPackage() {
    # macOS should install wget originally. Nothing to do
    echo "wget 正在安装中..."
    # try apt
    apt-get -h >/dev/null
    if [ $? -eq 0 ]; then
        # 先更新一下数据源，有些机器数据源比较老可能会404
        apt-get update -y >/dev/null
        apt-get install wget -y >/dev/null
    fi

    # try yum
    yum -h >/dev/null
    if [ $? -eq 0 ]; then
        yum -y update >/dev/null
        yum install wget -y >/dev/null
    fi

    # try dnf
    dnf -h >/dev/null
    if [ $? -eq 0 ]; then
        dnf check-update >/dev/null
        dnf install wget -y >/dev/null
    fi

    # try pacman
    pacman -h >/dev/null
    if [ $? -eq 0 ]; then
        pacman -Sy >/dev/null
        pacman -S wget >/dev/null
    fi

    wget -h >/dev/null
    if [ $? -ne 0 ]; then
        echo "wget 安装失败"
        exit 1
    fi
}

installJqPackage() {
    # macOS should install wget originally. Nothing to do
    echo "jq 正在安装中..."
    # try apt
    apt-get -h >/dev/null
    if [ $? -eq 0 ]; then
        # 先更新一下数据源，有些机器数据源比较老可能会404
        apt-get update -y >/dev/null
        apt-get install jq -y >/dev/null
    fi

    # try yum
    yum -h >/dev/null
    if [ $? -eq 0 ]; then
        yum -y update >/dev/null
        yum install jq -y >/dev/null
    fi

    # try dnf
    dnf -h >/dev/null
    if [ $? -eq 0 ]; then
        dnf check-update >/dev/null
        dnf install jq -y >/dev/null
    fi

    # try pacman
    pacman -h >/dev/null
    if [ $? -eq 0 ]; then
        pacman -Sy >/dev/null
        pacman -S jq >/dev/null
    fi

    jq -h >/dev/null
    if [ $? -ne 0 ]; then
        echo "jq 安装失败"
        exit 1
    fi
}

checkWgetPackage() {
    wget -h >/dev/null
    if [ $? -ne 0 ]; then
        read -r -p "您还没有安装wget，是否安装? (y/n)" input

        case $input in
        [yY][eE][sS] | [yY])
            installWgetPackage
            ;;

        [nN][oO] | [nN])
            echo "您选择了取消安装，脚本即将退出"
            exit 1
            ;;

        *)
            installWgetPackage
            ;;
        esac
    fi
}

checkVersion() {
    echo "正在检查版本..."
    version=$(curl -sL https://api.github.com/repos/xgadget-lab/nexttrace/releases/latest | jq -r '.tag_name')
    if [[ $version == "" ]]; then
        echo "获取版本失败，请检查网络连接"
        exit 1
    fi
    echo 当前最新release版本：${version}
    read -r -p "是否安装/更新软件? (y/n)" input
    case $input in
    [yY][eE][sS] | [yY])
        break
        ;;
    [nN][oO] | [nN])
        echo "您选择了取消安装/更新，脚本即将退出"
        exit 1
        ;;
    *)
        break
        ;;
    esac
}

checkJqPackage() {
    jq -h >/dev/null
    if [ $? -ne 0 ]; then
        echo "您还没有安装jq， 当您取消安装，我们会使用awk获取当前版本号。"
        read -r -p "但是如遇Github变更API，这可能会存在问题，是否安装? (y/n)" input

        case $input in
        [yY][eE][sS] | [yY])
            installJqPackage
            ;;

        [nN][oO] | [nN])
            echo "您选择了取消安装"
            return 0
            ;;

        *)
            installJqPackage
            ;;
        esac
    fi
    return 1
}

downloadBinrayFile() {
    echo "正在获取最新版的 NextTrace 发行版文件信息..."
    checkJqPackage
    # 简单说明一下，Github提供了一个API，可以获取最新发行版本的二进制文件下载地址（对应的是browser_download_url），根据刚刚测得的osDistribution、archParam，获取对应的下载地址
    if [[ $? -eq 1 ]]; then
        # 支持 jq 不回退
        # echo nexttrace_${osDistribution}_${archParam}
        latestURL=$(curl -s https://api.github.com/repos/xgadget-lab/nexttrace/releases/latest | jq ".assets[] | select(.name == \"nexttrace_${osDistribution}_${archParam}\") | .browser_download_url")
        latestURL=${latestURL:1:-1}
    else
        # 不支持 jq，用户拒绝安装，回退 awk
        latestURL=$(curl -s https://api.github.com/repos/xgadget-lab/nexttrace/releases/latest | grep -i "browser_download_url.*${osDistribution}.*${archParam}" | awk -F '"' '{print $4}')
    fi
    if [ "$countryCode" == "CN" ]; then
        read -r -p "检测到国内网络环境，是否使用镜像下载以加速(y/n)" input
        case $input in
        [yY][eE][sS] | [yY])
            latestURL="https://ghproxy.com/"$latestURL
            ;;

        [nN][oO] | [nN])
            echo "您选择了不使用镜像，下载可能会变得异常缓慢，或者失败"
            ;;

        *)
            latestURL="https://ghproxy.com/"$latestURL
            ;;
        esac
    fi

    echo "正在下载 NextTrace 二进制文件..."
    wget -O ${downPath} ${latestURL} >/dev/null
    if [ $? -eq 0 ]; then
        echo "NextTrace 现在已经在您的系统中可用"
        changeMode
        mv ${downPath} ${usrPath}
        if [[ ${osDistribution} == "macOS" ]]; then
            xattr -r -d com.apple.quarantine ${usrPath}/nexttrace
        fi
    else
        echo "NextTrace 下载失败，请检查您的网络是否正常"
        exit 1
    fi
}

changeMode() {
    chmod +x ${downPath} >/dev/null
}

runBinrayFileHelp() {
    if [ -e ${usrPath} ]; then
        ${usrPath}/nexttrace -h
    fi
}

# Check Procedure
checkRootPermit
checkSystemDistribution
checkSystemArch
checkWgetPackage
checkVersion

# Download Procedure
getLocation
downloadBinrayFile

# Run Procedure
runBinrayFileHelp