package main

import (
	"DockerST/task"
	"flag"
	"fmt"
)

var (
	VersionPrint bool
	Version      string
)

func init() {
	// checkUpdate()
	// 定义一个字符串类型的命令行标志
	flag.IntVar(&task.TcpPort, "tcp", 443, "TCP端口")
	flag.IntVar(&task.PingTimes, "t", 4, "Ping次数")
	flag.IntVar(&task.Routines, "r", 200, "存活检测并发数检测")
	flag.IntVar(&task.MinMS, "mis", 0, "只输出高于指定平均延迟的 IP")
	flag.IntVar(&task.MaxMS, "mxs", 1000, "只输出低于指定平均延迟的 IP")
	flag.IntVar(&task.DownloadNum, "dn", 10, "下载数量")
	flag.StringVar(&task.URL, "url", "https://cf.xiu2.xyz/url", "默认文件下载地址")
	flag.Float64Var(&task.MinSpeed, "md", 0, "最低下载速度")
	flag.BoolVar(&task.TestAll, "ta", false, "测试所有 IP")
	flag.BoolVar(&task.Disable, "dd", true, "禁止下载")
	flag.BoolVar(&VersionPrint, "v", false, "输出版本")
	flag.BoolVar(&task.IsOff, "om", false, "不下载子网列表")
	flag.Parse()
	if VersionPrint {
		fmt.Println("Version:", Version)
	}
}

func main() {
	if VersionPrint {
		return
	}
	// InitRandSeed 初始化随机数种子
	task.InitRandSeed()
	// 输出版本
	fmt.Printf("# DockerST %s \n", Version)
	pingData := task.CreateData().Run().ExcludeInvalid().SortNodesDesc()
	DownloadData := task.TestDownloadSpeed(pingData)
	for a, v := range DownloadData {
		if a == 10 {
			return
		}
		fmt.Printf("IP: %s, 延迟: %v, 下载速度: %.2f MB/s\n", v.IP.String(), v.Delay, v.DownloadSpeed)
	}
}