mirror of https://github.com/velour/catbase.git
114 lines
2.6 KiB
Go
114 lines
2.6 KiB
Go
package aoc
|
|
|
|
import (
|
|
"fmt"
|
|
"sort"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"code.chrissexton.org/cws/getaoc"
|
|
|
|
"github.com/velour/catbase/bot"
|
|
"github.com/velour/catbase/bot/msg"
|
|
"github.com/velour/catbase/config"
|
|
)
|
|
|
|
type AOC struct {
|
|
b bot.Bot
|
|
c *config.Config
|
|
}
|
|
|
|
func New(b bot.Bot) *AOC {
|
|
aoc := &AOC{
|
|
b: b,
|
|
c: b.Config(),
|
|
}
|
|
b.Register(aoc, bot.Message, aoc.message)
|
|
return aoc
|
|
}
|
|
|
|
func (p *AOC) message(c bot.Connector, kind bot.Kind, message msg.Message, args ...interface{}) bool {
|
|
if !message.Command {
|
|
return false
|
|
}
|
|
cleaned := strings.TrimSpace(strings.ToLower(message.Body))
|
|
fields := strings.Split(cleaned, " ")
|
|
if strings.HasPrefix(cleaned, "aoc") {
|
|
year := time.Now().Year()
|
|
if len(fields) > 1 {
|
|
year, _ = strconv.Atoi(fields[1])
|
|
}
|
|
boardId := p.c.GetInt("aoc.board", 0)
|
|
var err error
|
|
if len(fields) > 2 {
|
|
boardId, err = strconv.Atoi(fields[2])
|
|
if err != nil {
|
|
p.b.Send(c, bot.Message, message.Channel, fmt.Sprintf("Error getting leaderboard: %s", err))
|
|
return true
|
|
}
|
|
}
|
|
session := p.c.Get("aoc.session", "")
|
|
if session == "" {
|
|
p.b.Send(c, bot.Message, message.Channel, "Error getting leaderboard: must have a session ID.")
|
|
return true
|
|
}
|
|
if boardId == 0 {
|
|
p.b.Send(c, bot.Message, message.Channel, "Error getting leaderboard: must have a board ID.")
|
|
return true
|
|
}
|
|
board, err := getaoc.GetLeaderboard(session, year, boardId)
|
|
if err != nil {
|
|
p.b.Send(c, bot.Message, message.Channel, fmt.Sprintf("Error getting leaderboard: %s", err))
|
|
return true
|
|
}
|
|
|
|
members := []getaoc.Member{}
|
|
for _, m := range board.Members {
|
|
members = append(members, m)
|
|
}
|
|
sort.Slice(members, func(i, j int) bool {
|
|
return members[i].LocalScore > members[j].LocalScore
|
|
})
|
|
|
|
gold, silver, bronze := -1, -1, -1
|
|
goldID, silverID, bronzeID := "", "", ""
|
|
for _, m := range members {
|
|
if m.LocalScore > gold {
|
|
gold = m.LocalScore
|
|
goldID = m.ID
|
|
}
|
|
if m.LocalScore < gold && m.LocalScore > silver {
|
|
silver = m.LocalScore
|
|
silverID = m.ID
|
|
}
|
|
if m.LocalScore < silver && m.LocalScore > bronze {
|
|
bronze = m.LocalScore
|
|
bronzeID = m.ID
|
|
}
|
|
}
|
|
|
|
msg := "AoC Leaderboard:\n"
|
|
for _, m := range members {
|
|
if m.Stars == 0 {
|
|
continue
|
|
}
|
|
trophy := ""
|
|
switch m.ID {
|
|
case goldID:
|
|
trophy = ":gold-trophy:"
|
|
case silverID:
|
|
trophy = ":silver-trophy:"
|
|
case bronzeID:
|
|
trophy = ":bronze-trophy:"
|
|
}
|
|
msg += fmt.Sprintf("%s has %d :star: for a score of %d%s\n", m.Name, m.Stars, m.LocalScore, trophy)
|
|
}
|
|
msg = strings.TrimSpace(msg)
|
|
|
|
p.b.Send(c, bot.Message, message.Channel, msg)
|
|
return true
|
|
}
|
|
return false
|
|
}
|