package dashboard

import (
	"github.com/aerogo/aero"
	"github.com/aerogo/flow"
	"github.com/animenotifier/arn"
	"github.com/animenotifier/notify.moe/components"
	"github.com/animenotifier/notify.moe/pages/frontpage"
	"github.com/animenotifier/notify.moe/utils"
)

const maxPosts = 5
const maxFollowing = 5

// Get the dashboard or the frontpage when logged out.
func Get(ctx *aero.Context) string {
	user := utils.GetUser(ctx)

	if user == nil {
		return frontpage.Get(ctx)
	}

	return dashboard(ctx)
}

// Render the dashboard.
func dashboard(ctx *aero.Context) string {
	var posts []*arn.Post
	var err error
	var followIDList []string
	var userList interface{}
	var followingList []*arn.User

	user := utils.GetUser(ctx)

	flow.Parallel(func() {
		posts, err = arn.AllPostsSlice()
		arn.SortPostsLatestFirst(posts)

		if len(posts) > maxPosts {
			posts = posts[:maxPosts]
		}
	}, func() {
		followIDList = user.Following
		userList, err = arn.DB.GetMany("User", followIDList)
		followingList = userList.([]*arn.User)
		followingList = arn.SortUsersLastSeen(followingList)

		if len(followingList) > maxFollowing {
			followingList = followingList[:maxFollowing]
		}
	})

	if err != nil {
		return ctx.Error(500, "Error displaying dashboard", err)
	}

	return ctx.HTML(components.Dashboard(posts, followingList))
}