aljaz/python-nostr
1
0
Fork 0
mirror of https://github.com/aljazceru/python-nostr.git synced 2025-12-20 15:54:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

add bech32 key encoding

Browse Source
This commit is contained in:
jeffthibault
2022-10-26 19:37:29 -04:00
parent 751c1a3c7d
commit 629949db6a
2 changed files with 154 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
nostr/key.py
View File

@@ -4,6 +4,7 @@ from cffi import FFI
from secp256k1 import PrivateKey, PublicKey
from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
from cryptography.hazmat.primitives import padding
from . import bech32
def generate_private_key() -> str:
private_key = PrivateKey()
@@ -23,6 +24,22 @@ def get_key_pair() -> tuple:
public_key = private_key.pubkey.serialize().hex()
return (private_key.serialize(), public_key[2:])
def bech32_encode_private_key(private_key: str) -> str:
converted_bits = bech32.convertbits(bytes.fromhex(private_key), 8, 5)
return bech32.bech32_encode("nsec", converted_bits, bech32.Encoding.BECH32)
def bech32_decode_private_key(private_key_bech32: str) -> str:
data = bech32.bech32_decode(private_key_bech32)[1]
return bytes(bech32.convertbits(data, 5, 8, False)).hex()
def bech32_encode_public_key(public_key: str) -> str:
converted_bits = bech32.convertbits(bytes.fromhex(public_key), 8, 5)
return bech32.bech32_encode("npub", converted_bits, bech32.Encoding.BECH32)
def bech32_decode_public_key(public_key_bech32: str) -> str:
data = bech32.bech32_decode(public_key_bech32)[1]
return bytes(bech32.convertbits(data, 5, 8, False)).hex()
def tweak_add_private_key(private_key: str, scalar: bytes) -> str:
sk = PrivateKey(bytes.fromhex(private_key))
tweaked_secret = sk.tweak_add(scalar)