From 7d2faa4a5b09a1ec3f4e87552dacb8716ac235bb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BF=97=E5=AE=87?= <hello@evanlinjin.me>
Date: Tue, 3 Sep 2024 15:28:27 +0800
Subject: [PATCH] feat(core): add `TxUpdate::map_anchors`

---
 crates/core/src/tx_update.rs | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/crates/core/src/tx_update.rs b/crates/core/src/tx_update.rs
index a1ff75de7..29d6c2530 100644
--- a/crates/core/src/tx_update.rs
+++ b/crates/core/src/tx_update.rs
@@ -33,6 +33,23 @@ impl<A> Default for TxUpdate<A> {
 }
 
 impl<A: Ord> TxUpdate<A> {
+    /// Transforms the [`TxUpdate`] to have `anchors` (`A`) of another type (`A2`).
+    ///
+    /// This takes in a closure with signature `FnMut(A) -> A2` which is called for each anchor to
+    /// transform it.
+    pub fn map_anchors<A2: Ord, F: FnMut(A) -> A2>(self, mut map: F) -> TxUpdate<A2> {
+        TxUpdate {
+            txs: self.txs,
+            txouts: self.txouts,
+            anchors: self
+                .anchors
+                .into_iter()
+                .map(|(a, txid)| (map(a), txid))
+                .collect(),
+            seen_ats: self.seen_ats,
+        }
+    }
+
     /// Extend this update with `other`.
     pub fn extend(&mut self, other: TxUpdate<A>) {
         self.txs.extend(other.txs);