mirror of
https://github.com/TeaOSLab/EdgeAdmin.git
synced 2025-11-04 13:10:26 +08:00
131 lines
3.2 KiB
Go
131 lines
3.2 KiB
Go
package grants
|
|
|
|
import ( "github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
|
|
"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/clusters/grants/grantutils"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/langs/codes"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
|
"github.com/iwind/TeaGo/actions"
|
|
"github.com/iwind/TeaGo/maps"
|
|
"golang.org/x/crypto/ssh"
|
|
)
|
|
|
|
type UpdatePopupAction struct {
|
|
actionutils.ParentAction
|
|
}
|
|
|
|
func (this *UpdatePopupAction) Init() {
|
|
this.Nav("", "", "")
|
|
}
|
|
|
|
func (this *UpdatePopupAction) RunGet(params struct {
|
|
GrantId int64
|
|
}) {
|
|
this.Data["methods"] = grantutils.AllGrantMethods(this.LangCode())
|
|
|
|
grantResp, err := this.RPC().NodeGrantRPC().FindEnabledNodeGrant(this.AdminContext(), &pb.FindEnabledNodeGrantRequest{NodeGrantId: params.GrantId})
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
|
|
if grantResp.NodeGrant == nil {
|
|
this.WriteString("找不到要操作的对象")
|
|
return
|
|
}
|
|
|
|
grant := grantResp.NodeGrant
|
|
this.Data["grant"] = maps.Map{
|
|
"id": grant.Id,
|
|
"nodeId": grant.NodeId,
|
|
"method": grant.Method,
|
|
"name": grant.Name,
|
|
"username": grant.Username,
|
|
"password": grant.Password,
|
|
"description": grant.Description,
|
|
"privateKey": grant.PrivateKey,
|
|
"passphrase": grant.Passphrase,
|
|
"su": grant.Su,
|
|
}
|
|
|
|
this.Show()
|
|
}
|
|
|
|
func (this *UpdatePopupAction) RunPost(params struct {
|
|
GrantId int64
|
|
NodeId int64
|
|
Name string
|
|
Method string
|
|
Username string
|
|
Password string
|
|
PrivateKey string
|
|
Passphrase string
|
|
Description string
|
|
Su bool
|
|
|
|
Must *actions.Must
|
|
}) {
|
|
// 创建日志
|
|
defer this.CreateLogInfo(codes.NodeGrant_LogUpdateSSHGrant, params.GrantId)
|
|
|
|
params.Must.
|
|
Field("name", params.Name).
|
|
Require("请输入名称")
|
|
|
|
switch params.Method {
|
|
case "user":
|
|
if len(params.Username) == 0 {
|
|
this.FailField("username", "请输入SSH登录用户名")
|
|
}
|
|
case "privateKey":
|
|
if len(params.Username) == 0 {
|
|
this.FailField("username", "请输入SSH登录用户名")
|
|
}
|
|
if len(params.PrivateKey) == 0 {
|
|
this.FailField("privateKey", "请输入RSA私钥")
|
|
}
|
|
|
|
// 验证私钥
|
|
var err error
|
|
if len(params.Passphrase) > 0 {
|
|
_, err = ssh.ParsePrivateKeyWithPassphrase([]byte(params.PrivateKey), []byte(params.Passphrase))
|
|
} else {
|
|
_, err = ssh.ParsePrivateKey([]byte(params.PrivateKey))
|
|
}
|
|
if err != nil {
|
|
this.Fail("私钥验证失败，请检查格式：" + err.Error())
|
|
return
|
|
}
|
|
default:
|
|
this.Fail("请选择正确的认证方式")
|
|
}
|
|
|
|
// 执行修改
|
|
_, err := this.RPC().NodeGrantRPC().UpdateNodeGrant(this.AdminContext(), &pb.UpdateNodeGrantRequest{
|
|
NodeGrantId: params.GrantId,
|
|
Name: params.Name,
|
|
Method: params.Method,
|
|
Username: params.Username,
|
|
Password: params.Password,
|
|
PrivateKey: params.PrivateKey,
|
|
Passphrase: params.Passphrase,
|
|
Description: params.Description,
|
|
NodeId: params.NodeId,
|
|
Su: params.Su,
|
|
})
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
|
|
// 返回信息
|
|
this.Data["grant"] = maps.Map{
|
|
"id": params.GrantId,
|
|
"name": params.Name,
|
|
"method": params.Method,
|
|
"methodName": grantutils.FindGrantMethodName(params.Method, this.LangCode()),
|
|
"username": params.Username,
|
|
}
|
|
|
|
this.Success()
|
|
}
|