authentication: using a custom refresh function for Azure CLI Authent…

…ication fixes #22
hashicorp · Oct 20, 2019 · a2cb447 · a2cb447
1 parent c443704
commit a2cb447
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/authentication/auth_method_azure_cli_token.go b/authentication/auth_method_azure_cli_token.go
@@ -90,6 +90,21 @@ func (a azureCliTokenAuth) getAuthorizationToken(sender autorest.Sender, oauth *
 		return nil, err
 	}
 
+	var refreshFunc adal.TokenRefresh = func(ctx context.Context, resource string) (*adal.Token, error) {
+		token, err := obtainAuthorizationToken(resource, a.profile.subscriptionId)
+		if err != nil {
+			return nil, err
+		}
+
+		adalToken, err := token.ToADALToken()
+		if err != nil {
+			return nil, err
+		}
+
+		return &adalToken, nil
+	}
+	spt.SetCustomRefreshFunc(&refreshFunc)
+
 	auth := autorest.NewBearerAuthorizer(spt)
 	return auth, nil
 }