accept only json for api

vinceanalytics · Feb 8, 2024 · 18fea2e · 18fea2e
1 parent d21abc7
commit 18fea2e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/api/api.go b/api/api.go
@@ -72,7 +72,7 @@ func New(ctx context.Context, o *v1.Config) (*API, error) {
 				return
 			}
 		}
-		http.Error(w, http.StatusText(http.StatusNotFound), http.StatusNotFound)
+		request.Error(r.Context(), w, http.StatusNotFound, http.StatusText(http.StatusNotFound))
 	})
 
 	a.hand = base

diff --git a/request/request.go b/request/request.go
@@ -30,6 +30,10 @@ func Get(ctx context.Context) *protovalidate.Validator {
 
 func Read(w http.ResponseWriter, r *http.Request, o proto.Message) bool {
 	ctx := r.Context()
+	if r.Header.Get("Content-Type") != "application/json" {
+		Error(ctx, w, http.StatusBadRequest, http.StatusText(http.StatusBadRequest))
+		return false
+	}
 	if r.ContentLength == 0 || r.ContentLength > maxBodySize {
 		logger.Get(ctx).Error("Invalid content length", "contentLength", r.ContentLength)
 		Error(ctx, w, http.StatusBadRequest, http.StatusText(http.StatusBadRequest))