101 lines
3.4 KiB
Go
101 lines
3.4 KiB
Go
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
|
|
}
|