mirror of
https://github.com/TeaOSLab/EdgeAdmin.git
synced 2025-11-04 13:10:26 +08:00
103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
package waf
|
||
|
||
import (
|
||
"github.com/TeaOSLab/EdgeAdmin/internal/oplogs"
|
||
"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
|
||
"github.com/TeaOSLab/EdgeAdmin/internal/web/models"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/firewallconfigs"
|
||
"github.com/iwind/TeaGo/actions"
|
||
)
|
||
|
||
type CreateGroupPopupAction struct {
|
||
actionutils.ParentAction
|
||
}
|
||
|
||
func (this *CreateGroupPopupAction) Init() {
|
||
this.Nav("", "", "")
|
||
}
|
||
|
||
func (this *CreateGroupPopupAction) RunGet(params struct {
|
||
Type string
|
||
}) {
|
||
this.Data["type"] = params.Type
|
||
|
||
this.Show()
|
||
}
|
||
|
||
func (this *CreateGroupPopupAction) RunPost(params struct {
|
||
FirewallPolicyId int64
|
||
Type string
|
||
|
||
Name string
|
||
Description string
|
||
IsOn bool
|
||
|
||
Must *actions.Must
|
||
}) {
|
||
firewallPolicy, err := models.SharedHTTPFirewallPolicyDAO.FindEnabledPolicyConfig(this.AdminContext(), params.FirewallPolicyId)
|
||
if err != nil {
|
||
this.ErrorPage(err)
|
||
return
|
||
}
|
||
|
||
if firewallPolicy == nil {
|
||
this.NotFound("firewallPolicy", params.FirewallPolicyId)
|
||
}
|
||
|
||
params.Must.
|
||
Field("name", params.Name).
|
||
Require("请输入分组名称")
|
||
|
||
createResp, err := this.RPC().HTTPFirewallRuleGroupRPC().CreateHTTPFirewallRuleGroup(this.AdminContext(), &pb.CreateHTTPFirewallRuleGroupRequest{
|
||
IsOn: params.IsOn,
|
||
Name: params.Name,
|
||
Description: params.Description,
|
||
})
|
||
if err != nil {
|
||
this.ErrorPage(err)
|
||
return
|
||
}
|
||
groupId := createResp.FirewallRuleGroupId
|
||
|
||
switch params.Type {
|
||
case "inbound":
|
||
firewallPolicy.Inbound.GroupRefs = append(firewallPolicy.Inbound.GroupRefs, &firewallconfigs.HTTPFirewallRuleGroupRef{
|
||
IsOn: true,
|
||
GroupId: groupId,
|
||
})
|
||
default:
|
||
firewallPolicy.Outbound.GroupRefs = append(firewallPolicy.Outbound.GroupRefs, &firewallconfigs.HTTPFirewallRuleGroupRef{
|
||
IsOn: true,
|
||
GroupId: groupId,
|
||
})
|
||
}
|
||
|
||
inboundJSON, err := firewallPolicy.InboundJSON()
|
||
if err != nil {
|
||
this.ErrorPage(err)
|
||
return
|
||
}
|
||
|
||
outboundJSON, err := firewallPolicy.OutboundJSON()
|
||
if err != nil {
|
||
this.ErrorPage(err)
|
||
return
|
||
}
|
||
|
||
_, err = this.RPC().HTTPFirewallPolicyRPC().UpdateHTTPFirewallPolicyGroups(this.AdminContext(), &pb.UpdateHTTPFirewallPolicyGroupsRequest{
|
||
FirewallPolicyId: params.FirewallPolicyId,
|
||
InboundJSON: inboundJSON,
|
||
OutboundJSON: outboundJSON,
|
||
})
|
||
if err != nil {
|
||
this.ErrorPage(err)
|
||
return
|
||
}
|
||
|
||
// 日志
|
||
defer this.CreateLog(oplogs.LevelInfo, "创建规则分组 %d，名称：%s", groupId, params.Name)
|
||
|
||
this.Success()
|
||
}
|