// Copyright 2016 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_SERVICE_MANAGER_PUBLIC_CPP_INTERFACE_PROVIDER_SPEC_MOJOM_TRAITS_H_
#define SERVICES_SERVICE_MANAGER_PUBLIC_CPP_INTERFACE_PROVIDER_SPEC_MOJOM_TRAITS_H_

#include "services/service_manager/public/cpp/interface_provider_spec.h"
#include "services/service_manager/public/mojom/interface_provider_spec.mojom-shared.h"

namespace mojo {

template <>
struct COMPONENT_EXPORT(SERVICE_MANAGER_MOJOM)
    StructTraits<service_manager::mojom::InterfaceProviderSpecDataView,
                 service_manager::InterfaceProviderSpec> {
  static const std::map<service_manager::Capability,
                        service_manager::InterfaceSet>&
  provides(const service_manager::InterfaceProviderSpec& spec) {
    return spec.provides;
  }
  static const std::map<service_manager::Name, service_manager::CapabilitySet>&
  needs(const service_manager::InterfaceProviderSpec& spec) {
    return spec.needs;
  }
  static bool Read(service_manager::mojom::InterfaceProviderSpecDataView data,
                   service_manager::InterfaceProviderSpec* out) {
    return data.ReadProvides(&out->provides) && data.ReadNeeds(&out->needs);
  }
};

template <>
struct COMPONENT_EXPORT(SERVICE_MANAGER_MOJOM)
    StructTraits<service_manager::mojom::InterfaceSetDataView,
                 service_manager::InterfaceSet> {
  static std::vector<std::string> interfaces(
      const service_manager::InterfaceSet& spec) {
    std::vector<std::string> vec;
    for (const auto& interface_name : spec)
      vec.push_back(interface_name);
    return vec;
  }
  static bool Read(service_manager::mojom::InterfaceSetDataView data,
                   service_manager::InterfaceSet* out) {
    ArrayDataView<StringDataView> interfaces_data_view;
    data.GetInterfacesDataView(&interfaces_data_view);
    for (size_t i = 0; i < interfaces_data_view.size(); ++i) {
      std::string interface_name;
      if (!interfaces_data_view.Read(i, &interface_name))
        return false;
      out->insert(std::move(interface_name));
    }
    return true;
  }
};

template <>
struct COMPONENT_EXPORT(SERVICE_MANAGER_MOJOM)
    StructTraits<service_manager::mojom::CapabilitySetDataView,
                 service_manager::CapabilitySet> {
  static std::vector<std::string> capabilities(
      const service_manager::CapabilitySet& spec) {
    std::vector<std::string> vec;
    for (const auto& capability : spec)
      vec.push_back(capability);
    return vec;
  }
  static bool Read(service_manager::mojom::CapabilitySetDataView data,
                   service_manager::CapabilitySet* out) {
    ArrayDataView<StringDataView> capabilities_data_view;
    data.GetCapabilitiesDataView(&capabilities_data_view);
    for (size_t i = 0; i < capabilities_data_view.size(); ++i) {
      std::string capability;
      if (!capabilities_data_view.Read(i, &capability))
        return false;
      out->insert(std::move(capability));
    }
    return true;
  }
};

}  // namespace mojo

#endif  // SERVICES_SERVICE_MANAGER_PUBLIC_CPP_INTERFACE_PROVIDER_SPEC_MOJOM_TRAITS_H_