package handler

import (
	"astra-users/model"
	"astra-users/pkg/utils"
	"crypto/sha256"
	"fmt"
	"github.com/labstack/echo/v4"
	"net/http"
)

func (h *Handler) GetCurrentUser(c echo.Context) error {
	userId := c.Request().Header.Get("UserId")

	if userId == "" {
		return c.NoContent(http.StatusBadRequest)
	}

	u := new(model.User)

	tx := h.DB.First(&u, userId)

	if tx.RowsAffected == 0 {
		return c.NoContent(http.StatusNotFound)
	}

	u.Password = ""

	return c.JSON(http.StatusOK, &u)
}

func (h *Handler) GetUsers(c echo.Context) error {
	var users []model.User

	query := c.Request().URL.Query()
	limit, offset := utils.GetPaginationParams(query)
	sortCol, isDesc, filterCol, filterVal := utils.GetFilterParams(query)

	tx := h.DB.Limit(limit).Offset(offset)

	utils.ApplyIdFilter(c, tx)
	utils.ApplySort(tx, sortCol, isDesc)
	utils.ApplyFilters(tx, filterCol, filterVal)

	tx = tx.Find(&users)

	if tx.RowsAffected == 1 {
		return c.JSON(http.StatusOK, users[0])
	}

	return c.JSON(http.StatusOK, users)
}

func (h *Handler) UpdateUser(c echo.Context) (err error) {
	var user = new(model.User)
	var apiUser = new(model.APIUser)
	tx := h.DB

	id := c.Param("id")
	tx = tx.Find(&user, id)

	if tx.RowsAffected == 0 {
		return c.NoContent(http.StatusNotFound)
	}

	if err = c.Bind(apiUser); err != nil {
		return c.NoContent(http.StatusBadRequest)
	}

	user.FirstName = apiUser.FirstName
	user.LastName = apiUser.LastName

	tx = h.DB.Save(&user)

	if tx.RowsAffected == 0 {
		return c.NoContent(http.StatusBadGateway)
	}

	user.Password = ""

	return c.JSON(http.StatusOK, &user)
}

func (h *Handler) DeleteUser(c echo.Context) (err error) {
	id := c.Param("id")

	tx := h.DB.Delete(&model.User{}, id)

	if tx.RowsAffected == 0 {
		return c.NoContent(http.StatusBadGateway)
	}

	return c.NoContent(http.StatusOK)
}

func (h *Handler) CreateUser(c echo.Context) (err error) {
	var user = new(model.User)

	if err = c.Bind(user); err != nil {
		return c.NoContent(http.StatusBadRequest)
	}

	user.Password = fmt.Sprintf("%x", sha256.Sum256([]byte(user.Password)))

	tx := h.DB.Create(user)

	if tx.RowsAffected == 0 {
		return c.NoContent(http.StatusBadGateway)
	}

	user.Password = ""

	return c.JSON(http.StatusCreated, user)
}