// 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.

#include "services/network/public/cpp/first_party_sets_mojom_traits.h"

#include <algorithm>
#include <utility>

#include "base/containers/flat_map.h"
#include "base/containers/map_util.h"
#include "base/notreached.h"
#include "base/version.h"
#include "mojo/public/cpp/base/version_mojom_traits.h"
#include "mojo/public/cpp/bindings/enum_traits.h"
#include "mojo/public/cpp/bindings/struct_traits.h"
#include "net/base/schemeful_site.h"
#include "net/first_party_sets/first_party_set_entry.h"
#include "net/first_party_sets/first_party_set_entry_override.h"
#include "net/first_party_sets/first_party_set_metadata.h"
#include "net/first_party_sets/first_party_sets_cache_filter.h"
#include "net/first_party_sets/first_party_sets_context_config.h"
#include "net/first_party_sets/global_first_party_sets.h"
#include "services/network/public/cpp/schemeful_site_mojom_traits.h"
#include "services/network/public/mojom/first_party_sets.mojom-shared.h"

namespace mojo {

net::SiteType EnumTraits<network::mojom::SiteType, net::SiteType>::FromMojom(
    network::mojom::SiteType site_type) {
  switch (site_type) {
    case network::mojom::SiteType::kPrimary:
      return net::SiteType::kPrimary;
    case network::mojom::SiteType::kAssociated:
      return net::SiteType::kAssociated;
    case network::mojom::SiteType::kService:
      return net::SiteType::kService;
  }
  NOTREACHED();
}

network::mojom::SiteType
EnumTraits<network::mojom::SiteType, net::SiteType>::ToMojom(
    net::SiteType site_type) {
  switch (site_type) {
    case net::SiteType::kPrimary:
      return network::mojom::SiteType::kPrimary;
    case net::SiteType::kAssociated:
      return network::mojom::SiteType::kAssociated;
    case net::SiteType::kService:
      return network::mojom::SiteType::kService;
  }
  NOTREACHED();
}

bool StructTraits<network::mojom::FirstPartySetEntryDataView,
                  net::FirstPartySetEntry>::
    Read(network::mojom::FirstPartySetEntryDataView entry,
         net::FirstPartySetEntry* out) {
  net::SchemefulSite primary;
  if (!entry.ReadPrimary(&primary)) {
    return false;
  }

  net::SiteType site_type;
  if (!entry.ReadSiteType(&site_type)) {
    return false;
  }

  *out = net::FirstPartySetEntry(primary, site_type);
  return true;
}

bool StructTraits<network::mojom::FirstPartySetMetadataDataView,
                  net::FirstPartySetMetadata>::
    Read(network::mojom::FirstPartySetMetadataDataView metadata,
         net::FirstPartySetMetadata* out_metadata) {
  std::optional<net::FirstPartySetEntry> frame_entry;
  if (!metadata.ReadFrameEntry(&frame_entry)) {
    return false;
  }

  std::optional<net::FirstPartySetEntry> top_frame_entry;
  if (!metadata.ReadTopFrameEntry(&top_frame_entry)) {
    return false;
  }

  *out_metadata = net::FirstPartySetMetadata(std::move(frame_entry),
                                             std::move(top_frame_entry));

  return true;
}

bool StructTraits<network::mojom::GlobalFirstPartySetsDataView,
                  net::GlobalFirstPartySets>::
    Read(network::mojom::GlobalFirstPartySetsDataView sets,
         net::GlobalFirstPartySets* out_sets) {
  base::Version public_sets_version;
  if (!sets.ReadPublicSetsVersion(&public_sets_version)) {
    return false;
  }

  net::FirstPartySetsContextConfig public_config;
  if (!sets.ReadPublicConfig(&public_config)) {
    return false;
  }

  net::FirstPartySetsContextConfig manual_config;
  if (!sets.ReadManualConfig(&manual_config)) {
    return false;
  }

  *out_sets = net::GlobalFirstPartySets(std::move(public_sets_version),
                                        std::move(public_config),
                                        std::move(manual_config));

  return true;
}

bool StructTraits<network::mojom::FirstPartySetEntryOverrideDataView,
                  net::FirstPartySetEntryOverride>::
    Read(network::mojom::FirstPartySetEntryOverrideDataView override,
         net::FirstPartySetEntryOverride* out) {
  std::optional<net::FirstPartySetEntry> entry;
  if (!override.ReadEntry(&entry)) {
    return false;
  }

  if (entry.has_value()) {
    *out = net::FirstPartySetEntryOverride(entry.value());
  } else {
    *out = net::FirstPartySetEntryOverride();
  }
  return true;
}

bool StructTraits<network::mojom::FirstPartySetsContextConfigDataView,
                  net::FirstPartySetsContextConfig>::
    Read(network::mojom::FirstPartySetsContextConfigDataView view,
         net::FirstPartySetsContextConfig* out_config) {
  base::flat_map<net::SchemefulSite, net::FirstPartySetEntryOverride>
      customizations;
  base::flat_map<net::SchemefulSite, net::SchemefulSite> aliases;
  if (!view.ReadCustomizations(&customizations) ||
      !view.ReadAliases(&aliases)) {
    return false;
  }

  std::optional<net::FirstPartySetsContextConfig> config =
      net::FirstPartySetsContextConfig::Create(std::move(customizations),
                                               std::move(aliases));
  if (!config) {
    return false;
  }
  *out_config = std::move(config).value();
  return true;
}

bool StructTraits<network::mojom::FirstPartySetsCacheFilterDataView,
                  net::FirstPartySetsCacheFilter>::
    Read(network::mojom::FirstPartySetsCacheFilterDataView cache_filter,
         net::FirstPartySetsCacheFilter* out_cache_filter) {
  base::flat_map<net::SchemefulSite, int64_t> filter;
  if (!cache_filter.ReadFilter(&filter)) {
    return false;
  }

  *out_cache_filter = net::FirstPartySetsCacheFilter(
      std::move(filter), cache_filter.browser_run_id());

  return true;
}

}  // namespace mojo