Fix ETA with endBlock (#394)

enviodev · Dec 23, 2024 · 10b9dcd · 10b9dcd
1 parent 12f5cce
commit 10b9dcd
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/codegenerator/cli/templates/static/codegen/src/ink/components/SyncETA.res b/codegenerator/cli/templates/static/codegen/src/ink/components/SyncETA.res
@@ -11,12 +11,16 @@ let isIndexerFullySynced = (chains: array<ChainData.chainData>) => {
 }
 
 let getTotalRemainingBlocks = (chains: array<ChainData.chainData>) => {
-  chains->Array.reduce(0, (accum, {progress, currentBlockHeight, latestFetchedBlockNumber}) => {
+  chains->Array.reduce(0, (accum, {progress, currentBlockHeight, latestFetchedBlockNumber, endBlock}) => {
+    let finalBlock = switch endBlock {
+      | Some(endBlock) => endBlock
+      | None => currentBlockHeight
+    }
     switch progress {
     | Syncing({latestProcessedBlock})
     | Synced({latestProcessedBlock}) =>
-      currentBlockHeight - latestProcessedBlock + accum
-    | SearchingForEvents => currentBlockHeight - latestFetchedBlockNumber + accum
+      finalBlock - latestProcessedBlock + accum
+    | SearchingForEvents => finalBlock - latestFetchedBlockNumber + accum
     }
   })
 }