#include <reanimated/RuntimeDecorators/UIRuntimeDecorator.h>
#include <reanimated/Tools/ReaJSIUtils.h>

#include <utility>

namespace reanimated {

void UIRuntimeDecorator::decorate(
    jsi::Runtime &uiRuntime,
    const ObtainPropFunction &obtainPropFunction,
    const UpdatePropsFunction &updateProps,
    const MeasureFunction &measure,
    const DispatchCommandFunction &dispatchCommand,
    const GetAnimationTimestampFunction &getAnimationTimestamp,
    const SetGestureStateFunction &setGestureState,
    const ProgressLayoutAnimationFunction &progressLayoutAnimation,
    const EndLayoutAnimationFunction &endLayoutAnimation,
    const MaybeFlushUIUpdatesQueueFunction &maybeFlushUIUpdatesQueue) {

  jsi_utils::installJsiFunction(uiRuntime, "_updateProps", updateProps);
  jsi_utils::installJsiFunction(uiRuntime, "_dispatchCommand", dispatchCommand);
  jsi_utils::installJsiFunction(uiRuntime, "_measure", measure);
  jsi_utils::installJsiFunction(uiRuntime, "_getAnimationTimestamp", getAnimationTimestamp);
  jsi_utils::installJsiFunction(uiRuntime, "_notifyAboutProgress", progressLayoutAnimation);
  jsi_utils::installJsiFunction(uiRuntime, "_notifyAboutEnd", endLayoutAnimation);
  jsi_utils::installJsiFunction(uiRuntime, "_setGestureState", setGestureState);
  jsi_utils::installJsiFunction(uiRuntime, "_obtainProp", obtainPropFunction);

  subscribeForMicrotasksFinalization(uiRuntime, maybeFlushUIUpdatesQueue);
}

void UIRuntimeDecorator::subscribeForMicrotasksFinalization(
    jsi::Runtime &uiRuntime,
    const MaybeFlushUIUpdatesQueueFunction &maybeFlushUIUpdatesQueue) {
  auto maybeMicrotaskQueueFinalizers = uiRuntime.global().getProperty(uiRuntime, "_microtaskQueueFinalizers");

  if (maybeMicrotaskQueueFinalizers.isUndefined()) {
    throw std::runtime_error(
        "[Reanimated] Expected microtaskQueueFinalizers to be defined. "
        "Perhaps Worklets failed to initialize the UI Runtime?");
  }

  const auto microtaskQueueFinalizers = maybeMicrotaskQueueFinalizers.asObject(uiRuntime).asArray(uiRuntime);

  microtaskQueueFinalizers.getPropertyAsFunction(uiRuntime, "push")
      .callWithThis(
          uiRuntime,
          microtaskQueueFinalizers,
          jsi::Function::createFromHostFunction(
              uiRuntime,
              jsi::PropNameID::forAscii(uiRuntime, "_maybeFlushUIUpdatesQueue"),
              0,
              jsi_utils::createHostFunction(maybeFlushUIUpdatesQueue)));
}

} // namespace reanimated