2016-11-19 14:54:31 +00:00
|
|
|
package genre
|
|
|
|
|
|
|
|
import (
|
2017-11-30 00:47:12 +00:00
|
|
|
"strings"
|
|
|
|
|
2016-11-19 14:54:31 +00:00
|
|
|
"github.com/aerogo/aero"
|
|
|
|
"github.com/animenotifier/arn"
|
|
|
|
"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
|
|
|
)
|
|
|
|
|
2017-11-30 00:47:12 +00:00
|
|
|
const animePerPage = 100
|
|
|
|
|
2016-11-19 14:54:31 +00:00
|
|
|
// Get ...
|
|
|
|
func Get(ctx *aero.Context) string {
|
2017-11-12 10:18:16 +00:00
|
|
|
user := utils.GetUser(ctx)
|
2016-11-19 14:54:31 +00:00
|
|
|
genreName := ctx.Get("name")
|
2017-11-30 00:47:12 +00:00
|
|
|
animes := []*arn.Anime{}
|
|
|
|
|
|
|
|
for anime := range arn.StreamAnime() {
|
|
|
|
if containsLowerCase(anime.Genres, genreName) {
|
|
|
|
animes = append(animes, anime)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-03 10:13:38 +00:00
|
|
|
userScore := averageScore(user, animes)
|
2018-07-02 17:57:44 +00:00
|
|
|
|
2018-03-14 00:02:41 +00:00
|
|
|
arn.SortAnimeByQuality(animes)
|
2017-11-30 00:47:12 +00:00
|
|
|
|
|
|
|
if len(animes) > animePerPage {
|
|
|
|
animes = animes[:animePerPage]
|
|
|
|
}
|
|
|
|
|
2018-07-02 17:57:44 +00:00
|
|
|
return ctx.HTML(components.Genre(genreName, animes, user, userScore))
|
2017-11-30 00:47:12 +00:00
|
|
|
}
|
2016-11-19 14:54:31 +00:00
|
|
|
|
2017-11-30 00:47:12 +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
|
|
|
}
|
|
|
|
|
2017-11-30 00:47:12 +00:00
|
|
|
return false
|
2016-11-19 14:54:31 +00:00
|
|
|
}
|
2018-07-02 17:57:44 +00:00
|
|
|
|
2018-07-03 10:13:38 +00:00
|
|
|
// Counts users average score in genre
|
|
|
|
func averageScore(user *arn.User, animes []*arn.Anime) float64 {
|
2018-07-02 17:57:44 +00:00
|
|
|
if user == nil {
|
2018-07-03 10:13:38 +00:00
|
|
|
return 0
|
2018-07-02 17:57:44 +00:00
|
|
|
}
|
|
|
|
|
2018-07-03 10:13:38 +00:00
|
|
|
count := 0.0
|
2018-07-02 17:57:44 +00:00
|
|
|
scores := 0.0
|
|
|
|
|
2018-07-03 10:13:38 +00:00
|
|
|
animeList := user.AnimeList()
|
2018-07-02 17:57:44 +00:00
|
|
|
|
2018-07-03 10:13:38 +00:00
|
|
|
for _, anime := range animes {
|
|
|
|
userAnime := animeList.Find(anime.ID)
|
|
|
|
if userAnime != nil && !userAnime.Rating.IsNotRated() {
|
|
|
|
scores += userAnime.Rating.Overall
|
|
|
|
count += 1
|
2018-07-02 17:57:44 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-03 10:13:38 +00:00
|
|
|
if count == 0.0 || scores == 0.0 {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
|
|
|
return scores / count
|
2018-07-02 17:57:44 +00:00
|
|
|
}
|