mirror of
https://github.com/TeaOSLab/EdgeAdmin.git
synced 2025-11-04 13:10:26 +08:00
129 lines
3.5 KiB
Go
129 lines
3.5 KiB
Go
package setup
|
||
|
||
import (
|
||
"github.com/TeaOSLab/EdgeAdmin/internal/configs"
|
||
"github.com/TeaOSLab/EdgeAdmin/internal/rpc"
|
||
"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/configutils"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
||
"github.com/iwind/TeaGo/actions"
|
||
"github.com/iwind/TeaGo/maps"
|
||
"github.com/iwind/TeaGo/types"
|
||
"net"
|
||
"regexp"
|
||
"strings"
|
||
)
|
||
|
||
type ValidateApiAction struct {
|
||
actionutils.ParentAction
|
||
}
|
||
|
||
func (this *ValidateApiAction) RunPost(params struct {
|
||
Mode string
|
||
NewPort string
|
||
NewHost string
|
||
OldProtocol string
|
||
OldHost string
|
||
OldPort string
|
||
OldNodeId string
|
||
OldNodeSecret string
|
||
|
||
Must *actions.Must
|
||
}) {
|
||
params.OldNodeId = strings.Trim(params.OldNodeId, "\"' ")
|
||
params.OldNodeSecret = strings.Trim(params.OldNodeSecret, "\"' ")
|
||
|
||
this.Data["apiNode"] = maps.Map{
|
||
"mode": params.Mode,
|
||
|
||
"newPort": params.NewPort,
|
||
"newHost": params.NewHost,
|
||
|
||
"oldProtocol": params.OldProtocol,
|
||
"oldHost": params.OldHost,
|
||
"oldPort": params.OldPort,
|
||
"oldNodeId": params.OldNodeId,
|
||
"oldNodeSecret": params.OldNodeSecret,
|
||
}
|
||
|
||
if params.Mode == "new" {
|
||
params.Must.
|
||
Field("newPort", params.NewPort).
|
||
Require("请输入API节点端口").
|
||
Match(`^\d+$`, "API节点端口只能是数字").
|
||
MinLength(4, "请输入4位以上的数字端口号").
|
||
MaxLength(5, "请输入5位以下的数字端口号")
|
||
var newPort = types.Int(params.NewPort)
|
||
if newPort < 1024 {
|
||
this.FailField("newPort", "API端口号不能小于1024")
|
||
}
|
||
if newPort > 65534 {
|
||
this.FailField("newPort", "API端口号不能大于65534")
|
||
}
|
||
|
||
if net.ParseIP(params.NewHost) == nil {
|
||
// 检查是否为域名
|
||
var domainReg = regexp.MustCompile(`^[\w-]+$`) // 这里暂不支持 unicode 域名
|
||
var digitReg = regexp.MustCompile(`^\d+$`)
|
||
var isIP = true
|
||
for _, piece := range strings.Split(params.NewHost, ".") {
|
||
if !domainReg.MatchString(piece) {
|
||
this.FailField("newHost", "请输入正确的API节点主机地址")
|
||
return
|
||
}
|
||
if !digitReg.MatchString(piece) {
|
||
isIP = false
|
||
}
|
||
}
|
||
|
||
if isIP {
|
||
this.FailField("newHost", "请输入正确的API节点主机地址")
|
||
return
|
||
}
|
||
}
|
||
|
||
params.Must.
|
||
Field("newHost", params.NewHost).
|
||
Require("请输入API节点主机地址")
|
||
|
||
this.Success()
|
||
return
|
||
}
|
||
|
||
// 使用已有的API节点
|
||
params.Must.
|
||
Field("oldHost", params.OldHost).
|
||
Require("请输入主机地址").
|
||
Field("oldPort", params.OldPort).
|
||
Require("请输入服务端口").
|
||
Match(`^\d+$`, "服务端口只能是数字").
|
||
Field("oldNodeId", params.OldNodeId).
|
||
Require("请输入节点nodeId").
|
||
Field("oldNodeSecret", params.OldNodeSecret).
|
||
Require("请输入节点secret")
|
||
client, err := rpc.NewRPCClient(&configs.APIConfig{
|
||
RPC: struct {
|
||
Endpoints []string `yaml:"endpoints"`
|
||
DisableUpdate bool `yaml:"disableUpdate"`
|
||
}{
|
||
Endpoints: []string{params.OldProtocol + "://" + configutils.QuoteIP(params.OldHost) + ":" + params.OldPort},
|
||
},
|
||
NodeId: params.OldNodeId,
|
||
Secret: params.OldNodeSecret,
|
||
}, false)
|
||
if err != nil {
|
||
this.FailField("oldHost", "测试API节点时出错，请检查配置，错误信息："+err.Error())
|
||
}
|
||
|
||
defer func() {
|
||
_ = client.Close()
|
||
}()
|
||
|
||
_, err = client.APINodeRPC().FindCurrentAPINodeVersion(client.APIContext(0), &pb.FindCurrentAPINodeVersionRequest{})
|
||
if err != nil {
|
||
this.FailField("oldHost", "无法连接此API节点，错误信息："+err.Error())
|
||
}
|
||
|
||
this.Success()
|
||
}
|