111 lines
2.3 KiB
Go
Raw Normal View History

2016-11-19 14:54:31 +00:00
package genre
import (
"strings"
2016-11-19 14:54:31 +00:00
"github.com/aerogo/aero"
2019-06-03 09:32:43 +00:00
"github.com/animenotifier/notify.moe/arn"
2016-11-19 14:54:31 +00:00
"github.com/animenotifier/notify.moe/components"
2017-11-12 10:18:16 +00:00
"github.com/animenotifier/notify.moe/utils"
2016-11-19 14:54:31 +00:00
)
const animePerPage = 100
2018-10-10 09:37:31 +00:00
const animeRatingCountThreshold = 5
2018-10-10 09:37:31 +00:00
// Get renders the genre page.
2019-06-01 04:55:49 +00:00
func Get(ctx aero.Context) error {
2017-11-12 10:18:16 +00:00
user := utils.GetUser(ctx)
2016-11-19 14:54:31 +00:00
genreName := ctx.Get("name")
animes := []*arn.Anime{}
for anime := range arn.StreamAnime() {
if containsLowerCase(anime.Genres, genreName) {
animes = append(animes, anime)
}
}
userScore := averageUserScore(user, animes)
userCompleted := totalCompleted(user, animes)
globalScore := averageGlobalScore(animes)
2018-03-14 00:02:41 +00:00
arn.SortAnimeByQuality(animes)
if len(animes) > animePerPage {
animes = animes[:animePerPage]
}
return ctx.HTML(components.Genre(genreName, animes, user, userScore, userCompleted, globalScore))
}
2016-11-19 14:54:31 +00:00
// containsLowerCase tells you whether the given element exists when all elements are lowercased.
func containsLowerCase(array []string, search string) bool {
for _, item := range array {
if strings.ToLower(item) == search {
return true
}
2016-11-19 14:54:31 +00:00
}
return false
2016-11-19 14:54:31 +00:00
}
2018-10-10 09:37:31 +00:00
// averageUserScore counts the user's average score for a list of anime.
func averageUserScore(user *arn.User, animes []*arn.Anime) float64 {
if user == nil {
return 0
}
count := 0.0
scores := 0.0
animeList := user.AnimeList()
for _, anime := range animes {
userAnime := animeList.Find(anime.ID)
2018-07-03 13:33:35 +00:00
if userAnime != nil && !userAnime.Rating.IsNotRated() {
scores += userAnime.Rating.Overall
count++
}
}
2018-07-03 13:33:35 +00:00
if count == 0.0 {
return 0
}
return scores / count
}
2018-10-10 09:37:31 +00:00
// averageGlobalScore returns the average overall score for the given anime.
func averageGlobalScore(animes []*arn.Anime) float64 {
sum := 0.0
count := 0
for _, anime := range animes {
2018-10-10 09:37:31 +00:00
if anime.Rating.Count.Overall >= animeRatingCountThreshold {
sum += anime.Rating.Overall
count++
}
}
return sum / float64(count)
}
2018-10-10 09:37:31 +00:00
// totalCompleted counts the number of anime the user has completed from a given list of anime.
func totalCompleted(user *arn.User, animes []*arn.Anime) int {
if user == nil {
return 0
}
count := 0
completedList := user.AnimeList().FilterStatus(arn.AnimeListStatusCompleted)
for _, anime := range animes {
2018-07-28 14:47:25 +00:00
if completedList.Contains(anime.ID) {
count++
}
}
2018-07-28 14:47:25 +00:00
return count
}