mirror of
https://github.com/TeaOSLab/EdgeAdmin.git
synced 2025-11-04 21:50:28 +08:00
106 lines
2.9 KiB
Go
106 lines
2.9 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/rpc/pb"
|
|||
|
|
"github.com/iwind/TeaGo/actions"
|
|||
|
|
"github.com/iwind/TeaGo/maps"
|
|||
|
|
"github.com/iwind/TeaGo/types"
|
|||
|
|
"net"
|
|||
|
|
"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("请输入节点端口").
|
|||
|
|
Match(`^\d+$`, "节点端口只能是数字").
|
|||
|
|
MinLength(4, "请输入4位以上的数字").
|
|||
|
|
MaxLength(5, "请输入5位以下的数字")
|
|||
|
|
newPort := types.Int(params.NewPort)
|
|||
|
|
if newPort < 1024 {
|
|||
|
|
this.FailField("newPort", "端口号不能小于1024")
|
|||
|
|
}
|
|||
|
|
if newPort > 65534 {
|
|||
|
|
this.FailField("newPort", "端口号不能大于65534")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
conn, err := net.Dial("tcp", "127.0.0.1:"+params.NewPort)
|
|||
|
|
if err == nil {
|
|||
|
|
_ = conn.Close()
|
|||
|
|
this.FailField("newPort", "此端口已经被别的服务占用，请换一个")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
params.Must.
|
|||
|
|
Field("newHost", params.NewHost).
|
|||
|
|
Require("请输入节点主机地址")
|
|||
|
|
|
|||
|
|
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"`
|
|||
|
|
}{
|
|||
|
|
Endpoints: []string{params.OldProtocol + "://" + params.OldHost + ":" + params.OldPort},
|
|||
|
|
},
|
|||
|
|
NodeId: params.OldNodeId,
|
|||
|
|
Secret: params.OldNodeSecret,
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
this.FailField("oldHost", "测试API节点时出错，请检查配置，错误信息："+err.Error())
|
|||
|
|
}
|
|||
|
|
_, err = client.APINodeRPC().FindCurrentAPINodeVersion(client.APIContext(0), &pb.FindCurrentAPINodeVersionRequest{})
|
|||
|
|
if err != nil {
|
|||
|
|
this.FailField("oldHost", "无法连接此API节点，错误信息："+err.Error())
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
this.Success()
|
|||
|
|
}
|