t2/models/capture.go

90 lines
1.6 KiB
Go

package models
import (
"cmp"
"slices"
"time"
"github.com/mergestat/timediff"
)
type MapScoreList []MapScore
type Capture struct {
Id uint
Host string
Port string
Name string
Active bool
Start time.Time
MapScores MapScoreList
}
type MapScore struct {
Id uint
StartTime time.Time
Map string
ScoreList []Score
}
type Score struct {
Id uint
Name string
Score int
Ping int
}
type ResultTable struct {
Header []ResultTableHeader
Rows []ResultTableRow
}
type ResultTableHeader struct {
Title string
Subtitle string
}
type ResultTableRow struct {
Name string
Total int
Individual []int
}
func (msl MapScoreList) BuildTable() *ResultTable {
rt := &ResultTable{
Header: []ResultTableHeader{{Title: "Name"}, {Title: "Total"}},
}
userMapRows := make(map[string]*ResultTableRow)
for mapIndex, mapScore := range msl {
rt.Header = append(rt.Header, ResultTableHeader{
Title: mapScore.Map,
Subtitle: timediff.TimeDiff(mapScore.StartTime),
})
for _, score := range mapScore.ScoreList {
if score.Ping > 0 {
if _, ok := userMapRows[score.Name]; !ok {
rt.Rows = append(rt.Rows, ResultTableRow{
Name: score.Name,
Total: 0,
Individual: make([]int, len(msl)),
})
userMapRows[score.Name] = &rt.Rows[len(rt.Rows)-1]
}
row := userMapRows[score.Name]
row.Total = row.Total + score.Score
row.Individual[mapIndex] = score.Score
}
}
}
slices.SortFunc(rt.Rows, func(a, b ResultTableRow) int {
return cmp.Compare(b.Total, a.Total)
})
return rt
}