get_token_info

Author: jiasli

src/azure-cli-core/azure/cli/core/auth/credential_adaptor.py

@@ -34,10 +34,47 @@ def get_token(self, *scopes, **kwargs):
 
         return build_sdk_access_token(self._credential.acquire_token(list(scopes), **filtered_kwargs))
 
+
+    def get_token_info(self, *scopes, options=None):
+        """Get an access token from the main credential."""
+        logger.debug("CredentialAdaptor.get_token_info: scopes=%r, options=%r", scopes, options)
+
+        # Discard unsupported kwargs: tenant_id, enable_cae
+        msal_kwargs = {}
+        if 'claims' in options:
+            msal_kwargs['claims'] = options['claims']
+        if 'data' in options:
+            msal_kwargs['data'] = options['data']
+
+        return _build_sdk_access_token_info(self._credential.acquire_token(list(scopes), **msal_kwargs))
+
+
     def get_auxiliary_tokens(self, *scopes, **kwargs):
         """Get access tokens from auxiliary credentials."""
         # To test cross-tenant authentication, see https://github.com/Azure/azure-cli/issues/16691
         if self._auxiliary_credentials:
             return [build_sdk_access_token(cred.acquire_token(list(scopes), **kwargs))
                     for cred in self._auxiliary_credentials]
         return None
+
+
+def _build_sdk_access_token_info(token_entry):
+    # MSAL token entry sample:
+    # {
+    #     'access_token': 'eyJ0eXAiOiJKV...',
+    #     'token_type': 'Bearer',
+    #     'expires_in': 1618,
+    #     'token_source': 'cache'
+    # }
+
+    # Importing azure.core.credentials.AccessTokenInfo is expensive.
+    # This can slow down commands that doesn't need azure.core.
+    from .constants import ACCESS_TOKEN, EXPIRES_IN
+    from azure.core.credentials import AccessTokenInfo
+
+    return AccessTokenInfo(token_entry[ACCESS_TOKEN], _now_timestamp() + token_entry[EXPIRES_IN])
+
+
+def _now_timestamp():
+    import time
+    return int(time.time())