// Copyright (c) Facebook, Inc. and its affiliates.

// This source code is licensed under the MIT license found in the
// LICENSE file in the root directory of this source tree.

#pragma once

#include <memory>

#include <better/map.h>
#include <react/core/ComponentDescriptor.h>

namespace facebook {
namespace react {

class ComponentDescriptorRegistry;

using SharedComponentDescriptorRegistry =
    std::shared_ptr<const ComponentDescriptorRegistry>;

/*
 * Registry of particular `ComponentDescriptor`s.
 */
class ComponentDescriptorRegistry {
 public:
  void registerComponentDescriptor(
      SharedComponentDescriptor componentDescriptor);

  const ComponentDescriptor &at(ComponentName componentName) const;
  const ComponentDescriptor &at(ComponentHandle componentHandle) const;

  const SharedComponentDescriptor operator[](
      const SharedShadowNode &shadowNode) const;
  const SharedComponentDescriptor operator[](
      const ComponentName &componentName) const;
  SharedShadowNode createNode(
      Tag tag,
      const std::string &viewName,
      Tag rootTag,
      const folly::dynamic &props,
      const SharedEventTarget &eventTarget) const;
  void setFallbackComponentDescriptor(SharedComponentDescriptor descriptor);
  const SharedComponentDescriptor getFallbackComponentDescriptor() const;

 private:
  better::map<ComponentHandle, SharedComponentDescriptor> _registryByHandle;
  better::map<ComponentName, SharedComponentDescriptor> _registryByName;
  SharedComponentDescriptor _fallbackComponentDescriptor;
};

} // namespace react
} // namespace facebook