From 4d41c11cb8ac9c7c5f48c1dc1411433507101c44 Mon Sep 17 00:00:00 2001
From: Tony Giorgio <tonygiorgio@protonmail.com>
Date: Mon, 5 Feb 2024 17:46:41 -0600
Subject: [PATCH] Swap label on fedimint sweep payment

---
 mutiny-core/src/lib.rs | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/mutiny-core/src/lib.rs b/mutiny-core/src/lib.rs
index 31a5b8dab..fdf2ccdbf 100644
--- a/mutiny-core/src/lib.rs
+++ b/mutiny-core/src/lib.rs
@@ -1212,9 +1212,12 @@ impl<S: MutinyStorage> MutinyWallet<S> {
                 .node_manager
                 .create_invoice(amt, labels.clone())
                 .await?;
+
+            let bolt_11 = inv.bolt11.expect("create inv had one job");
             let pay_res = fedimint_client
-                .pay_invoice(inv.bolt11.expect("create inv had one job"), labels.clone())
+                .pay_invoice(bolt_11.clone(), labels.clone())
                 .await?;
+            self.storage.set_invoice_labels(bolt_11, labels)?;
             let total_fees_paid = pay_res.fees_paid.unwrap_or(0) + fee;
 
             return Ok(FedimintSweepResult {
@@ -1262,9 +1265,11 @@ impl<S: MutinyStorage> MutinyWallet<S> {
         };
 
         log_debug!(self.logger, "attempting payment from fedimint client");
+        let bolt_11 = inv_to_pay.bolt11.expect("create inv had one job");
         let first_invoice_res = fedimint_client
-            .pay_invoice(inv_to_pay.bolt11.expect("create inv had one job"), labels)
+            .pay_invoice(bolt_11.clone(), labels.clone())
             .await?;
+        self.storage.set_invoice_labels(bolt_11, labels)?;
 
         let remaining_balance = fedimint_client.get_balance().await?;
         if remaining_balance > 0 {