+ Add progress to client session.

1 files changed, 13 insertions, 2 deletions

diff --git a/src/Network/BitTorrent.hs b/src/Network/BitTorrent.hs
index 5fbc5ff6..61185d08 100644
--- a/src/Network/BitTorrent.hs
+++ b/src/Network/BitTorrent.hs
@@ -5,6 +5,7 @@
 --   Stability   :  experimental
 --   Portability :  portable
 --
+{-# LANGUAGE RecordWildCards #-}
 module Network.BitTorrent
        ( module BT
        , module Data.Torrent
@@ -15,8 +16,11 @@ module Network.BitTorrent
        , ClientSession, newClient
        , SwarmSession, newLeacher, newSeeder
        , PeerSession
+       , discover
        ) where
 
+import Data.IORef
+
 import Data.Torrent
 import Network.BitTorrent.Internal
 import Network.BitTorrent.Extension as BT
@@ -24,5 +28,12 @@ import Network.BitTorrent.Peer as BT
 import Network.BitTorrent.Exchange as BT
 import Network.BitTorrent.Tracker as BT
 
---discover :: SwarmSession -> ([PeerAddr] -> IO a) -> IO a
---discover = withTracker
+
+discover :: SwarmSession -> (TSession -> IO a) -> IO a
+discover SwarmSession {..} action = do
+  let conn = TConnection (tAnnounce torrentMeta) (tInfoHash torrentMeta)
+                         (clientPeerID clientSession) port
+  progress <- readIORef (currentProgress clientSession)
+  withTracker progress conn action
+
+port = 10000