Fix authStateChanged after app is reloaded (Fixed invertase#1677)

hayanmind · Nov 20, 2018 · 65db5cf · 65db5cf
1 parent 9486cdc
commit 65db5cf
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/ios/RNFirebase/auth/RNFirebaseAuth.h b/ios/RNFirebase/auth/RNFirebaseAuth.h
@@ -6,6 +6,7 @@
 #import <Firebase.h>
 #import <React/RCTBridgeModule.h>
 #import <React/RCTEventEmitter.h>
+#import <React/RCTInvalidating.h>
 
 @interface RNFirebaseAuth : RCTEventEmitter <RCTBridgeModule> {};
 @end

diff --git a/ios/RNFirebase/auth/RNFirebaseAuth.m b/ios/RNFirebase/auth/RNFirebaseAuth.m
@@ -62,6 +62,22 @@ - (void)dealloc {
   }
 }
 
+- (void)invalidate {
+  // dealloc sometimes is not called when app is reloaded.
+
+  for(NSString* key in authStateHandlers) {
+      FIRApp *firApp = [RNFirebaseUtil getApp:key];
+      [[FIRAuth authWithApp:firApp] removeAuthStateDidChangeListener:[authStateHandlers valueForKey:key]];
+      [authStateHandlers removeObjectForKey:key];
+  }
+
+  for(NSString* key in idTokenHandlers) {
+      FIRApp *firApp = [RNFirebaseUtil getApp:key];
+      [[FIRAuth authWithApp:firApp] removeIDTokenDidChangeListener:[idTokenHandlers valueForKey:key]];
+      [idTokenHandlers removeObjectForKey:key];
+  }
+}
+
 /**
  * addAuthStateListener
  *