From d1502e26b140cdd55c5fd93a79bf988234f0b14d Mon Sep 17 00:00:00 2001
From: ardocrat <ardocrat@gri.mw>
Date: Wed, 25 Jun 2025 11:32:26 +0300
Subject: [PATCH] wallet: cleanup broadcasting delay on repost

---
 src/wallet/wallet.rs | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/wallet/wallet.rs b/src/wallet/wallet.rs
index f43b7812..50f27114 100644
--- a/src/wallet/wallet.rs
+++ b/src/wallet/wallet.rs
@@ -1363,6 +1363,22 @@ async fn handle_task(w: &Wallet, t: WalletTask) {
                 None => &w.get_tx(*id).unwrap(),
                 Some(s) => s
             };
+            
+            // Cleanup broadcasting tx height.
+            let has_data = {
+                let r_data = w.data.read();
+                r_data.is_some()
+            };
+            if has_data {
+                let mut w_data = w.data.write();
+                for tx in w_data.as_mut().unwrap().txs.as_mut().unwrap() {
+                    if tx.data.id == *id {
+                        tx.broadcasting_height = None;
+                        break;
+                    }
+                } 
+            }
+
             match w.post(slate) {
                 Ok(_) => {
                     sync_wallet_data(&w, false);