mirror of https://github.com/dirtbags/moth.git
100 lines
1.9 KiB
Go
100 lines
1.9 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"math"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/dirtbags/moth/pkg/award"
|
||
|
|
"github.com/go-redis/redis/v8"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Redis struct {
|
||
|
|
rdb *redis.Client
|
||
|
|
prefix string
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewRedis returns a new Redis structure
|
||
|
|
func NewRedis(rdb *redis.Client, prefix string) Redis {
|
||
|
|
return Redis{
|
||
|
|
rdb: rdb,
|
||
|
|
prefix: prefix,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r Redis) key(path ...string) string {
|
||
|
|
path = append([]string{r.prefix}, path...)
|
||
|
|
return strings.Join(path, "/")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Messages returns all broadcast messages
|
||
|
|
func (r Redis) Messages() string {
|
||
|
|
messages, err := r.rdb.Get(
|
||
|
|
context.TODO(),
|
||
|
|
r.key("messages.html"),
|
||
|
|
).Result()
|
||
|
|
if err != nil {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
return messages
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r Redis) PointsLog() award.List {
|
||
|
|
pointsLog, err := r.rdb.ZRangeArgsWithScores(
|
||
|
|
context.TODO(),
|
||
|
|
redis.ZRangeArgs{
|
||
|
|
Key: r.key("points.log"),
|
||
|
|
Start: math.Inf(-1),
|
||
|
|
Stop: math.Inf(+1),
|
||
|
|
},
|
||
|
|
).Result()
|
||
|
|
if err != nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
ret := make(award.List, len(pointsLog), 0)
|
||
|
|
for _, entry := range pointsLog {
|
||
|
|
// XXX: Fix this kludge
|
||
|
|
line := fmt.Sprintf("%d %s", int64(math.Round(entry.Score)), entry.Member.(string))
|
||
|
|
aent, err := award.Parse(line)
|
||
|
|
if err != nil {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
ret = append(ret, aent)
|
||
|
|
}
|
||
|
|
return ret
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r Redis) TeamName(teamID string) (string, error) {
|
||
|
|
name, err := r.rdb.Get(
|
||
|
|
context.TODO(),
|
||
|
|
r.key("teams", teamID),
|
||
|
|
).Result()
|
||
|
|
if err != nil {
|
||
|
|
return "", fmt.Errorf("No such team")
|
||
|
|
}
|
||
|
|
return name, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r Redis) SetTeamName(teamID, teamName string) error {
|
||
|
|
_, err := r.rdb.SetArgs(
|
||
|
|
context.TODO(),
|
||
|
|
r.key("teams", teamID),
|
||
|
|
teamName,
|
||
|
|
redis.SetArgs{
|
||
|
|
Mode: "NX",
|
||
|
|
},
|
||
|
|
).Result()
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r Redis) AwardPoints(teamID string, cat string, points int) error {
|
||
|
|
// XXX: Add to award something that makes a string without timestamp
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r Redis) Maintain(updateInterval time.Duration) {
|
||
|
|
// No maintenance required
|
||
|
|
}
|