/*
 * 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 <react/renderer/consistency/ShadowTreeRevisionConsistencyManager.h>

namespace facebook::react {

/**
 * This is a RAII class that locks the shadow tree revisions during its
 * lifetime.
 *
 * @example
 *   {
 *     ScopedShadowTreeRevisionLock lock(consistencyManager);
 *     runJavaScriptTask(); // During this execution, the lock will be active.
 *   }
 */
class ScopedShadowTreeRevisionLock {
 public:
  explicit ScopedShadowTreeRevisionLock(
      ShadowTreeRevisionConsistencyManager* consistencyManager) noexcept
      : consistencyManager_(consistencyManager) {
    if (consistencyManager_ != nullptr) {
      consistencyManager_->lockRevisions();
    }
  }

  // Non-movable
  ScopedShadowTreeRevisionLock(const ScopedShadowTreeRevisionLock&) = delete;
  ScopedShadowTreeRevisionLock(ScopedShadowTreeRevisionLock&&) = delete;

  // Non-copyable
  ScopedShadowTreeRevisionLock& operator=(const ScopedShadowTreeRevisionLock&) =
      delete;
  ScopedShadowTreeRevisionLock& operator=(ScopedShadowTreeRevisionLock&&) =
      delete;

  ~ScopedShadowTreeRevisionLock() noexcept {
    if (consistencyManager_ != nullptr) {
      consistencyManager_->unlockRevisions();
    }
  }

 private:
  ShadowTreeRevisionConsistencyManager* consistencyManager_;
};

} // namespace facebook::react