// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef SERVICES_NETWORK_PUBLIC_CPP_NETWORK_ANONYMIZATION_KEY_MOJOM_TRAITS_H_
#define SERVICES_NETWORK_PUBLIC_CPP_NETWORK_ANONYMIZATION_KEY_MOJOM_TRAITS_H_

#include <optional>

#include "base/no_destructor.h"
#include "mojo/public/cpp/bindings/struct_traits.h"
#include "net/base/network_anonymization_key.h"
#include "net/base/network_isolation_partition.h"
#include "net/base/schemeful_site.h"
#include "services/network/public/cpp/network_isolation_partition_mojom_traits.h"
#include "services/network/public/cpp/schemeful_site_mojom_traits.h"
#include "services/network/public/mojom/network_anonymization_key.mojom-shared.h"

namespace base {
class UnguessableToken;
}  // namespace base

namespace mojo {

template <>
struct COMPONENT_EXPORT(NETWORK_CPP_BASE)
    StructTraits<network::mojom::EmptyNetworkAnonymizationKeyDataView,
                 net::NetworkAnonymizationKey> {
  static bool Read(network::mojom::EmptyNetworkAnonymizationKeyDataView data,
                   net::NetworkAnonymizationKey* out);
};

template <>
struct COMPONENT_EXPORT(NETWORK_CPP_BASE)
    StructTraits<network::mojom::NonEmptyNetworkAnonymizationKeyDataView,
                 net::NetworkAnonymizationKey> {
  static const net::SchemefulSite& top_frame_site(
      const net::NetworkAnonymizationKey& input) {
    return input.GetTopFrameSite().value();
  }

  static bool is_cross_site(const net::NetworkAnonymizationKey& input) {
    return input.IsCrossSite();
  }

  static const std::optional<base::UnguessableToken>& nonce(
      const net::NetworkAnonymizationKey& input) {
    return input.GetNonce();
  }

  static net::NetworkIsolationPartition network_isolation_partition(
      const net::NetworkAnonymizationKey& input) {
    return input.network_isolation_partition();
  }

  static bool Read(network::mojom::NonEmptyNetworkAnonymizationKeyDataView data,
                   net::NetworkAnonymizationKey* out);
};

template <>
struct COMPONENT_EXPORT(NETWORK_CPP_BASE)
    UnionTraits<network::mojom::NetworkAnonymizationKeyDataView,
                net::NetworkAnonymizationKey> {
  static const net::NetworkAnonymizationKey& empty(
      const net::NetworkAnonymizationKey& input) {
    return input;
  }

  static const net::NetworkAnonymizationKey& non_empty(
      const net::NetworkAnonymizationKey& input) {
    return input;
  }

  static bool Read(network::mojom::NetworkAnonymizationKeyDataView data,
                   net::NetworkAnonymizationKey* out);

  static network::mojom::NetworkAnonymizationKeyDataView::Tag GetTag(
      const net::NetworkAnonymizationKey& input);
};

}  // namespace mojo

#endif  // SERVICES_NETWORK_PUBLIC_CPP_NETWORK_ANONYMIZATION_KEY_MOJOM_TRAITS_H_