From b158d437299c017d4ff7874f62e551eba1219420 Mon Sep 17 00:00:00 2001
From: Marcin Czenko <marcin.czenko@pm.me>
Date: Tue, 17 Jun 2025 02:27:48 +0200
Subject: [PATCH] adds missing mapFuture operation to SafeAsyncIter

---
 codex/utils/safeasynciter.nim | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/codex/utils/safeasynciter.nim b/codex/utils/safeasynciter.nim
index d582fec3..3db0c70f 100644
--- a/codex/utils/safeasynciter.nim
+++ b/codex/utils/safeasynciter.nim
@@ -34,6 +34,7 @@ import ./iter
 ## - next - to get the next item from the async iterator
 ## - items - to iterate over the async iterator
 ## - pairs - to iterate over the async iterator and return the index of each item
+## - mapFuture - to convert a (raising) Future[T] to a (raising) Future[U] using a function fn: auto -> Future[U] - we use auto to handle both raising and non-raising futures
 ## - mapAsync - to convert a regular sync iterator (Iter) to an async iter (SafeAsyncIter)
 ## - map - to convert one async iterator (SafeAsyncIter) to another async iter (SafeAsyncIter)
 ## - mapFilter - to convert one async iterator (SafeAsyncIter) to another async iter (SafeAsyncIter) and apply filtering at the same time
@@ -150,6 +151,12 @@ iterator pairs*[T](self: SafeAsyncIter[T]): auto {.inline.} =
     yield (i, self.next())
     inc(i)
 
+proc mapFuture*[T, U](
+    fut: auto, fn: SafeFunction[T, U]
+): Future[U] {.async: (raises: [CancelledError]).} =
+  let t = await fut
+  await fn(t)
+
 proc mapAsync*[T, U](
     iter: Iter[T], fn: SafeFunction[T, ?!U], finishOnErr: bool = true
 ): SafeAsyncIter[U] =