141 lines
3.6 KiB
Go
141 lines
3.6 KiB
Go
package waf
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/userconfigs"
|
|
"github.com/TeaOSLab/EdgeUser/internal/web/actions/actionutils"
|
|
"github.com/iwind/TeaGo/maps"
|
|
timeutil "github.com/iwind/TeaGo/utils/time"
|
|
)
|
|
|
|
type IndexAction struct {
|
|
actionutils.ParentAction
|
|
}
|
|
|
|
func (this *IndexAction) Init() {
|
|
this.Nav("", "", "")
|
|
}
|
|
|
|
func (this *IndexAction) RunGet(params struct {
|
|
ServerId int64
|
|
}) {
|
|
if !this.ValidateFeature(userconfigs.UserFeatureCodeServerWAF, params.ServerId) {
|
|
return
|
|
}
|
|
|
|
this.Data["serverId"] = params.ServerId
|
|
this.Data["path"] = this.Request.URL.Path
|
|
|
|
// 所有的服务列表
|
|
serversResp, err := this.RPC().ServerRPC().ListEnabledServersMatch(this.UserContext(), &pb.ListEnabledServersMatchRequest{
|
|
Offset: 0,
|
|
Size: 100, // 我们这里最多显示前100个
|
|
ServerGroupId: 0,
|
|
Keyword: "",
|
|
ProtocolFamily: "http",
|
|
UserId: this.UserId(),
|
|
IgnoreSSLCerts: true,
|
|
})
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
serverMaps := []maps.Map{}
|
|
for _, server := range serversResp.Servers {
|
|
if !server.IsOn {
|
|
continue
|
|
}
|
|
|
|
// 域名列表
|
|
serverNames := []*serverconfigs.ServerNameConfig{}
|
|
if server.IsAuditing || (server.AuditingResult != nil && !server.AuditingResult.IsOk) {
|
|
server.ServerNamesJSON = server.AuditingServerNamesJSON
|
|
}
|
|
if len(server.ServerNamesJSON) > 0 {
|
|
err = json.Unmarshal(server.ServerNamesJSON, &serverNames)
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
}
|
|
if len(serverNames) == 0 {
|
|
continue
|
|
}
|
|
|
|
serverName := server.Name
|
|
if len(serverNames) > 0 {
|
|
if len(serverNames[0].SubNames) == 0 {
|
|
serverName = serverNames[0].Name
|
|
} else {
|
|
serverName = serverNames[0].SubNames[0]
|
|
}
|
|
}
|
|
|
|
serverMaps = append(serverMaps, maps.Map{
|
|
"id": server.Id,
|
|
"serverName": serverName,
|
|
})
|
|
}
|
|
this.Data["servers"] = serverMaps
|
|
|
|
// 统计数据
|
|
resp, err := this.RPC().ServerHTTPFirewallDailyStatRPC().ComposeServerHTTPFirewallDashboard(this.UserContext(), &pb.ComposeServerHTTPFirewallDashboardRequest{
|
|
Day: timeutil.Format("Ymd"),
|
|
UserId: this.UserId(),
|
|
ServerId: params.ServerId,
|
|
})
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
|
|
this.Data["countDailyLog"] = resp.CountDailyLog
|
|
this.Data["countDailyBlock"] = resp.CountDailyBlock
|
|
this.Data["countDailyCaptcha"] = resp.CountDailyCaptcha
|
|
this.Data["countWeeklyBlock"] = resp.CountWeeklyBlock
|
|
this.Data["countMonthlyBlock"] = resp.CountMonthlyBlock
|
|
|
|
// 分组
|
|
groupStatMaps := []maps.Map{}
|
|
for _, group := range resp.HttpFirewallRuleGroups {
|
|
groupStatMaps = append(groupStatMaps, maps.Map{
|
|
"group": maps.Map{
|
|
"id": group.HttpFirewallRuleGroup.Id,
|
|
"name": group.HttpFirewallRuleGroup.Name,
|
|
},
|
|
"count": group.Count,
|
|
})
|
|
}
|
|
this.Data["groupStats"] = groupStatMaps
|
|
|
|
// 每日趋势
|
|
logStatMaps := []maps.Map{}
|
|
blockStatMaps := []maps.Map{}
|
|
captchaStatMaps := []maps.Map{}
|
|
for _, stat := range resp.LogDailyStats {
|
|
logStatMaps = append(logStatMaps, maps.Map{
|
|
"day": stat.Day,
|
|
"count": stat.Count,
|
|
})
|
|
}
|
|
for _, stat := range resp.BlockDailyStats {
|
|
blockStatMaps = append(blockStatMaps, maps.Map{
|
|
"day": stat.Day,
|
|
"count": stat.Count,
|
|
})
|
|
}
|
|
for _, stat := range resp.CaptchaDailyStats {
|
|
captchaStatMaps = append(captchaStatMaps, maps.Map{
|
|
"day": stat.Day,
|
|
"count": stat.Count,
|
|
})
|
|
}
|
|
this.Data["logDailyStats"] = logStatMaps
|
|
this.Data["blockDailyStats"] = blockStatMaps
|
|
this.Data["captchaDailyStats"] = captchaStatMaps
|
|
|
|
this.Show()
|
|
}
|