102 lines
2.3 KiB
Go
Raw Normal View History

package recommended
import (
"net/http"
"sort"
"github.com/aerogo/aero"
"github.com/animenotifier/arn"
"github.com/animenotifier/notify.moe/components"
2018-02-24 14:35:34 +00:00
"github.com/animenotifier/notify.moe/utils"
)
2018-02-24 13:51:47 +00:00
const (
2018-03-27 08:13:27 +00:00
maxRecommendations = 10
bestGenreCount = 3
genreBonusCompletedAnimeThreshold = 40
2018-02-24 13:51:47 +00:00
)
// Anime shows a list of recommended anime.
func Anime(ctx *aero.Context) string {
2018-02-24 14:35:34 +00:00
user := utils.GetUser(ctx)
nick := ctx.Get("nick")
2018-02-24 14:35:34 +00:00
viewUser, err := arn.GetUserByNick(nick)
if err != nil {
return ctx.Error(http.StatusUnauthorized, "Not logged in", err)
}
2018-02-24 14:35:34 +00:00
animeList := viewUser.AnimeList()
2018-03-27 08:13:27 +00:00
completed := animeList.FilterStatus(arn.AnimeListStatusCompleted)
// Genre affinity
bestGenres := getBestGenres(animeList)
2018-03-27 08:13:27 +00:00
// Get all anime
var tv []*arn.Anime
var movies []*arn.Anime
allAnime := arn.AllAnime()
// Affinity maps an anime ID to a number that indicates how likely a user is going to enjoy that anime.
affinity := map[string]float64{}
// Calculate affinity for each anime
for _, anime := range allAnime {
2018-03-04 00:02:30 +00:00
// Skip anime that are upcoming or tba
if anime.Status == "upcoming" || anime.Status == "tba" {
2018-02-24 13:51:47 +00:00
continue
}
if anime.Type == "tv" {
tv = append(tv, anime)
} else if anime.Type == "movie" {
movies = append(movies, anime)
} else {
continue
}
// Skip anime from my list (except planned anime)
existing := animeList.Find(anime.ID)
if existing != nil && existing.Status != arn.AnimeListStatusPlanned {
continue
}
affinity[anime.ID] = getAnimeAffinity(anime, existing, completed, bestGenres)
}
// Sort
sort.Slice(tv, func(i, j int) bool {
affinityA := affinity[tv[i].ID]
affinityB := affinity[tv[j].ID]
if affinityA == affinityB {
return tv[i].Title.Canonical < tv[j].Title.Canonical
}
return affinityA > affinityB
})
sort.Slice(movies, func(i, j int) bool {
affinityA := affinity[movies[i].ID]
affinityB := affinity[movies[j].ID]
2018-02-24 13:51:47 +00:00
if affinityA == affinityB {
return movies[i].Title.Canonical < movies[j].Title.Canonical
2018-02-24 13:51:47 +00:00
}
return affinityA > affinityB
})
// Take the top 10
if len(tv) > maxRecommendations {
tv = tv[:maxRecommendations]
}
if len(movies) > maxRecommendations {
movies = movies[:maxRecommendations]
}
return ctx.HTML(components.RecommendedAnime(tv, movies, viewUser, user))
}