From 8fe8eb787c57adf73b6970ddc410155dd8da8156 Mon Sep 17 00:00:00 2001
From: Shaurya Rehan <shaurya.rehan@navi.com>
Date: Fri, 21 Mar 2025 19:33:08 +0530
Subject: [PATCH] NTP-39209 | lite balance consistency logic (#15486)

---
 .../pay/common/usecase/LinkedAccountsUseCase.kt    |  9 ++++++++-
 .../pay/common/usecase/LiteAccountSyncUseCase.kt   | 14 ++++++++++++--
 2 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/android/navi-pay/src/main/kotlin/com/navi/pay/common/usecase/LinkedAccountsUseCase.kt b/android/navi-pay/src/main/kotlin/com/navi/pay/common/usecase/LinkedAccountsUseCase.kt
index a0e353698c..8a5baee4d1 100644
--- a/android/navi-pay/src/main/kotlin/com/navi/pay/common/usecase/LinkedAccountsUseCase.kt
+++ b/android/navi-pay/src/main/kotlin/com/navi/pay/common/usecase/LinkedAccountsUseCase.kt
@@ -20,6 +20,7 @@ import com.navi.pay.common.bankuptime.model.view.BankUptimeEntity
 import com.navi.pay.common.bankuptime.model.view.BankUptimeStatus
 import com.navi.pay.common.bankuptime.repository.BankUptimeRepository
 import com.navi.pay.common.model.config.NaviPayDefaultAccountSelectionConfig
+import com.navi.pay.common.model.view.PspType
 import com.navi.pay.common.theme.color.NaviPayColor
 import com.navi.pay.management.common.sendmoney.model.view.EligibilityState
 import com.navi.pay.management.lite.util.UpiLiteClHelper
@@ -192,6 +193,12 @@ constructor(
         val accountWithActiveUpiLite =
             updatedLinkedAccounts.singleOrNull { it.hasActiveLiteAccount }
         if (accountWithActiveUpiLite != null) {
+            val buid =
+                accountWithActiveUpiLite
+                    .getBuidByPsp(
+                        psp = accountWithActiveUpiLite.activeLiteAccountPsp ?: PspType.JUSPAY_AXIS
+                    )
+                    .orEmpty()
             val updatedAccountWithActiveUpiLite =
                 accountWithActiveUpiLite.copy(
                     accountId =
@@ -204,7 +211,7 @@ constructor(
                     bankIconImageUrl = NAVI_PAY_UPI_LITE_LOGO_URL,
                     maskedAccountNumber = EMPTY,
                     originalMaskedAccountNumber = EMPTY,
-                    balance = upiLiteBalanceUseCase.execute(),
+                    balance = upiLiteCLHelper.getUpiLiteBalance(buid = buid),
                 )
 
             // For UPI Lite, Bank uptime will always be UP
diff --git a/android/navi-pay/src/main/kotlin/com/navi/pay/common/usecase/LiteAccountSyncUseCase.kt b/android/navi-pay/src/main/kotlin/com/navi/pay/common/usecase/LiteAccountSyncUseCase.kt
index 92e6909ce1..3a462e7dab 100644
--- a/android/navi-pay/src/main/kotlin/com/navi/pay/common/usecase/LiteAccountSyncUseCase.kt
+++ b/android/navi-pay/src/main/kotlin/com/navi/pay/common/usecase/LiteAccountSyncUseCase.kt
@@ -240,6 +240,18 @@ constructor(
         liteAccount: UPILiteEntity,
         linkedAccountEntity: LinkedAccountEntity,
     ) {
+        val buid = linkedAccountEntity.getBuidByPsp(liteAccount.psp).orEmpty()
+        val upiLiteBoundStatus = upiLiteClHelper.checkUpiLiteBoundStatus(buid = buid)
+        val liteAccountInfoFromCache =
+            naviCacheRepository.get(KEY_UPI_LITE_ACTIVE_ACCOUNT_INFO)?.value.orEmpty()
+
+        // If CL is bound, load data in cache if empty
+        if (
+            upiLiteBoundStatus?.isLiteInSyncedState() == true && liteAccountInfoFromCache.isBlank()
+        ) {
+            checkIfClIsBoundUpdateBalance(buid = buid)
+        }
+
         if (!isUserOnboarded(psp = liteAccount.psp)) {
             naviPayAnalytics.onDevGenericEvent(
                 event = "Lite_Sync_execution_started",
@@ -252,8 +264,6 @@ constructor(
             )
             return
         }
-        val buid = linkedAccountEntity.getBuidByPsp(liteAccount.psp).orEmpty()
-        val upiLiteBoundStatus = upiLiteClHelper.checkUpiLiteBoundStatus(buid = buid)
         if (upiLiteBoundStatus?.status == null) {
             naviPayAnalytics.onDevGenericEvent(
                 event = "Sync_CL_For_Upi_Lite_Failed",