// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
// IMPORTANT: Before updating this file
// please read react-native-windows repo:
// vnext/Microsoft.ReactNative.Cxx/README.md

#pragma once

#include <TurboModuleRegistry.h>
#include <react/bridging/LongLivedObject.h>
#include "Base/FollyIncludes.h"
#include "winrt/Microsoft.ReactNative.h"

namespace winrt::Microsoft::ReactNative {

class TurboModulesProvider final : public facebook::react::TurboModuleRegistry {
 public: // TurboModuleRegistry implementation
  std::shared_ptr<facebook::react::TurboModule> getModule(
      const std::string &moduleName,
      const std::shared_ptr<facebook::react::CallInvoker> &callInvoker) noexcept override;
  std::vector<std::string> getEagerInitModuleNames() noexcept override;

 public:
  void SetReactContext(const IReactContext &reactContext) noexcept;
  void AddModuleProvider(
      winrt::hstring const &moduleName,
      ReactModuleProvider const &moduleProvider,
      bool overwriteExisting) noexcept;
  std::shared_ptr<facebook::react::LongLivedObjectCollection> const &LongLivedObjectCollection() noexcept;

 private:
  // To keep a list of deferred asynchronous callbacks and promises.
  std::shared_ptr<facebook::react::LongLivedObjectCollection> m_longLivedObjectCollection{
      std::make_shared<facebook::react::LongLivedObjectCollection>()};
  std::unordered_map<std::string, ReactModuleProvider> m_moduleProviders;
  IReactContext m_reactContext;
};

} // namespace winrt::Microsoft::ReactNative