Add Client.hs module

2 files changed, 41 insertions, 0 deletions

diff --git a/bittorrent.cabal b/bittorrent.cabal
index dbb3c3be..f1a4f545 100644
--- a/bittorrent.cabal
+++ b/bittorrent.cabal
@@ -50,6 +50,7 @@ library
                        Data.Torrent.Progress
                        Data.Torrent.Tree
 --                     Network.BitTorrent
+                       Network.BitTorrent.Client
                        Network.BitTorrent.Core
                        Network.BitTorrent.Core.Fingerprint
                        Network.BitTorrent.Core.PeerId
diff --git a/src/Network/BitTorrent/Client.hs b/src/Network/BitTorrent/Client.hs
new file mode 100644
index 00000000..fc2b904a
--- /dev/null
+++ b/src/Network/BitTorrent/Client.hs
@@ -0,0 +1,40 @@
+module Network.BitTorrent.Client
+       ( Options (..)
+       , Client (..)
+       ) where
+
+import Data.Default
+import Data.Function
+import Data.Ord
+import Data.Text
+import Network
+
+import Data.Torrent
+import Data.Torrent.InfoHash
+import Network.BitTorrent.Core
+import Network.BitTorrent.Exchange.Message
+
+
+data Options = Options
+  { fingerprint :: Fingerprint
+  , name        :: Text
+  , port        :: PortNumber
+  }
+
+instance Default Options where
+  def = Options
+    { fingerprint = def
+    , name        = "hs-bittorrent"
+    , port        = 6882
+    }
+
+data Client = Client
+  { clientPeerId      :: !PeerId
+  , allowedExtensions :: !Caps
+  }
+
+instance Eq Client where
+  (==) = (==) `on` clientPeerId
+
+instance Ord Client where
+  compare = comparing clientPeerId