mirror of
https://github.com/crunchy-labs/crunchy-cli.git
synced 2026-01-21 04:02:00 -06:00
123 lines
3.2 KiB
Go
123 lines
3.2 KiB
Go
package crunchyroll
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"regexp"
|
|
)
|
|
|
|
// Season contains information about an anime season.
|
|
type Season struct {
|
|
crunchy *Crunchyroll
|
|
|
|
children []*Episode
|
|
|
|
ID string `json:"id"`
|
|
ChannelID string `json:"channel_id"`
|
|
|
|
Title string `json:"title"`
|
|
SlugTitle string `json:"slug_title"`
|
|
|
|
SeriesID string `json:"series_id"`
|
|
SeasonNumber int `json:"season_number"`
|
|
|
|
IsComplete bool `json:"is_complete"`
|
|
|
|
Description string `json:"description"`
|
|
Keywords []string `json:"keywords"`
|
|
SeasonTags []string `json:"season_tags"`
|
|
IsMature bool `json:"is_mature"`
|
|
MatureBlocked bool `json:"mature_blocked"`
|
|
IsSubbed bool `json:"is_subbed"`
|
|
IsDubbed bool `json:"is_dubbed"`
|
|
IsSimulcast bool `json:"is_simulcast"`
|
|
|
|
SeoTitle string `json:"seo_title"`
|
|
SeoDescription string `json:"seo_description"`
|
|
|
|
AvailabilityNotes string `json:"availability_notes"`
|
|
|
|
// the locales are always empty, idk why this may change in the future
|
|
AudioLocales []LOCALE
|
|
SubtitleLocales []LOCALE
|
|
}
|
|
|
|
// SeasonFromID returns a season by its api id.
|
|
func SeasonFromID(crunchy *Crunchyroll, id string) (*Season, error) {
|
|
resp, err := crunchy.Client.Get(fmt.Sprintf("https://beta-api.crunchyroll.com/cms/v2/%s/seasons?series_id=%s&locale=%s&Signature=%s&Policy=%s&Key-Pair-Id=%s",
|
|
crunchy.Config.Bucket,
|
|
id,
|
|
crunchy.Locale,
|
|
crunchy.Config.Signature,
|
|
crunchy.Config.Policy,
|
|
crunchy.Config.KeyPairID))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer resp.Body.Close()
|
|
var jsonBody map[string]interface{}
|
|
json.NewDecoder(resp.Body).Decode(&jsonBody)
|
|
|
|
season := &Season{
|
|
crunchy: crunchy,
|
|
ID: id,
|
|
}
|
|
if err := decodeMapToStruct(jsonBody, season); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return season, nil
|
|
}
|
|
|
|
// AudioLocale returns the audio locale of the season.
|
|
func (s *Season) AudioLocale() (LOCALE, error) {
|
|
episodes, err := s.Episodes()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return episodes[0].AudioLocale()
|
|
}
|
|
|
|
// Episodes returns all episodes which are available for the season.
|
|
func (s *Season) Episodes() (episodes []*Episode, err error) {
|
|
if s.children != nil {
|
|
return s.children, nil
|
|
}
|
|
|
|
resp, err := s.crunchy.request(fmt.Sprintf("https://beta-api.crunchyroll.com/cms/v2/%s/episodes?season_id=%s&locale=%s&Signature=%s&Policy=%s&Key-Pair-Id=%s",
|
|
s.crunchy.Config.Bucket,
|
|
s.ID,
|
|
s.crunchy.Locale,
|
|
s.crunchy.Config.Signature,
|
|
s.crunchy.Config.Policy,
|
|
s.crunchy.Config.KeyPairID))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer resp.Body.Close()
|
|
var jsonBody map[string]interface{}
|
|
json.NewDecoder(resp.Body).Decode(&jsonBody)
|
|
|
|
for _, item := range jsonBody["items"].([]interface{}) {
|
|
episode := &Episode{
|
|
crunchy: s.crunchy,
|
|
}
|
|
if err = decodeMapToStruct(item, episode); err != nil {
|
|
return nil, err
|
|
}
|
|
if episode.Playback != "" {
|
|
streamHref := item.(map[string]interface{})["__links__"].(map[string]interface{})["streams"].(map[string]interface{})["href"].(string)
|
|
if match := regexp.MustCompile(`(?m)(\w+)/streams$`).FindAllStringSubmatch(streamHref, -1); len(match) > 0 {
|
|
episode.StreamID = match[0][1]
|
|
} else {
|
|
fmt.Println()
|
|
}
|
|
}
|
|
episodes = append(episodes, episode)
|
|
}
|
|
|
|
if s.crunchy.cache {
|
|
s.children = episodes
|
|
}
|
|
return
|
|
}
|