#!/bin/bash
echo "==================== Server Bench Script ===================="
if ! command -v bc &> /dev/null;then
    echo "检测到缺少bc，自动安装..."
    if command -v apt &> /dev/null;then
        apt update && apt install bc -y
    elif command -v yum &> /dev/null;then
        yum install bc -y
    elif command -v dnf &> /dev/null;then
        dnf install bc -y
    fi
fi

echo "CPU Bench Test..."
cpu_start=$(date +%s.%N)
for i in $(seq 1 300000); do echo $i > /dev/null; done
cpu_end=$(date +%s.%N)
cpu_time=$(echo "$cpu_end - $cpu_start" | bc)
cpu_score=$(echo "scale=0; 1000 / $cpu_time" | bc)

echo "Disk IO Bench Test..."
dd if=/dev/zero of=benchtest.tmp bs=1M count=50 oflag=direct 2>/dev/null
disk_start=$(date +%s.%N)
dd if=benchtest.tmp of=/dev/null bs=1M count=50 iflag=direct 2>/dev/null
disk_end=$(date +%s.%N)
rm -f benchtest.tmp
disk_time=$(echo "$disk_end - $disk_start" | bc)
disk_score=$(echo "scale=0; 950 / $disk_time" | bc)

echo "Network Delay Test..."
ping_sum=0
for i in $(seq 1 4); do
    t=$(ping -c1 qq.com 2>/dev/null | grep time= | awk '{print $7}' | cut -d= -f2)
    if [ -n "$t" ];then
        ping_sum=$(echo "$ping_sum + $t" | bc)
    fi
done
avg_ping=$(echo "scale=2; $ping_sum / 4" | bc)
net_score=$(echo "scale=0; 300 - $avg_ping" | bc)
if [ $net_score -lt 10 ]; then net_score=10; fi

total=$(echo "$cpu_score + $disk_score + $net_score" | bc)
echo "============================================================"
echo "CPU分数: $cpu_score"
echo "磁盘分数: $disk_score"
echo "网络分数: $net_score"
echo "综合总分: $total"
echo "============================================================"
echo "复制上面分数，回到网站手动录入提交榜单"