@@ -47,7 +47,7 @@ use crate::io::{
4747} ;
4848use crate :: logger:: { log_error, LdkLogger , Logger } ;
4949use crate :: peer_store:: PeerStore ;
50- use crate :: types:: { Broadcaster , DynStore , KeysManager , MnemonicWordCount , Sweeper } ;
50+ use crate :: types:: { Broadcaster , DynStore , KeysManager , Sweeper , WordCount } ;
5151use crate :: wallet:: ser:: { ChangeSetDeserWrapper , ChangeSetSerWrapper } ;
5252use crate :: { Error , EventQueue , NodeMetrics , PaymentDetails } ;
5353
@@ -63,12 +63,9 @@ pub const EXTERNAL_PATHFINDING_SCORES_CACHE_KEY: &str = "external_pathfinding_sc
6363/// [BIP 39]: https://github.com/bitcoin/bips/blob/master/bip-0039.mediawiki
6464/// [`Node`]: crate::Node
6565/// [`Builder::set_entropy_bip39_mnemonic`]: crate::Builder::set_entropy_bip39_mnemonic
66- pub fn generate_entropy_mnemonic ( word_count : Option < MnemonicWordCount > ) -> Mnemonic {
67- let word_count = word_count. unwrap_or ( MnemonicWordCount :: Words24 ) ;
68- let entropy_bytes = word_count. entropy_bytes ( ) ;
69- let mut entropy = vec ! [ 0u8 ; entropy_bytes] ;
70- OsRng . try_fill_bytes ( & mut entropy) . expect ( "Failed to generate entropy" ) ;
71- Mnemonic :: from_entropy ( & entropy) . unwrap ( )
66+ pub fn generate_entropy_mnemonic ( word_count : Option < WordCount > ) -> Mnemonic {
67+ let word_count = word_count. unwrap_or ( WordCount :: Words24 ) . word_count ( ) ;
68+ Mnemonic :: generate ( word_count) . expect ( "Failed to generate mnemonic" )
7269}
7370
7471pub ( crate ) fn read_or_generate_seed_file < L : Deref > (
@@ -638,11 +635,11 @@ mod tests {
638635
639636 // Test with different word counts
640637 let word_counts = [
641- MnemonicWordCount :: Words12 ,
642- MnemonicWordCount :: Words15 ,
643- MnemonicWordCount :: Words18 ,
644- MnemonicWordCount :: Words21 ,
645- MnemonicWordCount :: Words24 ,
638+ WordCount :: Words12 ,
639+ WordCount :: Words15 ,
640+ WordCount :: Words18 ,
641+ WordCount :: Words21 ,
642+ WordCount :: Words24 ,
646643 ] ;
647644
648645 for word_count in word_counts {
@@ -652,11 +649,11 @@ mod tests {
652649
653650 // Verify expected word count
654651 let expected_words = match word_count {
655- MnemonicWordCount :: Words12 => 12 ,
656- MnemonicWordCount :: Words15 => 15 ,
657- MnemonicWordCount :: Words18 => 18 ,
658- MnemonicWordCount :: Words21 => 21 ,
659- MnemonicWordCount :: Words24 => 24 ,
652+ WordCount :: Words12 => 12 ,
653+ WordCount :: Words15 => 15 ,
654+ WordCount :: Words18 => 18 ,
655+ WordCount :: Words21 => 21 ,
656+ WordCount :: Words24 => 24 ,
660657 } ;
661658 assert_eq ! ( mnemonic. word_count( ) , expected_words) ;
662659 }
0 commit comments