Reviewed By: passy Differential Revision: D17863711 fbshipit-source-id: 259dc77826fb803ff1b88c88529d7f679d3b74d8
59 lines
1.5 KiB
Plaintext
59 lines
1.5 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.
|
|
*/
|
|
|
|
#if FB_SONARKIT_ENABLED
|
|
|
|
#import <UIKit/UIKit.h>
|
|
#import <objc/runtime.h>
|
|
|
|
#import "SKInvalidation.h"
|
|
#import "SKSwizzle.h"
|
|
#import "UIView+SKInvalidation.h"
|
|
|
|
FB_LINKABLE(UIView_SKInvalidation)
|
|
@implementation UIView (SKInvalidation)
|
|
|
|
+ (void)enableInvalidation {
|
|
static dispatch_once_t onceToken;
|
|
dispatch_once(&onceToken, ^{
|
|
swizzleMethods([self class], @selector(setHidden:), @selector(swizzle_setHidden:));
|
|
swizzleMethods([self class], @selector(addSubview:), @selector(swizzle_addSubview:));
|
|
swizzleMethods([self class], @selector(removeFromSuperview), @selector(swizzle_removeFromSuperview));
|
|
});
|
|
}
|
|
|
|
- (void)swizzle_setHidden:(BOOL)hidden {
|
|
[self swizzle_setHidden: hidden];
|
|
|
|
id<SKInvalidationDelegate> delegate = [SKInvalidation sharedInstance].delegate;
|
|
if (delegate != nil) {
|
|
[delegate invalidateNode: self.superview];
|
|
}
|
|
}
|
|
|
|
- (void)swizzle_addSubview:(UIView *)view {
|
|
[self swizzle_addSubview: view];
|
|
|
|
id<SKInvalidationDelegate> delegate = [SKInvalidation sharedInstance].delegate;
|
|
if (delegate != nil) {
|
|
[delegate invalidateNode: view];
|
|
}
|
|
}
|
|
|
|
- (void)swizzle_removeFromSuperview {
|
|
id<SKInvalidationDelegate> delegate = [SKInvalidation sharedInstance].delegate;
|
|
if (delegate != nil && self.superview != nil) {
|
|
[delegate invalidateNode: self.superview];
|
|
}
|
|
|
|
[self swizzle_removeFromSuperview];
|
|
}
|
|
|
|
@end
|
|
|
|
#endif
|