Update datastructures.py

hivemind/p2p/p2p_daemon_bindings/datastructures.py

@@ -17,10 +17,24 @@
 # NOTE: On inlining...
 # See: https://github.com/libp2p/specs/issues/138
 # NOTE: enabling to be interoperable w/ the Go implementation
-ENABLE_INLINING = False
+# ENABLE_INLINING = False
 MAX_INLINE_KEY_LENGTH = 42
 
-IDENTITY_MULTIHASH_CODE = 0x00
+# IDENTITY_MULTIHASH_CODE = 0x00
+
+# if ENABLE_INLINING:
+
+#     class IdentityHash:
+#         def __init__(self) -> None:
+#             self._digest = bytearray()
+
+#         def update(self, input: bytes) -> None:
+#             self._digest += input
+
+#         def digest(self) -> bytes:
+#             return self._digest
+
+#     multihash.FuncReg.register(IDENTITY_MULTIHASH_CODE, "identity", hash_new=IdentityHash)
 
 
 class PeerID:
@@ -82,7 +96,6 @@ def from_identity(cls, data: bytes) -> "PeerID":
 
         [1] https://github.com/libp2p/specs/blob/master/peer-ids/peer-ids.md#peer-ids
         """
-
         key_data = crypto_pb2.PrivateKey.FromString(data).data
         private_key = serialization.load_der_private_key(key_data, password=None)
 
@@ -95,8 +108,11 @@ def from_identity(cls, data: bytes) -> "PeerID":
             data=encoded_public_key,
         ).SerializeToString()
 
-        algo = multihash.Func.sha2_256
-        encoded_digest = multihash.digest(encoded_public_key, algo).encode()
+        #algo = multihash.Func.sha2_256
+        #if ENABLE_INLINING and len(encoded_public_key) <= MAX_INLINE_KEY_LENGTH:
+        #    algo = IDENTITY_MULTIHASH_CODE
+        #encoded_digest = multihash.digest(encoded_public_key, algo).encode()
+        encoded_digest = multihash.encode(hashlib.sha256(encoded_public_key).digest(), multihash.coerce_code("sha2-256"))
         return cls(encoded_digest)