#!/bin/bash

# 检查并安装依赖项 jq 和 bc
if ! command -v jq &> /dev/null
then
    echo "jq could not be found, installing now..."
    sudo apt-get update && sudo apt-get install -y jq
fi

if ! command -v bc &> /dev/null
then
    echo "bc could not be found, installing now..."
    sudo apt-get update && sudo apt-get install -y bc
fi

# 输出设置 crontab 的命令
echo "To run this script every day at 3 AM, add the following line to your crontab:"
echo "0 3 * * * /path/to/this/script.sh"
# 测速脚本部分
# 使用 dig 命令查询 DNS 记录并写入 ip.txt
dig cf.youoffer.net A +short > ip.txt

# 创建一个新的文件 final_ips.txt
touch final_ips.txt

# 读取 ip.txt 中的每个 IP 地址进行下载测试
while IFS= read -r ip
do
    echo "Starting download test for $ip"
    # 使用 curl 命令进行下载测试
    start_time=$(date +%s.%N)
    curl -s --resolve speed.cloudflare.com:443:"$ip" https://speed.cloudflare.com/__down?bytes=10000000 -o /dev/null --connect-timeout 5 --max-time 10
    curl_status=$?
    end_time=$(date +%s.%N)
    elapsed_time=$(echo "$end_time - $start_time" | bc)
    
    # 检查 curl 命令的返回状态
    if [ $curl_status -eq 0 ]; then
        # 计算下载速度
        download_speed=$(echo "scale=2; (10000000*8)/(($elapsed_time)*1000000)" | bc)
        
        # 显示下载速度
        echo "$ip has a download speed of $download_speed Mbps"
        
        # 如果下载速度大于等于 50Mbps
        if (( $(echo "$download_speed >= 50" | bc -l) )); then
            # 将大于等于 50Mbps 的 IP 写入 final_ips.txt
            echo "$ip" >> final_ips.txt
        fi
    else
        echo "Download test for $ip failed with status $curl_status"
    fi
done < ip.txt

echo "Download tests completed. IPs with download speed >= 50Mbps are written to final_ips.txt"

# Cloudflare DDNS 更新部分
# 以下变量需要填写
API_KEY="your_cloudflare_api_key"
EMAIL="your_email@example.com"
ZONE_ID="your_zone_id"
DOMAIN="your_domain.com"
SUBDOMAIN="sub.your_domain.com"
TTL=300

# 使用Cloudflare API获取子域名的所有A记录
records=$(curl -s -X GET "https://api.cloudflare.com/client/v4/zones/${ZONE_ID}/dns_records?type=A&name=${SUBDOMAIN}" \
-H "X-Auth-Email: ${EMAIL}" \
-H "X-Auth-Key: ${API_KEY}" \
-H "Content-Type: application/json")

# 从返回值中提取出所有A记录的ID
record_ids=$(echo "$records" | jq -r '.result[].id')

# 遍历所有A记录的ID，并使用Cloudflare API删除DNS记录
for record_id in $record_ids
do
    response=$(curl -s -X DELETE "https://api.cloudflare.com/client/v4/zones/${ZONE_ID}/dns_records/${record_id}" \
    -H "X-Auth-Email: ${EMAIL}" \
    -H "X-Auth-Key: ${API_KEY}" \
    -H "Content-Type: application/json")

    if echo "$response" | grep -q '"success":true'; then
        echo "DNS record with ID $record_id deleted successfully"
    else
        echo "Failed to delete DNS record with ID $record_id"
    fi
done

# 从final_ips.txt文件中读取IP地址，并使用Cloudflare API添加DNS记录
while IFS= read -r ip
do
    # 使用Cloudflare API添加DNS记录
    response=$(curl -s -X POST "https://api.cloudflare.com/client/v4/zones/${ZONE_ID}/dns_records" \
    -H "X-Auth-Email: ${EMAIL}" \
    -H "X-Auth-Key: ${API_KEY}" \
    -H "Content-Type: application/json" \
    --data '{"type":"A","name":"'${SUBDOMAIN}'","content":"'${ip}'","ttl":'${TTL}',"proxied":false}')

    # 检查返回值中的success字段是否为true
    if echo "$response" | grep -q '"success":true'; then
        echo "DNS record for IP $ip added successfully"
    else
        echo "Failed to add DNS record for IP $ip"
    fi
done < final_ips.txt