snapshot of https://github.com/ory/keto.git (master, 2026-09-16; blobs > 131072B omitted)

oryx/httpx/gzip_server.go

@@ -0,0 +1,50 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package httpx
+
+import (
+	"compress/gzip"
+	"fmt"
+	"io"
+	"net/http"
+	"strings"
+)
+
+type CompressionRequestReader struct {
+	ErrHandler func(w http.ResponseWriter, r *http.Request, err error)
+}
+
+func defaultCompressionErrorHandler(w http.ResponseWriter, r *http.Request, err error) {
+	http.Error(w, err.Error(), http.StatusBadRequest)
+}
+
+func NewCompressionRequestReader(eh func(w http.ResponseWriter, r *http.Request, err error)) *CompressionRequestReader {
+	if eh == nil {
+		eh = defaultCompressionErrorHandler
+	}
+
+	return &CompressionRequestReader{
+		ErrHandler: eh,
+	}
+}
+
+func (c *CompressionRequestReader) ServeHTTP(w http.ResponseWriter, r *http.Request, next http.HandlerFunc) {
+	for _, enc := range strings.Split(r.Header.Get("Content-Encoding"), ",") {
+		switch enc = strings.TrimSpace(enc); enc {
+		case "gzip":
+			reader, err := gzip.NewReader(r.Body)
+			if err != nil {
+				c.ErrHandler(w, r, err)
+				return
+			}
+			r.Body = io.NopCloser(reader)
+		case "identity", "":
+			// nothing to do
+		default:
+			c.ErrHandler(w, r, fmt.Errorf("%s content encoding not supported", enc))
+		}
+	}
+
+	next(w, r)
+}