datalake_sas_builder.hpp

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
 
#pragma once
 
#include "azure/storage/files/datalake/datalake_responses.hpp"
 
#include <azure/core/nullable.hpp>
#include <azure/storage/common/account_sas_builder.hpp>
 
#include <cstdint>
#include <string>
#include <type_traits>
 
namespace Azure { namespace Storage { namespace Sas {
 
  enum class DataLakeSasResource
  {
    FileSystem,
 
    File,
 
    Directory,
  };
 
  enum class DataLakeFileSystemSasPermissions
  {
    Read = 1,
 
    Write = 2,
 
    Delete = 4,
 
    List = 8,
 
    Add = 16,
 
    Create = 32,
 
    All = ~0,
  };
 
  inline DataLakeFileSystemSasPermissions operator|(
      DataLakeFileSystemSasPermissions lhs,
      DataLakeFileSystemSasPermissions rhs)
  {
    using type = std::underlying_type_t<DataLakeFileSystemSasPermissions>;
    return static_cast<DataLakeFileSystemSasPermissions>(
        static_cast<type>(lhs) | static_cast<type>(rhs));
  }
 
  inline DataLakeFileSystemSasPermissions operator&(
      DataLakeFileSystemSasPermissions lhs,
      DataLakeFileSystemSasPermissions rhs)
  {
    using type = std::underlying_type_t<DataLakeFileSystemSasPermissions>;
    return static_cast<DataLakeFileSystemSasPermissions>(
        static_cast<type>(lhs) & static_cast<type>(rhs));
  }
 
  enum class DataLakeSasPermissions
  {
    Read = 1,
 
    Write = 2,
 
    Delete = 4,
 
    Add = 8,
 
    Create = 16,
 
    List = 32,
 
    Move = 64,
 
    Execute = 128,
 
    ManageOwnership = 256,
 
    ManageAccessControl = 512,
 
    All = ~0,
  };
 
  inline DataLakeSasPermissions operator|(DataLakeSasPermissions lhs, DataLakeSasPermissions rhs)
  {
    using type = std::underlying_type_t<DataLakeSasPermissions>;
    return static_cast<DataLakeSasPermissions>(static_cast<type>(lhs) | static_cast<type>(rhs));
  }
 
  inline DataLakeSasPermissions operator&(DataLakeSasPermissions lhs, DataLakeSasPermissions rhs)
  {
    using type = std::underlying_type_t<DataLakeSasPermissions>;
    return static_cast<DataLakeSasPermissions>(static_cast<type>(lhs) & static_cast<type>(rhs));
  }
 
  struct DataLakeSasBuilder final
  {
    SasProtocol Protocol;
 
    Azure::Nullable<Azure::DateTime> StartsOn;
 
    Azure::DateTime ExpiresOn;
 
    Azure::Nullable<std::string> IPRange;
 
    std::string Identifier;
 
    std::string FileSystemName;
 
    std::string Path;
 
    bool IsDirectory = false;
 
    Azure::Nullable<int32_t> DirectoryDepth;
 
    DataLakeSasResource Resource;
 
    std::string CacheControl;
 
    std::string ContentDisposition;
 
    std::string ContentEncoding;
 
    std::string ContentLanguage;
 
    std::string ContentType;
 
    std::string PreauthorizedAgentObjectId;
 
    std::string AgentObjectId;
 
    std::string CorrelationId;
 
    std::string EncryptionScope;
 
    void SetPermissions(DataLakeFileSystemSasPermissions permissions);
 
    void SetPermissions(DataLakeSasPermissions permissions);
 
    std::string GenerateSasToken(const StorageSharedKeyCredential& credential);
 
    void SetPermissions(std::string rawPermissions) { Permissions = std::move(rawPermissions); }
 
    std::string GenerateSasToken(
        const Files::DataLake::Models::UserDelegationKey& userDelegationKey,
        const std::string& accountName);
 
    std::string GenerateSasStringToSign(const StorageSharedKeyCredential& credential);
 
    std::string GenerateSasStringToSign(
        const Blobs::Models::UserDelegationKey& userDelegationKey,
        const std::string& accountName);
 
  private:
    std::string Permissions;
  };
 
}}} // namespace Azure::Storage::Sas