// Copyright 2004-present Facebook. All Rights Reserved.
// 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 "EventBeatManager.h"
#include <jsi/jsi.h>
#include <react/events/EventBeat.h>

namespace facebook {
namespace react {

namespace {

 class AsyncEventBeat:
    public EventBeat {

  private:
    EventBeatManager *eventBeatManager_;
    jsi::Runtime *runtime_;
    jni::global_ref<jobject> javaUIManager_;

  public:

    friend class EventBeatManager;

    AsyncEventBeat(EventBeatManager* eventBeatManager, jsi::Runtime *runtime, jni::global_ref<jobject> javaUIManager) {
      eventBeatManager_ = eventBeatManager;
      runtime_ = runtime;
      javaUIManager_ = javaUIManager;
      eventBeatManager->registerEventBeat(this);
    }

    ~AsyncEventBeat() {
      eventBeatManager_->unregisterEventBeat(this);
    }

    void induce() const override {
      beat(*runtime_);
    }

    void request() const override {
      bool alreadyRequested = isRequested_;
      EventBeat::request();
      if (!alreadyRequested) {
        // Notifies java side that an event will be dispatched (e.g. LayoutEvent)
        static auto onRequestEventBeat =
           jni::findClassStatic("com/facebook/react/fabric/FabricUIManager")
              ->getMethod<void()>("onRequestEventBeat");
        onRequestEventBeat(javaUIManager_);
      }
    }
  };

}

}
}