add bip32 word list iterator

aspectron · Sep 20, 2023 · c1a40b1 · c1a40b1
1 parent 59e965f
commit c1a40b1
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/wallet/bip32/src/mnemonic/language.rs b/wallet/bip32/src/mnemonic/language.rs
@@ -54,6 +54,29 @@ impl WordList {
     pub fn get_word(&self, bits: Bits11) -> &'static str {
         self.inner[bits.bits() as usize]
     }
+
+    pub fn iter(&self) -> WordListIterator<'_> {
+        WordListIterator { wordlist: self, index: 0 }
+    }
+}
+
+pub struct WordListIterator<'a> {
+    wordlist: &'a WordList,
+    index: usize,
+}
+
+impl Iterator for WordListIterator<'_> {
+    type Item = &'static str;
+
+    fn next(&mut self) -> Option<Self::Item> {
+        if self.index < self.wordlist.inner.len() {
+            let word = self.wordlist.inner[self.index];
+            self.index += 1;
+            Some(word)
+        } else {
+            None
+        }
+    }
 }
 
 // TODO(tarcieri): use `const fn` instead of `Lazy`