pages.go

package main

import (
	"log"
	"net/http"

	"github.com/gin-gonic/gin"
)

const (
	IsUserLinked     = `SELECT EXISTS(SELECT 1 FROM users WHERE email = $1)`
	GetWiiNumberUser = `SELECT wii_number FROM users WHERE email = $1`
	GetLinkedDominos = `SELECT dominos_linked FROM users WHERE email = $1`
)

func HomePage(c *gin.Context) {
	if username, ok := c.Get("username"); ok {
		if email, ok := c.Get("email"); ok {
			var exists bool
			err := pool.QueryRow(ctx, IsUserLinked, email.(string)).Scan(&exists)

			if err != nil {
				c.HTML(http.StatusBadRequest, "error.html", gin.H{
					"Error": err.Error(),
				})
				return
			}

			if exists {
				log.Printf("User with username %s is linked!!!", username)
				var wiiNumber string
				var linked_dominos bool
				err := pool.QueryRow(ctx, GetWiiNumberUser, email.(string)).Scan(&wiiNumber)
				if err != nil {
					c.HTML(http.StatusBadRequest, "error.html", gin.H{
						"Error": err.Error(),
					})
					return
				}
				err = pool.QueryRow(ctx, GetLinkedDominos, email.(string)).Scan(&linked_dominos)
				if err != nil {
					c.HTML(http.StatusBadRequest, "error.html", gin.H{
						"Error": err.Error(),
					})
					return
				}

				if pfp, ok := c.Get("picture"); ok {
					c.HTML(http.StatusOK, "linked.html", gin.H{
						"username":       username,
						"email":          email,
						"pfp":            pfp,
						"wiinumber":      wiiNumber,
						"linked_dominos": linked_dominos,
					})
				} else {
					c.HTML(http.StatusOK, "linked.html", gin.H{
						"username":       username,
						"email":          email,
						"wiinumber":      wiiNumber,
						"linked_dominos": linked_dominos,
					})
				}
			} else {
				log.Printf("User with username %s is not linked", username)
				if pfp, ok := c.Get("picture"); ok {
					c.HTML(http.StatusOK, "not_linked.html", gin.H{
						"username": username,
						"pfp":      pfp,
					})
				} else {
					c.HTML(http.StatusOK, "not_linked.html", gin.H{
						"username": username,
					})
				}
			}

			return
		}
	}

	c.HTML(http.StatusBadRequest, "error.html", gin.H{
		"Error": "Username not found in context",
	})
}

func NotLinkedPage(c *gin.Context) {
	email, _ := c.Get("email")
	log.Printf("User with email %s is not linked", email)
	c.HTML(http.StatusOK, "not_linked.html", gin.H{
		"email": email,
	})
}