forked from cloudfoundry/cli
-
Notifications
You must be signed in to change notification settings - Fork 0
/
endpoints.go
122 lines (98 loc) · 3.54 KB
/
endpoints.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
package api
import (
"cf/configuration"
"cf/net"
"regexp"
"strings"
)
type EndpointRepository interface {
UpdateEndpoint(endpoint string) (finalEndpoint string, apiResponse net.ApiResponse)
GetLoggregatorEndpoint() (endpoint string, apiResponse net.ApiResponse)
GetUAAEndpoint() (endpoint string, apiResponse net.ApiResponse)
GetCloudControllerEndpoint() (endpoint string, apiResponse net.ApiResponse)
}
type RemoteEndpointRepository struct {
config configuration.ReadWriter
gateway net.Gateway
}
func NewEndpointRepository(config configuration.ReadWriter, gateway net.Gateway) (repo RemoteEndpointRepository) {
repo.config = config
repo.gateway = gateway
return
}
func (repo RemoteEndpointRepository) UpdateEndpoint(endpoint string) (finalEndpoint string, apiResponse net.ApiResponse) {
endpointMissingScheme := !strings.HasPrefix(endpoint, "https://") && !strings.HasPrefix(endpoint, "http://")
if endpointMissingScheme {
finalEndpoint = "https://" + endpoint
apiResponse = repo.attemptUpdate(finalEndpoint)
if apiResponse.IsNotSuccessful() {
finalEndpoint = "http://" + endpoint
apiResponse = repo.attemptUpdate(finalEndpoint)
}
return
}
finalEndpoint = endpoint
apiResponse = repo.attemptUpdate(finalEndpoint)
return
}
func (repo RemoteEndpointRepository) attemptUpdate(endpoint string) (apiResponse net.ApiResponse) {
request, apiResponse := repo.gateway.NewRequest("GET", endpoint+"/v2/info", "", nil)
if apiResponse.IsNotSuccessful() {
return
}
serverResponse := new(struct {
ApiVersion string `json:"api_version"`
AuthorizationEndpoint string `json:"authorization_endpoint"`
LoggregatorEndpoint string `json:"logging_endpoint"`
})
_, apiResponse = repo.gateway.PerformRequestForJSONResponse(request, &serverResponse)
if apiResponse.IsNotSuccessful() {
return
}
if endpoint != repo.config.ApiEndpoint() {
repo.config.ClearSession()
}
repo.config.SetApiEndpoint(endpoint)
repo.config.SetApiVersion(serverResponse.ApiVersion)
repo.config.SetAuthorizationEndpoint(serverResponse.AuthorizationEndpoint)
repo.config.SetLoggregatorEndpoint(serverResponse.LoggregatorEndpoint)
return
}
func (repo RemoteEndpointRepository) GetLoggregatorEndpoint() (endpoint string, apiResponse net.ApiResponse) {
if repo.config.LoggregatorEndpoint() == "" {
if repo.config.ApiEndpoint() == "" {
apiResponse = net.NewApiResponseWithMessage("Loggregator endpoint missing from config file")
} else {
endpoint = defaultLoggregatorEndpoint(repo.config.ApiEndpoint())
}
} else {
endpoint = repo.config.LoggregatorEndpoint()
}
return
}
func (repo RemoteEndpointRepository) GetCloudControllerEndpoint() (endpoint string, apiResponse net.ApiResponse) {
if repo.config.ApiEndpoint() == "" {
apiResponse = net.NewApiResponseWithMessage("Target endpoint missing from config file")
return
}
endpoint = repo.config.ApiEndpoint()
return
}
func (repo RemoteEndpointRepository) GetUAAEndpoint() (endpoint string, apiResponse net.ApiResponse) {
if repo.config.AuthorizationEndpoint() == "" {
apiResponse = net.NewApiResponseWithMessage("UAA endpoint missing from config file")
return
}
endpoint = strings.Replace(repo.config.AuthorizationEndpoint(), "login", "uaa", 1)
return
}
// FIXME: needs semantic versioning
func defaultLoggregatorEndpoint(apiEndpoint string) string {
url := endpointDomainRegex.ReplaceAllString(apiEndpoint, "ws${1}://loggregator.${2}")
if url[0:3] == "wss" {
return url + ":4443"
} else {
return url + ":80"
}
}
var endpointDomainRegex = regexp.MustCompile(`^http(s?)://[^\.]+\.(.+)\/?`)