mirror of https://github.com/portainer/portainer
60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
|
package middlewares
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/gorilla/mux"
|
||
|
|
httperror "github.com/portainer/libhttp/error"
|
||
|
|
"github.com/portainer/libhttp/request"
|
||
|
|
bolterrors "github.com/portainer/portainer/api/dataservices/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ItemContextKey string
|
||
|
|
|
||
|
|
type ItemGetter[TId ~int, TObject any] func(id TId) (*TObject, error)
|
||
|
|
|
||
|
|
func WithItem[TId ~int, TObject any](getter ItemGetter[TId, TObject], idParam string, contextKey ItemContextKey) mux.MiddlewareFunc {
|
||
|
|
if idParam == "" {
|
||
|
|
idParam = "id"
|
||
|
|
}
|
||
|
|
|
||
|
|
return func(next http.Handler) http.Handler {
|
||
|
|
return http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
itemId, err := request.RetrieveNumericRouteVariableValue(req, idParam)
|
||
|
|
if err != nil {
|
||
|
|
httperror.WriteError(rw, http.StatusBadRequest, "Invalid identifier route variable", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
item, err := getter(TId(itemId))
|
||
|
|
if err != nil {
|
||
|
|
statusCode := http.StatusInternalServerError
|
||
|
|
if err == bolterrors.ErrObjectNotFound {
|
||
|
|
statusCode = http.StatusNotFound
|
||
|
|
}
|
||
|
|
httperror.WriteError(rw, statusCode, "Unable to find a object with the specified identifier inside the database", err)
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|
||
|
|
ctx := context.WithValue(req.Context(), contextKey, item)
|
||
|
|
next.ServeHTTP(rw, req.WithContext(ctx))
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func FetchItem[T any](request *http.Request, contextKey string) (*T, error) {
|
||
|
|
contextData := request.Context().Value(contextKey)
|
||
|
|
if contextData == nil {
|
||
|
|
return nil, errors.New("unable to find item in request context")
|
||
|
|
}
|
||
|
|
|
||
|
|
item, ok := contextData.(*T)
|
||
|
|
if !ok {
|
||
|
|
return nil, errors.New("unable to cast context item")
|
||
|
|
}
|
||
|
|
|
||
|
|
return item, nil
|
||
|
|
}
|