storage_credential.hpp

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
 
#pragma once
 
#include <azure/core/http/http.hpp>
 
#include <memory>
#include <mutex>
#include <string>
 
namespace Azure { namespace Storage {
 
  namespace Sas {
    struct AccountSasBuilder;
    struct BlobSasBuilder;
    struct ShareSasBuilder;
    struct DataLakeSasBuilder;
    struct QueueSasBuilder;
  } // namespace Sas
 
  namespace _internal {
    class SharedKeyPolicy;
  }
 
  class StorageSharedKeyCredential final {
  public:
    explicit StorageSharedKeyCredential(std::string accountName, std::string accountKey)
        : AccountName(std::move(accountName)), m_accountKey(std::move(accountKey))
    {
    }
 
    void Update(std::string accountKey)
    {
      std::lock_guard<std::mutex> guard(m_mutex);
      m_accountKey = std::move(accountKey);
    }
 
    const std::string AccountName;
 
  private:
    friend class _internal::SharedKeyPolicy;
    friend struct Sas::BlobSasBuilder;
    friend struct Sas::ShareSasBuilder;
    friend struct Sas::DataLakeSasBuilder;
    friend struct Sas::QueueSasBuilder;
    friend struct Sas::AccountSasBuilder;
    std::string GetAccountKey() const
    {
      std::lock_guard<std::mutex> guard(m_mutex);
      return m_accountKey;
    }
 
    mutable std::mutex m_mutex;
    std::string m_accountKey;
  };
 
  namespace _internal {
 
    struct ConnectionStringParts
    {
      std::string AccountName;
      std::string AccountKey;
      Azure::Core::Url BlobServiceUrl;
      Azure::Core::Url FileServiceUrl;
      Azure::Core::Url QueueServiceUrl;
      Azure::Core::Url DataLakeServiceUrl;
      std::shared_ptr<StorageSharedKeyCredential> KeyCredential;
    };
 
    ConnectionStringParts ParseConnectionString(const std::string& connectionString);
 
    std::string GetDefaultScopeForAudience(const std::string& audience);
 
  } // namespace _internal
 
}} // namespace Azure::Storage