grpc-web/go/grpcweb/helpers.go

//Copyright 2017 Improbable. All Rights Reserved.
// See LICENSE for licensing terms.

package grpcweb

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.
func ListGRPCResources(server *grpc.Server) []string {
	ret := []string{}
	for serviceName, serviceInfo := range server.GetServiceInfo() {
		for _, methodInfo := range serviceInfo.Methods {
			fullResource := fmt.Sprintf("/%s/%s", serviceName, methodInfo.Name)
			ret = append(ret, fullResource)
		}
	}
	return ret
}

// WebsocketRequestOrigin returns the host from which a websocket request made by a web browser
// originated.
func WebsocketRequestOrigin(req *http.Request) (string, error) {
	origin := req.Header.Get("Origin")
	parsed, err := url.ParseRequestURI(origin)
	if err != nil {
		return "", fmt.Errorf("failed to parse url for grpc-websocket origin check: %q. error: %v", origin, err)
	}
	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
}