package compare

import (
	"net/http"
	"sort"

	"github.com/animenotifier/notify.moe/components"
	"github.com/animenotifier/notify.moe/utils"

	"github.com/aerogo/aero"
	"github.com/animenotifier/notify.moe/arn"
)

// AnimeList ...
func AnimeList(ctx aero.Context) error {
	user := arn.GetUserFromContext(ctx)
	nickA := ctx.Get("nick-1")
	nickB := ctx.Get("nick-2")

	a, err := arn.GetUserByNick(nickA)

	if err != nil || a == nil {
		return ctx.Error(http.StatusNotFound, "User not found: "+nickA, err)
	}

	b, err := arn.GetUserByNick(nickB)

	if err != nil || b == nil {
		return ctx.Error(http.StatusNotFound, "User not found: "+nickB, err)
	}

	comparisons := []*utils.Comparison{}
	countA := 0
	countB := 0

	animeListA := a.AnimeList()
	animeListB := b.AnimeList()

	for _, item := range animeListA.Items {
		if item.Status == arn.AnimeListStatusPlanned {
			continue
		}

		countA++

		comparisons = append(comparisons, &utils.Comparison{
			Anime: item.Anime(),
			ItemA: item,
			ItemB: animeListB.Find(item.AnimeID),
		})
	}

	for _, item := range animeListB.Items {
		if item.Status == arn.AnimeListStatusPlanned {
			continue
		}

		countB++

		if Contains(comparisons, item.AnimeID) {
			continue
		}

		comparisons = append(comparisons, &utils.Comparison{
			Anime: item.Anime(),
			ItemA: animeListA.Find(item.AnimeID),
			ItemB: item,
		})
	}

	sort.Slice(comparisons, func(i, j int) bool {
		aPopularity := comparisons[i].Anime.Popularity.Total()
		bPopularity := comparisons[j].Anime.Popularity.Total()

		if aPopularity == bPopularity {
			return comparisons[i].Anime.Title.Canonical < comparisons[j].Anime.Title.Canonical
		}

		return aPopularity > bPopularity
	})

	return ctx.HTML(components.CompareAnimeList(a, b, countA, countB, comparisons, user))
}

// Contains ...
func Contains(comparisons []*utils.Comparison, animeID string) bool {
	for _, comparison := range comparisons {
		if comparison.Anime.ID == animeID {
			return true
		}
	}

	return false
}