Initial Commit
This commit is contained in:
commit
a3d9eb97a9
144
aoc.go
Normal file
144
aoc.go
Normal file
@ -0,0 +1,144 @@
|
|||||||
|
package aoc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
|
"net/http"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type AoC struct {
|
||||||
|
session string
|
||||||
|
boardId string
|
||||||
|
boards map[int]*Leaderboard
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewAoC(boardId, session string) (*AoC, error) {
|
||||||
|
if boardId == "" {
|
||||||
|
return nil, errors.New("Board ID is required")
|
||||||
|
}
|
||||||
|
if session == "" {
|
||||||
|
return nil, errors.New("Session key is required")
|
||||||
|
}
|
||||||
|
return &AoC{
|
||||||
|
session: session,
|
||||||
|
boardId: boardId,
|
||||||
|
boards: make(map[int]*Leaderboard),
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AoC) GetCachedLeaderboard(year int) (*Leaderboard, error) {
|
||||||
|
if year < 2015 || year > time.Now().Year() {
|
||||||
|
return nil, errors.New("Invalid Year")
|
||||||
|
}
|
||||||
|
if board, ok := a.boards[year]; ok {
|
||||||
|
return board, nil
|
||||||
|
}
|
||||||
|
return nil, errors.New(fmt.Sprintf("Leaderboard (%d) is not cached", year))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AoC) GetLeaderboard(year int) (*Leaderboard, error) {
|
||||||
|
if year < 2015 || year > time.Now().Year() {
|
||||||
|
return nil, errors.New("Invalid Year")
|
||||||
|
}
|
||||||
|
if board, ok := a.boards[year]; ok {
|
||||||
|
if time.Since(board.LastFetch) < (time.Minute * 10) {
|
||||||
|
return board, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
var err error
|
||||||
|
a.boards[year], err = a.fetchLeaderboard(year)
|
||||||
|
if err != nil {
|
||||||
|
if err.Error() == "invalid character '<' looking for beginning of value" {
|
||||||
|
return nil, errors.New("Invalid Session Cookie")
|
||||||
|
}
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return a.boards[year], nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AoC) fetchLeaderboard(year int) (*Leaderboard, error) {
|
||||||
|
var err error
|
||||||
|
var req *http.Request
|
||||||
|
var resp *http.Response
|
||||||
|
var body []byte
|
||||||
|
leaderboard := new(Leaderboard)
|
||||||
|
leaderboard.Members = make(map[string]Member)
|
||||||
|
|
||||||
|
client := &http.Client{}
|
||||||
|
boardString := fmt.Sprintf("https://adventofcode.com/%d/leaderboard/private/view/%s.json", year, a.boardId)
|
||||||
|
|
||||||
|
req, err = http.NewRequest("GET", boardString, nil)
|
||||||
|
req.Header.Add("Cookie", "session="+a.session)
|
||||||
|
resp, err = client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
body, err = ioutil.ReadAll(resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
strBody := string(body)
|
||||||
|
strBody = strings.ReplaceAll(strBody, "\"last_star_ts\":0", "\"last_star_ts\":\"0\"")
|
||||||
|
err = json.Unmarshal([]byte(strBody), &leaderboard)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
for k, mbr := range leaderboard.Members {
|
||||||
|
starTs, err := strconv.ParseInt(mbr.RawStarTs, 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
mbr.LastStarTs = time.Unix(starTs, 0)
|
||||||
|
leaderboard.Members[k] = mbr
|
||||||
|
}
|
||||||
|
leaderboard.LastFetch = time.Now()
|
||||||
|
a.boards[year] = leaderboard
|
||||||
|
return leaderboard, err
|
||||||
|
}
|
||||||
|
|
||||||
|
type Leaderboard struct {
|
||||||
|
OwnerID string `json:"owner_id"`
|
||||||
|
Event string `json:"event"`
|
||||||
|
Members map[string]Member `json:"members"`
|
||||||
|
LastFetch time.Time
|
||||||
|
}
|
||||||
|
|
||||||
|
type Member struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Stars int `json:"stars"`
|
||||||
|
RawStarTs string `json:"last_star_ts"`
|
||||||
|
LocalScore int `json:"local_score"`
|
||||||
|
GlobalScore int `json:"global_score"`
|
||||||
|
LastStarTs time.Time
|
||||||
|
SlackID string
|
||||||
|
}
|
||||||
|
|
||||||
|
type ByLocalScore []Member
|
||||||
|
|
||||||
|
func (a ByLocalScore) Len() int { return len(a) }
|
||||||
|
func (a ByLocalScore) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
|
||||||
|
func (a ByLocalScore) Less(i, j int) bool { return a[i].LocalScore < a[j].LocalScore }
|
||||||
|
|
||||||
|
type ByStars []Member
|
||||||
|
|
||||||
|
func (a ByStars) Len() int { return len(a) }
|
||||||
|
func (a ByStars) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
|
||||||
|
func (a ByStars) Less(i, j int) bool { return a[i].Stars < a[j].Stars }
|
||||||
|
|
||||||
|
type ByStarsThenScore []Member
|
||||||
|
|
||||||
|
func (a ByStarsThenScore) Len() int { return len(a) }
|
||||||
|
func (a ByStarsThenScore) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
|
||||||
|
func (a ByStarsThenScore) Less(i, j int) bool {
|
||||||
|
if a[i].Stars == a[j].Stars {
|
||||||
|
return a[i].LocalScore < a[j].LocalScore
|
||||||
|
}
|
||||||
|
return a[i].Stars < a[j].Stars
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user