From 4159c3121675ad7497af5358244cdd9b593ebe71 Mon Sep 17 00:00:00 2001
From: Felix Lange <fjl@twurst.com>
Date: Wed, 14 May 2025 09:59:15 +0200
Subject: [PATCH] internal/era/eradl: add DownloadEpochRange

---
 internal/era/eradl/eradl.go | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/internal/era/eradl/eradl.go b/internal/era/eradl/eradl.go
index 1dbd30c254..30bd2bc0d5 100644
--- a/internal/era/eradl/eradl.go
+++ b/internal/era/eradl/eradl.go
@@ -86,21 +86,25 @@ func (l *Loader) DownloadAll(destDir string) error {
 
 // DownloadBlockRange fetches the era1 files for the given block range.
 func (l *Loader) DownloadBlockRange(start, end uint64, destDir string) error {
-	startEpoch := int(start / uint64(era.MaxEra1Size))
-	endEpoch := int(end / uint64(era.MaxEra1Size))
+	startEpoch := start / uint64(era.MaxEra1Size)
+	endEpoch := end / uint64(era.MaxEra1Size)
+	return l.DownloadEpochRange(startEpoch, endEpoch, destDir)
+}
+
+// DownloadEpochRange fetches the era1 files in the given epoch range.
+func (l *Loader) DownloadEpochRange(start, end uint64, destDir string) error {
 	pat := regexp.MustCompile(regexp.QuoteMeta(l.network) + "-([0-9]+)-[0-9a-f]+\\.era1")
 	for file := range l.csdb.Files() {
 		m := pat.FindStringSubmatch(file)
 		if len(m) == 2 {
 			fileEpoch, _ := strconv.Atoi(m[1])
-			if fileEpoch >= startEpoch && fileEpoch <= endEpoch {
+			if uint64(fileEpoch) >= start && uint64(fileEpoch) <= end {
 				if err := l.download(file, destDir); err != nil {
 					return err
 				}
 			}
 		}
 	}
-	// TODO: detect if blocks are out available era1 range.
 	return nil
 }