67 lines
1.4 KiB
Go
67 lines
1.4 KiB
Go
package azure
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"log"
|
|
|
|
"net/http"
|
|
|
|
"github.com/hashicorp/go-retryablehttp"
|
|
)
|
|
|
|
type Client struct {
|
|
logger *log.Logger
|
|
|
|
subscriptionID string
|
|
resourceManagerEndpoint string
|
|
|
|
tokenRequester *tokenRequester
|
|
httpClient *retryablehttp.Client
|
|
}
|
|
|
|
func NewClient(creds *AzureResourceManagerCredentials) (*Client, error) {
|
|
defaultLogger := log.New(ioutil.Discard, "", 0)
|
|
|
|
httpClient := retryablehttp.NewClient()
|
|
httpClient.Logger = defaultLogger
|
|
|
|
if creds.ResourceManagerEndpoint == "" {
|
|
creds.ResourceManagerEndpoint = defaultResourceManagerEndpoint
|
|
}
|
|
if creds.ActiveDirectoryEndpoint == "" {
|
|
creds.ActiveDirectoryEndpoint = defaultActiveDirectoryEndpoint
|
|
}
|
|
|
|
tr := newTokenRequester(httpClient, creds)
|
|
|
|
return &Client{
|
|
subscriptionID: creds.SubscriptionID,
|
|
resourceManagerEndpoint: creds.ResourceManagerEndpoint,
|
|
httpClient: httpClient,
|
|
tokenRequester: tr,
|
|
logger: defaultLogger,
|
|
}, nil
|
|
}
|
|
|
|
func (c *Client) SetRequestLoggingHook(hook func(*log.Logger, *http.Request, int)) {
|
|
c.httpClient.RequestLogHook = hook
|
|
}
|
|
|
|
func (c *Client) SetLogger(newLogger *log.Logger) {
|
|
c.logger = newLogger
|
|
c.httpClient.Logger = newLogger
|
|
}
|
|
|
|
func (c *Client) NewRequest() *Request {
|
|
return &Request{
|
|
client: c,
|
|
}
|
|
}
|
|
|
|
func (c *Client) NewRequestForURI(resourceURI string) *Request {
|
|
return &Request{
|
|
URI: &resourceURI,
|
|
client: c,
|
|
}
|
|
}
|