userHandler.go

package handlers

import (
	"github.com/Arjuna-Ragil/Localbase/Internal/core/services"
	"github.com/gin-gonic/gin"
)

type UserHandler struct {
	Service *services.UserService
}

func NewUserHandler(s *services.UserService) *UserHandler {
	return &UserHandler{Service: s}
}

func (uh *UserHandler) GetUserHandler(c *gin.Context) {
	userID, exist := c.Get("userID")
	if !exist {
		c.JSON(400, gin.H{
			"message": "Unable to find user",
			"data":    exist,
		})
		return
	}
	user, err := uh.Service.GetUser(userID.(uint))
	if err != nil {
		c.JSON(400, gin.H{
			"message": "Unable to get user data",
			"data":    err,
		})
		return
	}
	c.JSON(200, gin.H{
		"message": "Successfully get user data",
		"data":    user,
	})
}

func (uh *UserHandler) AllUserHandler(c *gin.Context) {
	userRole, _ := c.Get("userRole")
	if userRole != "admin" {
		c.JSON(400, gin.H{
			"message": "Not allowed to get all users",
			"data":    false,
		})
		return
	}
	users, err := uh.Service.GetAllUser()
	if err != nil {
		c.JSON(400, gin.H{
			"message": "Unable to get all users",
			"data":    false,
		})
	}
	c.JSON(200, gin.H{
		"message": "Successfully get all users",
		"data":    users,
	})
}

func (uh *UserHandler) UpdateRoleHandler(c *gin.Context) {
	var input services.RoleInput
	//userRole, _ := c.Get("userRole")
	//if userRole != "admin" {
	//	c.JSON(400, gin.H{
	//		"message": "Not allowed to update role",
	//		"data":    false,
	//	})
	//	return
	//}
	if err := c.ShouldBindJSON(&input); err != nil {
		c.JSON(400, gin.H{
			"message": "input not valid",
			"data":    err.Error(),
		})
		return
	}
	user, err := uh.Service.UpdateRole(&input)
	if err != nil {
		c.JSON(400, gin.H{
			"message": "Unable to update role",
			"data":    err,
		})
		return
	}
	c.JSON(200, gin.H{
		"message": "Successfully update role",
		"data":    user,
	})
}