fix: refresh paykit on channel events

Author: ben-kaufman

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

@@ -460,6 +460,7 @@ class AppViewModel @Inject constructor(
     private suspend fun handleChannelReady(event: Event.ChannelReady) {
         transferRepo.syncTransferStates()
         walletRepo.syncBalances()
+        refreshPublicPaykitEndpointsIfEnabled()
         notifyChannelReady(event)
     }
 
@@ -476,6 +477,7 @@ class AppViewModel @Inject constructor(
         }
         transferRepo.syncTransferStates()
         walletRepo.syncBalances()
+        refreshPublicPaykitEndpointsIfEnabled()
     }
 
     private suspend fun createTransferForCounterpartyClose(channelId: String, isForceClose: Boolean) {

app/src/test/java/to/bitkit/viewmodels/AppViewModelSendFlowTest.kt

@@ -12,6 +12,7 @@ import org.junit.Test
 import org.lightningdevkit.ldknode.Event
 import org.mockito.kotlin.any
 import org.mockito.kotlin.anyOrNull
+import org.mockito.kotlin.clearInvocations
 import org.mockito.kotlin.mock
 import org.mockito.kotlin.verify
 import org.mockito.kotlin.whenever
@@ -83,6 +84,8 @@ class AppViewModelSendFlowTest : BaseUnitTest() {
     private val formatMoneyValue = mock<FormatMoneyValue>()
 
     private val balanceState = MutableStateFlow(BalanceState())
+    private val settingsData = MutableStateFlow(SettingsData())
+    private val walletState = MutableStateFlow(WalletState())
     private val nodeEvents = MutableSharedFlow<Event>()
 
     private val timedSheetManager = mock<TimedSheetManager>()
@@ -100,9 +103,9 @@ class AppViewModelSendFlowTest : BaseUnitTest() {
         whenever(lightningRepo.lightningState).thenReturn(MutableStateFlow(LightningState()))
         whenever(lightningRepo.nodeEvents).thenReturn(nodeEvents)
         whenever(walletRepo.balanceState).thenReturn(balanceState)
-        whenever(walletRepo.walletState).thenReturn(MutableStateFlow(WalletState()))
+        whenever(walletRepo.walletState).thenReturn(walletState)
         whenever(walletRepo.walletExists()).thenReturn(true)
-        whenever(settingsStore.data).thenReturn(flowOf(SettingsData()))
+        whenever(settingsStore.data).thenReturn(settingsData)
         whenever(cacheStore.data).thenReturn(flowOf(AppCacheData()))
         whenever(transferRepo.activeTransfers).thenReturn(flowOf(emptyList()))
         whenever(timedSheetManager.currentSheet).thenReturn(MutableStateFlow(null))
@@ -310,6 +313,44 @@ class AppViewModelSendFlowTest : BaseUnitTest() {
         assertNull(pendingContactPaymentContext(paymentHash))
     }
 
+    @Test
+    fun `channel ready refreshes public Paykit endpoints when sharing enabled`() = test {
+        enablePublicPaykitSharing()
+        advanceUntilIdle()
+        clearInvocations(publicPaykitRepo)
+
+        nodeEvents.emit(
+            Event.ChannelReady(
+                channelId = "testChannelId",
+                userChannelId = "testUserChannelId",
+                counterpartyNodeId = null,
+                fundingTxo = null,
+            ),
+        )
+        advanceUntilIdle()
+
+        verify(publicPaykitRepo).syncCurrentPublishedEndpoints()
+    }
+
+    @Test
+    fun `channel closed refreshes public Paykit endpoints when sharing enabled`() = test {
+        enablePublicPaykitSharing()
+        advanceUntilIdle()
+        clearInvocations(publicPaykitRepo)
+
+        nodeEvents.emit(
+            Event.ChannelClosed(
+                channelId = "testChannelId",
+                userChannelId = "testUserChannelId",
+                counterpartyNodeId = null,
+                reason = null,
+            ),
+        )
+        advanceUntilIdle()
+
+        verify(publicPaykitRepo).syncCurrentPublishedEndpoints()
+    }
+
     @Test
     fun `amount change clears confirmedWarnings`() = test {
         setUnifiedState(amount = 1000u)
@@ -363,6 +404,12 @@ class AppViewModelSendFlowTest : BaseUnitTest() {
         assertEquals(0L, sut.sendUiState.value.lastLightningFee)
     }
 
+    private suspend fun enablePublicPaykitSharing() {
+        settingsData.value = SettingsData(sharesPublicPaykitEndpoints = true)
+        walletState.value = WalletState(onchainAddress = "bc1qtest")
+        whenever { publicPaykitRepo.syncCurrentPublishedEndpoints() }.thenReturn(Result.success(Unit))
+    }
+
     @Suppress("UNCHECKED_CAST")
     private fun setPendingContactPaymentContext(paymentHash: String, publicKey: String) {
         val field = AppViewModel::class.java.getDeclaredField("pendingContactPaymentContexts")