1.0.0 版本

README.md

@@ -5,15 +5,3 @@ Docker官方使用的是CloudFlare的CDN 优选CloudFlare IP后即可实现高
 
 ## 安装本程序
 
-```bash
-
-```
-
-## 使用代理优选加速(推荐)
-
-```bash
-https://docker.sxh.workers.dev/
-```
-
-## 直接加速
-

main.go

@@ -38,12 +38,14 @@ func init() {
 	flag.StringVar(&task.IPFile, "f", "ip.txt", "IP段数据文件")
 	flag.StringVar(&task.IPText, "ip", "", "指定IP段数据")
 	flag.StringVar(&utils.Output, "o", "result.csv", "输出结果文件")
-	flag.BoolVar(&task.IsOff, "om", false, "关闭在线读取列表")
+
+	flag.BoolVar(&task.IsOff, "off", false, "关闭在线读取列表")
 	flag.BoolVar(&task.Disable, "dd", false, "禁用下载测速")
 	flag.BoolVar(&task.TestAll, "allip", false, "测速全部 IP")
 
 	flag.BoolVar(&printVersion, "v", false, "打印程序版本")
-	flag.Parse()
+	flag.StringVar(&utils.DefaultDockerUrl, "docker-url", utils.DefaultDockerUrl, "Docker 地址")
+	flag.BoolVar(&task.IsOff, "om", false, "关闭在线读取列表")
 
 	if task.MinSpeed > 0 && time.Duration(maxDelay)*time.Millisecond == utils.InputMaxDelay {
 		fmt.Println("[小提示] 在使用 [-sl] 参数时，建议搭配 [-tl] 参数，以避免因凑不够 [-dn] 数量而一直测速...")
@@ -69,15 +71,13 @@ func init() {
 
 func main() {
 	task.InitRandSeed() // 置随机数种子
-
+	fmt.Printf("# sxhoio/DockerST %s \n\n", version)
-	fmt.Printf("# XIU2/CloudflareSpeedTest %s \n\n", version)
-
 	// 开始延迟测速 + 过滤延迟/丢包
 	pingData := task.NewPing().Run().FilterDelay().FilterLossRate()
 	// 开始下载测速
 	speedData := task.TestDownloadSpeed(pingData)
-	utils.ExportCsv(speedData) // 输出文件
+
-	speedData.Print()          // 打印结果
+	speedData.DockerSet() // 替换节点
 
 	if versionNew != "" {
 		fmt.Printf("\n*** 发现新版本 [%s]！请前往 [https://github.com/sxhoio/DockerST] 更新！ ***\n", versionNew)

utils/DockerRep.go

@@ -0,0 +1,197 @@
+package utils
+
+import (
+	"bufio"
+	"encoding/json"
+	"fmt"
+	"os"
+	"runtime"
+	"strings"
+)
+
+var (
+	DefaultDockerUrl = "http://docker.sxh.workers.dev"
+)
+
+func (s DownloadSpeedSet) DockerSet() {
+	// 选择最优的节点
+	bestIP := s[0].IP
+	bestSpeed := s[0].DownloadSpeed
+	// 自动优选节点 最高速度为 0 时，不进行优选
+	if bestSpeed == 0 {
+		fmt.Println("\n[信息] 未找到最优节点，跳过优选节点。")
+		return
+	} else {
+		fmt.Println("\n[信息] 最优节点：", bestIP, " 速度：", bestSpeed, "MB/s")
+	}
+	// 输出结果
+	fmt.Println("\n[信息] 开始写入 hosts 文件...")
+	err := WriteHosts(bestIP.String())
+	if err != nil {
+		fmt.Println("\n[错误] 写入 hosts 文件失败：", err)
+		return
+	}
+
+	err = SetDockerAccelerator(DefaultDockerUrl)
+	if err != nil {
+		fmt.Println("\n[错误] 设置 Docker 加速器失败：", err)
+	}
+}
+
+func WriteHosts(bestIP string) error {
+	// 获取当前系统
+	system := runtime.GOOS
+	switch system {
+	case "windows":
+		return writeHostsWindows(bestIP)
+	case "darwin":
+		return writeHostsMac(bestIP)
+	case "linux":
+		return writeHostsLinux(bestIP)
+	default:
+		return fmt.Errorf("不支持的操作系统：%s", system)
+	}
+}
+
+func writeHostsWindows(bestIP string) error {
+	hostsFilePath := "C:\\Windows\\System32\\drivers\\etc\\hosts"
+	return writeHostsFile(hostsFilePath, bestIP)
+}
+
+func writeHostsMac(bestIP string) error {
+	hostsFilePath := "/etc/hosts"
+	return writeHostsFile(hostsFilePath, bestIP)
+}
+
+func writeHostsLinux(bestIP string) error {
+	hostsFilePath := "/etc/hosts"
+	return writeHostsFile(hostsFilePath, bestIP)
+}
+
+func writeHostsFile(hostsFilePath, bestIP string) error {
+	// 检查文件权限
+	file, err := os.OpenFile(hostsFilePath, os.O_RDWR, 0644)
+	if err != nil {
+		return fmt.Errorf("无法打开 hosts 文件: %w", err)
+	}
+	defer func(file *os.File) {
+		_ = file.Close()
+	}(file)
+
+	// 读取文件内容
+	var lines []string
+	scanner := bufio.NewScanner(file)
+	inBlock := false
+	for scanner.Scan() {
+		line := scanner.Text()
+		if strings.Contains(line, "# DockerST Start") {
+			inBlock = true
+		}
+		if !inBlock {
+			lines = append(lines, line)
+		}
+		if strings.Contains(line, "# DockerST End") {
+			inBlock = false
+		}
+	}
+	if err := scanner.Err(); err != nil {
+		return fmt.Errorf("读取 hosts 文件出错: %w", err)
+	}
+
+	// 重写文件内容
+	file, err = os.Create(hostsFilePath)
+	if err != nil {
+		return fmt.Errorf("无法重写 hosts 文件: %w", err)
+	}
+	defer func(file *os.File) {
+		_ = file.Close()
+	}(file)
+
+	writer := bufio.NewWriter(file)
+	for _, line := range lines {
+		_, _ = fmt.Fprintln(writer, line)
+	}
+	_, _ = fmt.Fprintln(writer, "# DockerST Start")
+	domain := strings.Split(strings.Split(DefaultDockerUrl, "//")[1], "/")[0]
+	_, _ = fmt.Fprintln(writer, bestIP+" "+domain)
+	_, _ = fmt.Fprintln(writer, "# DockerST End")
+
+	err = writer.Flush()
+	if err != nil {
+		return fmt.Errorf("写入 hosts 文件出错: %w", err)
+	}
+
+	fmt.Println("\n[信息] 成功写入 hosts 文件。")
+	return nil
+}
+
+func SetDockerAccelerator(dockerUrl string) error {
+	system := runtime.GOOS
+	var err error
+
+	switch system {
+	case "windows":
+		err = setDockerAcceleratorWindows(dockerUrl)
+	case "darwin":
+		err = setDockerAcceleratorMac(dockerUrl)
+	case "linux":
+		err = setDockerAcceleratorLinux(dockerUrl)
+	default:
+		err = fmt.Errorf("不支持的操作系统：%s", system)
+	}
+
+	if err != nil {
+		return fmt.Errorf("设置 Docker 加速器失败: %w", err)
+	}
+	fmt.Println("\n[信息] Docker 加速器已设置为：", dockerUrl)
+	return nil
+}
+
+func setDockerAcceleratorWindows(dockerUrl string) error {
+	configPath := os.Getenv("USERPROFILE") + "\\.docker\\daemon.json"
+	return updateDockerConfig(configPath, dockerUrl)
+}
+
+func setDockerAcceleratorMac(dockerUrl string) error {
+	configPath := os.Getenv("HOME") + "/.docker/daemon.json"
+	return updateDockerConfig(configPath, dockerUrl)
+}
+
+func setDockerAcceleratorLinux(dockerUrl string) error {
+	configPath := "/etc/docker/daemon.json"
+	return updateDockerConfig(configPath, dockerUrl)
+}
+
+func updateDockerConfig(configPath, dockerUrl string) error {
+	file, err := os.OpenFile(configPath, os.O_RDWR|os.O_CREATE, 0644)
+	if err != nil {
+		return fmt.Errorf("无法打开 Docker 配置文件: %w", err)
+	}
+	defer func(file *os.File) {
+		_ = file.Close()
+	}(file)
+
+	var config map[string]interface{}
+	if err := json.NewDecoder(file).Decode(&config); err != nil {
+		config = make(map[string]interface{})
+	}
+
+	config["registry-mirrors"] = []string{dockerUrl}
+
+	file, err = os.Create(configPath)
+	if err != nil {
+		return fmt.Errorf("无法重写 Docker 配置文件: %w", err)
+	}
+	defer func(file *os.File) {
+		_ = file.Close()
+	}(file)
+
+	encoder := json.NewEncoder(file)
+	encoder.SetIndent("", "  ")
+	if err := encoder.Encode(config); err != nil {
+		return fmt.Errorf("写入 Docker 配置文件出错: %w", err)
+	}
+
+	fmt.Println("\n[信息] 成功更新 Docker 配置文件。")
+	return nil
+}