Updates

Also custom errors

aoc.go

@@ -2,9 +2,8 @@ package aoc
 
 import (
 	"encoding/json"
-	"errors"
 	"fmt"
-	"io/ioutil"
+	"io"
 	"net/http"
 	"strconv"
 	"time"
@@ -14,14 +13,16 @@ type AoC struct {
 	session string
 	boardId string
 	boards  map[int]*Leaderboard
+
+	sessionErr bool
 }
 
 func NewAoC(boardId, session string) (*AoC, error) {
 	if boardId == "" {
-		return nil, errors.New("Board ID is required")
+		return nil, BoardIdRequiredError
 	}
 	if session == "" {
-		return nil, errors.New("Session key is required")
+		return nil, SessionRequiredError
 	}
 	return &AoC{
 		session: session,
@@ -30,19 +31,39 @@ func NewAoC(boardId, session string) (*AoC, error) {
 	}, nil
 }
 
+func (a *AoC) GetSession() string { return a.session }
+func (a *AoC) SetSession(session string) {
+	a.session = session
+	a.sessionErr = false
+}
+func (a *AoC) NeedsNewSession() bool { return a.sessionErr }
+
+func (a *AoC) SetCachedLeaderboard(l *Leaderboard) error {
+	yr, err := strconv.Atoi(l.Event)
+	if err != nil {
+		return InvalidYearError
+	}
+
+	a.boards[yr] = l
+	return nil
+}
+
 func (a *AoC) GetCachedLeaderboard(year int) (*Leaderboard, error) {
 	if year < 2015 || year > time.Now().Year() {
-		return nil, errors.New("Invalid Year")
+		return nil, InvalidYearError
 	}
 	if board, ok := a.boards[year]; ok {
 		return board, nil
 	}
-	return nil, errors.New(fmt.Sprintf("Leaderboard (%d) is not cached", year))
+	return nil, LeaderboardNotCachedError
 }
 
 func (a *AoC) GetLeaderboard(year int) (*Leaderboard, error) {
 	if year < 2015 || year > time.Now().Year() {
-		return nil, errors.New("Invalid Year")
+		return nil, InvalidYearError
+	}
+	if a.sessionErr {
+		return nil, SessionError
 	}
 	if board, ok := a.boards[year]; ok {
 		if time.Since(board.LastFetch) < (time.Minute * 15) {
@@ -53,7 +74,8 @@ func (a *AoC) GetLeaderboard(year int) (*Leaderboard, 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")
+			a.sessionErr = true
+			return nil, SessionError
 		}
 		return nil, err
 	}
@@ -71,23 +93,27 @@ func (a *AoC) fetchLeaderboard(year int) (*Leaderboard, error) {
 	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)
+	if err != nil {
+		return nil, err
+	}
 	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)
+	body, err = io.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 {
-		fmt.Println("Error parsing board")
-		fmt.Println(err)
-		return nil, err
+		if err.Error() == "invalid character '<' looking for beginning of value" {
+			a.sessionErr = true
+			return nil, SessionError
+		}
+		return nil, fmt.Errorf("error parsing board: %w", err)
 	}
 	for k, mbr := range leaderboard.Members {
 		starTs, err := strconv.ParseInt(mbr.RawStarTs, 10, 64)
@@ -144,7 +170,6 @@ func (a ByStarsThenScore) Less(i, j int) bool {
 }
 
 func (m *Member) UnmarshalJSON(data []byte) error {
-
 	var v map[string]interface{}
 	var err error
 

aoc_errors.go

@@ -0,0 +1,13 @@
+package aoc
+
+import (
+	"errors"
+)
+
+var (
+	SessionRequiredError      = errors.New("session key is required")
+	SessionError              = errors.New("invalid session")
+	BoardIdRequiredError      = errors.New("board id is required")
+	InvalidYearError          = errors.New("invalid year")
+	LeaderboardNotCachedError = errors.New("leaderboard is not cached")
+)