[Electric-Coin-Company#1475] Adopt transaction data requests

- SpendsFromAddress endHeight is an optional UInt32 [Electric-Coin-Company#1475] Adopt transaction data requests - ignoring the getSubtreeRoots error for now so I can test tex send [Electric-Coin-Company#1475] Adopt transaction data requests - ignoring error of getSubtreeRoots [Electric-Coin-Company#1475] Adopt transaction data requests - reverting the DemoAppConfig
LukasKorba · Aug 20, 2024 · f25c714 · f25c714
1 parent e5e1fdf
commit f25c714
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 4 deletions.
diff --git a/Sources/ZcashLightClientKit/Block/Actions/UpdateSubtreeRootsAction.swift b/Sources/ZcashLightClientKit/Block/Actions/UpdateSubtreeRootsAction.swift
@@ -30,7 +30,7 @@ extension UpdateSubtreeRootsAction: Action {
 
         logger.debug("Attempt to get subtree roots, this may fail because lightwalletd may not support Spend before Sync.")
         let stream = service.getSubtreeRoots(request)
-        
+
         var saplingRoots: [SubtreeRoot] = []
 
         do {
@@ -39,6 +39,8 @@ extension UpdateSubtreeRootsAction: Action {
             }
         } catch ZcashError.serviceSubtreeRootsStreamFailed(LightWalletServiceError.timeOut) {
             throw ZcashError.serviceSubtreeRootsStreamFailed(LightWalletServiceError.timeOut)
+        } catch {
+            await context.update(state: .updateChainTip)
         }
 
         logger.debug("Sapling tree has \(saplingRoots.count) subtrees")

diff --git a/Sources/ZcashLightClientKit/Block/Enhance/BlockEnhancer.swift b/Sources/ZcashLightClientKit/Block/Enhance/BlockEnhancer.swift
@@ -113,9 +113,15 @@ extension BlockEnhancerImpl: BlockEnhancer {
                             }
 
                         case .spendsFromAddress(let sfa):
+                            guard let blockRangeEnd = sfa.blockRangeEnd else {
+                                logger.error("spendsFromAddress \(sfa) is missing blockRangeEnd, ignoring the request.")
+                                continue
+                            }
+
                             var filter = TransparentAddressBlockFilter()
                             filter.address = sfa.address
-                            filter.range = BlockRange(startHeight: Int(sfa.blockRangeStart), endHeight: Int(sfa.blockRangeEnd - 1))
+                            filter.range = BlockRange(startHeight: Int(sfa.blockRangeStart), endHeight: Int(blockRangeEnd - 1))
+
                             let stream = service.getTaddressTxids(filter)
 
                             for try await rawTransaction in stream {

diff --git a/Sources/ZcashLightClientKit/Model/TransactionDataRequest.swift b/Sources/ZcashLightClientKit/Model/TransactionDataRequest.swift
@@ -10,7 +10,7 @@ import Foundation
 struct SpendsFromAddress: Equatable {
     let address: String
     let blockRangeStart: UInt32
-    let blockRangeEnd: Int64
+    let blockRangeEnd: UInt32?
 }
 
 enum TransactionDataRequest: Equatable {

diff --git a/Sources/ZcashLightClientKit/Rust/ZcashRustBackend.swift b/Sources/ZcashLightClientKit/Rust/ZcashRustBackend.swift
@@ -871,11 +871,14 @@ struct ZcashRustBackend: ZcashRustBackendWelding {
             } else if tDataRequestPtr.tag == 1 {
                 tDataRequest = TransactionDataRequest.enhancement(FfiTxId(tuple: tDataRequestPtr.enhancement).array)
             } else if tDataRequestPtr.tag == 2, let address = String(validatingUTF8: tDataRequestPtr.spends_from_address.address) {
+                let end = tDataRequestPtr.spends_from_address.block_range_end
+                let blockRangeEnd: UInt32? = end > UInt32.max || end == -1 ? nil : UInt32(end)
+
                 tDataRequest = TransactionDataRequest.spendsFromAddress(
                     SpendsFromAddress(
                         address: address,
                         blockRangeStart: tDataRequestPtr.spends_from_address.block_range_start,
-                        blockRangeEnd: tDataRequestPtr.spends_from_address.block_range_end
+                        blockRangeEnd: blockRangeEnd
                     )
                 )
             }