// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
/**
 * @file
 * @brief Defines the Key Vault Secret paged responses.
 *
 */

#pragma once

#include "azure/keyvault/secrets/keyvault_deleted_secret.hpp"
#include "azure/keyvault/secrets/keyvault_secret_properties.hpp"

#include <azure/core/paged_response.hpp>

#include <memory>
#include <vector>

namespace Azure { namespace Security { namespace KeyVault { namespace Secrets {

  // forward definition
  class SecretClient;

  /**
   * @brief Define a single page to list the secrets from the Key Vault.
   *
   */
  class SecretPropertiesPagedResponse final
      : public Azure::Core::PagedResponse<SecretPropertiesPagedResponse> {
  private:
    friend class SecretClient;
    friend class Azure::Core::PagedResponse<SecretPropertiesPagedResponse>;

    std::string m_secretName;
    std::shared_ptr<SecretClient> m_secretClient;

    void OnNextPage(const Azure::Core::Context& context);

    SecretPropertiesPagedResponse(
        SecretPropertiesPagedResponse&& secretProperties,
        std::unique_ptr<Azure::Core::Http::RawResponse> rawResponse,
        std::shared_ptr<SecretClient> secretClient,
        std::string const& secretName = std::string())
        : PagedResponse(std::move(secretProperties)), m_secretName(secretName),
          m_secretClient(secretClient), Items(std::move(secretProperties.Items))
    {
      RawResponse = std::move(rawResponse);
    }

  public:
    /**
     * @brief Construct a new SecretPropertiesPagedResponse object.
     *
     */
    SecretPropertiesPagedResponse() = default;

    /**
     * @brief Each #Azure::Security::KeyVault::Secrets::SecretProperties represent a Secret in the
     * Key Vault.
     *
     */
    std::vector<SecretProperties> Items;
  };

  /**
   * @brief Define a single page containing the deleted keys from the Key Vault.
   *
   */
  class DeletedSecretPagedResponse final
      : public Azure::Core::PagedResponse<DeletedSecretPagedResponse> {
  private:
    friend class SecretClient;
    friend class Azure::Core::PagedResponse<DeletedSecretPagedResponse>;

    std::shared_ptr<SecretClient> m_secretClient;
    void OnNextPage(const Azure::Core::Context& context);

    DeletedSecretPagedResponse(
        DeletedSecretPagedResponse&& deletedKeyProperties,
        std::unique_ptr<Azure::Core::Http::RawResponse> rawResponse,
        std::shared_ptr<SecretClient> secretClient)
        : PagedResponse(std::move(deletedKeyProperties)), m_secretClient(secretClient),
          Items(std::move(deletedKeyProperties.Items))
    {
      RawResponse = std::move(rawResponse);
    }

  public:
    /**
     * @brief Construct a new Deleted Key Single Page object
     *
     */
    DeletedSecretPagedResponse() = default;

    /**
     * @brief Each #Azure::Security::KeyVault::Secrets::DeletedSecret represent a deleted secret in
     * the Key Vault.
     *
     */
    std::vector<DeletedSecret> Items;
  };
}}}} // namespace Azure::Security::KeyVault::Secrets
