feat: add conformsToProtocol forwarding to MPAppDelegateProxy (#144)

mParticle · May 16, 2022 · fc3dce4 · fc3dce4
1 parent 0244451
commit fc3dce4
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/mParticle-Apple-SDK/AppNotifications/MPAppDelegateProxy.m b/mParticle-Apple-SDK/AppNotifications/MPAppDelegateProxy.m
@@ -13,9 +13,18 @@ @interface MPAppDelegateProxy() {
 
 @implementation MPAppDelegateProxy
 
+static MPAppDelegateProxy* _defaultProxy = nil;
++ (MPAppDelegateProxy *)defaultProxy {
+    return _defaultProxy;
+}
+
 - (instancetype)initWithOriginalAppDelegate:(id)originalAppDelegate {
     _originalAppDelegate = originalAppDelegate;
     originalAppDelegateSelector = @selector(originalAppDelegate);
+    static dispatch_once_t pred = 0;
+    dispatch_once(&pred, ^{
+        _defaultProxy = self;
+    });
     return self;
 }
 
@@ -87,6 +96,12 @@ - (BOOL)respondsToSelector:(SEL)aSelector {
     return respondsToSelector;
 }
 
++ (BOOL)conformsToProtocol:(Protocol *)aProtocol {
+    id original = [[MPAppDelegateProxy defaultProxy] originalAppDelegate];
+    id surrogate = [[MPAppDelegateProxy defaultProxy] surrogateAppDelegate];
+    return [original conformsToProtocol:aProtocol] || [surrogate conformsToProtocol:aProtocol];
+}
+
 #pragma mark Public accessors
 - (MPSurrogateAppDelegate *)surrogateAppDelegate {
     if (_surrogateAppDelegate) {