mirror of
https://github.com/crunchy-labs/crunchy-cli.git
synced 2026-01-21 12:12:00 -06:00
Moved url parsing methods from utils to standard library
This commit is contained in:
parent
5219f18417
commit
d6347a6b6b
1 changed files with 15 additions and 16 deletions
110
url.go
Normal file
110
url.go
Normal file
|
|
@ -0,0 +1,110 @@
|
|||
package crunchyroll
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
)
|
||||
|
||||
// ExtractEpisodesFromUrl extracts all episodes from an url.
|
||||
// If audio is not empty, the episodes gets filtered after the given locale
|
||||
func (c *Crunchyroll) ExtractEpisodesFromUrl(url string, audio ...LOCALE) ([]*Episode, error) {
|
||||
series, episodes, err := c.ParseUrl(url)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var eps []*Episode
|
||||
|
||||
if series != nil {
|
||||
seasons, err := series.Seasons()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
for _, season := range seasons {
|
||||
if audio != nil {
|
||||
locale, err := season.AudioLocale()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var found bool
|
||||
for _, l := range audio {
|
||||
if locale == l {
|
||||
found = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !found {
|
||||
continue
|
||||
}
|
||||
}
|
||||
e, err := season.Episodes()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
eps = append(eps, e...)
|
||||
}
|
||||
} else if episodes != nil {
|
||||
if audio == nil {
|
||||
return episodes, nil
|
||||
}
|
||||
|
||||
for _, episode := range episodes {
|
||||
locale, err := episode.AudioLocale()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if audio != nil {
|
||||
var found bool
|
||||
for _, l := range audio {
|
||||
if locale == l {
|
||||
found = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !found {
|
||||
continue
|
||||
}
|
||||
}
|
||||
|
||||
eps = append(eps, episode)
|
||||
}
|
||||
}
|
||||
|
||||
if len(eps) == 0 {
|
||||
return nil, fmt.Errorf("could not find any matching episode")
|
||||
}
|
||||
|
||||
return eps, nil
|
||||
}
|
||||
|
||||
// ParseUrl parses the given url into a series or episode.
|
||||
// The returning episode is a slice because non-beta urls have the same episode with different languages
|
||||
func (c *Crunchyroll) ParseUrl(url string) (*Series, []*Episode, error) {
|
||||
if seriesId, ok := ParseBetaSeriesURL(url); ok {
|
||||
series, err := SeriesFromID(c, seriesId)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
return series, nil, nil
|
||||
} else if episodeId, ok := ParseBetaEpisodeURL(url); ok {
|
||||
episode, err := EpisodeFromID(c, episodeId)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
return nil, []*Episode{episode}, nil
|
||||
} else if seriesName, ok := ParseVideoURL(url); ok {
|
||||
video, err := c.FindVideoByName(seriesName)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
return video.(*Series), nil, nil
|
||||
} else if seriesName, title, _, _, ok := ParseEpisodeURL(url); ok {
|
||||
episodes, err := c.FindEpisodeByName(seriesName, title)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
return nil, episodes, nil
|
||||
} else {
|
||||
return nil, nil, fmt.Errorf("invalid url %s", url)
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue