feat(routing/http/client): allow custom User-Agent

routing/http/client/client.go

@@ -40,6 +40,10 @@ type client struct {
 	afterSignCallback func(req *types.WriteBitswapProviderRecord)
 }
 
+// defaultUserAgent is used as a fallback to inform HTTP server which library
+// version sent a request
+var defaultUserAgent = moduleVersion()
+
 var _ contentrouter.Client = &client{}
 
 type httpClient interface {
@@ -60,6 +64,23 @@ func WithHTTPClient(h httpClient) option {
 	}
 }
 
+func WithUserAgent(ua string) option {
+	return func(c *client) {
+		if ua == "" {
+			return
+		}
+		httpClient, ok := c.httpClient.(*http.Client)
+		if !ok {
+			return
+		}
+		transport, ok := httpClient.Transport.(*ResponseBodyLimitedTransport)
+		if !ok {
+			return
+		}
+		transport.UserAgent = ua
+	}
+}
+
 func WithProviderInfo(peerID peer.ID, addrs []multiaddr.Multiaddr) option {
 	return func(c *client) {
 		c.peerID = peerID
@@ -76,6 +97,7 @@ func New(baseURL string, opts ...option) (*client, error) {
 		Transport: &ResponseBodyLimitedTransport{
 			RoundTripper: http.DefaultTransport,
 			LimitBytes:   1 << 20,
+			UserAgent:    defaultUserAgent,
 		},
 	}
 	client := &client{

routing/http/client/client_test.go

@@ -48,15 +48,17 @@ type testDeps struct {
 }
 
 func makeTestDeps(t *testing.T) testDeps {
+	const testUserAgent = "testUserAgent"
 	peerID, addrs, identity := makeProviderAndIdentity()
 	router := &mockContentRouter{}
 	server := httptest.NewServer(server.Handler(router))
 	t.Cleanup(server.Close)
 	serverAddr := "http://" + server.Listener.Addr().String()
-	c, err := New(serverAddr, WithProviderInfo(peerID, addrs), WithIdentity(identity))
+	c, err := New(serverAddr, WithProviderInfo(peerID, addrs), WithIdentity(identity), WithUserAgent(testUserAgent))
 	if err != nil {
 		panic(err)
 	}
+	assertUserAgentOverride(t, c, testUserAgent)
 	return testDeps{
 		router: router,
 		server: server,
@@ -66,6 +68,20 @@ func makeTestDeps(t *testing.T) testDeps {
 	}
 }
 
+func assertUserAgentOverride(t *testing.T, c *client, expected string) {
+	httpClient, ok := c.httpClient.(*http.Client)
+	if !ok {
+		t.Error("invalid c.httpClient")
+	}
+	transport, ok := httpClient.Transport.(*ResponseBodyLimitedTransport)
+	if !ok {
+		t.Error("invalid httpClient.Transport")
+	}
+	if transport.UserAgent != expected {
+		t.Error("invalid httpClient.Transport.UserAgent")
+	}
+}
+
 func makeCID() cid.Cid {
 	buf := make([]byte, 63)
 	_, err := rand.Read(buf)

routing/http/client/transport.go

@@ -4,14 +4,19 @@ import (
 	"fmt"
 	"io"
 	"net/http"
+	"runtime/debug"
 )
 
 type ResponseBodyLimitedTransport struct {
 	http.RoundTripper
 	LimitBytes int64
+	UserAgent  string
 }
 
 func (r *ResponseBodyLimitedTransport) RoundTrip(req *http.Request) (*http.Response, error) {
+	if r.UserAgent != "" {
+		req.Header.Set("User-Agent", r.UserAgent)
+	}
 	resp, err := r.RoundTripper.RoundTrip(req)
 	if resp != nil && resp.Body != nil {
 		resp.Body = &limitReadCloser{
@@ -36,3 +41,35 @@ func (l *limitReadCloser) Read(p []byte) (int, error) {
 	}
 	return n, err
 }
+
+// ImportPath is the canonical import path that allows us to identify
+// official client builds vs modified forks, and use that info in User-Agent header.
+const ImportPath = "github.com/ipfs/go-libipfs/routing/http/client"
+
+// moduleVersion returns a useful user agent version string allowing us to
+// identify requests coming from officialxl releases of this module.
+func moduleVersion() string {
+	var module *debug.Module
+	bi, ok := debug.ReadBuildInfo()
+	if !ok {
+		return ""
+	}
+
+	// find this client in the dependency list
+	// of the app that has it in go.mod
+	for _, dep := range bi.Deps {
+		if dep.Path == ImportPath {
+			module = dep
+			break
+		}
+	}
+
+	if module == nil {
+		return ""
+	}
+	ua := ImportPath + "/" + module.Version
+	if module.Sum != "" {
+		ua += "/" + module.Sum
+	}
+	return ua
+}