nodeloc_vps_test/nodeloc_vps_autotest_speed.sh

215 lines
6.5 KiB
Bash
Raw Normal View History

2024-06-27 15:15:11 +00:00
#!/bin/bash
# 定义版本
VERSION="1.0.0"
# 定义颜色
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color
# 检查 root 权限并获取 sudo 权限
if [ "$(id -u)" != "0" ]; then
echo "此脚本需要 root 权限运行。"
if ! sudo -v; then
echo "无法获取 sudo 权限,退出脚本。"
exit 1
fi
echo "已获取 sudo 权限。"
fi
# 检查并安装依赖
install_dependencies() {
echo -e "${YELLOW}正在检查并安装必要的依赖项...${NC}"
# 更新包列表
if ! sudo apt-get update; then
echo -e "${RED}无法更新包列表。请检查您的网络连接和系统设置。${NC}"
exit 1
fi
# 安装依赖
local dependencies=(
"curl"
"wget"
)
for dep in "${dependencies[@]}"; do
if ! command -v "$dep" &> /dev/null; then
echo -e "${YELLOW}正在安装 $dep...${NC}"
if ! sudo apt-get install -y "$dep"; then
echo -e "${RED}无法安装 $dep。请手动安装此依赖项。${NC}"
fi
else
echo -e "${GREEN}$dep 已安装。${NC}"
fi
done
echo -e "${GREEN}依赖项检查和安装完成。${NC}"
clear
}
# 统计使用次数
sum_run_times() {
local COUNT
COUNT=$(wget --no-check-certificate -qO- --tries=2 --timeout=2 "https://hits.seeyoufarm.com/api/count/incr/badge.svg?url=https://github.com/everett7623/nodeloc_vps_test/blob/main/nodeloc_vps_autotest.sh" 2>&1 | grep -m1 -oE "[0-9]+[ ]+/[ ]+[0-9]+")
if [[ -n "$COUNT" ]]; then
daily_count=$(cut -d " " -f1 <<< "$COUNT")
total_count=$(cut -d " " -f3 <<< "$COUNT")
else
echo "Failed to fetch usage counts."
daily_count=0
total_count=0
fi
}
# 调用函数获取统计数据
sum_run_times
# 输出欢迎信息
show_welcome() {
echo ""
echo -e "${RED}---------------------------------By'Jensfrank---------------------------------${NC}"
echo ""
echo "Nodeloc_VPS_自动脚本测试 $VERSION"
echo "GitHub地址: https://github.com/everett7623/nodeloc_vps_test"
echo "VPS选购: https://www.nodeloc.com/vps"
echo ""
echo -e "${YELLOW}# # ##### #### ###### # #### #### # # #### ####${NC}"
echo -e "${YELLOW}## # # # # # # # # # # # # # # # # #${NC}"
echo -e "${YELLOW}# # # # # # # ##### # # # # # # #### ####${NC}"
echo -e "${YELLOW}# # # # # # # # # # # # # # # #${NC}"
echo -e "${YELLOW}# # # # # # # # # # # # # # # # # #${NC}"
echo -e "${YELLOW}# ## ##### #### ###### ###### #### #### #### # ####${NC}"
echo ""
echo "支持Ubuntu/Debian"
echo ""
echo -e "今日运行次数: ${RED}$daily_count${NC} 次,累计运行次数: ${RED}$total_count${NC}"
echo ""
echo -e "${RED}---------------------------------By'Jensfrank---------------------------------${NC}"
echo ""
echo "一键脚本将测试以下项目:"
2024-06-27 16:34:28 +00:00
echo "多线程测试(调试中)"
echo "单线程测试(调试中)"
2024-06-27 15:15:11 +00:00
echo ""
echo -e "${RED}按任意键开始测试...${NC}"
read -n 1 -s
clear
}
# 定义一个数组来存储每个命令的输出
declare -a test_results
# 在每个命令执行后保存结果
run_and_capture() {
local command_output
command_output=$(eval "$1" 2>&1)
test_results+=("$command_output")
echo "$command_output"
}
2024-06-28 14:15:09 +00:00
# 去除三网测速板块板块ANSI转义码并截取
2024-06-28 13:26:13 +00:00
process_speedtest_output() {
2024-06-27 16:34:28 +00:00
local input="$1"
2024-06-28 13:26:13 +00:00
local keyword="$2"
2024-06-28 14:04:22 +00:00
echo "Debug: Processing for keyword: $keyword" >&2
2024-06-28 13:26:13 +00:00
# 去除 ANSI 转义码
2024-06-27 17:39:04 +00:00
local no_ansi
no_ansi=$(echo "$input" | sed -E 's/\x1b\[[0-9;]*[a-zA-Z]//g')
2024-06-28 12:46:43 +00:00
2024-06-28 14:04:22 +00:00
echo "Debug: Input after ANSI removal (first 500 chars):" >&2
echo "${no_ansi:0:500}" >&2
2024-06-28 13:26:13 +00:00
# 计算关键字出现的次数
2024-06-28 12:46:43 +00:00
local count
2024-06-28 13:26:13 +00:00
count=$(echo "$no_ansi" | grep -Fc "$keyword")
2024-06-28 12:46:43 +00:00
2024-06-28 14:04:22 +00:00
echo "Debug: Keyword count: $count" >&2
2024-06-28 13:26:13 +00:00
# 提取所需的测试结果并过滤进度条
2024-06-28 14:04:22 +00:00
local result
result=$(echo "$no_ansi" | awk -v count="$count" -v key="$keyword" '
BEGIN { found = 0 }
2024-06-28 14:15:09 +00:00
$0 ~ key {
2024-06-28 14:04:22 +00:00
print "Debug: Found keyword, remaining count: " count > "/dev/stderr"
2024-06-28 12:46:43 +00:00
if (--count == 0) {
2024-06-28 14:04:22 +00:00
found = 1
print "Debug: Starting to capture output" > "/dev/stderr"
2024-06-28 12:46:43 +00:00
}
}
2024-06-28 14:04:22 +00:00
found && !/测试进行中/ && NF {
print "Debug: Capturing line: " $0 > "/dev/stderr"
print
}
2024-06-28 14:15:09 +00:00
/------------------------ 多功能 自更新 测速脚本 ------------------------/ && count == 0 {
2024-06-28 14:04:22 +00:00
print "Debug: Reached end marker" > "/dev/stderr"
exit
}
')
echo "Debug: Result (first 500 chars):" >&2
echo "${result:0:500}" >&2
echo "$result"
2024-06-28 13:26:13 +00:00
}
2024-06-28 14:15:09 +00:00
speedtest_process_output() {
echo "Debug: Calling speedtest process" >&2
process_speedtest_output "$1" "多功能 自更新 测速脚本"
2024-06-27 16:34:28 +00:00
}
2024-06-27 15:15:11 +00:00
# 运行所有测试
run_all_tests() {
echo -e "${RED}开始测试,测试时间较长,请耐心等待...${NC}"
# 三网测速
echo -e "运行${YELLOW}三网测速(多线程/单线程)...${NC}"
2024-06-27 16:54:36 +00:00
echo -e "目前默认选择${YELLOW}大陆三网+教育网 IPv4多线程/单线程)测试...${NC}"
2024-06-27 15:15:11 +00:00
speedtest_multi_result=$(run_and_capture "echo '1' | bash <(curl -sL bash.icu/speedtest)")
speedtest_single_result=$(run_and_capture "echo '2' | bash <(curl -sL bash.icu/speedtest)")
# 格式化结果
echo -e "${YELLOW}此报告由Nodeloc_VPS_自动脚本测试生成...${NC}"
format_results
}
# 格式化结果为 Markdown
format_results() {
2024-06-27 16:34:28 +00:00
# 处理三网测速结果
2024-06-28 14:15:09 +00:00
local processed_speedtest_multi_result=$(speedtest_process_output "$speedtest_multi_result")
local processed_speedtest_single_result=$(speedtest_process_output "$speedtest_single_result")
2024-06-27 16:34:28 +00:00
result="[tabs]
2024-06-27 15:15:11 +00:00
[tab=\"多线程测速\"]
\`\`\`
$processed_speedtest_multi_result
\`\`\`
[/tab]
[tab=\"单线程测速\"]
\`\`\`
$processed_speedtest_single_result
\`\`\`
[/tab]
[/tabs]"
echo "$result" > results.md
2024-06-27 16:47:57 +00:00
echo -e "${YELLOW}结果已保存到 results.md 文件中。${NC}"
2024-06-27 15:15:11 +00:00
}
# 主函数
main() {
install_dependencies
show_welcome
run_all_tests
2024-06-27 16:47:57 +00:00
echo -e "${YELLOW}所有测试完成可到results.md复制到Nodeloc使用。${NC}"
2024-06-27 15:15:11 +00:00
read -n 1 -s
2024-06-27 16:47:57 +00:00
echo "最终测试结果如下:" >&2
2024-06-27 15:15:11 +00:00
cat results.md >&2
}
main