Spaces:
Running
Running
File size: 2,791 Bytes
b110593 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
// _ _
// __ _____ __ ___ ___ __ _| |_ ___
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
// \ V V / __/ (_| |\ V /| | (_| | || __/
// \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
//
// Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
//
// CONTACT: [email protected]
//
package extensions
import (
"context"
"io"
"net/http"
"github.com/weaviate/weaviate/entities/models"
)
type RESTUserFacingHandlers struct {
proxy Proxy
}
func newRESTUserFacingHandlers(proxy Proxy) *RESTUserFacingHandlers {
return &RESTUserFacingHandlers{
proxy: proxy,
}
}
func (h *RESTHandlers) UserFacingHandler() http.Handler {
return newRESTUserFacingHandlers(h.proxy).Handler()
}
func (h *RESTUserFacingHandlers) Handler() http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case http.MethodPost:
h.post(w, r)
default:
w.WriteHeader(http.StatusMethodNotAllowed)
}
})
}
func (h *RESTUserFacingHandlers) post(w http.ResponseWriter, r *http.Request) {
ct := r.Header.Get("content-type")
if ct != "application/json" {
w.WriteHeader(http.StatusUnsupportedMediaType)
return
}
defer r.Body.Close()
body, err := io.ReadAll(r.Body)
if err != nil {
h.writeError(w, err, http.StatusInternalServerError)
return
}
var ext models.C11yExtension
if err := (&ext).UnmarshalBinary(body); err != nil {
h.writeError(w, err, http.StatusUnprocessableEntity)
return
}
if err := h.proxy.AddExtension(r.Context(), &ext); err != nil {
h.writeError(w, err, http.StatusBadRequest)
return
}
resBody, err := ext.MarshalBinary()
if err != nil {
h.writeError(w, err, http.StatusInternalServerError)
return
}
w.Header().Add("content-type", "application/json")
w.WriteHeader(http.StatusOK)
w.Write(resBody)
}
// C11yProxy proxies the request through the separate container, only for it to
// come back here for the storage. This is legacy from the pre-module times.
// TODO: cleanup, there does not need to be a separation between user-facing
// and internal storage endpoint in the long-term
type Proxy interface {
AddExtension(ctx context.Context, extension *models.C11yExtension) error
}
func (h *RESTUserFacingHandlers) writeError(w http.ResponseWriter, err error, code int) {
res := &models.ErrorResponse{Error: []*models.ErrorResponseErrorItems0{{
Message: err.Error(),
}}}
json, mErr := res.MarshalBinary()
if mErr != nil {
// fallback to text
w.Header().Add("content-type", "text/plain")
w.WriteHeader(code)
w.Write([]byte(err.Error()))
}
w.Header().Add("content-type", "application/json")
w.WriteHeader(code)
w.Write(json)
}
|