error_handler.go

package middlewares

import (
	"errors"
	"net/http"

	"github.com/go-chi/render"

	serviceErrors "github.com/tink3rlabs/magic/errors"
	"github.com/tink3rlabs/magic/storage"
	"github.com/tink3rlabs/magic/types"
)

type ErrorHandler struct{}

func (e *ErrorHandler) Wrap(handler func(w http.ResponseWriter, r *http.Request) error) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		var notFoundError *serviceErrors.NotFound
		var badRequestError *serviceErrors.BadRequest
		var serviceUnavailable *serviceErrors.ServiceUnavailable
		var forbiddenError *serviceErrors.Forbidden
		var unauthorizedError *serviceErrors.Unauthorized
		var methodNotAllowedError *serviceErrors.MethodNotAllowed
		var conflictError *serviceErrors.Conflict
		var goneError *serviceErrors.Gone
		var unsupportedMediaTypeError *serviceErrors.UnsupportedMediaType
		var unprocessableEntityError *serviceErrors.UnprocessableEntity
		var tooManyRequestsError *serviceErrors.TooManyRequests
		var internalServerError *serviceErrors.InternalServerError
		var badGatewayError *serviceErrors.BadGateway
		var gatewayTimeoutError *serviceErrors.GatewayTimeout
		var requestTimeoutError *serviceErrors.RequestTimeout
		var notImplementedError *serviceErrors.NotImplemented

		err := handler(w, r)
		if err == nil {
			return
		}

		status := http.StatusInternalServerError
		message := err.Error()

		switch {
		case errors.As(err, &notFoundError) || errors.Is(err, storage.ErrNotFound):
			status = http.StatusNotFound
		case errors.As(err, &badRequestError):
			status = http.StatusBadRequest
		case errors.As(err, &serviceUnavailable):
			status = http.StatusServiceUnavailable
		case errors.As(err, &forbiddenError):
			status = http.StatusForbidden
		case errors.As(err, &unauthorizedError):
			status = http.StatusUnauthorized
		case errors.As(err, &methodNotAllowedError):
			status = http.StatusMethodNotAllowed
		case errors.As(err, &conflictError):
			status = http.StatusConflict
		case errors.As(err, &goneError):
			status = http.StatusGone
		case errors.As(err, &unsupportedMediaTypeError):
			status = http.StatusUnsupportedMediaType
		case errors.As(err, &unprocessableEntityError):
			status = http.StatusUnprocessableEntity
		case errors.As(err, &tooManyRequestsError):
			status = http.StatusTooManyRequests
		case errors.As(err, &internalServerError):
			status = http.StatusInternalServerError
		case errors.As(err, &badGatewayError):
			status = http.StatusBadGateway
		case errors.As(err, &gatewayTimeoutError):
			status = http.StatusGatewayTimeout
		case errors.As(err, &requestTimeoutError):
			status = http.StatusRequestTimeout
		case errors.As(err, &notImplementedError):
			status = http.StatusNotImplemented
		default:
			message = "encountered an unexpected server error: " + err.Error()
		}

		render.Status(r, status)
		render.JSON(w, r, types.ErrorResponse{
			Status: http.StatusText(status),
			Error:  message,
		})
	}
}