From 7b846740c14e57852e7f2c55a6160151253531fe Mon Sep 17 00:00:00 2001
From: Neeraj Gupta <254676+ua741@users.noreply.github.com>
Date: Thu, 16 Feb 2023 15:42:20 +0530
Subject: [PATCH] Referesh UI after apply code

---
 lib/ui/growth/referral_screen.dart | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/lib/ui/growth/referral_screen.dart b/lib/ui/growth/referral_screen.dart
index 2d632decf..7b2f693a6 100644
--- a/lib/ui/growth/referral_screen.dart
+++ b/lib/ui/growth/referral_screen.dart
@@ -28,6 +28,11 @@ class ReferralScreen extends StatefulWidget {
 
 class _ReferralScreenState extends State<ReferralScreen> {
   bool canApplyCode = true;
+  void _safeUIUpdate() {
+    if (mounted) {
+      setState(() => {});
+    }
+  }
 
   @override
   void initState() {
@@ -71,6 +76,7 @@ class _ReferralScreenState extends State<ReferralScreen> {
                         return ReferralWidget(
                           snapshot.data!,
                           UserService.instance.getCachedUserDetails()!,
+                          _safeUIUpdate,
                         );
                       } else if (snapshot.hasError) {
                         return const Center(
@@ -101,8 +107,14 @@ class _ReferralScreenState extends State<ReferralScreen> {
 class ReferralWidget extends StatelessWidget {
   final ReferralView referralView;
   final UserDetails userDetails;
+  final Function notifyParent;
 
-  const ReferralWidget(this.referralView, this.userDetails, {super.key});
+  const ReferralWidget(
+    this.referralView,
+    this.userDetails,
+    this.notifyParent, {
+    super.key,
+  });
 
   @override
   Widget build(BuildContext context) {
@@ -233,10 +245,11 @@ class ReferralWidget extends StatelessWidget {
                 alignCaptionedTextToLeft: true,
                 isBottomBorderRadiusRemoved: true,
                 onTap: () async {
-                  routeToPage(
+                  await routeToPage(
                     context,
                     const ApplyCodeScreen(),
                   );
+                  notifyParent();
                 },
               )
             : const SizedBox.shrink(),