waf-platform/EdgeUser/internal/web/actions/default/finance/bills/bill.go

// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved.

package bills

import (
	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
	"github.com/TeaOSLab/EdgeCommon/pkg/userconfigs"
	"github.com/TeaOSLab/EdgeUser/internal/utils/numberutils"
	"github.com/TeaOSLab/EdgeUser/internal/web/actions/actionutils"
	"github.com/iwind/TeaGo/maps"
	timeutil "github.com/iwind/TeaGo/utils/time"
)

type BillAction struct {
	actionutils.ParentAction
}

func (this *BillAction) Init() {
	this.Nav("", "", "index")
}

func (this *BillAction) RunGet(params struct {
	Code string
}) {
	userBillResp, err := this.RPC().UserBillRPC().FindUserBill(this.UserContext(), &pb.FindUserBillRequest{Code: params.Code})
	if err != nil {
		this.ErrorPage(err)
		return
	}
	var bill = userBillResp.UserBill
	if bill == nil {
		this.NotFound("user bill", 0)
		return
	}

	var userMap maps.Map = nil
	if bill.User != nil {
		userMap = maps.Map{
			"id":       bill.User.Id,
			"fullname": bill.User.Fullname,
			"username": bill.User.Username,
		}
	}

	var month = bill.Month
	var dayFrom = bill.DayFrom
	var dayTo = bill.DayTo

	this.Data["bill"] = maps.Map{
		"id":              bill.Id,
		"isPaid":          bill.IsPaid,
		"month":           month,
		"dayFrom":         dayFrom,
		"dayTo":           dayTo,
		"amount":          numberutils.FormatFloat(bill.Amount, 2),
		"typeName":        bill.TypeName,
		"user":            userMap,
		"description":     bill.Description,
		"code":            bill.Code,
		"canPay":          bill.CanPay,
		"pricePeriodName": userconfigs.PricePeriodName(bill.PricePeriod),
		"pricePeriod":     bill.PricePeriod,
	}

	// 服务账单
	var serverBillMaps = []maps.Map{}
	if (bill.Type == "traffic" || bill.Type == "trafficAndBandwidth") && bill.User != nil {
		countResp, err := this.RPC().ServerBillRPC().CountAllServerBills(this.UserContext(), &pb.CountAllServerBillsRequest{
			UserId: bill.User.Id,
			Month:  bill.Month,
		})
		if err != nil {
			this.ErrorPage(err)
			return
		}
		var count = countResp.Count
		var page = this.NewPage(count)
		this.Data["page"] = page.AsHTML()

		serverBillsResp, err := this.RPC().ServerBillRPC().ListServerBills(this.UserContext(), &pb.ListServerBillsRequest{
			UserId: bill.User.Id,
			Month:  bill.Month,
			Offset: page.Offset,
			Size:   page.Size,
		})
		if err != nil {
			this.ErrorPage(err)
			return
		}
		for _, serverBill := range serverBillsResp.ServerBills {
			// server
			var serverMap = maps.Map{"id": 0}
			if serverBill.Server != nil {
				serverMap = maps.Map{
					"id":   serverBill.Server.Id,
					"name": serverBill.Server.Name,
				}
			}

			// plan
			var planMap = maps.Map{"id": 0}
			if serverBill.Plan != nil {
				planMap = maps.Map{
					"id":        serverBill.Plan.Id,
					"name":      serverBill.Plan.Name,
					"priceType": serverBill.Plan.PriceType,
				}
			}

			serverBillMaps = append(serverBillMaps, maps.Map{
				"id":                      serverBill.Id,
				"server":                  serverMap,
				"plan":                    planMap,
				"traffic":                 numberutils.FormatBytes(serverBill.TotalTrafficBytes),
				"bandwidthPercentile":     serverBill.BandwidthPercentile,
				"bandwidthPercentileSize": numberutils.FormatBytes(serverBill.BandwidthPercentileBytes),
				"createdTime":             timeutil.FormatTime("Y-m-d H:i:s", serverBill.CreatedAt),
				"amount":                  numberutils.FormatFloat(serverBill.Amount, 2),
				"priceType":               serverBill.PriceType,
			})
		}
	}
	this.Data["serverBills"] = serverBillMaps

	// traffic bills
	var trafficBillMaps = []maps.Map{}
	trafficBillsResp, err := this.RPC().UserTrafficBillRPC().FindUserTrafficBills(this.UserContext(), &pb.FindUserTrafficBillsRequest{UserBillId: bill.Id})
	if err != nil {
		this.ErrorPage(err)
		return
	}
	for _, trafficBill := range trafficBillsResp.UserTrafficBills {
		var regionMap = maps.Map{"id": 0}
		var nodeRegion = trafficBill.NodeRegion
		if nodeRegion != nil {
			regionMap = maps.Map{
				"id":   nodeRegion.Id,
				"name": nodeRegion.Name,
			}
		} else if trafficBill.NodeRegionId > 0 {
			regionMap = maps.Map{
				"id":   trafficBill.NodeRegionId,
				"name": "[已取消]",
			}
		}

		trafficBillMaps = append(trafficBillMaps, maps.Map{
			"priceType":           trafficBill.PriceType,
			"priceTypeName":       userconfigs.PriceTypeName(trafficBill.PriceType),
			"trafficGB":           numberutils.FormatBytes(int64(trafficBill.TrafficGB * (1 << 30))),
			"trafficPackageGB":    numberutils.FormatBytes(int64(trafficBill.TrafficPackageGB * (1 << 30))),
			"bandwidthMB":         numberutils.FormatBits(int64(trafficBill.BandwidthMB * (1 << 20))),
			"bandwidthPercentile": trafficBill.BandwidthPercentile,
			"amount":              numberutils.FormatFloat(trafficBill.Amount, 2),
			"pricePerUnit":        trafficBill.PricePerUnit,
			"region":              regionMap,
		})
	}
	this.Data["trafficBills"] = trafficBillMaps

	this.Show()
}