// Copyright 2019 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_ISOLATION_KEY_MOJOM_TRAITS_H_
#define SERVICES_NETWORK_PUBLIC_CPP_NETWORK_ISOLATION_KEY_MOJOM_TRAITS_H_

#include <optional>

#include "base/no_destructor.h"
#include "mojo/public/cpp/bindings/struct_traits.h"
#include "net/base/network_isolation_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_isolation_key.mojom-shared.h"

namespace base {
class UnguessableToken;
}  // namespace base

namespace mojo {

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

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

  static const net::SchemefulSite& frame_site(
      const net::NetworkIsolationKey& input) {
    return input
        .GetFrameSiteForSerialization(
            net::NetworkIsolationKey::SerializationPassKey())
        .value();
  }

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

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

  static bool Read(network::mojom::NonEmptyNetworkIsolationKeyDataView data,
                   net::NetworkIsolationKey* out);
};

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

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

  static bool Read(network::mojom::NetworkIsolationKeyDataView data,
                   net::NetworkIsolationKey* out);

  static network::mojom::NetworkIsolationKeyDataView::Tag GetTag(
      const net::NetworkIsolationKey& input);
};

}  // namespace mojo

#endif  // SERVICES_NETWORK_PUBLIC_CPP_NETWORK_ISOLATION_KEY_MOJOM_TRAITS_H_