// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
//go:build plus

package boards

import (
	"fmt"
	teaconst "github.com/TeaOSLab/EdgeAdmin/internal/const"
	"github.com/TeaOSLab/EdgeAdmin/internal/utils/numberutils"
	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/dashboard/boards/boardutils"
	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
	"github.com/iwind/TeaGo/maps"
)

type WafAction struct {
	actionutils.ParentAction
}

func (this *WafAction) Init() {
	this.Nav("", "", "waf")
}

func (this *WafAction) RunGet(params struct{}) {
	if !teaconst.IsPlus {
		this.RedirectURL("/dashboard")
		return
	}

	// 将badge置为已读
	boardutils.CountTodayAttacksRead = boardutils.CountTodayAttacks

	// 初始化
	err := boardutils.InitBoard(this.Parent())
	if err != nil {
		this.ErrorPage(err)
		return
	}

	resp, err := this.RPC().FirewallRPC().ComposeFirewallGlobalBoard(this.AdminContext(), &pb.ComposeFirewallGlobalBoardRequest{})
	if err != nil {
		this.ErrorPage(err)
		return
	}
	this.Data["board"] = maps.Map{
		"countDailyLogs":    resp.CountDailyLogs,
		"countDailyBlocks":  resp.CountDailyBlocks,
		"countDailyCaptcha": resp.CountDailyCaptcha,
		"countWeeklyBlocks": resp.CountWeeklyBlocks,
	}

	{
		var statMaps = []maps.Map{}
		for _, stat := range resp.HourlyStats {
			statMaps = append(statMaps, maps.Map{
				"hour":         stat.Hour,
				"countLogs":    stat.CountLogs,
				"countCaptcha": stat.CountCaptcha,
				"countBlocks":  stat.CountBlocks,
			})
		}
		this.Data["hourlyStats"] = statMaps
	}

	{
		var statMaps = []maps.Map{}
		for _, stat := range resp.DailyStats {
			statMaps = append(statMaps, maps.Map{
				"day":          stat.Day,
				"countLogs":    stat.CountLogs,
				"countCaptcha": stat.CountCaptcha,
				"countBlocks":  stat.CountBlocks,
			})
		}
		this.Data["dailyStats"] = statMaps
	}

	{
		var statMaps = []maps.Map{}
		for _, stat := range resp.HttpFirewallRuleGroups {
			statMaps = append(statMaps, maps.Map{
				"name":  stat.HttpFirewallRuleGroup.Name,
				"count": stat.Count,
			})
		}
		this.Data["groupStats"] = statMaps
	}

	// 节点排行
	{
		var statMaps = []maps.Map{}
		for _, stat := range resp.TopNodeStats {
			statMaps = append(statMaps, maps.Map{
				"nodeId":        stat.NodeId,
				"nodeName":      stat.NodeName,
				"countRequests": stat.CountAttackRequests,
				"bytes":         stat.AttackBytes,
			})
		}
		this.Data["topNodeStats"] = statMaps
	}

	// 域名排行
	{
		var statMaps = []maps.Map{}
		for _, stat := range resp.TopDomainStats {
			statMaps = append(statMaps, maps.Map{
				"serverId":      stat.ServerId,
				"domain":        stat.Domain,
				"countRequests": stat.CountAttackRequests,
				"bytes":         stat.AttackBytes,
			})
		}
		this.Data["topDomainStats"] = statMaps
	}

	// 地区排行
	{
		var countryMaps = []maps.Map{}
		for _, stat := range resp.TopCountryStats {
			countryMaps = append(countryMaps, maps.Map{
				"name":                stat.CountryName,
				"bytes":               stat.Bytes,
				"formattedBytes":      numberutils.FormatBytes(stat.Bytes),
				"countRequests":       stat.CountRequests,
				"countAttackRequests": stat.CountAttackRequests,
				"percent":             fmt.Sprintf("%.2f", stat.Percent),
			})
		}
		this.Data["topCountryStats"] = countryMaps
	}

	this.Show()
}