//go:build plus

package fee

import (
	"encoding/json"
	"github.com/TeaOSLab/EdgeAdmin/internal/utils/numberutils"
	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/finance/financeutils"
	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
	"github.com/TeaOSLab/EdgeCommon/pkg/userconfigs"
	"github.com/iwind/TeaGo/maps"
)

type BandwidthAction struct {
	actionutils.ParentAction
}

func (this *BandwidthAction) Init() {
	this.Nav("", "", "bandwidth")
}

func (this *BandwidthAction) RunGet(params struct{}) {
	// 财务相关配置
	config, err := financeutils.ReadPriceConfig(this.AdminContext())
	if err != nil {
		this.ErrorPage(err)
		return
	}
	this.Data["config"] = config

	// 所有价格项目
	itemsResp, err := this.RPC().NodePriceItemRPC().FindAllAvailableNodePriceItems(this.AdminContext(), &pb.FindAllAvailableNodePriceItemsRequest{Type: userconfigs.PriceTypeBandwidth})
	if err != nil {
		this.ErrorPage(err)
		return
	}
	var itemMaps = []maps.Map{}
	for _, item := range itemsResp.NodePriceItems {
		var maxSize string
		if item.BitsTo == 0 {
			maxSize = "∞"
		} else {
			maxSize = numberutils.FormatBits(item.BitsTo)
		}

		itemMaps = append(itemMaps, maps.Map{
			"id":      item.Id,
			"name":    item.Name,
			"minSize": numberutils.FormatBits(item.BitsFrom),
			"maxSize": maxSize,
		})
	}
	this.Data["items"] = itemMaps

	// 所有区域
	regionsResp, err := this.RPC().NodeRegionRPC().FindAllAvailableNodeRegions(this.AdminContext(), &pb.FindAllAvailableNodeRegionsRequest{})
	if err != nil {
		this.ErrorPage(err)
		return
	}
	var regionMaps = []maps.Map{}
	for _, region := range regionsResp.NodeRegions {
		pricesMap := map[string]float32{}
		if len(region.PricesJSON) > 0 {
			err = json.Unmarshal(region.PricesJSON, &pricesMap)
			if err != nil {
				this.ErrorPage(err)
				return
			}
		}
		regionMaps = append(regionMaps, maps.Map{
			"id":     region.Id,
			"isOn":   region.IsOn,
			"name":   region.Name,
			"prices": pricesMap,
		})
	}
	this.Data["regions"] = regionMaps

	this.Show()
}