package recommended import ( "net/http" "sort" "github.com/aerogo/aero" "github.com/animenotifier/arn" "github.com/animenotifier/notify.moe/components" "github.com/animenotifier/notify.moe/utils" ) const ( maxRecommendations = 10 ) // Anime shows a list of recommended anime. func Anime(ctx *aero.Context) string { user := utils.GetUser(ctx) nick := ctx.Get("nick") viewUser, err := arn.GetUserByNick(nick) if err != nil { return ctx.Error(http.StatusUnauthorized, "Not logged in", err) } animeList := viewUser.AnimeList() // 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 { // Skip anime that are upcoming or tba if anime.Status == "upcoming" || anime.Status == "tba" { continue } if anime.Type == "tv" { tv = append(tv, anime) } else if anime.Type == "movie" { movies = append(movies, anime) } // Skip anime from my list (except planned anime) existing := animeList.Find(anime.ID) if existing != nil && existing.Status != arn.AnimeListStatusPlanned { continue } animeAffinity := anime.Score() // Planned anime go higher if existing != nil && existing.Status == arn.AnimeListStatusPlanned { animeAffinity += 15.0 } affinity[anime.ID] = animeAffinity } // 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] if affinityA == affinityB { return movies[i].Title.Canonical < movies[j].Title.Canonical } 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)) }