issue-183

Author: reez

LDKNodeMonday/View Model/Home/Send/SendViewModel.swift

@@ -20,6 +20,8 @@ class SendViewModel: ObservableObject {
     @Published var sendError: MondayError?
     var price: Double
     @Published var balances: BalanceDetails
+    @Published var paymentSendState: PaymentSendState = .pending
+    private var sentPaymentId: String?
 
     init(
         lightningClient: LightningNodeClient,
@@ -31,14 +33,39 @@ class SendViewModel: ObservableObject {
         self.sendViewState = sendViewState
         self.price = price
         self.balances = balances
+        NotificationCenter.default.addObserver(
+            self,
+            selector: #selector(handleLdkEvent(_:)),
+            name: .ldkEventReceived,
+            object: nil
+        )
+    }
+
+    @objc private func handleLdkEvent(_ notification: Notification) {
+        guard let event = notification.object as? Event else { return }
+        guard let paymentId = sentPaymentId else { return }
+        switch event {
+        case .paymentSuccessful(_, let paymentHash, _, _):
+            if paymentHash == paymentId {
+                paymentSendState = .succeeded
+            }
+        case .paymentFailed(_, let paymentHash, _):
+            if let hash = paymentHash, hash == paymentId {
+                paymentSendState = .failed
+            }
+        default:
+            break
+        }
     }
 
     func send() async throws {
         do {
+            var paymentId: String? = nil
             switch paymentAddress?.type {
             case .bip21:
                 let uriString = paymentAddress?.address ?? ""
-                _ = try await lightningClient.send(uriString)
+                let result = try await lightningClient.send(uriString)
+                if let id = (result as? QrPaymentResult)?.id { paymentId = id }
             case .onchain:
                 let uriString = unifiedQRString(
                     onchainAddress: paymentAddress?.address ?? "",
@@ -47,12 +74,14 @@ class SendViewModel: ObservableObject {
                     bolt11: nil,
                     bolt12: nil
                 )
-                _ = try await lightningClient.send(uriString)
+                let result = try await lightningClient.send(uriString)
+                if let id = (result as? QrPaymentResult)?.id { paymentId = id }
             case .bolt11:
-                _ = try await lightningClient.sendBolt11Payment(
+                let result = try await lightningClient.sendBolt11Payment(
                     Bolt11Invoice(paymentAddress?.address ?? ""),
                     nil
                 )
+                paymentId = result.value
             default:
                 debugPrint("Unhandled paymentAddress type")
                 DispatchQueue.main.async {
@@ -62,6 +91,10 @@ class SendViewModel: ObservableObject {
                     )
                 }
             }
+            if let paymentId = paymentId {
+                self.sentPaymentId = paymentId
+                self.paymentSendState = .pending
+            }
         } catch let error as NodeError {
             NotificationCenter.default.post(name: .ldkErrorReceived, object: error)
             let errorString = handleNodeError(error)