From 8845942f32a3673d48894089e1d4e501431023f5 Mon Sep 17 00:00:00 2001
From: Sosthene <contact@sosthene.net>
Date: Mon, 25 Nov 2024 22:26:50 +0100
Subject: [PATCH] confirm_secret_for_address returns the previous secret

---
 src/secrets.rs | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/secrets.rs b/src/secrets.rs
index 30113cb..acb478c 100644
--- a/src/secrets.rs
+++ b/src/secrets.rs
@@ -92,15 +92,16 @@ impl SecretsStore {
         self.unconfirmed_secrets.push(new_secret);
     }
 
-    pub fn confirm_secret_for_address(&mut self, secret: AnkSharedSecretHash, address: SilentPaymentAddress) {
+    /// Returns the previous secret for this address, if any
+    pub fn confirm_secret_for_address(&mut self, secret: AnkSharedSecretHash, address: SilentPaymentAddress) -> Option<AnkSharedSecretHash> {
         if let Some(pos) = self.unconfirmed_secrets.iter()
             .position(|s| *s == secret)
         {
-            self.shared_secrets.insert(address, self.unconfirmed_secrets.swap_remove(pos));
+            self.shared_secrets.insert(address, self.unconfirmed_secrets.swap_remove(pos))
         } else {
             // We didn't know about that secret, just add it
             // TODO if we already had a secret for this address we just replace it for now
-            self.shared_secrets.insert(address, secret);
+            self.shared_secrets.insert(address, secret)
         }
     }