1 files changed, 30 insertions, 3 deletions

diff --git a/src/Network/BitTorrent/Client.hs b/src/Network/BitTorrent/Client.hs
index fc2b904a..a6151857 100644
--- a/src/Network/BitTorrent/Client.hs
+++ b/src/Network/BitTorrent/Client.hs
@@ -1,16 +1,21 @@
 module Network.BitTorrent.Client
        ( Options (..)
-       , Client (..)
+       , Client
+       , newClient
+       , addTorrent
        ) where
 
+import Control.Concurrent.STM
 import Data.Default
 import Data.Function
+import Data.HashMap.Strict as HM
 import Data.Ord
 import Data.Text
 import Network
 
 import Data.Torrent
 import Data.Torrent.InfoHash
+import Network.BitTorrent.Client.Swarm
 import Network.BitTorrent.Core
 import Network.BitTorrent.Exchange.Message
 
@@ -19,6 +24,7 @@ data Options = Options
   { fingerprint :: Fingerprint
   , name        :: Text
   , port        :: PortNumber
+  , extensions  :: [Extension]
   }
 
 instance Default Options where
@@ -26,11 +32,14 @@ instance Default Options where
     { fingerprint = def
     , name        = "hs-bittorrent"
     , port        = 6882
+    , extensions  = []
     }
 
 data Client = Client
-  { clientPeerId      :: !PeerId
-  , allowedExtensions :: !Caps
+  { clientPeerId       :: !PeerId
+  , clientListenerPort :: !PortNumber
+  , allowedExtensions  :: !Caps
+  , torrents           :: TVar (HashMap InfoHash Swarm)
   }
 
 instance Eq Client where
@@ -38,3 +47,21 @@ instance Eq Client where
 
 instance Ord Client where
   compare = comparing clientPeerId
+
+newClient :: Options -> IO Client
+newClient Options {..} = do
+  pid <- genPeerId
+  ts  <- newTVarIO HM.empty
+  return Client
+    { clientPeerId       = pid
+    , clientListenerPort = port
+    , allowedExtensions  = toCaps extensions
+    , torrents           = ts
+    }
+
+addTorrent :: Torrent -> Client -> IO ()
+addTorrent t Client {..} = do
+  leecher <- newLeecher clientPeerId clientListenerPort t
+  let ih = idInfoHash (tInfoDict t)
+  atomically $ modifyTVar' torrents (HM.insert ih leecher)
+  askPeers leecher >>= print
\ No newline at end of file