Added statistics

pages/profile/stats.go

@@ -0,0 +1,64 @@
+package profile
+
+import (
+	"net/http"
+	"strconv"
+	"time"
+
+	"github.com/aerogo/aero"
+	"github.com/animenotifier/arn"
+	"github.com/animenotifier/notify.moe/components"
+	"github.com/animenotifier/notify.moe/utils"
+)
+
+type stats map[string]float64
+
+// GetStatsByUser shows statistics for a given user.
+func GetStatsByUser(ctx *aero.Context) string {
+	nick := ctx.Get("nick")
+	viewUser, err := arn.GetUserByNick(nick)
+	userStats := utils.UserStats{}
+	ratings := stats{}
+	status := stats{}
+	types := stats{}
+	years := stats{}
+
+	if err != nil {
+		return ctx.Error(http.StatusNotFound, "User not found", err)
+	}
+
+	animeList, err := arn.GetAnimeList(viewUser)
+	animeList.PrefetchAnime()
+
+	if err != nil {
+		return ctx.Error(http.StatusInternalServerError, "Anime list not found", err)
+	}
+
+	for _, item := range animeList.Items {
+		duration := time.Duration(item.Episodes * item.Anime().EpisodeLength)
+		userStats.AnimeWatchingTime += duration * time.Minute
+
+		ratings[strconv.Itoa(int(item.Rating.Overall+0.5))]++
+		status[item.Status]++
+		types[item.Anime().Type]++
+
+		if item.Anime().StartDate != "" {
+			year := item.Anime().StartDate[:4]
+
+			if year < "2000" {
+				year = "Before 2000"
+			}
+
+			years[year]++
+		}
+	}
+
+	userStats.PieCharts = []*arn.PieChart{
+		arn.NewPieChart("Ratings", ratings),
+		arn.NewPieChart("Status", status),
+		arn.NewPieChart("Types", types),
+		arn.NewPieChart("Years", years),
+	}
+
+	return ctx.HTML(components.ProfileStats(&userStats, viewUser, utils.GetUser(ctx), ctx.URI()))
+}