fix(auth): remove listener on dispose and only call setState when mou…

…nted (#419)
firebase · Nov 13, 2024 · 91c982b · 91c982b
1 parent 93bfe60
commit 91c982b
Showing 1 changed file with 21 additions and 5 deletions.
diff --git a/packages/firebase_ui_auth/lib/src/screens/profile_screen.dart b/packages/firebase_ui_auth/lib/src/screens/profile_screen.dart
@@ -351,11 +351,27 @@ class _EmailVerificationBadge extends StatefulWidget {
 }
 
 class _EmailVerificationBadgeState extends State<_EmailVerificationBadge> {
-  late final service = EmailVerificationController(widget.auth)
-    ..addListener(() {
-      setState(() {});
-    })
-    ..reload();
+  late final EmailVerificationController service;
+  late final VoidCallback listener;
+
+  @override
+  void initState() {
+    super.initState();
+    listener = () {
+      if (mounted) {
+        setState(() {});
+      }
+    };
+    service = EmailVerificationController(widget.auth)
+      ..addListener(listener)
+      ..reload();
+  }
+
+  @override
+  void dispose() {
+    service.removeListener(listener);
+    super.dispose();
+  }
 
   EmailVerificationState get state => service.state;