goldgorilla/controllers/helpers.go

71 lines
1.7 KiB
Go
Raw Normal View History

package controllers
import (
"github.com/gin-gonic/gin"
"sourcecode.social/greatape/goldgorilla/models"
)
type ResponseHelper struct {
}
func NewResponseHelper() *ResponseHelper {
return &ResponseHelper{}
}
func (r *ResponseHelper) Response(ctx *gin.Context, body any, code int) {
if body == nil {
body = struct{}{}
}
ctx.JSON(code, body)
}
func (r *ResponseHelper) ResponseError(ctx *gin.Context, errObj any, code *int) {
errCode := 500
if code != nil {
errCode = *code
} else if bigErr, isBigErr := errObj.(models.BigError); isBigErr {
errCode = bigErr.ErrCode()
}
var errResp models.MessageResponse
if err, isError := errObj.(error); isError {
errResp.Message = err.Error()
} else if err, isMessage := errObj.(models.MessageResponse); isMessage {
errResp = err
} else {
errResp.Message = "unhandled error, contact support"
}
ctx.JSON(errCode, errResp)
}
func (r *ResponseHelper) ResponseBadReq(ctx *gin.Context) {
st := 400
r.ResponseError(ctx, models.MessageResponse{Message: "bad request, invalid input"}, &st)
}
func (r *ResponseHelper) ResponseUnprocessableEntity(ctx *gin.Context) {
st := 422
r.ResponseError(ctx, models.MessageResponse{Message: "unprocessable entity, invalid input"}, &st)
}
func (r *ResponseHelper) HandleIfErr(ctx *gin.Context, err error, status *int) bool {
if err == nil {
return false
}
var statusCode = 500
if bigErr, extends := err.(models.BigError); extends {
if status != nil {
statusCode = *status
} else {
statusCode = bigErr.ErrCode()
}
r.ResponseError(ctx, models.MessageResponse{Message: bigErr.Error()}, &statusCode)
} else {
if status != nil {
statusCode = *status
}
r.ResponseError(ctx, models.MessageResponse{Message: err.Error()}, &statusCode)
}
return true
}