61 lines
1.1 KiB
Go
61 lines
1.1 KiB
Go
package anilist
|
|
|
|
import (
|
|
"time"
|
|
)
|
|
|
|
type GetSeasonKind int
|
|
|
|
const (
|
|
GetSeasonKindCurrent GetSeasonKind = iota
|
|
GetSeasonKindNext
|
|
GetSeasonKindPrevious
|
|
)
|
|
|
|
func GetSeasonInfo(now time.Time, kind GetSeasonKind) (MediaSeason, int) {
|
|
month, year := now.Month(), now.Year()
|
|
|
|
getSeasonIndex := func(m time.Month) int {
|
|
switch {
|
|
case m >= 3 && m <= 5: // spring: 3, 4, 5
|
|
return 1
|
|
case m >= 6 && m <= 8: // summer: 6, 7, 8
|
|
return 2
|
|
case m >= 9 && m <= 11: // fall: 9, 10, 11
|
|
return 3
|
|
default: // winter: 12, 1, 2
|
|
return 0
|
|
}
|
|
}
|
|
|
|
seasons := []MediaSeason{MediaSeasonWinter, MediaSeasonSpring, MediaSeasonSummer, MediaSeasonFall}
|
|
var index int
|
|
|
|
switch kind {
|
|
case GetSeasonKindCurrent:
|
|
index = getSeasonIndex(month)
|
|
|
|
case GetSeasonKindNext:
|
|
nextMonth := month + 3
|
|
nextYear := year
|
|
if nextMonth > 12 {
|
|
nextMonth -= 12
|
|
nextYear++
|
|
}
|
|
index = getSeasonIndex(nextMonth)
|
|
year = nextYear
|
|
|
|
case GetSeasonKindPrevious:
|
|
prevMonth := month - 3
|
|
prevYear := year
|
|
if prevMonth <= 0 {
|
|
prevMonth += 12
|
|
prevYear--
|
|
}
|
|
index = getSeasonIndex(prevMonth)
|
|
year = prevYear
|
|
}
|
|
|
|
return seasons[index], year
|
|
}
|