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

oryx/httpx/transports.go

@@ -0,0 +1,68 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package httpx
+
+import "net/http"
+
+// WrapTransportWithHeader wraps a http.Transport to always use the values from the given header.
+func WrapTransportWithHeader(parent http.RoundTripper, h http.Header) *TransportWithHeader {
+	return &TransportWithHeader{
+		RoundTripper: parent,
+		h:            h,
+	}
+}
+
+// NewTransportWithHeader returns a new http.Transport that always uses the values from the given header.
+func NewTransportWithHeader(h http.Header) *TransportWithHeader {
+	return &TransportWithHeader{
+		RoundTripper: http.DefaultTransport,
+		h:            h,
+	}
+}
+
+// TransportWithHeader is an http.RoundTripper that always uses the values from the given header.
+type TransportWithHeader struct {
+	http.RoundTripper
+	h http.Header
+}
+
+// RoundTrip implements http.RoundTripper.
+func (ct *TransportWithHeader) RoundTrip(req *http.Request) (*http.Response, error) {
+	for k := range ct.h {
+		req.Header.Set(k, ct.h.Get(k))
+	}
+	return ct.RoundTripper.RoundTrip(req)
+}
+
+// NewTransportWithHost returns a new http.Transport that always uses the given host.
+func NewTransportWithHost(host string) *TransportWithHost {
+	return &TransportWithHost{
+		RoundTripper: http.DefaultTransport,
+		host:         host,
+	}
+}
+
+// WrapRoundTripperWithHost wraps a http.RoundTripper that always uses the given host.
+func WrapRoundTripperWithHost(parent http.RoundTripper, host string) *TransportWithHost {
+	return &TransportWithHost{
+		RoundTripper: parent,
+		host:         host,
+	}
+}
+
+// TransportWithHost is an http.RoundTripper that always uses the given host.
+type TransportWithHost struct {
+	http.RoundTripper
+	host string
+}
+
+// RoundTrip implements http.RoundTripper.
+func (ct *TransportWithHost) RoundTrip(req *http.Request) (*http.Response, error) {
+	req.Host = ct.host
+	return ct.RoundTripper.RoundTrip(req)
+}
+
+type TransportFunc func(*http.Request) (*http.Response, error)
+
+func (f TransportFunc) RoundTrip(req *http.Request) (*http.Response, error) { return f(req) }