mirror of
https://github.com/aljazceru/nutshell.git
synced 2025-12-20 10:34:20 +01:00
55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
import base64
|
|
import hashlib
|
|
from typing import Dict, List
|
|
|
|
from cashu.core.secp import PrivateKey, PublicKey
|
|
from cashu.core.settings import settings
|
|
|
|
# entropy = bytes([random.getrandbits(8) for i in range(16)])
|
|
# mnemonic = bip39.mnemonic_from_bytes(entropy)
|
|
# seed = bip39.mnemonic_to_seed(mnemonic)
|
|
# root = bip32.HDKey.from_seed(seed, version=NETWORKS["main"]["xprv"])
|
|
|
|
# bip44_xprv = root.derive("m/44h/1h/0h")
|
|
# bip44_xpub = bip44_xprv.to_public()
|
|
|
|
|
|
def derive_keys(master_key: str, derivation_path: str = ""):
|
|
"""
|
|
Deterministic derivation of keys for 2^n values.
|
|
TODO: Implement BIP32.
|
|
"""
|
|
return {
|
|
2
|
|
** i: PrivateKey(
|
|
hashlib.sha256((master_key + derivation_path + str(i)).encode("utf-8"))
|
|
.hexdigest()
|
|
.encode("utf-8")[:32],
|
|
raw=True,
|
|
)
|
|
for i in range(settings.max_order)
|
|
}
|
|
|
|
|
|
def derive_pubkey(master_key: str):
|
|
return PrivateKey(
|
|
hashlib.sha256((master_key).encode("utf-8")).hexdigest().encode("utf-8")[:32],
|
|
raw=True,
|
|
).pubkey
|
|
|
|
|
|
def derive_pubkeys(keys: Dict[int, PrivateKey]):
|
|
return {
|
|
amt: keys[amt].pubkey for amt in [2**i for i in range(settings.max_order)]
|
|
}
|
|
|
|
|
|
def derive_keyset_id(keys: Dict[int, PublicKey]):
|
|
"""Deterministic derivation keyset_id from set of public keys."""
|
|
# sort public keys by amount
|
|
sorted_keys = dict(sorted(keys.items()))
|
|
pubkeys_concat = "".join([p.serialize().hex() for _, p in sorted_keys.items()])
|
|
return base64.b64encode(
|
|
hashlib.sha256((pubkeys_concat).encode("utf-8")).digest()
|
|
).decode()[:12]
|