package logic

import (
	"context"

	"git.milar.in/milarin/anilist"
	"git.milar.in/milarin/channel"
)

func GetAnimeListByAnimeID(statuses []anilist.MediaListStatus) (map[anilist.MediaID]*anilist.MediaList, error) {
	animeListChannel, err := GetCurrentlyWatchingAnimes(statuses...)
	if err != nil {
		return nil, err
	}

	toMapFunc := func(entry *anilist.MediaList) (anilist.MediaID, *anilist.MediaList) { return entry.MediaID, entry }
	return channel.ToMap(animeListChannel, toMapFunc), nil
}

func GetCurrentlyWatchingAnimes(statuses ...anilist.MediaListStatus) (<-chan *anilist.MediaList, error) {
	return GetCurrentlyWatchingAnimesContext(context.Background(), statuses...)
}

func GetCurrentlyWatchingAnimesContext(ctx context.Context, statuses ...anilist.MediaListStatus) (<-chan *anilist.MediaList, error) {
	token, err := GetAnilistAccessToken()
	if err != nil {
		return nil, ErrAnimeListNotObtainable.Wrap(err, "access token acquisition failed")
	}

	media := channel.Map(channel.Of(statuses...), func(status anilist.MediaListStatus) <-chan *anilist.MediaList {
		return anilist.NewApi(token).GetMediaList(ctx, anilist.MediaListQuery{
			UserName: AnilistUsername,
			Type:     anilist.MediaTypeAnime,
			Status:   status,
		}, nil).Chan
	})

	return channel.FlatChan(media), nil
}

var (
	animeByTitleCache = map[string]*anilist.Media{}
)

func SearchAnimeByTitle(title string) (anime *anilist.Media, err error) {
	// caching
	if cacheEntry, ok := animeByTitleCache[title]; ok {
		return cacheEntry, nil
	}
	defer func() {
		if err != nil && anime != nil {
			animeByTitleCache[title] = anime
		}
	}()

	token, err := GetAnilistAccessToken()
	if err != nil {
		return nil, err
	}

	anime = anilist.NewApi(token).GetMedia(context.Background(), anilist.MediaQuery{
		Search: title,
		Type:   anilist.MediaTypeAnime,
	}, nil).First()

	if anime == nil {
		return nil, ErrAnimeNotFound.New(title)
	}

	return anime, nil
}