wms-be/modules/role/controller/role_controller.go

package controller

import (
	"net/http"

	"github.com/Caknoooo/go-gin-clean-starter/modules/role/dto"
	"github.com/Caknoooo/go-gin-clean-starter/modules/role/query"
	"github.com/Caknoooo/go-gin-clean-starter/modules/role/service"
	"github.com/Caknoooo/go-gin-clean-starter/pkg/constants"
	"github.com/Caknoooo/go-gin-clean-starter/pkg/utils"
	"github.com/Caknoooo/go-pagination"
	"github.com/gin-gonic/gin"
	"github.com/samber/do"
	"github.com/sirupsen/logrus"
	"gorm.io/gorm"
)

type (
	RoleController interface {
		CreateRole(ctx *gin.Context)
		GetRoles(ctx *gin.Context)
		GetRoleByID(ctx *gin.Context)
		UpdateRole(ctx *gin.Context)
		DeleteRole(ctx *gin.Context)
		AssignPermissionsToRole(ctx *gin.Context)
		RemovePermissionsFromRole(ctx *gin.Context)
		AssignRolesToUser(ctx *gin.Context)
		RemoveRolesFromUser(ctx *gin.Context)
		GetRolesByUserID(ctx *gin.Context)
	}

	roleController struct {
		roleService service.RoleService
		db          *gorm.DB
	}
)

// PingExample godoc
// @Summary ping example
// @Schemes
// @Description do ping
// @Tags example
// @Accept json
// @Produce json
// @Success 200 {string} Helloworld
// @Router /example/helloworld [get]
func (r *roleController) AssignPermissionsToRole(ctx *gin.Context) {
	var req dto.AssignPermissionRequest
	roleId := ctx.Param("id")
	if err := ctx.ShouldBind(&req); err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_DATA_FROM_BODY, err.Error(), nil)
		ctx.AbortWithStatusJSON(http.StatusBadRequest, res)
		return
	}
	if err := r.roleService.AssignPermissionsToRole(ctx.Request.Context(), roleId, req.PermissionID); err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_ASSIGN_PERMISSION, err.Error(), nil)
		ctx.JSON(http.StatusBadRequest, res)
		return
	}
	res := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_ASSIGN_PERMISSION, nil)
	ctx.JSON(http.StatusOK, res)
}

// AssignRoleToUser implements RoleController.
func (r *roleController) AssignRolesToUser(ctx *gin.Context) {
	var req dto.AssignRoleRequest
	userId := ctx.Param("id")
	if err := ctx.ShouldBind(&req); err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_DATA_FROM_BODY, err.Error(), nil)
		ctx.AbortWithStatusJSON(http.StatusBadRequest, res)
		return
	}

	if err := r.roleService.AssignRolesToUser(ctx.Request.Context(), userId, req.RoleID); err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_ASSIGN_ROLE, err.Error(), nil)
		ctx.JSON(http.StatusBadRequest, res)
		return
	}

	res := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_ASSIGN_ROLE, nil)
	ctx.JSON(http.StatusOK, res)
}

// CreateRole implements RoleController.
func (r *roleController) CreateRole(ctx *gin.Context) {
	var role dto.RoleCreateRequest
	if err := ctx.ShouldBind(&role); err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_DATA_FROM_BODY, err.Error(), nil)
		ctx.AbortWithStatusJSON(http.StatusBadRequest, res)
		return
	}

	result, err := r.roleService.CreateRole(ctx.Request.Context(), role)
	if err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_REGISTER_ROLE, err.Error(), nil)
		ctx.JSON(http.StatusBadRequest, res)
		return
	}

	res := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_REGISTER_ROLE, result)
	ctx.JSON(http.StatusOK, res)

}

// DeleteRole implements RoleController.
func (r *roleController) DeleteRole(ctx *gin.Context) {
	var roleID string = ctx.Param("id")
	err := r.roleService.DeleteRole(ctx.Request.Context(), roleID)
	if err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_DELETE_ROLE, err.Error(), nil)
		ctx.JSON(http.StatusBadRequest, res)
		return
	}
	res := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_DELETE_ROLE, nil)
	ctx.JSON(http.StatusOK, res)
}

// GetRoleByID implements RoleController.
func (r *roleController) GetRoleByID(ctx *gin.Context) {
	var roleID string = ctx.Param("id")
	result, err := r.roleService.GetRoleByID(ctx.Request.Context(), roleID)
	if err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_ROLE, err.Error(), nil)
		ctx.JSON(http.StatusBadRequest, res)
		return
	}

	res := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_GET_ROLE, result)
	ctx.JSON(http.StatusOK, res)
}

// GetRoles implements RoleController.
func (r *roleController) GetRoles(ctx *gin.Context) {
	clientId := ctx.MustGet("client_id").(string)
	// logrus.Info("Client ID: ", clientId)
	var filter = &query.RoleFilter{
		ClientID: clientId,
		Name:     ctx.Query("name"),
	}
	// logrus.Info("Filter: ", filter)
	filter.BindPagination(ctx)

	ctx.ShouldBindQuery(filter)

	roles, total, err := pagination.PaginatedQueryWithIncludable[query.M_Role](r.db, filter)
	if err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_ROLE, err.Error(), nil)
		ctx.JSON(http.StatusBadRequest, res)
		return
	}

	paginationResponse := pagination.CalculatePagination(filter.Pagination, total)
	response := pagination.NewPaginatedResponse(http.StatusOK, dto.MESSAGE_SUCCESS_GET_LIST_ROLE, roles, paginationResponse)
	ctx.JSON(http.StatusOK, response)
}

// GetRolesByUserID implements RoleController.
func (r *roleController) GetRolesByUserID(ctx *gin.Context) {
	userId := ctx.Param("id")
	logrus.Info("Fetching roles for User ID: ", userId)
	roles, err := r.roleService.GetRolesByUserID(ctx.Request.Context(), userId)
	if err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_ROLE, err.Error(), nil)
		ctx.JSON(http.StatusBadRequest, res)
		return
	}

	res := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_GET_ROLE, roles)
	ctx.JSON(http.StatusOK, res)
}

// RemovePermissionsFromRole implements RoleController.
func (r *roleController) RemovePermissionsFromRole(ctx *gin.Context) {
	var req dto.RemovePermissionRequest
	roleId := ctx.Param("id")
	if err := ctx.ShouldBind(&req); err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_DATA_FROM_BODY, err.Error(), nil)
		ctx.AbortWithStatusJSON(http.StatusBadRequest, res)
		return
	}
	if err := r.roleService.RemovePermissionsFromRole(ctx.Request.Context(), roleId, req.PermissionID); err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_REMOVE_ROLE, err.Error(), nil)
		ctx.JSON(http.StatusBadRequest, res)
		return
	}
	res := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_REMOVE_ROLE, nil)
	ctx.JSON(http.StatusOK, res)
}

// RemoveRoleFromUser implements RoleController.
func (r *roleController) RemoveRolesFromUser(ctx *gin.Context) {
	var req dto.RemoveRoleRequest
	userId := ctx.Param("id")
	if err := ctx.ShouldBind(&req); err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_DATA_FROM_BODY, err.Error(), nil)
		ctx.AbortWithStatusJSON(http.StatusBadRequest, res)
		return
	}
	if err := r.roleService.RemoveRolesFromUser(ctx.Request.Context(), userId, req.RoleID); err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_REMOVE_ROLE, err.Error(), nil)
		ctx.JSON(http.StatusBadRequest, res)
		return
	}
	res := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_REMOVE_ROLE, nil)
	ctx.JSON(http.StatusOK, res)
}

// UpdateRole implements RoleController.
func (r *roleController) UpdateRole(ctx *gin.Context) {
	var req dto.RoleUpdateRequest
	if err := ctx.ShouldBind(&req); err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_DATA_FROM_BODY, err.Error(), nil)
		ctx.AbortWithStatusJSON(http.StatusBadRequest, res)
		return
	}

	roleId := ctx.Param("id")
	result, err := r.roleService.UpdateRole(ctx.Request.Context(), roleId, req)
	if err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_UPDATE_ROLE, err.Error(), nil)
		ctx.JSON(http.StatusBadRequest, res)
		return
	}

	res := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_UPDATE_ROLE, result)
	ctx.JSON(http.StatusOK, res)

}

func NewRoleController(injector *do.Injector, rs service.RoleService) RoleController {
	db := do.MustInvokeNamed[*gorm.DB](injector, constants.DB)
	return &roleController{
		roleService: rs,
		db:          db,
	}
}