23 lines
663 B
Python
23 lines
663 B
Python
|
from onionrtypes import OnionAddressString
|
||
|
|
from typing import Iterable
|
||
|
|
from .extracted25519 import extract_ed25519_from_onion_address
|
||
|
|
|
||
|
|
|
||
|
|
def identify_neighbors(
|
||
|
|
address: OnionAddressString,
|
||
|
|
peers: Iterable[OnionAddressString],
|
||
|
|
closest_n: int) -> OnionAddressString:
|
||
|
|
|
||
|
|
address = extract_ed25519_from_onion_address(address)
|
||
|
|
address_int = int.from_bytes(address, "big")
|
||
|
|
|
||
|
|
def _calc_closeness(y):
|
||
|
|
return abs(address_int - int.from_bytes(y, "big"))
|
||
|
|
|
||
|
|
|
||
|
|
peer_ed_keys = list(map(extract_ed25519_from_onion_address, peers))
|
||
|
|
differences = list(map(_calc_closeness, peer_ed_keys))
|
||
|
|
|
||
|
|
return sorted(differences)[:closest_n]
|
||
|
|
|