Fix Claude comments

Author: ben-kaufman

app/src/main/java/to/bitkit/repositories/SweepRepo.kt

@@ -100,6 +100,16 @@ class SweepRepo @Inject constructor(
 
     suspend fun getFeeRates(): Result<FeeRates> = coreService.blocktank.getFees()
 
+    suspend fun hasSweepableFunds(): Result<Boolean> = checkSweepableBalances().map { balances ->
+        val hasFunds = balances.totalBalance > 0u
+        if (hasFunds) {
+            Logger.info("Found ${balances.totalBalance} sats to sweep", context = TAG)
+        } else {
+            Logger.debug("No sweepable funds found", context = TAG)
+        }
+        hasFunds
+    }
+
     companion object {
         private const val TAG = "SweepRepo"
     }

app/src/main/java/to/bitkit/viewmodels/AppViewModel.kt

@@ -457,10 +457,8 @@ class AppViewModel @Inject constructor(
 
     fun checkForSweepableFunds() {
         viewModelScope.launch(bgDispatcher) {
-            val hasSweepableFunds = SweepViewModel.checkForSweepableFundsAfterMigration(sweepRepo)
-            if (hasSweepableFunds) {
-                showSheet(Sheet.SweepPrompt)
-            }
+            sweepRepo.hasSweepableFunds()
+                .onSuccess { hasFunds -> if (hasFunds) showSheet(Sheet.SweepPrompt) }
         }
     }
 

app/src/main/java/to/bitkit/viewmodels/SweepViewModel.kt

@@ -40,7 +40,7 @@ class SweepViewModel @Inject constructor(
                 }
             },
             onFailure = { error ->
-                Logger.error("Failed to check sweepable balance", error, TAG)
+                Logger.error("Failed to check sweepable balance", error, context = TAG)
                 _uiState.update { it.copy(checkState = CheckState.Error(error.message ?: "Unknown error")) }
             }
         )
@@ -65,7 +65,7 @@ class SweepViewModel @Inject constructor(
                     _uiState.update { it.copy(destinationAddress = address) }
                 },
                 onFailure = { error ->
-                    Logger.error("Failed to get destination address", error, TAG)
+                    Logger.error("Failed to get destination address", error, context = TAG)
                     val errorMsg = "Failed to get destination address"
                     _uiState.update {
                         it.copy(sweepState = SweepState.Error(errorMsg), errorMessage = errorMsg)
@@ -90,7 +90,7 @@ class SweepViewModel @Inject constructor(
                 }
             },
             onFailure = { error ->
-                Logger.error("Failed to prepare sweep", error, TAG)
+                Logger.error("Failed to prepare sweep", error, context = TAG)
                 _uiState.update {
                     it.copy(
                         sweepState = SweepState.Error(error.message ?: "Unknown error"),
@@ -120,7 +120,7 @@ class SweepViewModel @Inject constructor(
                 }
             },
             onFailure = { error ->
-                Logger.error("Failed to broadcast sweep", error, TAG)
+                Logger.error("Failed to broadcast sweep", error, context = TAG)
                 _uiState.update {
                     it.copy(
                         sweepState = SweepState.Error(error.message ?: "Unknown error"),
@@ -153,7 +153,7 @@ class SweepViewModel @Inject constructor(
                 }
             },
             onFailure = { error ->
-                Logger.error("Failed to load fee estimates", error, TAG)
+                Logger.error("Failed to load fee estimates", error, context = TAG)
                 _uiState.update { it.copy(errorMessage = error.message) }
             }
         )
@@ -165,24 +165,6 @@ class SweepViewModel @Inject constructor(
 
     companion object {
         private const val TAG = "SweepViewModel"
-
-        suspend fun checkForSweepableFundsAfterMigration(sweepRepo: SweepRepo): Boolean {
-            return sweepRepo.checkSweepableBalances().fold(
-                onSuccess = { balances ->
-                    if (balances.totalBalance > 0u) {
-                        Logger.info("Found ${balances.totalBalance} sats to sweep after migration", context = TAG)
-                        true
-                    } else {
-                        Logger.debug("No sweepable funds found after migration", context = TAG)
-                        false
-                    }
-                },
-                onFailure = { error ->
-                    Logger.error("Failed to check sweepable funds after migration", error, TAG)
-                    false
-                }
-            )
-        }
     }
 }