package controllers

import (
	"git.preston-baxter.com/Preston_PLB/capstone/frontend-service/db/models"
	"git.preston-baxter.com/Preston_PLB/capstone/frontend-service/templates"
	"github.com/gin-gonic/gin"
)

func LandingPage(c *gin.Context) {
	if raw, exists := c.Get(USER_OBJ_KEY); exists {
		if user, ok := raw.(*models.User); ok {
			renderTempl(c, templates.LandingPage(user))
			return
		}
	}
	renderTempl(c, templates.LandingPage(nil))
}

func LoginPage(c *gin.Context) {
	renderTempl(c, templates.LoginPage(""))
}

func SignUpPage(c *gin.Context) {
	renderTempl(c, templates.SignupPage(""))
}

func DashboardPage(c *gin.Context) {
	user := getUserFromContext(c)

	if user == nil {
		log.Error("No user found in context")
		c.AbortWithStatus(502)
		return
	}

	vendors, err := mongo.FindVendorAccountByUser(user.MongoId())
	if err != nil {
		log.WithError(err).Error("Failed to lookup vendor accounts for user")
		c.AbortWithStatus(502)
		return
	}

	renderTempl(c, templates.DashboardPage(user, vendors))
}