#!/bin/bash

# constant
date_time_str="$(TZ="UST-8" date +%Y-%m-%d_%H-%M-%S)"
markdown_log_file="BenchResult_${date_time_str}.md"
api_url="https://bench.nodeloc.com/up2.php"

function green_color(){
    echo -ne "\e[1;32m"
}
function white_color(){
    echo -ne "\e[m"
}
function yellow_color(){
    echo -ne "\e[1;33m"
}

function print_help(){
    green_color
    cat <<- EOF
		BenchScript: 主机测试脚本,用于在NodeLoc.com发表测评文章
		Author: James@nodeloc.com

	EOF
    white_color
}

function install(){
    if [ -n "$(command -v apt)" ] ; then
        cmd1="apt-get"
        cmd2="apt-get install -y"
    elif [ -n "$(command -v yum)" ] ; then
        cmd1="yum"
        cmd2="yum install -y"
    elif [ -n "$(command -v dnf)" ] ; then
        cmd1="dnf"
        cmd2="dnf install -y"
    elif [ -n "$(command -v apk)" ] ; then
        cmd1="apk"
        cmd2="apk add"
    else
        echo "Error: OS Not Support"
        exit 1
    fi
    if [ -z "$updated" ] ; then
        $cmd1 update
        updated=1
    fi
    $cmd2 "$@"
}

function pre_install(){
    if [ -z "$(command -v wget)" ] ; then
        install wget
    fi
    if [ -z "$(command -v curl)" ] ; then
        install curl
    fi
}

function fetch(){
    download_url="$1"
    if [ -n "$(command -v wget)" ] ; then
        wget -qO- $download_url
    elif [ -n "$(command -v curl)" ] ; then
        curl -sLo- $download_url
    else
        install wget
        wget -qO- $download_url
    fi
}

function print_header(){
    header=$1

    if [ -n "$2" ] ; then
        title_level=$2
    else
        title_level=1
    fi

    title_prefix=" "
    for _ in $(seq 1 $title_level) ; do
        title_prefix="#"$title_prefix
    done


    green_color | tee -a $markdown_log_file
    echo -ne "$title_prefix" | tee -a $markdown_log_file
    echo -ne $header | tee -a $markdown_log_file
    white_color | tee -a $markdown_log_file
    echo -e '\n' | tee -a $markdown_log_file
}

function print_code_block(){
    echo '```' >> $markdown_log_file
    tee -a $markdown_log_file
    echo '```' >> $markdown_log_file
}

function print_markdown_block(){
    print_header $1
    print_code_block
}

function yabs(){
    fetch 'https://yabs.sh' | bash -s -- -5 -6 | print_markdown_block Yabs测试
}

function backtrace(){
    fetch 'https://raw.githubusercontent.com/zhucaidan/mtr_trace/main/mtr_trace.sh' | \
        bash  2>&1 | print_markdown_block 三网回程路由测试
}

function RegionRestrictionCheck(){
    num=0 bash <(fetch https://raw.githubusercontent.com/lmc999/RegionRestrictionCheck/main/check.sh | \
        sed -E '/请输入正确数字或直接按回车:/d') | sed -n '/正在测试/,/测试已结束/p' | \
        print_markdown_block 流媒体平台及游戏区域限制测试
}

function IPCheck(){
    fetch https://raw.githubusercontent.com/spiritLHLS/ecs/main/qzcheck.sh | \
        bash  2>&1 | print_markdown_block IP质量测试
}

function hyperspeed(){
    bash <(fetch https://bench.im/hyperspeed) | \
        print_markdown_block 单线程测速
}

function benchsh(){
    fetch bench.sh | bash | \
        print_markdown_block bench.sh测试
}

function main(){
    print_help
    pre_install

    yabs
    backtrace
    RegionRestrictionCheck

    echo -ne "\e[1;33m是否进行IP质量测试(Y/n) default Y: \e[m"
    read ans
    if [[ -z "$ans" ]]; then
        ans=y
    fi
    if [[ "$ans" == 'y' ]] || [[ "$ans" == 'Y' ]]; then
        IPCheck
    fi

    echo -ne "\e[1;33m是否进行单线程测速(Y/n) default Y: \e[m"
    read ans
    if [[ -z "$ans" ]]; then
        ans=y
    fi
    if [[ "$ans" == 'y' ]] || [[ "$ans" == 'Y' ]]; then
        hyperspeed
    fi

    echo -ne "\e[1;33m是否补充测试bench.sh(y/N) default n: \e[m"
    read ans
    if [[ "$ans" == 'y' ]] || [[ "$ans" == 'Y' ]]; then
        benchsh
    fi


if [ -n "$(command -v wget)" ] ; then
        wget -qO- --post-file "$markdown_log_file" --header="Content-Disposition: attachment; filename=\"$markdown_log_file\"" $api_url | cat
    else
        curl -X POST --header "Content-Disposition: attachment; filename=\"$markdown_log_file\"" --data-binary @"$markdown_log_file" $api_url?filename=$markdown_log_file
    fi
    echo
}

main