128 lines
2.9 KiB
Go
128 lines
2.9 KiB
Go
package allanime
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"mal/internal/domain"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type AvailableEpisodes struct {
|
|
Sub []string
|
|
Dub []string
|
|
Raw []string
|
|
Titles map[int]string
|
|
}
|
|
|
|
func (c *AllAnimeProvider) GetEpisodeAvailability(ctx context.Context, animeID int, titleCandidates []string) (domain.EpisodeAvailability, error) {
|
|
showID, err := c.ResolveEpisodeProviderID(ctx, animeID, titleCandidates)
|
|
if err != nil {
|
|
return domain.EpisodeAvailability{}, err
|
|
}
|
|
return c.GetEpisodeAvailabilityByProviderID(ctx, showID)
|
|
}
|
|
|
|
func (c *AllAnimeProvider) GetEpisodeAvailabilityByProviderID(ctx context.Context, showID string) (domain.EpisodeAvailability, error) {
|
|
available, err := c.GetAvailableEpisodes(ctx, showID)
|
|
if err != nil {
|
|
return domain.EpisodeAvailability{}, err
|
|
}
|
|
|
|
sub := episodeNums(append(available.Sub, available.Raw...))
|
|
dub := episodeNums(available.Dub)
|
|
return domain.EpisodeAvailability{Sub: sub, Dub: dub, Titles: available.Titles}, nil
|
|
}
|
|
|
|
func (c *AllAnimeProvider) GetAvailableEpisodes(ctx context.Context, showID string) (AvailableEpisodes, error) {
|
|
graphqlQuery := `query($showId: String!, $start: Float!, $end: Float!) {
|
|
show(_id: $showId) {
|
|
availableEpisodesDetail
|
|
}
|
|
episodeInfos(showId: $showId, episodeNumStart: $start, episodeNumEnd: $end) {
|
|
episodeIdNum
|
|
notes
|
|
}
|
|
}`
|
|
|
|
result, err := c.graphqlRequest(ctx, graphqlQuery, map[string]any{
|
|
"showId": showID,
|
|
"start": 1,
|
|
"end": 100000,
|
|
})
|
|
if err != nil {
|
|
return AvailableEpisodes{}, err
|
|
}
|
|
|
|
data, ok := result["data"].(map[string]any)
|
|
if !ok {
|
|
return AvailableEpisodes{}, errors.New("invalid response")
|
|
}
|
|
|
|
show, ok := data["show"].(map[string]any)
|
|
if !ok || show == nil {
|
|
return AvailableEpisodes{}, errors.New("show not found")
|
|
}
|
|
|
|
detail, ok := show["availableEpisodesDetail"].(map[string]any)
|
|
if !ok {
|
|
return AvailableEpisodes{}, errors.New("invalid detail")
|
|
}
|
|
|
|
titles := map[int]string{}
|
|
infos, _ := data["episodeInfos"].([]any)
|
|
for _, value := range infos {
|
|
info, ok := value.(map[string]any)
|
|
if !ok {
|
|
continue
|
|
}
|
|
number := numericInt(info["episodeIdNum"])
|
|
title := plainText(stringValue(info["notes"]))
|
|
if number > 0 && title != "" {
|
|
titles[number] = title
|
|
}
|
|
}
|
|
|
|
return AvailableEpisodes{
|
|
Sub: stringsFrom(detail["sub"]),
|
|
Dub: stringsFrom(detail["dub"]),
|
|
Raw: stringsFrom(detail["raw"]),
|
|
Titles: titles,
|
|
}, nil
|
|
}
|
|
|
|
// episode ids
|
|
func episodeNums(raw []string) []int {
|
|
seen := make(map[int]bool, len(raw))
|
|
out := make([]int, 0, len(raw))
|
|
for _, value := range raw {
|
|
n, err := strconv.Atoi(strings.TrimSpace(value))
|
|
if err != nil || n <= 0 || seen[n] {
|
|
continue
|
|
}
|
|
seen[n] = true
|
|
out = append(out, n)
|
|
}
|
|
return out
|
|
}
|
|
|
|
// graphql list
|
|
func stringsFrom(value any) []string {
|
|
items, ok := value.([]any)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
|
|
values := make([]string, 0, len(items))
|
|
for _, item := range items {
|
|
str, ok := item.(string)
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
values = append(values, str)
|
|
}
|
|
|
|
return values
|
|
}
|