83 строки
1.6 KiB
Plaintext
83 строки
1.6 KiB
Plaintext
|
/**
|
||
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
||
|
*
|
||
|
* This source code is licensed under the MIT license found in the
|
||
|
* LICENSE file in the root directory of this source tree.
|
||
|
*/
|
||
|
|
||
|
#import "RNGenericDelegateSplitter.h"
|
||
|
|
||
|
@implementation RNGenericDelegateSplitter {
|
||
|
NSMutableArray *_delegates;
|
||
|
}
|
||
|
|
||
|
#pragma mark - Public
|
||
|
|
||
|
- (instancetype)initWithDelegateUpdateBlock:(void (^)(id _Nullable delegate))block
|
||
|
{
|
||
|
if (self = [super init]) {
|
||
|
_delegateUpdateBlock = block;
|
||
|
_delegates = [NSMutableArray new];
|
||
|
}
|
||
|
|
||
|
return self;
|
||
|
}
|
||
|
|
||
|
- (void)addDelegate:(id)delegate
|
||
|
{
|
||
|
[_delegates addObject:delegate];
|
||
|
[self _updateDelegate];
|
||
|
}
|
||
|
|
||
|
- (void)removeDelegate:(id)delegate
|
||
|
{
|
||
|
[_delegates removeObject:delegate];
|
||
|
[self _updateDelegate];
|
||
|
}
|
||
|
|
||
|
#pragma mark - Private
|
||
|
|
||
|
- (void)_updateDelegate
|
||
|
{
|
||
|
_delegateUpdateBlock(nil);
|
||
|
if (_delegates.count == 0) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
_delegateUpdateBlock(_delegates.count == 1 ? _delegates.firstObject : self);
|
||
|
}
|
||
|
|
||
|
#pragma mark - Fast Forwarding
|
||
|
|
||
|
- (BOOL)respondsToSelector:(SEL)selector
|
||
|
{
|
||
|
for (id delegate in _delegates) {
|
||
|
if ([delegate respondsToSelector:selector]) {
|
||
|
return YES;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return NO;
|
||
|
}
|
||
|
|
||
|
- (NSMethodSignature *)methodSignatureForSelector:(SEL)selector
|
||
|
{
|
||
|
for (id delegate in _delegates) {
|
||
|
if ([delegate respondsToSelector:selector]) {
|
||
|
return [delegate methodSignatureForSelector:selector];
|
||
|
}
|
||
|
}
|
||
|
return nil;
|
||
|
}
|
||
|
|
||
|
- (void)forwardInvocation:(NSInvocation *)invocation
|
||
|
{
|
||
|
for (id delegate in _delegates) {
|
||
|
if ([delegate respondsToSelector:[invocation selector]]) {
|
||
|
[invocation invokeWithTarget:delegate];
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
@end
|