mirror of https://github.com/hashicorp/consul
71 lines
2.2 KiB
Go
71 lines
2.2 KiB
Go
|
// Package endpoints validates regional endpoints for services.
|
||
|
|
package endpoints
|
||
|
|
|
||
|
|
//go:generate go run -tags codegen ../model/cli/gen-endpoints/main.go endpoints.json endpoints_map.go
|
||
|
|
//go:generate gofmt -s -w endpoints_map.go
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"regexp"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// NormalizeEndpoint takes and endpoint and service API information to return a
|
||
|
|
// normalized endpoint and signing region. If the endpoint is not an empty string
|
||
|
|
// the service name and region will be used to look up the service's API endpoint.
|
||
|
|
// If the endpoint is provided the scheme will be added if it is not present.
|
||
|
|
func NormalizeEndpoint(endpoint, serviceName, region string, disableSSL, useDualStack bool) (normEndpoint, signingRegion string) {
|
||
|
|
if endpoint == "" {
|
||
|
|
return EndpointForRegion(serviceName, region, disableSSL, useDualStack)
|
||
|
|
}
|
||
|
|
|
||
|
|
return AddScheme(endpoint, disableSSL), ""
|
||
|
|
}
|
||
|
|
|
||
|
|
// EndpointForRegion returns an endpoint and its signing region for a service and region.
|
||
|
|
// if the service and region pair are not found endpoint and signingRegion will be empty.
|
||
|
|
func EndpointForRegion(svcName, region string, disableSSL, useDualStack bool) (endpoint, signingRegion string) {
|
||
|
|
dualStackField := ""
|
||
|
|
if useDualStack {
|
||
|
|
dualStackField = "/dualstack"
|
||
|
|
}
|
||
|
|
|
||
|
|
derivedKeys := []string{
|
||
|
|
region + "/" + svcName + dualStackField,
|
||
|
|
region + "/*" + dualStackField,
|
||
|
|
"*/" + svcName + dualStackField,
|
||
|
|
"*/*" + dualStackField,
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, key := range derivedKeys {
|
||
|
|
if val, ok := endpointsMap.Endpoints[key]; ok {
|
||
|
|
ep := val.Endpoint
|
||
|
|
ep = strings.Replace(ep, "{region}", region, -1)
|
||
|
|
ep = strings.Replace(ep, "{service}", svcName, -1)
|
||
|
|
|
||
|
|
endpoint = ep
|
||
|
|
signingRegion = val.SigningRegion
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return AddScheme(endpoint, disableSSL), signingRegion
|
||
|
|
}
|
||
|
|
|
||
|
|
// Regular expression to determine if the endpoint string is prefixed with a scheme.
|
||
|
|
var schemeRE = regexp.MustCompile("^([^:]+)://")
|
||
|
|
|
||
|
|
// AddScheme adds the HTTP or HTTPS schemes to a endpoint URL if there is no
|
||
|
|
// scheme. If disableSSL is true HTTP will be added instead of the default HTTPS.
|
||
|
|
func AddScheme(endpoint string, disableSSL bool) string {
|
||
|
|
if endpoint != "" && !schemeRE.MatchString(endpoint) {
|
||
|
|
scheme := "https"
|
||
|
|
if disableSSL {
|
||
|
|
scheme = "http"
|
||
|
|
}
|
||
|
|
endpoint = fmt.Sprintf("%s://%s", scheme, endpoint)
|
||
|
|
}
|
||
|
|
|
||
|
|
return endpoint
|
||
|
|
}
|