71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
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
|
|
}
|