Improved nid command JID parsing.

1 files changed, 14 insertions, 1 deletions

diff --git a/src/Network/Tox/NodeId.hs b/src/Network/Tox/NodeId.hs
index dab1b14a..f3013142 100644
--- a/src/Network/Tox/NodeId.hs
+++ b/src/Network/Tox/NodeId.hs
@@ -29,6 +29,7 @@ module Network.Tox.NodeId
     , sampleNodeId
     , NoSpam(..)
     , NoSpamId(..)
+    , noSpamIdToHex
     , parseNoSpamId
     , nospam64
     , verifyChecksum) where
@@ -520,6 +521,16 @@ verifyChecksum _ _ = return () -- TODO
 data NoSpamId = NoSpamId NoSpam PublicKey
  deriving (Eq,Ord)
 
+noSpamIdToHex :: NoSpamId -> String
+noSpamIdToHex (NoSpamId nspam pub) = C8.unpack (Base16.encode $ BA.convert pub)
+                                     ++ nospam16 nspam
+
+nospam16 :: NoSpam -> String
+nospam16 (NoSpam w32 Nothing) = n ++ "????"
+    where n = take 8 $ nospam16 (NoSpam w32 (Just 0))
+nospam16 (NoSpam w32 (Just w16)) = C8.unpack $ Base16.encode $ runPut $ do
+    put w32
+    put w16
 
 nospam64 :: NoSpam -> String
 nospam64 (NoSpam w32 Nothing) = n ++ "???"
@@ -549,7 +560,9 @@ parseNoSpamId spec | Text.length spec == 76
 
 parseNoSpamJID :: Text -> Either String NoSpamId
 parseNoSpamJID jid = do
-    (Just u,h,_) <- Right $ splitJID jid
+    (u,h) <- maybe (Left "Invalid JID.") Right
+                $ let (mu,h,_) = splitJID jid
+                  in fmap (, h) mu
     base64 <- case splitAt 43 $ Text.unpack h of
                 (base64,".tox") -> Right base64
                 _               -> Left "Hostname should be 43 base64 digits followed by .tox."