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 { 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) } }