From d8797823bb3cbf91eabad48c400632dcecfec245 Mon Sep 17 00:00:00 2001
From: Joe Crayne <joe@jerkface.net>
Date: Wed, 28 Nov 2018 22:19:33 -0500
Subject: Use getPublicKey/putPublicKey more liberally.

---
 src/Network/Tox/Crypto/Transport.hs | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

(limited to 'src/Network/Tox/Crypto/Transport.hs')

diff --git a/src/Network/Tox/Crypto/Transport.hs b/src/Network/Tox/Crypto/Transport.hs
index 555164f2..1c641584 100644
--- a/src/Network/Tox/Crypto/Transport.hs
+++ b/src/Network/Tox/Crypto/Transport.hs
@@ -170,12 +170,12 @@ instance Sized HandshakeData where
 
 instance Serialize HandshakeData where
     get = HandshakeData <$> get
-                        <*> (id2key <$> get)
+                        <*> getPublicKey
                         <*> (fromJust . digestFromByteString <$> getBytes 64)
                         <*> get
     put (HandshakeData n k h c) = do
         put n
-        put $ key2id k
+        putPublicKey k
         putByteString (convert h)
         put c
 
@@ -724,15 +724,15 @@ instance HasPeerNumber PeerInfo where
 instance Serialize PeerInfo where
     get = do
         w16 <- get
-        ukey <- id2key <$> get
-        dkey <- id2key <$> get
+        ukey <- getPublicKey
+        dkey <- getPublicKey
         w8 <- get :: Get Word8
         PeerInfo w16 ukey dkey <$> getBytes (fromIntegral w8)
 
     put (PeerInfo w16 ukey dkey bs) = do
         put w16
-        put $ key2id ukey
-        put $ key2id dkey
+        putPublicKey ukey
+        putPublicKey dkey
         let sz :: Word8
             sz = case B.length bs of
                     n | n <= 255  -> fromIntegral n
-- 
cgit v1.2.3