初始化项目

2025-03-27 16:09:20 +08:00
parent e09a32d1e8
commit fc2d585489
709 changed files with 516391 additions and 0 deletions
--- a/agent-wdd/cmd/Download.go
+++ b/agent-wdd/cmd/Download.go
@@ -0,0 +1,122 @@
+package cmd
+
+import (
+	"agent-wdd/log"
+	"agent-wdd/utils"
+	"context"
+	"fmt"
+	"net"
+	"net/http"
+	"net/url"
+	"time"
+
+	"github.com/spf13/cobra"
+	"golang.org/x/net/proxy"
+)
+
+// 示例：添加download子命令
+func addDownloadSubcommands(cmd *cobra.Command) {
+	proxyCmd := &cobra.Command{
+		Use:   "proxy [proxyUrl] [url] [dest]",
+		Short: "使用代理下载 socks5://[username]:[password]@ip:port http://[username]:[password]@ip:port",
+		Args:  cobra.ExactArgs(3),
+		Run: func(cmd *cobra.Command, args []string) {
+			proxyURL := args[0]
+			fileURL := args[1]
+			destPath := args[2]
+
+			log.Info("Downloading using proxy: %s -> from %s to %s\n", proxyURL, fileURL, destPath)
+
+			// 创建带代理的HTTP客户端
+			client, err := createProxyClient(proxyURL)
+			if err != nil {
+				log.Error("创建代理客户端失败: %v", err)
+			}
+
+			// 执行下载
+			downloadOk, resultLog := utils.DownloadFileWithClient(client, fileURL, destPath)
+			if !downloadOk {
+				log.Error("下载失败: %v", resultLog)
+			} else {
+				log.Info("文件下载完成")
+			}
+
+		},
+	}
+
+	cmd.Run = func(cmd *cobra.Command, args []string) {
+
+		if len(args) == 0 {
+			log.Error("请输入下载地址")
+			return
+		} else if len(args) == 1 {
+			timeString := utils.CurrentTimeString()
+			args = append(args, "./"+timeString+".txt")
+		}
+
+		log.Info("Downloading without proxy: %s -> %s\n", args[0], args[1])
+
+		downloadOk, resultLog := utils.DownloadFile(args[0], args[1])
+		if !downloadOk {
+			log.Error("下载失败: %s", resultLog)
+		} else {
+			log.Info("下载成功: %s", args[1])
+		}
+
+	}
+
+	cmd.AddCommand(proxyCmd)
+}
+
+// 创建带代理的HTTP客户端
+func createProxyClient(proxyURL string) (*http.Client, error) {
+	parsedURL, err := url.Parse(proxyURL)
+	if err != nil {
+		return nil, fmt.Errorf("无效的代理URL: %w", err)
+	}
+
+	switch parsedURL.Scheme {
+	case "socks5":
+		return createSocks5Client(parsedURL)
+	case "http", "https":
+		return createHTTPClient(parsedURL), nil
+	default:
+		return nil, fmt.Errorf("不支持的代理协议: %s", parsedURL.Scheme)
+	}
+}
+
+// 创建SOCKS5代理客户端
+func createSocks5Client(proxyURL *url.URL) (*http.Client, error) {
+	var auth *proxy.Auth
+	if proxyURL.User != nil {
+		password, _ := proxyURL.User.Password()
+		auth = &proxy.Auth{
+			User:     proxyURL.User.Username(),
+			Password: password,
+		}
+	}
+
+	dialer, err := proxy.SOCKS5("tcp", proxyURL.Host, auth, proxy.Direct)
+	if err != nil {
+		return nil, fmt.Errorf("创建SOCKS5拨号器失败: %w", err)
+	}
+
+	return &http.Client{
+		Transport: &http.Transport{
+			DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
+				return dialer.Dial(network, addr)
+			},
+		},
+		Timeout: 10 * time.Second,
+	}, nil
+}
+
+// 创建HTTP代理客户端
+func createHTTPClient(proxyURL *url.URL) *http.Client {
+	return &http.Client{
+		Transport: &http.Transport{
+			Proxy: http.ProxyURL(proxyURL),
+		},
+		Timeout: 30 * time.Minute,
+	}
+}