amis-rpc-design/node_modules/react-native/React/Base/RCTViewRegistry.m

70 lines
1.5 KiB
Mathematica
Raw Normal View History

2023-10-07 19:42:30 +08:00
/*
* 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.
*/
#import <React/RCTSurfacePresenterStub.h>
#import <React/RCTUIManager.h>
#import "RCTBridge.h"
#import "RCTBridgeModule.h"
@implementation RCTViewRegistry {
RCTBridgelessComponentViewProvider _bridgelessComponentViewProvider;
__weak RCTBridge *_bridge;
}
- (void)setBridge:(RCTBridge *)bridge
{
_bridge = bridge;
}
- (void)setBridgelessComponentViewProvider:(RCTBridgelessComponentViewProvider)bridgelessComponentViewProvider
{
_bridgelessComponentViewProvider = bridgelessComponentViewProvider;
}
- (UIView *)viewForReactTag:(NSNumber *)reactTag
{
UIView *view = nil;
RCTBridge *bridge = _bridge;
if (bridge) {
view = [bridge.uiManager viewForReactTag:reactTag];
}
if (view == nil && _bridgelessComponentViewProvider) {
view = _bridgelessComponentViewProvider(reactTag);
}
return view;
}
- (void)addUIBlock:(RCTViewRegistryUIBlock)block
{
if (!block) {
return;
}
__weak __typeof(self) weakSelf = self;
if (_bridge) {
[_bridge.uiManager addUIBlock:^(RCTUIManager *uiManager, NSDictionary<NSNumber *, UIView *> *viewRegistry) {
__typeof(self) strongSelf = weakSelf;
if (strongSelf) {
block(strongSelf);
}
}];
} else {
RCTExecuteOnMainQueue(^{
__typeof(self) strongSelf = weakSelf;
if (strongSelf) {
block(strongSelf);
}
});
}
}
@end