Initial commit (code only without large binaries)

EdgeUser/internal/web/actions/default/servers/traffic-stats/download.go

@@ -0,0 +1,105 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package trafficstats
+
+import (
+	"bytes"
+	"fmt"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeUser/internal/configloaders"
+	"github.com/TeaOSLab/EdgeUser/internal/web/actions/actionutils"
+	"github.com/iwind/TeaGo/lists"
+	"github.com/iwind/TeaGo/types"
+	"github.com/tealeg/xlsx/v3"
+	"strconv"
+	"strings"
+)
+
+type DownloadAction struct {
+	actionutils.ParentAction
+}
+
+func (this *DownloadAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *DownloadAction) RunGet(params struct {
+	DayFrom  string
+	DayTo    string
+	ServerId int64
+}) {
+	// 是否显示带宽
+	config, err := configloaders.LoadUIConfig()
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	if config == nil || !config.ShowBandwidthCharts {
+		return
+	}
+
+	// 用户名
+	var username = "unknown"
+	userResp, err := this.RPC().UserRPC().FindEnabledUser(this.UserContext(), &pb.FindEnabledUserRequest{UserId: this.UserId()})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	if userResp.User != nil {
+		username = userResp.User.Username
+	}
+
+	params.DayFrom = strings.ReplaceAll(params.DayFrom, "-", "")
+	params.DayTo = strings.ReplaceAll(params.DayTo, "-", "")
+
+	bandwidthStatsResp, err := this.RPC().ServerBandwidthStatRPC().FindDailyServerBandwidthStatsBetweenDays(this.UserContext(), &pb.FindDailyServerBandwidthStatsBetweenDaysRequest{
+		ServerId: params.ServerId,
+		DayFrom:  params.DayFrom,
+		DayTo:    params.DayTo,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	var wb = xlsx.NewFile()
+	sheet, err := wb.AddSheet("default")
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	// 头部
+	{
+		var row = sheet.AddRow()
+		row.SetHeight(25)
+		row.AddCell().SetString("时间")
+		row.AddCell().SetString("带宽（Mbps）")
+		row.AddCell().SetString("带宽（Gbps）")
+	}
+
+	var stats = bandwidthStatsResp.Stats
+	lists.Reverse(stats)
+	for _, stat := range stats {
+		var row = sheet.AddRow()
+		row.SetHeight(25)
+		row.AddCell().SetString(stat.Day[:4] + "-" + stat.Day[4:6] + "-" + stat.Day[6:] + " " + stat.TimeAt[:2] + ":" + stat.TimeAt[2:])
+		row.AddCell().SetString(fmt.Sprintf("%.4f", float64(stat.Bits)/(1<<20)))
+		row.AddCell().SetString(fmt.Sprintf("%.4f", float64(stat.Bits)/(1<<30)))
+	}
+
+	this.AddHeader("Content-Type", "application/vnd.ms-excel")
+	this.AddHeader("Content-Disposition", "attachment; filename=\"BANDWIDTH-"+username+"-"+types.String(params.ServerId)+"-"+params.DayFrom+"-"+params.DayTo+".xlsx\"")
+	this.AddHeader("Cache-Control", "max-age=0")
+
+	var buf = bytes.NewBuffer([]byte{})
+	err = wb.Write(buf)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.AddHeader("Content-Length", strconv.Itoa(buf.Len()))
+	_, _ = this.Write(buf.Bytes())
+}