-
Notifications
You must be signed in to change notification settings - Fork 3
/
faclient.go
54 lines (45 loc) · 1.73 KB
/
faclient.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
package faclient
import (
"crypto/tls"
"fmt"
"net/http"
"github.com/PureStorage-OpenConnect/terraform-provider-cbs/cbs/internal/array/faclient/2.4/client"
"github.com/PureStorage-OpenConnect/terraform-provider-cbs/cbs/internal/array/faclient/auth"
"github.com/PureStorage-OpenConnect/terraform-provider-cbs/version"
httptransport "github.com/go-openapi/runtime/client"
"github.com/go-openapi/strfmt"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/logging"
)
type customUserAgentTransport struct {
innerTransport http.RoundTripper
userAgent string
}
func (t *customUserAgentTransport) RoundTrip(r *http.Request) (*http.Response, error) {
r.Header.Add("User-Agent", t.userAgent)
return t.innerTransport.RoundTrip(r)
}
func New(host string, tokenConfig auth.TokenConfig) (*client.Flasharray, error) {
identityToken, err := tokenConfig.BuildToken()
if err != nil {
return nil, fmt.Errorf("Error creating client: %s", err)
}
tex := auth.Tex{TexHost: host}
accessToken, err := tex.ExchangeToken(identityToken)
if err != nil {
return nil, fmt.Errorf("Error exchanging tokens: %s", err)
}
cfg := client.DefaultTransportConfig().WithHost(host).WithSchemes([]string{"https"})
httpClient := http.Client{
Transport: &customUserAgentTransport{
innerTransport: logging.NewLoggingHTTPTransport(
&http.Transport{
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
}),
userAgent: fmt.Sprintf("%s/%s", "terraform-provider-cbs", version.ProviderVersion),
},
}
transport := httptransport.NewWithClient(cfg.Host, cfg.BasePath, cfg.Schemes, &httpClient)
// TODO implement token refreshing
transport.DefaultAuthentication = httptransport.BearerToken(accessToken)
return client.New(transport, strfmt.Default), nil
}