mirror of
https://github.com/TeaOSLab/EdgeAdmin.git
synced 2025-11-04 05:00:25 +08:00
109 lines
2.8 KiB
Go
109 lines
2.8 KiB
Go
// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
|
|
|
|
package remoteAddr
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/dao"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
|
|
"github.com/iwind/TeaGo/actions"
|
|
"github.com/iwind/TeaGo/types"
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
type IndexAction struct {
|
|
actionutils.ParentAction
|
|
}
|
|
|
|
func (this *IndexAction) Init() {
|
|
this.Nav("", "setting", "index")
|
|
this.SecondMenu("remoteAddr")
|
|
}
|
|
|
|
func (this *IndexAction) RunGet(params struct {
|
|
ServerId int64
|
|
}) {
|
|
// 服务分组设置
|
|
groupResp, err := this.RPC().ServerGroupRPC().FindEnabledServerGroupConfigInfo(this.AdminContext(), &pb.FindEnabledServerGroupConfigInfoRequest{
|
|
ServerId: params.ServerId,
|
|
})
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
this.Data["hasGroupConfig"] = groupResp.HasRemoteAddrConfig
|
|
this.Data["groupSettingURL"] = "/servers/groups/group/settings/remoteAddr?groupId=" + types.String(groupResp.ServerGroupId)
|
|
|
|
webConfig, err := dao.SharedHTTPWebDAO.FindWebConfigWithServerId(this.AdminContext(), params.ServerId)
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
|
|
this.Data["webId"] = webConfig.Id
|
|
this.Data["remoteAddrConfig"] = webConfig.RemoteAddr
|
|
|
|
this.Show()
|
|
}
|
|
|
|
func (this *IndexAction) RunPost(params struct {
|
|
WebId int64
|
|
RemoteAddrJSON []byte
|
|
|
|
Must *actions.Must
|
|
CSRF *actionutils.CSRF
|
|
}) {
|
|
var remoteAddrConfig = &serverconfigs.HTTPRemoteAddrConfig{}
|
|
err := json.Unmarshal(params.RemoteAddrJSON, remoteAddrConfig)
|
|
if err != nil {
|
|
this.Fail("参数校验失败：" + err.Error())
|
|
return
|
|
}
|
|
|
|
remoteAddrConfig.Value = strings.TrimSpace(remoteAddrConfig.Value)
|
|
|
|
switch remoteAddrConfig.Type {
|
|
case serverconfigs.HTTPRemoteAddrTypeRequestHeader:
|
|
if len(remoteAddrConfig.RequestHeaderName) == 0 {
|
|
this.FailField("requestHeaderName", "请输入请求报头")
|
|
return
|
|
}
|
|
if !regexp.MustCompile(`^[\w-_]+$`).MatchString(remoteAddrConfig.RequestHeaderName) {
|
|
this.FailField("requestHeaderName", "请求报头中只能含有数字、英文字母、下划线、中划线")
|
|
return
|
|
}
|
|
remoteAddrConfig.Value = "${header." + remoteAddrConfig.RequestHeaderName + "}"
|
|
case serverconfigs.HTTPRemoteAddrTypeVariable:
|
|
if len(remoteAddrConfig.Value) == 0 {
|
|
this.FailField("value", "请输入自定义变量")
|
|
return
|
|
}
|
|
}
|
|
|
|
err = remoteAddrConfig.Init()
|
|
if err != nil {
|
|
this.Fail("配置校验失败：" + err.Error())
|
|
return
|
|
}
|
|
|
|
remoteAddrJSON, err := json.Marshal(remoteAddrConfig)
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
|
|
_, err = this.RPC().HTTPWebRPC().UpdateHTTPWebRemoteAddr(this.AdminContext(), &pb.UpdateHTTPWebRemoteAddrRequest{
|
|
HttpWebId: params.WebId,
|
|
RemoteAddrJSON: remoteAddrJSON,
|
|
})
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
|
|
this.Success()
|
|
}
|