update

utils/utils.go

@@ -0,0 +1,133 @@
+package utils
+
+import (
+	"bytes"
+	"fmt"
+	"github.com/imroc/req/v3"
+	"github.com/oneclickvirt/UnlockTests/uts"
+	"github.com/oneclickvirt/basics/system"
+	"github.com/oneclickvirt/security/network"
+	"os"
+	"strings"
+	"sync"
+	"unicode/utf8"
+)
+
+// PrintCenteredTitle 根据指定的宽度打印居中标题
+func PrintCenteredTitle(title string, width int) {
+	// 计算字符串的字符数
+	titleLength := utf8.RuneCountInString(title)
+	totalPadding := width - titleLength
+	padding := totalPadding / 2
+	paddingStr := strings.Repeat("-", padding)
+	fmt.Println(paddingStr + title + paddingStr + strings.Repeat("-", totalPadding%2))
+}
+
+// PrintHead 根据语言打印头部信息
+func PrintHead(language string, width int, ecsVersion string) {
+	if language == "zh" {
+		PrintCenteredTitle("融合怪测试", width)
+		fmt.Printf("版本：%s\n", ecsVersion)
+		fmt.Println("测评频道: https://t.me/vps_reviews\n" +
+			"Go项目地址：https://github.com/oneclickvirt/ecs\n" +
+			"Shell项目地址：https://github.com/spiritLHLS/ecs")
+	} else {
+		PrintCenteredTitle("Fusion Monster Test", width)
+		fmt.Printf("Version: %s\n", ecsVersion)
+		fmt.Println("Review Channel: https://t.me/vps_reviews\n" +
+			"Go Project URL: https://github.com/oneclickvirt/ecs\n" +
+			"Shell Project URL: https://github.com/spiritLHLS/ecs")
+	}
+}
+
+// SecurityCheck 执行安全检查
+func SecurityCheck(language, nt3CheckType string) (string, string, string) {
+	var wgt sync.WaitGroup
+	var ipInfo, securityInfo, systemInfo string
+	var err error
+	wgt.Add(2)
+	go func() {
+		defer wgt.Done()
+		ipInfo, securityInfo, err = network.NetworkCheck("both", true, language)
+		if err != nil {
+			fmt.Println(err.Error())
+		}
+	}()
+	go func() {
+		defer wgt.Done()
+		systemInfo = system.CheckSystemInfo(language)
+	}()
+	wgt.Wait()
+	basicInfo := systemInfo + ipInfo
+	if strings.Contains(ipInfo, "IPV4") && strings.Contains(ipInfo, "IPV6") {
+		uts.IPV4 = true
+		uts.IPV6 = true
+		if nt3CheckType == "" {
+			nt3CheckType = "ipv4"
+		}
+	} else if strings.Contains(ipInfo, "IPV4") {
+		uts.IPV4 = true
+		uts.IPV6 = false
+		if nt3CheckType == "" {
+			nt3CheckType = "ipv4"
+		}
+	} else if strings.Contains(ipInfo, "IPV6") {
+		uts.IPV6 = true
+		uts.IPV4 = false
+		if nt3CheckType == "" {
+			nt3CheckType = "ipv6"
+		}
+	}
+	if nt3CheckType == "ipv4" && !strings.Contains(ipInfo, "IPV4") && strings.Contains(ipInfo, "IPV6") {
+		nt3CheckType = "ipv6"
+	} else if nt3CheckType == "ipv6" && !strings.Contains(ipInfo, "IPV6") && strings.Contains(ipInfo, "IPV4") {
+		nt3CheckType = "ipv4"
+	}
+	basicInfo = strings.ReplaceAll(basicInfo, "\n\n", "\n")
+	return basicInfo, securityInfo, nt3CheckType
+}
+
+// CaptureOutput 捕获函数输出并返回字符串
+func CaptureOutput(f func()) string {
+	old := os.Stdout
+	r, w, _ := os.Pipe()
+	os.Stdout = w
+	f()
+	w.Close()
+	os.Stdout = old
+	var buf bytes.Buffer
+	buf.ReadFrom(r)
+	return buf.String()
+}
+
+// PrintAndCapture 捕获函数输出的同时打印内容
+func PrintAndCapture(f func(), tempOutput, output string) string {
+	tempOutput = CaptureOutput(f)
+	fmt.Printf(tempOutput)
+	output += tempOutput
+	return output
+}
+
+// UploadText 上传文本内容到指定URL
+func UploadText(textContent string) (string, error) {
+	url := "https://paste.spiritlhl.net/api/upload"
+	token := network.SecurityUploadToken
+	client := req.C().SetTimeout(10 * 1000 * 1000) // 10 seconds timeout
+	resp, err := client.R().
+		SetHeader("Authorization", token).
+		SetHeader("Format", "RANDOM").
+		SetHeader("Max-Views", "0").
+		SetHeader("UploadText", "true").
+		SetHeader("Content-Type", "multipart/form-data").
+		SetHeader("No-JSON", "true").
+		SetBodyString(textContent).
+		Post(url)
+	if err != nil {
+		return "", err
+	}
+	if resp.StatusCode >= 200 && resp.StatusCode <= 299 {
+		return resp.String(), nil
+	} else {
+		return "", fmt.Errorf("upload failed with status code: %d", resp.StatusCode)
+	}
+}