Commit

internal/app/config/safeline.go

@@ -0,0 +1,100 @@
+package config
+
+import (
+	"SafelineAPI/internal/app/logger"
+	"SafelineAPI/pkg/utils"
+)
+
+type SafeLine struct {
+	Host     `json:"Host"`
+	ApiToken `json:"ApiToken"`
+}
+type ApiToken string
+
+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
+}