oauth2-proxy/pkg/requests/http.go

package requests

import (
	"net/http"

	"github.com/oauth2-proxy/oauth2-proxy/v7/pkg/version"
)

type userAgentTransport struct {
	Next      http.RoundTripper
	userAgent string
}

func (t *userAgentTransport) RoundTrip(req *http.Request) (*http.Response, error) {
	r := req.Clone(req.Context())
	setDefaultUserAgent(r.Header, t.userAgent)
	return t.Next.RoundTrip(r)
}

var DefaultHTTPClient = &http.Client{Transport: &DefaultTransport}

var DefaultTransport = userAgentTransport{
	Next:      http.DefaultTransport,
	userAgent: "oauth2-proxy/" + version.VERSION,
}

func setDefaultUserAgent(header http.Header, userAgent string) {
	if header != nil && len(header.Values("User-Agent")) == 0 {
		header.Set("User-Agent", userAgent)
	}
}