token_credential_impl.hpp

Go to the documentation of this file.

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
 
#pragma once
 
#include <azure/core/credentials/token_credential_options.hpp>
#include <azure/core/http/http.hpp>
#include <azure/core/http/http_status_code.hpp>
#include <azure/core/http/raw_response.hpp>
#include <azure/core/internal/http/pipeline.hpp>
#include <azure/core/io/body_stream.hpp>
 
#include <functional>
#include <memory>
#include <string>
#include <utility>
#include <vector>
 
namespace Azure { namespace Identity { namespace _detail {
  class TokenCredentialImpl {
  private:
    Core::Http::_internal::HttpPipeline m_httpPipeline;
 
  public:
    virtual ~TokenCredentialImpl() = default;
 
    explicit TokenCredentialImpl(Core::Credentials::TokenCredentialOptions const& options);
 
    static std::string FormatScopes(
        std::vector<std::string> const& scopes,
        bool asResource,
        bool urlEncode = true);
 
    static Core::Credentials::AccessToken ParseToken(
        std::string const& jsonString,
        std::string const& accessTokenPropertyName,
        std::string const& expiresInPropertyName,
        std::vector<std::string> const& expiresOnPropertyNames,
        std::string const& refreshInPropertyName = "",
        bool proactiveRenewal = false,
        int utcDiffSeconds = 0);
 
    static Core::Credentials::AccessToken ParseToken(
        std::string const& jsonString,
        std::string const& accessTokenPropertyName,
        std::string const& expiresInPropertyName,
        std::string const& expiresOnPropertyName,
        std::string const& refreshInPropertyName = "",
        bool proactiveRenewal = false)
    {
      return ParseToken(
          jsonString,
          accessTokenPropertyName,
          expiresInPropertyName,
          std::vector<std::string>{expiresOnPropertyName},
          refreshInPropertyName,
          proactiveRenewal);
    }
 
    class TokenRequest final {
    private:
      std::unique_ptr<std::string> m_body;
      std::unique_ptr<Core::IO::MemoryBodyStream> m_memoryBodyStream;
 
    public:
      Core::Http::Request HttpRequest;
 
      explicit TokenRequest(Core::Http::HttpMethod httpMethod, Core::Url url, std::string body)
          : m_body(new std::string(std::move(body))),
            m_memoryBodyStream(new Core::IO::MemoryBodyStream(
                reinterpret_cast<uint8_t const*>(m_body->data()),
                m_body->size())),
            HttpRequest(std::move(httpMethod), std::move(url), m_memoryBodyStream.get())
      {
        HttpRequest.SetHeader("Content-Type", "application/x-www-form-urlencoded");
        HttpRequest.SetHeader("Content-Length", std::to_string(m_body->size()));
      }
 
      explicit TokenRequest(Core::Http::Request httpRequest) : HttpRequest(std::move(httpRequest))
      {
      }
    };
 
    Core::Credentials::AccessToken GetToken(
        Core::Context const& context,
        bool proactiveRenewal,
        std::function<std::unique_ptr<TokenRequest>()> const& createRequest,
        std::function<std::unique_ptr<TokenRequest>(
            Core::Http::HttpStatusCode statusCode,
            Core::Http::RawResponse const& response)> const& shouldRetry
        = [](auto const, auto const&) { return nullptr; }) const;
  };
}}} // namespace Azure::Identity::_detail