75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
|
package server
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"net/http"
|
||
|
|
||
|
"github.com/go-chi/render"
|
||
|
"gitlab.com/1f320/pronouns/backend/log"
|
||
|
)
|
||
|
|
||
|
// WrapHandler wraps a modified http.HandlerFunc into a stdlib-compatible one.
|
||
|
// The inner HandlerFunc additionally returns an error.
|
||
|
func WrapHandler(hn func(w http.ResponseWriter, r *http.Request) error) http.HandlerFunc {
|
||
|
return func(w http.ResponseWriter, r *http.Request) {
|
||
|
err := hn(w, r)
|
||
|
if err != nil {
|
||
|
// if the function returned an API error, just render that verbatim
|
||
|
// we can assume that it also logged the error (if that was needed)
|
||
|
if apiErr, ok := err.(APIError); ok {
|
||
|
apiErr.prepare()
|
||
|
|
||
|
render.Status(r, apiErr.Status)
|
||
|
render.JSON(w, r, apiErr)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// otherwise, we log the error and return an internal server error message
|
||
|
log.Errorf("error in http handler: %v", err)
|
||
|
|
||
|
apiErr := APIError{Code: ErrInternalServerError}
|
||
|
apiErr.prepare()
|
||
|
|
||
|
render.Status(r, apiErr.Status)
|
||
|
render.JSON(w, r, apiErr)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// APIError is an object returned by the API when an error occurs.
|
||
|
// It implements the error interface and can be returned by handlers.
|
||
|
type APIError struct {
|
||
|
Code int `json:"code"`
|
||
|
Message string `json:"message,omitempty"`
|
||
|
|
||
|
// Status is
|
||
|
Status int `json:"-"`
|
||
|
}
|
||
|
|
||
|
func (e APIError) Error() string {
|
||
|
return fmt.Sprintf("%s (code: %d)", e.Message, e.Code)
|
||
|
}
|
||
|
|
||
|
func (e *APIError) prepare() {
|
||
|
if e.Status == 0 {
|
||
|
e.Status = errCodeStatuses[e.Code]
|
||
|
}
|
||
|
|
||
|
if e.Message == "" {
|
||
|
e.Message = errCodeMessages[e.Code]
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Error code constants
|
||
|
const (
|
||
|
ErrInternalServerError = 500 // catch-all code for unknown errors
|
||
|
)
|
||
|
|
||
|
var errCodeMessages = map[int]string{
|
||
|
ErrInternalServerError: "Internal server error",
|
||
|
}
|
||
|
|
||
|
var errCodeStatuses = map[int]int{
|
||
|
ErrInternalServerError: http.StatusInternalServerError,
|
||
|
}
|