wms-be/modules/assignment/controller/assignment_controller.go

package controller

import (
	"net/http"

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

type AssignmentController interface {
	Create(ctx *gin.Context)
	Update(ctx *gin.Context)
	Delete(ctx *gin.Context)
	GetById(ctx *gin.Context)
	GetAll(ctx *gin.Context)
}

type assignmentController struct {
	assignmentService service.AssignmentService
	db                *gorm.DB
}

func (c *assignmentController) Create(ctx *gin.Context) {
	var req dto.AssignmentCreateRequest
	if err := ctx.ShouldBindJSON(&req); err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_DATA_FROM_BODY, err.Error(), nil)
		ctx.JSON(http.StatusBadRequest, res)
		return
	}
	created, err := c.assignmentService.Create(ctx, req)
	if err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_CREATE_ASSIGNMENT, err.Error(), nil)
		ctx.JSON(http.StatusInternalServerError, res)
		return
	}
	res := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_CREATE_ASSIGNMENT, created)
	ctx.JSON(http.StatusOK, res)
}

func (c *assignmentController) Update(ctx *gin.Context) {
	id := ctx.Param("id")
	var req dto.AssignmentUpdateRequest
	if err := ctx.ShouldBindJSON(&req); err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_DATA_FROM_BODY, err.Error(), nil)
		ctx.JSON(http.StatusBadRequest, res)
		return
	}
	updated, err := c.assignmentService.Update(ctx, req, id)
	if err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_UPDATE_ASSIGNMENT, err.Error(), nil)
		ctx.JSON(http.StatusInternalServerError, res)
		return
	}
	res := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_UPDATE_ASSIGNMENT, updated)
	ctx.JSON(http.StatusOK, res)
}

func (c *assignmentController) Delete(ctx *gin.Context) {
	id := ctx.Param("id")
	if err := c.assignmentService.Delete(ctx, id); err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_DELETE_ASSIGNMENT, err.Error(), nil)
		ctx.JSON(http.StatusInternalServerError, res)
		return
	}
	res := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_DELETE_ASSIGNMENT, nil)
	ctx.JSON(http.StatusOK, res)
}

func (c *assignmentController) GetById(ctx *gin.Context) {
	id := ctx.Param("id")
	assignment, err := c.assignmentService.GetById(ctx, id)
	if err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_ASSIGNMENT, err.Error(), nil)
		ctx.JSON(http.StatusNotFound, res)
		return
	}
	res := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_GET_ASSIGNMENT, assignment)
	ctx.JSON(http.StatusOK, res)
}

func (c *assignmentController) GetAll(ctx *gin.Context) {
	clientId := ctx.DefaultQuery("client_id", "")
	var filter query.AssignmentFilter
	filter.ClientID = clientId
	if err := ctx.ShouldBindQuery(&filter); err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_ASSIGNMENT, err.Error(), nil)
		ctx.JSON(http.StatusBadRequest, res)
		return
	}
	getAll := ctx.Query("get_all")
	if getAll != "" {
		assignments, _, err := c.assignmentService.GetAll(ctx, filter)
		if err != nil {
			res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_ASSIGNMENT, err.Error(), nil)
			ctx.JSON(http.StatusBadRequest, res)
			return
		}
		response := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_GET_ASSIGNMENT, assignments)
		ctx.JSON(http.StatusOK, response)
		return
	}
	perPage := utils.ParseInt(ctx.DefaultQuery("per_page", "10"))
	page := utils.ParseInt(ctx.DefaultQuery("page", "1"))
	filter.PerPage = perPage
	filter.Page = (page - 1) * perPage
	assignments, total, err := c.assignmentService.GetAll(ctx, filter)
	if err != nil {
		res := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_ASSIGNMENT, err.Error(), nil)
		ctx.JSON(http.StatusInternalServerError, res)
		return
	}
	paginationResponse := utils.BuildPaginationResponse(perPage, page, total)
	res := utils.BuildResponseSuccessWithPagination(http.StatusOK, dto.MESSAGE_SUCCESS_GET_ASSIGNMENT, assignments, paginationResponse)
	ctx.JSON(http.StatusOK, res)
}

func NewAssignmentController(i *do.Injector, assignmentService service.AssignmentService) AssignmentController {
	db := do.MustInvokeNamed[*gorm.DB](i, constants.DB)
	return &assignmentController{
		assignmentService: assignmentService,
		db:                db,
	}
}