#!/bin/sh
# vim: set noexpandtab tabstop=4 shiftwidth=4 softtabstop=4 :
FORCEVPS=false
if [ "$1" = "forcevps" ]; then
	INTERFACE="$2"
	FORCEVPS=true
else
	INTERFACE="$1"
fi

[ -n "$INTERFACE" ] && [ ! -d "/sys/class/net/$INTERFACE" ] && {
	echo "You must use a real interface. You wan find them using 'ip a' for example"
	exit 0
}

echo "Select best test server..."
HOSTLST="https://repo.huaweicloud.com/debian-cd/10.0.0/amd64/iso-dvd/debian-10.0.0-amd64-DVD-2.iso https://mirrors.aliyun.com/ubuntu-releases/impish/ubuntu-21.10-desktop-amd64.iso http://mirrors.163.com/ubuntukylin-cd/20.04/ubuntukylin-20.04-pro-amd64.iso https://mirrors.cloud.tencent.com/ubuntu-cdimage/ubuntu/daily-canary/20220209/jammy-desktop-canary-amd64.iso"
bestping="9999"
for pinghost in $HOSTLST; do
	domain=$(echo $pinghost | awk -F/ '{print $3}')
	if [ -z "$INTERFACE" ] || [ "$FORCEVPS" = true ]; then
		ping=$(ping -c1 -w2 $domain | cut -d "/" -s -f5 | cut -d "." -f1 | tr -d '\n')
	else
		ping=$(ping -c1 -w2 -I $INTERFACE -B $domain | cut -d "/" -s -f5 | cut -d "." -f1 | tr -d '\n')
	fi
	echo "host: $domain - ping: $ping"
	if [ -n "$ping" ] && [ "$ping" -lt "$bestping" ]; then
		bestping=$ping
		HOST=$pinghost
	fi
done

[ -z "$HOST" ] && HOST="https://speed.hetzner.de/10GB.bin"

echo "Best server is $HOST, running test:"
trap : HUP INT TERM
if [ -z "$INTERFACE" ]; then
	curl -4 -o /dev/null $HOST || echo
else
	domain=$(echo $HOST | awk -F/ '{print $3}')
	hostip=$(dig +nocmd +noall +answer A $domain | grep -v CNAME | awk '{print $5}' | tr '\n' ' ')
	if [ -n "$(ipset list 2>/dev/null | grep ss_rules)" ]; then
		for ip in $hostip; do
			ipset add ss_rules_dst_bypass_all $ip
		done
	fi
	curl -4 -o /dev/null --interface $INTERFACE $HOST || echo
	if [ -n "$(ipset list 2>/dev/null | grep ss_rules)" ]; then
		for ip in $hostip; do
			ipset del ss_rules_dst_bypass_all $ip
		done
	fi
fi