package admin

import (
	"net/http"
	"sort"

	"github.com/aerogo/aero"
	"github.com/animenotifier/arn"
	"github.com/animenotifier/notify.moe/components"
	"github.com/animenotifier/notify.moe/utils"
	"github.com/fatih/color"
)

// UserRegistrations ...
func UserRegistrations(ctx *aero.Context) string {
	user := utils.GetUser(ctx)

	if user == nil {
		return ctx.Error(http.StatusUnauthorized, "Not logged in", nil)
	}

	if user.Role != "admin" {
		return ctx.Error(http.StatusUnauthorized, "Not authorized", nil)
	}

	yearInfo := map[int]utils.YearRegistrations{}
	years := []int{}

	for user := range arn.StreamUsers() {
		if user.Registered == "" {
			color.Red("%s %s", user.ID, user.Nick)
			user.Registered = user.LastLogin
			user.Save()
		}

		registered := user.RegisteredTime()
		year := registered.Year()
		yearRegistrations := yearInfo[year]
		yearRegistrations.Total++

		if yearRegistrations.Months == nil {
			yearRegistrations.Months = map[int]int{}
		}

		yearRegistrations.Months[int(registered.Month())]++
		yearInfo[year] = yearRegistrations
	}

	for year := range yearInfo {
		years = append(years, year)
	}

	sort.Ints(years)

	return ctx.HTML(components.UserRegistrations(years, yearInfo))
}