3 files changed, 48 insertions, 22 deletions

diff --git a/src/Network/BitTorrent.hs b/src/Network/BitTorrent.hs
index c37129cb..8f0f42ce 100644
--- a/src/Network/BitTorrent.hs
+++ b/src/Network/BitTorrent.hs
@@ -7,35 +7,50 @@
 --
 {-# LANGUAGE RecordWildCards #-}
 module Network.BitTorrent
-       ( module BT
-       , module Data.Torrent
+       (
+         module Data.Torrent
 
-         -- * Tracker
+         -- * Session
+       , ClientSession
+       , newClient
 
-         -- * P2P
-       , ClientSession, newClient
-       , SwarmSession, newLeacher, newSeeder
-       , PeerSession
+       , SwarmSession
+       , newLeacher, newSeeder
+
+         -- * Discovery
        , discover
+
+         -- * Peer to Peer
+       , P2P, PeerSession
+            ( connectedPeerAddr, enabledExtensions
+            , peerBitfield, peerSessionStatus
+            )
+
+       , awaitEvent, signalEvent
        ) where
 
+import Control.Monad
 import Data.IORef
 
 import Data.Torrent
 import Network.BitTorrent.Internal
-import Network.BitTorrent.Extension as BT
-import Network.BitTorrent.Peer as BT
-import Network.BitTorrent.Exchange as BT
-import Network.BitTorrent.Tracker as BT
+import Network.BitTorrent.Exchange
+import Network.BitTorrent.Tracker
+
+
 
 -- discover should hide tracker and DHT communication under the hood
 -- thus we can obtain unified interface
 
-discover :: SwarmSession -> (TSession -> IO a) -> IO a
-discover SwarmSession {..} action = do
+discover :: SwarmSession -> P2P () -> IO ()
+discover swarm @ SwarmSession {..} action = do
   let conn = TConnection (tAnnounce torrentMeta) (tInfoHash torrentMeta)
                          (clientPeerID clientSession) port
   progress <- readIORef (currentProgress clientSession)
-  withTracker progress conn action
+  withTracker progress conn $ \tses -> do
+    forever $ do
+      addr <- getPeerAddr tses
+      withPeer swarm addr action
+
 
 port = 10000
diff --git a/src/Network/BitTorrent/Exchange.hs b/src/Network/BitTorrent/Exchange.hs
index dd1f9d0b..2173cf8b 100644
--- a/src/Network/BitTorrent/Exchange.hs
+++ b/src/Network/BitTorrent/Exchange.hs
@@ -45,6 +45,7 @@ import Data.Torrent
 data Event = Available Bitfield
            | Want
            | Block
+             deriving Show
 
 {-----------------------------------------------------------------------
     Peer wire
diff --git a/src/Network/BitTorrent/Internal.hs b/src/Network/BitTorrent/Internal.hs
index e231fb2c..39e10ce2 100644
--- a/src/Network/BitTorrent/Internal.hs
+++ b/src/Network/BitTorrent/Internal.hs
@@ -85,11 +85,16 @@ instance Ord ClientSession where
   compare = comparing clientPeerID
 
 newClient :: [Extension] -> IO ClientSession
-newClient exts = ClientSession <$> newPeerID
-                               <*> pure exts
-                               <*> newTVarIO S.empty
-                               <*> Ev.new
-                               <*> newIORef (startProgress 0)
+newClient exts = do
+  mgr <- Ev.new
+  forkIO $ loop mgr
+
+  ClientSession
+    <$> newPeerID
+    <*> pure exts
+    <*> newTVarIO S.empty
+    <*> pure mgr
+    <*> newIORef (startProgress 0)
 
 {-----------------------------------------------------------------------
     Swarm session
@@ -207,7 +212,7 @@ maxIncomingTime :: Int
 maxIncomingTime = 120 * sec
 
 maxOutcomingTime :: Int
-maxOutcomingTime = 60 * sec
+maxOutcomingTime = 1 * sec
 
 -- | Should be called after we have received any message from a peer.
 updateIncoming :: PeerSession -> IO ()
@@ -221,8 +226,13 @@ updateOutcoming PeerSession {..}  =
   updateTimeout (eventManager (clientSession swarmSession))
     outcomingTimeout maxOutcomingTime
 
-sendKA :: Socket -> IO ()
-sendKA sock = sendAll sock (encode BT.KeepAlive)
+sendKA :: Socket -> SwarmSession -> IO ()
+sendKA sock SwarmSession {..} = do
+  print "I'm sending keep alive."
+  sendAll sock (encode BT.KeepAlive)
+  let mgr = eventManager clientSession
+  updateTimeout mgr
+  print "Done.."
 
 abortSession :: IO ()
 abortSession = error "abortSession: not implemented"