79023ee190
Reviewed By: bhamodi Differential Revision: D33331422 fbshipit-source-id: 016e8dcc0c0c7f1fc353a348b54fda0d5e2ddc01
119 lines
3.6 KiB
Plaintext
119 lines
3.6 KiB
Plaintext
/*
|
|
* 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.
|
|
*/
|
|
|
|
#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(bringSubviewToFront:),
|
|
@selector(swizzle_bringSubviewToFront:));
|
|
swizzleMethods(
|
|
[self class],
|
|
@selector(sendSubviewToBack:),
|
|
@selector(swizzle_sendSubviewToBack:));
|
|
swizzleMethods(
|
|
[self class],
|
|
@selector(insertSubview:atIndex:),
|
|
@selector(swizzle_insertSubview:atIndex:));
|
|
swizzleMethods(
|
|
[self class],
|
|
@selector(insertSubview:aboveSubview:),
|
|
@selector(swizzle_insertSubview:aboveSubview:));
|
|
swizzleMethods(
|
|
[self class],
|
|
@selector(insertSubview:belowSubview:),
|
|
@selector(swizzle_insertSubview:belowSubview:));
|
|
swizzleMethods(
|
|
[self class],
|
|
@selector(exchangeSubviewAtIndex:withSubviewAtIndex:),
|
|
@selector(swizzle_exchangeSubviewAtIndex:withSubviewAtIndex:));
|
|
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];
|
|
[[SKInvalidation sharedInstance].delegate invalidateNode:self];
|
|
}
|
|
|
|
- (void)swizzle_bringSubviewToFront:(UIView*)subview {
|
|
[self swizzle_bringSubviewToFront:subview];
|
|
[[SKInvalidation sharedInstance].delegate invalidateNode:self];
|
|
}
|
|
|
|
- (void)swizzle_sendSubviewToBack:(UIView*)subview {
|
|
[self swizzle_sendSubviewToBack:subview];
|
|
[[SKInvalidation sharedInstance].delegate invalidateNode:self];
|
|
}
|
|
|
|
- (void)swizzle_insertSubview:(UIView*)subview atIndex:(NSInteger)index {
|
|
[self swizzle_insertSubview:subview atIndex:index];
|
|
[[SKInvalidation sharedInstance].delegate invalidateNode:self];
|
|
}
|
|
|
|
- (void)swizzle_insertSubview:(UIView*)subview
|
|
aboveSubview:(UIView*)siblingSubview {
|
|
[self swizzle_insertSubview:subview aboveSubview:siblingSubview];
|
|
[[SKInvalidation sharedInstance].delegate invalidateNode:self];
|
|
}
|
|
|
|
- (void)swizzle_insertSubview:(UIView*)subview
|
|
belowSubview:(UIView*)siblingSubview {
|
|
[self swizzle_insertSubview:subview belowSubview:siblingSubview];
|
|
[[SKInvalidation sharedInstance].delegate invalidateNode:self];
|
|
}
|
|
|
|
- (void)swizzle_exchangeSubviewAtIndex:(NSInteger)index1
|
|
withSubviewAtIndex:(NSInteger)index2 {
|
|
[self swizzle_exchangeSubviewAtIndex:index1 withSubviewAtIndex:index2];
|
|
[[SKInvalidation sharedInstance].delegate invalidateNode:self];
|
|
}
|
|
|
|
- (void)swizzle_removeFromSuperview {
|
|
UIView* oldSuperview = self.superview;
|
|
// Be careful that we always call the swizzled implementation
|
|
// before any early returns or mischief below!
|
|
[self swizzle_removeFromSuperview];
|
|
|
|
if (oldSuperview) {
|
|
[[SKInvalidation sharedInstance].delegate invalidateNode:oldSuperview];
|
|
}
|
|
}
|
|
|
|
@end
|
|
|
|
#endif
|