/*
 * Copyright (c) Meta Platforms, Inc. and 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 <mutex>
#include <unordered_map>
#include <unordered_set>

#include <fbjni/fbjni.h>
#include <react/fabric/JFabricUIManager.h>
#include <react/renderer/uimanager/primitives.h>

namespace facebook::react {

class MountingTransaction;
struct ShadowView;

class FabricMountingManager final {
 public:
  FabricMountingManager(jni::global_ref<JFabricUIManager::javaobject> &javaUIManager);
  FabricMountingManager(const FabricMountingManager &) = delete;
  ~FabricMountingManager();

  void onSurfaceStart(SurfaceId surfaceId);

  void onSurfaceStop(SurfaceId surfaceId);

  void maybePreallocateShadowNode(const ShadowNode &shadowNode);

  void destroyUnmountedShadowNode(const ShadowNodeFamily &family);

  /*
   * Drains preallocatedViewsQueue_ by calling preallocateShadowView on each
   * item in the queue. Can be called by any thread.
   */
  void drainPreallocateViewsQueue();

  void executeMount(const MountingTransaction &transaction);

  void dispatchCommand(const ShadowView &shadowView, const std::string &commandName, const folly::dynamic &args);

  void sendAccessibilityEvent(const ShadowView &shadowView, const std::string &eventType);

  void setIsJSResponder(const ShadowView &shadowView, bool isJSResponder, bool blockNativeResponder);

  void onAnimationStarted();

  void onAllAnimationsComplete();

  void synchronouslyUpdateViewOnUIThread(Tag viewTag, const folly::dynamic &props);

  void scheduleReactRevisionMerge(SurfaceId surfaceId);

 private:
  bool isOnMainThread();

  jni::global_ref<JFabricUIManager::javaobject> javaUIManager_;

  std::recursive_mutex commitMutex_;

  /*
   * Protects preallocatedViewsQueue_.
   */
  std::mutex preallocateMutex_;

  /*
   * A queue of views to be preallocated on the Java side.
   */
  std::vector<ShadowView> preallocatedViewsQueue_{};

  std::unordered_map<SurfaceId, std::unordered_set<Tag>> allocatedViewRegistry_{};
  std::recursive_mutex allocatedViewsMutex_;

  /*
   * Calls FabricUIManager.preallocateView() on the Java side if view needs to
   * be preallocated.
   */
  void preallocateShadowView(const ShadowView &shadowView);
};

} // namespace facebook::react
