123 lines
2.9 KiB
Go
Raw Normal View History

2017-07-03 17:33:52 +00:00
package listimportanilist
import (
2019-06-01 04:55:49 +00:00
"errors"
"fmt"
2017-07-03 17:33:52 +00:00
"net/http"
2018-04-05 20:36:59 +00:00
"strconv"
2017-07-03 17:33:52 +00:00
"github.com/aerogo/aero"
2017-07-09 13:56:40 +00:00
"github.com/animenotifier/anilist"
2019-06-03 09:32:43 +00:00
"github.com/animenotifier/notify.moe/arn"
2017-07-03 17:33:52 +00:00
"github.com/animenotifier/notify.moe/components"
)
2017-07-07 23:18:21 +00:00
// Preview shows an import preview.
2019-06-01 04:55:49 +00:00
func Preview(ctx aero.Context) error {
2019-11-17 07:59:34 +00:00
user := arn.GetUserFromContext(ctx)
2017-07-03 17:33:52 +00:00
if user == nil {
2018-07-07 03:42:00 +00:00
return ctx.Error(http.StatusBadRequest, "Not logged in")
2017-07-03 17:33:52 +00:00
}
2019-06-01 04:55:49 +00:00
matches, err := getMatches(ctx)
2017-07-04 22:40:03 +00:00
2019-06-01 04:55:49 +00:00
if err != nil {
return ctx.Error(http.StatusBadRequest, err)
2017-07-04 22:40:03 +00:00
}
2017-07-04 08:26:18 +00:00
return ctx.HTML(components.ImportAnilist(user, matches))
}
2017-07-04 22:40:03 +00:00
// Finish ...
2019-06-01 04:55:49 +00:00
func Finish(ctx aero.Context) error {
2019-11-17 07:59:34 +00:00
user := arn.GetUserFromContext(ctx)
2017-07-07 23:18:21 +00:00
if user == nil {
2018-07-07 03:42:00 +00:00
return ctx.Error(http.StatusBadRequest, "Not logged in")
2017-07-07 23:18:21 +00:00
}
2019-06-01 04:55:49 +00:00
matches, err := getMatches(ctx)
2017-07-04 22:40:03 +00:00
2019-06-01 04:55:49 +00:00
if err != nil {
return ctx.Error(http.StatusBadRequest, err)
2017-07-04 22:40:03 +00:00
}
animeList := user.AnimeList()
for _, match := range matches {
if match.ARNAnime == nil || match.AniListItem == nil {
continue
}
item := &arn.AnimeListItem{
AnimeID: match.ARNAnime.ID,
2017-07-09 13:56:40 +00:00
Status: arn.AniListAnimeListStatus(match.AniListItem),
2018-04-05 20:36:59 +00:00
Episodes: match.AniListItem.Progress,
2017-07-04 22:40:03 +00:00
Notes: match.AniListItem.Notes,
2018-03-27 06:19:01 +00:00
Rating: arn.AnimeListItemRating{
2017-07-04 22:40:03 +00:00
Overall: float64(match.AniListItem.ScoreRaw) / 10.0,
},
2018-04-05 20:36:59 +00:00
RewatchCount: match.AniListItem.Repeat,
Private: match.AniListItem.Private,
2017-07-04 22:40:03 +00:00
Created: arn.DateTimeUTC(),
Edited: arn.DateTimeUTC(),
}
animeList.Import(item)
}
2017-10-27 07:11:56 +00:00
animeList.Save()
2018-11-15 03:42:10 +00:00
return ctx.HTML(components.ImportFinished(user))
2017-07-04 22:40:03 +00:00
}
2017-07-07 23:18:21 +00:00
// getMatches finds and returns all matches for the logged in user.
2019-06-01 04:55:49 +00:00
func getMatches(ctx aero.Context) ([]*arn.AniListMatch, error) {
2019-11-17 07:59:34 +00:00
user := arn.GetUserFromContext(ctx)
2017-07-07 23:18:21 +00:00
if user == nil {
2019-06-01 04:55:49 +00:00
return nil, errors.New("Not logged in")
2017-07-07 23:18:21 +00:00
}
2018-04-05 20:36:59 +00:00
anilistUser, err := anilist.GetUser(user.Accounts.AniList.Nick)
2017-07-07 23:18:21 +00:00
2018-04-05 20:36:59 +00:00
if err != nil {
2019-06-01 04:55:49 +00:00
return nil, fmt.Errorf("User doesn't exist on AniList: %s", err.Error())
2017-07-07 23:18:21 +00:00
}
2018-04-05 20:36:59 +00:00
anilistAnimeList, err := anilist.GetAnimeList(anilistUser.ID)
2017-07-07 23:18:21 +00:00
if err != nil {
2019-06-01 04:55:49 +00:00
return nil, fmt.Errorf("Couldn't load your anime list from AniList: %s", err.Error())
2017-07-07 23:18:21 +00:00
}
2018-04-05 20:36:59 +00:00
// Find matches
matches := findAllMatches(anilistAnimeList)
2017-07-07 23:18:21 +00:00
2019-06-01 04:55:49 +00:00
return matches, nil
2017-07-07 23:18:21 +00:00
}
2017-07-04 08:26:18 +00:00
// findAllMatches returns all matches for the anime inside an anilist anime list.
2018-04-05 20:36:59 +00:00
func findAllMatches(animeList *anilist.AnimeList) []*arn.AniListMatch {
finder := arn.NewAniListAnimeFinder()
2017-07-03 17:33:52 +00:00
matches := []*arn.AniListMatch{}
2018-04-05 20:36:59 +00:00
for _, list := range animeList.Lists {
matches = importList(matches, finder, list.Entries)
2017-07-03 17:33:52 +00:00
}
2017-07-04 08:26:18 +00:00
return matches
2017-07-03 17:33:52 +00:00
}
2017-07-04 08:26:18 +00:00
// importList imports a single list inside an anilist anime list collection.
2018-04-05 20:36:59 +00:00
func importList(matches []*arn.AniListMatch, finder *arn.AniListAnimeFinder, animeListItems []*anilist.AnimeListItem) []*arn.AniListMatch {
2017-07-03 17:33:52 +00:00
for _, item := range animeListItems {
matches = append(matches, &arn.AniListMatch{
2017-07-04 22:40:03 +00:00
AniListItem: item,
2018-04-05 20:36:59 +00:00
ARNAnime: finder.GetAnime(strconv.Itoa(item.Anime.ID), strconv.Itoa(item.Anime.MALID)),
2017-07-03 17:33:52 +00:00
})
}
return matches
}