package handler

import (
	"github.com/ditatompel/xmr-nodes/internal/database"
	"github.com/ditatompel/xmr-nodes/internal/repo"

	"github.com/gofiber/fiber/v2"
)

func CookieProtected(c *fiber.Ctx) error {
	cookie := c.Cookies("xmr-nodes-ui")
	if cookie == "" {
		return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{
			"status":  "error",
			"message": "Unauthorized",
			"data":    nil,
		})
	}

	return c.Next()
}

func CheckProber(c *fiber.Ctx) error {
	key := c.Get("X-Prober-Api-Key")
	if key == "" {
		return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{
			"status":  "error",
			"message": "Unauthorized",
			"data":    nil,
		})
	}

	proberRepo := repo.NewProberRepo(database.GetDB())

	prober, err := proberRepo.CheckApi(key)
	if err != nil {
		return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{
			"status":  "error",
			"message": "No API key match",
			"data":    nil,
		})
	}

	c.Locals("prober", prober)
	return c.Next()
}