Allow non root resources

go/grpcweb/helpers.go

@@ -7,10 +7,14 @@ import (
 	"fmt"
 	"net/http"
 	"net/url"
+	"regexp"
+	"strings"
 
 	"google.golang.org/grpc"
 )
 
+var pathMatcher = regexp.MustCompile(`/[^/]*/[^/]*$`)
+
 // ListGRPCResources is a helper function that lists all URLs that are registered on gRPC server.
 //
 // This makes it easy to register all the relevant routes in your HTTP router of choice.
@@ -35,3 +39,12 @@ func WebsocketRequestOrigin(req *http.Request) (string, error) {
 	}
 	return parsed.Host, nil
 }
+
+func GetGRPCEndpoint(req *http.Request) string {
+	endpoint := pathMatcher.FindString(strings.TrimRight(req.URL.Path, "/"))
+	if len(endpoint) == 0 {
+		return req.URL.Path
+	}
+
+	return endpoint
+}

go/grpcweb/helpers_test.go

@@ -4,6 +4,7 @@
 package grpcweb_test
 
 import (
+	"net/http/httptest"
 	"sort"
 	"testing"
 
@@ -34,3 +35,26 @@ func TestListGRPCResources(t *testing.T) {
 		actual,
 		"list grpc resources must provide an exhaustive list of all registered handlers")
 }
+
+func TestGetGRPCEndpoint(t *testing.T) {
+	cases := []struct {
+		input  string
+		output string
+	}{
+		{input: "/", output: "/"},
+		{input: "/resource", output: "/resource"},
+		{input: "/improbable.grpcweb.test.TestService/PingEmpty", output: "/improbable.grpcweb.test.TestService/PingEmpty"},
+		{input: "/improbable.grpcweb.test.TestService/PingEmpty/", output: "/improbable.grpcweb.test.TestService/PingEmpty"},
+		{input: "/a/b/c/improbable.grpcweb.test.TestService/PingEmpty", output: "/improbable.grpcweb.test.TestService/PingEmpty"},
+		{input: "/a/b/c/improbable.grpcweb.test.TestService/PingEmpty/", output: "/improbable.grpcweb.test.TestService/PingEmpty"},
+	}
+
+	for _, c := range cases {
+		req := httptest.NewRequest("GET", c.input, nil)
+
+		result := grpcweb.GetGRPCEndpoint(req)
+
+		assert.Equal(t, c.output, result)
+	}
+
+}

go/grpcweb/options.go

@@ -10,6 +10,7 @@ var (
 		allowedRequestHeaders:          []string{"*"},
 		corsForRegisteredEndpointsOnly: true,
 		originFunc:                     func(origin string) bool { return false },
+		allowNonRootResources:          false,
 	}
 )
 
@@ -19,6 +20,7 @@ type options struct {
 	originFunc                     func(origin string) bool
 	enableWebsockets               bool
 	websocketOriginFunc            func(req *http.Request) bool
+	allowNonRootResources          bool
 }
 
 func evaluateOptions(opts []Option) *options {
@@ -99,3 +101,9 @@ func WithWebsocketOriginFunc(websocketOriginFunc func(req *http.Request) bool) O
 		o.websocketOriginFunc = websocketOriginFunc
 	}
 }
+
+func WithAllowNonRootResource(allowNonRootResources bool) Option {
+	return func(o *options) {
+		o.allowNonRootResources = allowNonRootResources
+	}
+}

go/grpcweb/wrapper.go

@@ -35,6 +35,7 @@ type WrappedGrpcServer struct {
 	originFunc          func(origin string) bool
 	enableWebsockets    bool
 	websocketOriginFunc func(req *http.Request) bool
+	endpointFunc        func(req *http.Request) string
 }
 
 // WrapServer takes a gRPC Server in Go and returns a WrappedGrpcServer that provides gRPC-Web Compatibility.
@@ -56,13 +57,20 @@ func WrapServer(server *grpc.Server, options ...Option) *WrappedGrpcServer {
 	if websocketOriginFunc == nil {
 		websocketOriginFunc = defaultWebsocketOriginFunc
 	}
+
+	endpointFunc := func(req *http.Request) string { return req.URL.Path }
+	if opts.allowNonRootResources {
+		endpointFunc = GetGRPCEndpoint
+	}
+
 	return &WrappedGrpcServer{
 		server:              server,
 		opts:                opts,
 		corsWrapper:         corsWrapper,
 		originFunc:          opts.originFunc,
 		enableWebsockets:    opts.enableWebsockets,
 		websocketOriginFunc: websocketOriginFunc,
+		endpointFunc:        endpointFunc,
 	}
 }
 
@@ -105,6 +113,7 @@ func (w *WrappedGrpcServer) IsGrpcWebSocketRequest(req *http.Request) bool {
 func (w *WrappedGrpcServer) HandleGrpcWebRequest(resp http.ResponseWriter, req *http.Request) {
 	intReq, isTextFormat := hackIntoNormalGrpcRequest(req)
 	intResp := newGrpcWebResponse(resp, isTextFormat)
+	req.URL.Path = w.endpointFunc(req)
 	w.server.ServeHTTP(intResp, intReq)
 	intResp.finishRequest(req)
 }
@@ -161,6 +170,7 @@ func (w *WrappedGrpcServer) handleWebSocket(wsConn *websocket.Conn, req *http.Re
 		grpclog.Errorf("web socket text format requests not yet supported")
 		return
 	}
+	req.URL.Path = w.endpointFunc(req)
 	w.server.ServeHTTP(respWriter, interceptedRequest)
 }
 
@@ -187,7 +197,7 @@ func (w *WrappedGrpcServer) IsAcceptableGrpcCorsRequest(req *http.Request) bool
 
 func (w *WrappedGrpcServer) isRequestForRegisteredEndpoint(req *http.Request) bool {
 	registeredEndpoints := ListGRPCResources(w.server)
-	requestedEndpoint := req.URL.Path
+	requestedEndpoint := w.endpointFunc(req)
 	for _, v := range registeredEndpoints {
 		if v == requestedEndpoint {
 			return true