From 933e7d37aeafac38eae806fb4556d59803a03270 Mon Sep 17 00:00:00 2001
From: Sam Truzjan <pxqr.sta@gmail.com>
Date: Fri, 14 Feb 2014 22:28:15 +0400
Subject: Move piece manager to separate module

---
 src/Network/BitTorrent/Client/Handle.hs | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

(limited to 'src/Network/BitTorrent/Client')

diff --git a/src/Network/BitTorrent/Client/Handle.hs b/src/Network/BitTorrent/Client/Handle.hs
index 7aaaf5aa..f539d53a 100644
--- a/src/Network/BitTorrent/Client/Handle.hs
+++ b/src/Network/BitTorrent/Client/Handle.hs
@@ -22,6 +22,7 @@ import Control.Applicative
 import Control.Concurrent
 import Control.Monad
 import Control.Monad.Trans
+import Data.List as L
 import Data.HashMap.Strict as HM
 
 import Data.Torrent
@@ -77,7 +78,8 @@ openTorrent rootPath t @ Torrent {..} = do
   allocHandle ih $ do
     c @ Client {..} <- getClient
     tses <- liftIO $ Tracker.newSession ih (trackerList t)
-    eses <- liftIO $ Exchange.newSession (externalAddr c) rootPath tInfoDict
+    eses <- liftIO $ Exchange.newSession clientLogger (externalAddr c) rootPath
+                                         tInfoDict
     return $ Handle ih (idPrivate tInfoDict) tses eses
 
 -- | Use 'nullMagnet' to open handle from 'InfoHash'.
@@ -109,9 +111,11 @@ start Handle {..} = do
   liftIO $ Tracker.notify trackerManager trackers Tracker.Started
   unless private $ do
     liftDHT $ DHT.insert topic undefined
-  peers <- liftIO $ askPeers trackerManager trackers
-  forM_ peers $ \ peer -> do
-    liftIO $ Exchange.insert peer exchange
+  liftIO $ do
+    peers <- askPeers trackerManager trackers
+    print $ "got: " ++ show (L.length peers) ++ " peers"
+    forM_ peers $ \ peer -> do
+      Exchange.insert peer exchange
 
 -- | Stop downloading this torrent.
 pause :: Handle -> BitTorrent ()
-- 
cgit v1.2.3