neko-custom/server/internal/http/endpoint/endpoint.go

package endpoint

import (
	"encoding/json"
	"fmt"
	"net/http"
	"runtime/debug"

	"github.com/go-chi/chi/middleware"
	"github.com/rs/zerolog/log"
)

type (
	Endpoint func(http.ResponseWriter, *http.Request) error

	ErrResponse struct {
		Status    int    `json:"status,omitempty"`
		Err       string `json:"error,omitempty"`
		Message   string `json:"message,omitempty"`
		Details   string `json:"details,omitempty"`
		Code      string `json:"code,omitempty"`
		RequestID string `json:"request,omitempty"`
	}
)

func Handle(handler Endpoint) http.HandlerFunc {
	fn := func(w http.ResponseWriter, r *http.Request) {
		if err := handler(w, r); err != nil {
			WriteError(w, r, err)
		}
	}

	return http.HandlerFunc(fn)
}

var nonErrorsCodes = map[int]bool{
	404: true,
}

func errResponse(input interface{}) *ErrResponse {
	var res *ErrResponse
	var err interface{}

	switch input.(type) {
	case *HandlerError:
		e := input.(*HandlerError)
		res = &ErrResponse{
			Status:  e.Status,
			Err:     http.StatusText(e.Status),
			Message: e.Message,
		}
		err = e.Err
	default:
		res = &ErrResponse{
			Status: http.StatusInternalServerError,
			Err:    http.StatusText(http.StatusInternalServerError),
		}
		err = input
	}

	if err != nil {
		switch err.(type) {
		case *error:
			e := err.(error)
			res.Details = e.Error()
			break
		default:
			res.Details = fmt.Sprintf("%+v", err)
			break
		}
	}

	return res
}

func WriteError(w http.ResponseWriter, r *http.Request, err interface{}) {
	hlog := log.With().
		Str("module", "http").
		Logger()

	res := errResponse(err)

	if reqID := middleware.GetReqID(r.Context()); reqID != "" {
		res.RequestID = reqID
	}

	w.Header().Set("Content-Type", "application/json")
	w.WriteHeader(res.Status)

	if err := json.NewEncoder(w).Encode(res); err != nil {
		hlog.Warn().Err(err).Msg("Failed writing json error response")
	}

	if !nonErrorsCodes[res.Status] {
		logEntry := middleware.GetLogEntry(r)
		if logEntry != nil {
			logEntry.Panic(err, debug.Stack())
		} else {
			hlog.Error().Str("stack", string(debug.Stack())).Msgf("%+v", err)
		}
	}
}