Expose UIManager from Scheduler (facebook#33545)

Summary: Pull Request resolved: facebook#33545 Differential Revision: D35314058 fbshipit-source-id: fe7a0feb37da3a691735c4315c64c7bc11c0a89d
ShikaSD · Apr 1, 2022 · 36e50f2 · 36e50f2
1 parent be11895
commit 36e50f2
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 0 deletions.
diff --git a/React/Fabric/RCTScheduler.h b/React/Fabric/RCTScheduler.h
@@ -16,6 +16,7 @@
 #import <react/renderer/mounting/MountingCoordinator.h>
 #import <react/renderer/scheduler/SchedulerToolbox.h>
 #import <react/renderer/scheduler/SurfaceHandler.h>
+#import <react/renderer/uimanager/UIManager.h>
 #import <react/utils/ContextContainer.h>
 
 NS_ASSUME_NONNULL_BEGIN
@@ -48,6 +49,7 @@ NS_ASSUME_NONNULL_BEGIN
 @interface RCTScheduler : NSObject
 
 @property (atomic, weak, nullable) id<RCTSchedulerDelegate> delegate;
+@property (readonly, nullable) facebook::react::UIManager const *uiManager;
 
 - (instancetype)initWithToolbox:(facebook::react::SchedulerToolbox)toolbox;
 

diff --git a/React/Fabric/RCTScheduler.mm b/React/Fabric/RCTScheduler.mm
@@ -203,4 +203,9 @@ - (void)removeEventListener:(std::shared_ptr<EventListener>)listener
   return _scheduler->removeEventListener(listener);
 }
 
+- (facebook::react::UIManager const *)uiManager
+{
+  return _scheduler->getUIManager().get();
+}
+
 @end
diff --git a/ReactCommon/react/renderer/scheduler/Scheduler.cpp b/ReactCommon/react/renderer/scheduler/Scheduler.cpp
@@ -375,6 +375,10 @@ ContextContainer::Shared Scheduler::getContextContainer() const {
   return contextContainer_;
 }
 
+std::shared_ptr<UIManager> Scheduler::getUIManager() const {
+  return uiManager_;
+}
+
 void Scheduler::addEventListener(std::shared_ptr<EventListener const> listener) {
   eventListeners_.addListener(listener);
 }

diff --git a/ReactCommon/react/renderer/scheduler/Scheduler.h b/ReactCommon/react/renderer/scheduler/Scheduler.h
@@ -108,6 +108,9 @@ class Scheduler final : public UIManagerDelegate {
 #pragma mark - ContextContainer
   ContextContainer::Shared getContextContainer() const;
 
+#pragma mark - UIManager
+  std::shared_ptr<UIManager> getUIManager() const;
+
 #pragma mark - Event listeners
   void addEventListener(std::shared_ptr<EventListener const> listener);
   void removeEventListener(std::shared_ptr<EventListener const> listener);