/*
 * 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 <folly/dynamic.h>
#include <react/renderer/core/ReactPrimitives.h>
#include <react/renderer/uimanager/UIManager.h>
#include <react/renderer/uimanager/UIManagerAnimationBackend.h>
#include <functional>
#include <vector>
#include "AnimatedProps.h"
#include "AnimatedPropsBuilder.h"

namespace facebook::react {

struct AnimationMutation {
  Tag tag;
  const ShadowNodeFamily *family;
  AnimatedProps props;
};

using AnimationMutations = std::vector<AnimationMutation>;

class AnimationBackend : public UIManagerAnimationBackend {
 public:
  using Callback = std::function<AnimationMutations(float)>;
  using StartOnRenderCallback = std::function<void(std::function<void()> &&, bool /* isAsync */)>;
  using StopOnRenderCallback = std::function<void(bool /* isAsync */)>;
  using DirectManipulationCallback = std::function<void(Tag, const folly::dynamic &)>;
  using FabricCommitCallback = std::function<void(std::unordered_map<Tag, folly::dynamic> &)>;

  std::vector<Callback> callbacks;
  const StartOnRenderCallback startOnRenderCallback_;
  const StopOnRenderCallback stopOnRenderCallback_;
  const DirectManipulationCallback directManipulationCallback_;
  const FabricCommitCallback fabricCommitCallback_;
  UIManager *uiManager_;

  AnimationBackend(
      StartOnRenderCallback &&startOnRenderCallback,
      StopOnRenderCallback &&stopOnRenderCallback,
      DirectManipulationCallback &&directManipulationCallback,
      FabricCommitCallback &&fabricCommitCallback,
      UIManager *uiManager);
  void commitUpdatesWithFamilies(
      const std::unordered_set<const ShadowNodeFamily *> &families,
      std::unordered_map<Tag, AnimatedProps> &updates);
  void synchronouslyUpdateProps(const std::unordered_map<Tag, AnimatedProps> &updates);

  void onAnimationFrame(double timestamp) override;
  void start(const Callback &callback, bool isAsync);
  void stop(bool isAsync) override;
};
} // namespace facebook::react