+ add THP

1 files changed, 15 insertions, 4 deletions

diff --git a/tests/client.hs b/tests/client.hs
index 2b626241..bab11b11 100644
--- a/tests/client.hs
+++ b/tests/client.hs
@@ -1,12 +1,19 @@
 module Main (main) where
 
-import Data.Torrent
-import Network.Torrent.THP
 import System.Environment
 import Data.ByteString as B
 import Data.ByteString.Lazy as L
 import Data.BEncode
 
+import Data.Torrent
+import Network.Torrent.THP
+import Network.Torrent.PeerID
+
+showWarn :: TResponse -> IO ()
+showWarn resp
+  | Just msg <- respWarning resp = print msg
+  | otherwise = return ()
+
 main :: IO ()
 main = do
   [path] <- getArgs
@@ -14,6 +21,7 @@ main = do
 
   let Right contents' = decode contents >>= return . L.toStrict . encode
   print (contents' == contents)
+
 --  let (a, b) = showInfos contents
 --  print b
 --  print a
@@ -22,7 +30,10 @@ main = do
   let Right b = decode contents
   let Right t = fromBEncode b
 
+  peerID <- newPeerID
+
+  let req = defaultRequest (tAnnounce t) (tInfoHash t) (peerID)
+  Right resp <- sendRequest req
 
-  let req = defaultRequest (tAnnounce t) (tInfoHash t)
-  resp <- sendRequest req
   print resp
+  showWarn resp