Add CustomRootHandler option

p2p/http/options.go

@@ -3,14 +3,15 @@ package libp2phttp
 import (
 	"crypto/tls"
 	"fmt"
+	"net/http"
 
 	"github.com/libp2p/go-libp2p/core/host"
 	ma "github.com/multiformats/go-multiaddr"
 )
 
 type HTTPHostOption func(*HTTPHost) error
 
-// WithStreamHost sets the stream host to use for HTTP over libp2p streams.
+// ListenAddrs sets the listen addresses for the HTTP transport.
 func ListenAddrs(addrs []ma.Multiaddr) HTTPHostOption {
 	return func(h *HTTPHost) error {
 		// assert that each addr contains a /http component
@@ -26,6 +27,7 @@ func ListenAddrs(addrs []ma.Multiaddr) HTTPHostOption {
 	}
 }
 
+// TLSConfig sets the server TLS config for the HTTP transport.
 func TLSConfig(tlsConfig *tls.Config) HTTPHostOption {
 	return func(h *HTTPHost) error {
 		h.httpTransport.tlsConfig = tlsConfig
@@ -48,3 +50,14 @@ func WithTLSClientConfig(tlsConfig *tls.Config) HTTPHostOption {
 		return nil
 	}
 }
+
+// CustomRootHandler sets the root handler for the HTTP transport. It *does not*
+// set up a .well-known/libp2p handler. Users assume the responsibility of
+// handling .well-known/libp2p, as well as keeping it up to date.
+func CustomRootHandler(mux http.Handler) HTTPHostOption {
+	return func(h *HTTPHost) error {
+		h.rootHandler = http.ServeMux{}
+		h.rootHandler.Handle("/", mux)
+		return nil
+	}
+}