重构项目

2025-10-30 18:26:02 +08:00
parent fbc8bcd089
commit 7e10e2d2da
39 changed files with 728 additions and 464 deletions
--- a/internal/class/Safeline/Method.go
+++ b/internal/class/Safeline/Method.go
@ -0,0 +1,119 @@
+package Safeline
+
+import (
+	"SafelineAPI/internal/app/SafelineRequest"
+	"SafelineAPI/internal/app/logger"
+	"SafelineAPI/pkg/utils"
+	"fmt"
+	"net/url"
+)
+
+func (apiToken ApiToken) GetApiToken() (string, string) {
+	return "X-SLCE-API-TOKEN", apiToken.String()
+}
+
+func (apiToken ApiToken) String() string {
+	return string(apiToken)
+}
+
+func (apiToken ApiToken) Verify() bool {
+	if apiToken.String() == "" {
+		logger.Warning.Printf("未设置 %sSafeLine API Token%s : 请检查配置文件中的 %sSafeLine.ApiToken%s 参数", logger.Cyan, logger.Reset, logger.Yellow, logger.Reset)
+		return true
+	}
+	return false
+}
+
+func (safeLine SafeLine) Verify() bool {
+	a := safeLine.ApiToken.Verify()
+	b := safeLine.Host.Verify()
+	if a || b {
+		return true
+	}
+
+	NoLoginAuthTokenResp, NoLoginStatusCode, NoLoginErr := utils.AuthSafeLine(*safeLine.Host.Url())
+	if NoLoginErr != nil {
+		logger.Error.Printf("请求服务端时发生错误: %s%s%s", logger.Red, NoLoginErr.Error(), logger.Reset)
+		return true
+	}
+	LoginAuthTokenResp, LoginStatusCode, LoginErr := utils.VerifyAuthToken(*safeLine.Host.Url(), safeLine.ApiToken.String())
+	if LoginErr != nil {
+		logger.Error.Printf("验证 %sSafeLine API Token%s 时发生错误: %s%s%s", logger.Cyan, logger.Reset, logger.Red, LoginErr.Error(), logger.Reset)
+		return true
+	}
+
+	if !(NoLoginAuthTokenResp.Err == "login-required" && NoLoginStatusCode == 401) {
+		logger.Warning.Printf("服务端接口 %s/open/auth/token%s 请求有误: 请检查配置文件中的 %sSafeLine.Host%s 参数", logger.Cyan, logger.Reset, logger.Yellow, logger.Reset)
+		return true
+	}
+
+	if LoginAuthTokenResp.Err == "login-required" && LoginStatusCode == 401 {
+		logger.Warning.Printf("%sSafeLine API Token%s 有误: 请检查后重试", logger.Cyan, logger.Reset)
+		return true
+	}
+
+	logger.Success.Printf("%sSafeLine%s 相关配置检验完成!", logger.Cyan, logger.Reset)
+	return false
+}
+
+func (apiToken ApiToken) VerifyCommand() bool {
+	if apiToken.String() == "" {
+		logger.Warning.Printf("未设置 %sSafeLine API Token%s : 请检查命令中的 %s-t%s 参数", logger.Cyan, logger.Reset, logger.Yellow, logger.Reset)
+		return true
+	}
+	return false
+}
+
+func (safeLine SafeLine) VerifyCommand() bool {
+	a := safeLine.ApiToken.VerifyCommand()
+	b := safeLine.Host.VerifyCommand()
+	if a || b {
+		return true
+	}
+
+	NoLoginAuthTokenResp, NoLoginStatusCode, NoLoginErr := utils.AuthSafeLine(*safeLine.Host.Url())
+	if NoLoginErr != nil {
+		logger.Error.Printf("请求服务端时发生错误: %s%s%s", logger.Red, NoLoginErr.Error(), logger.Reset)
+		return true
+	}
+	LoginAuthTokenResp, LoginStatusCode, LoginErr := utils.VerifyAuthToken(*safeLine.Host.Url(), safeLine.ApiToken.String())
+	if LoginErr != nil {
+		logger.Error.Printf("验证 %sSafeLine API Token%s 时发生错误: %s%s%s", logger.Cyan, logger.Reset, logger.Red, LoginErr.Error(), logger.Reset)
+		return true
+	}
+
+	if !(NoLoginAuthTokenResp.Err == "login-required" && NoLoginStatusCode == 401) {
+		logger.Warning.Printf("服务端接口 %s/open/auth/token%s 请求有误: 请检查命令中的 %s-h%s 参数", logger.Cyan, logger.Reset, logger.Yellow, logger.Reset)
+		return true
+	}
+
+	if LoginAuthTokenResp.Err == "login-required" && LoginStatusCode == 401 {
+		logger.Warning.Printf("%sSafeLine API Token%s 有误: 请检查后重试", logger.Cyan, logger.Reset)
+		return true
+	}
+
+	logger.Success.Printf("%sSafeLine%s 相关配置检验完成!", logger.Cyan, logger.Reset)
+	return false
+}
+
+func (host Host) Verify() bool {
+	if host == "" {
+		logger.Warning.Printf("未设置 %s服务器主机%s: 请检查配置文件中的 %sSafeLine.Host%s 参数", logger.Cyan, logger.Reset, logger.Yellow, logger.Reset)
+		return true
+	}
+	return false
+}
+
+func (host Host) VerifyCommand() bool {
+	if host == "" {
+		logger.Warning.Printf("未设置 %s主机名称%s: 请检查命令中的 %s-h%s 参数", logger.Cyan, logger.Reset, logger.Yellow, logger.Reset)
+		return true
+	}
+	return false
+}
+
+func (host Host) Url() *SafelineRequest.URL {
+	var u *url.URL
+	u, _ = url.Parse(fmt.Sprintf("https://%s", host))
+	return (*SafelineRequest.URL)(u)
+}