package main

import (
	"errors"
	"github.com/gin-gonic/gin"
	"net/http"
	"strconv"
)

type ViewModel struct {
	Users       []User
	CurrentUser int
	Allowances  []Allowance
	Tasks       []Task
	History     []History
}

func renderLite(c *gin.Context) {
	if c.Query("user") != "" {
		c.SetCookie("user", c.Query("user"), 3600, "/", "localhost", false, true)
		c.Redirect(http.StatusFound, "/")
		return
	}

	currentUserStr, err := c.Cookie("user")
	if errors.Is(err, http.ErrNoCookie) {
		renderNoUser(c)
		return
	}

	if err != nil {
		unsetUserCookie(c)
		return
	}
	currentUser, err := strconv.Atoi(currentUserStr)
	if err != nil {
		unsetUserCookie(c)
		return
	}
	userExists, err := db.UserExists(currentUser)
	if !userExists || err != nil {
		unsetUserCookie(c)
		return
	}
	renderWithUser(c, currentUser)
}

func unsetUserCookie(c *gin.Context) {
	c.SetCookie("user", "", -1, "/", "localhost", false, true)
	c.Redirect(http.StatusFound, "/")
}

func renderNoUser(c *gin.Context) {
	users, err := db.GetUsers()
	if err != nil {
		c.HTML(http.StatusInternalServerError, "error.gohtml", gin.H{
			"error": err.Error(),
		})
		return
	}

	c.HTML(http.StatusOK, "lite.gohtml", ViewModel{
		Users: users,
	})
}

func renderWithUser(c *gin.Context, currentUser int) {
	users, err := db.GetUsers()
	if err != nil {
		c.HTML(http.StatusInternalServerError, "error.gohtml", gin.H{
			"error": err.Error(),
		})
		return
	}

	allowances, err := db.GetUserAllowances(currentUser)
	if err != nil {
		c.HTML(http.StatusInternalServerError, "error.gohtml", gin.H{
			"error": err.Error(),
		})
		return
	}

	tasks, err := db.GetTasks()
	if err != nil {
		c.HTML(http.StatusInternalServerError, "error.gohtml", gin.H{
			"error": err.Error(),
		})
		return
	}

	history, err := db.GetHistory(currentUser)
	if err != nil {
		c.HTML(http.StatusInternalServerError, "error.gohtml", gin.H{
			"error": err.Error(),
		})
		return
	}

	c.HTML(http.StatusOK, "lite.gohtml", ViewModel{
		Users:       users,
		CurrentUser: currentUser,
		Allowances:  allowances,
		Tasks:       tasks,
		History:     history,
	})
}