py-libp2p/peer/id.py

67 lines
1.6 KiB
Python
Raw Normal View History

2018-11-19 00:22:56 +08:00
import base58
import multihash
# MaxInlineKeyLength is the maximum length a key can be for it to be inlined in
# the peer ID.
# * When `len(pubKey.Bytes()) <= MaxInlineKeyLength`, the peer ID is the
# identity multihash hash of the public key.
# * When `len(pubKey.Bytes()) > MaxInlineKeyLength`, the peer ID is the
# sha2-256 multihash of the public key.
MAX_INLINE_KEY_LENGTH = 42
2018-11-19 00:22:56 +08:00
class ID:
def __init__(self, id_str):
self._id_str = id_str
def pretty(self):
return base58.b58encode(self._id_str).decode()
def __str__(self):
pid = self.pretty()
if len(pid) <= 10:
return "<peer.ID %s>" % pid
return "<peer.ID %s*%s>" % (pid[:2], pid[len(pid)-6:])
__repr__ = __str__
def __eq__(self, other):
#pylint: disable=protected-access
return self._id_str == other._id_str
def __hash__(self):
return hash(self._id_str)
def id_b58_encode(peer_id):
"""
return a b58-encoded string
"""
#pylint: disable=protected-access
return base58.b58encode(peer_id._id_str).decode()
def id_b58_decode(peer_id_str):
"""
return a base58-decoded peer ID
"""
return ID(base58.b58decode(peer_id_str))
def id_from_public_key(key):
# export into binary format
key_bin = key.exportKey("DER")
algo = multihash.Func.sha2_256
# TODO: seems identity is not yet supported in pymultihash
# if len(b) <= MAX_INLINE_KEY_LENGTH:
# algo multihash.func.identity
mh_digest = multihash.digest(key_bin, algo)
return ID(mh_digest.encode())
def id_from_private_key(key):
return id_from_public_key(key.publickey())