2019-05-31 18:37:43 -04:00
|
|
|
package client
|
2019-01-10 18:21:16 -05:00
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"github.com/Azure/go-autorest/autorest/adal"
|
|
|
|
|
"github.com/Azure/go-autorest/autorest/azure"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
// for managed identity auth
|
|
|
|
|
type msiOAuthTokenProvider struct {
|
|
|
|
|
env azure.Environment
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func NewMSIOAuthTokenProvider(env azure.Environment) oAuthTokenProvider {
|
|
|
|
|
return &msiOAuthTokenProvider{env}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (tp *msiOAuthTokenProvider) getServicePrincipalToken() (*adal.ServicePrincipalToken, error) {
|
|
|
|
|
return tp.getServicePrincipalTokenWithResource(tp.env.ResourceManagerEndpoint)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (tp *msiOAuthTokenProvider) getServicePrincipalTokenWithResource(resource string) (*adal.ServicePrincipalToken, error) {
|
|
|
|
|
return adal.NewServicePrincipalTokenFromMSI("http://169.254.169.254/metadata/identity/oauth2/token", resource)
|
|
|
|
|
}
|
