feat: add mnemonic generation from password

tonsdk_ng/crypto/__init__.py

@@ -1,14 +1,20 @@
 from ._keystore import generate_keystore_key, generate_new_keystore
-from ._mnemonic import mnemonic_is_valid, mnemonic_new, mnemonic_to_wallet_key
+from ._mnemonic import (
+ mnemonic_from_password,
+ mnemonic_is_valid,
+ mnemonic_new,
+ mnemonic_to_wallet_key,
+)
 from ._utils import private_key_to_public_key, verify_sign
 
 __all__ = [
+ "generate_key_pair",
+ "generate_keystore_key",
+ "generate_new_keystore",
+ "mnemonic_from_password",
+ "mnemonic_is_valid",
  "mnemonic_new",
  "mnemonic_to_wallet_key",
- "mnemonic_is_valid",
- "generate_new_keystore",
- "generate_keystore_key",
  "private_key_to_public_key",
  "verify_sign",
- "generate_key_pair",
 ]

tonsdk_ng/crypto/_mnemonic.py

@@ -1,5 +1,6 @@
 import hashlib
 import hmac
+import random
 
 from nacl.bindings import crypto_sign_seed_keypair
 
@@ -14,44 +15,35 @@ def mnemonic_is_valid(mnemo_words: list[str]) -> bool:
  )
 
 
-def mnemonic_to_entropy(mnemo_words: list[str], password: str | None = None):
- # TODO: implement password
+def mnemonic_to_entropy(mnemo_words: list[str]):
  sign = hmac.new(
  (" ".join(mnemo_words)).encode("utf-8"), bytes(0), hashlib.sha512
  ).digest()
  return sign
 
 
-def mnemonic_to_seed(
- mnemo_words: list[str], seed: str, password: str | None = None
-):
- entropy = mnemonic_to_entropy(mnemo_words, password)
+def mnemonic_to_seed(mnemo_words: list[str], seed: str):
+ entropy = mnemonic_to_entropy(mnemo_words)
  return hashlib.pbkdf2_hmac("sha512", entropy, seed, PBKDF_ITERATIONS)
 
 
-def mnemonic_to_private_key(
- mnemo_words: list[str], password: str | None = None
-) -> tuple[bytes, bytes]:
+def mnemonic_to_private_key(mnemo_words: list[str]) -> tuple[bytes, bytes]:
  """
  :rtype: (bytes(public_key), bytes(secret_key))
  """
- seed = mnemonic_to_seed(mnemo_words, b"TON default seed", password)
+ seed = mnemonic_to_seed(mnemo_words, b"TON default seed")
  return crypto_sign_seed_keypair(seed[:32])
 
 
-def mnemonic_to_wallet_key(
- mnemo_words: list[str], password: str | None = None
-) -> tuple[bytes, bytes]:
+def mnemonic_to_wallet_key(mnemo_words: list[str]) -> tuple[bytes, bytes]:
  """
  :rtype: (bytes(public_key), bytes(secret_key))
  """
- _, priv_k = mnemonic_to_private_key(mnemo_words, password)
+ _, priv_k = mnemonic_to_private_key(mnemo_words)
  return crypto_sign_seed_keypair(priv_k[:32])
 
 
-def mnemonic_new(
- words_count: int = 24, password: str | None = None
-) -> list[str]:
+def mnemonic_new(words_count: int = 24) -> list[str]:
  while True:
  mnemo_arr = []
 
@@ -63,5 +55,18 @@ def mnemonic_new(
  continue
 
  break
+ return mnemo_arr
+
+
+def mnemonic_from_password(password: str, words_count: int = 24) -> list[str]:
+ rnd = random.Random(password)
+
+ while True:
+ mnemo_arr = [rnd.choice(english) for _ in range(words_count)]
+
+ if not is_basic_seed(mnemonic_to_entropy(mnemo_arr)):
+ continue
+
+ break
 
  return mnemo_arr