Add canDownload & canUpload functions

2 files changed, 18 insertions, 2 deletions

diff --git a/src/Data/Torrent/Progress.hs b/src/Data/Torrent/Progress.hs
index bbfe6e47..c42af248 100644
--- a/src/Data/Torrent/Progress.hs
+++ b/src/Data/Torrent/Progress.hs
@@ -27,6 +27,10 @@ module Data.Torrent.Progress
        , enqueuedProgress
        , uploadedProgress
        , dequeuedProgress
+
+         -- * Query
+       , canDownload
+       , canUpload
        ) where
 
 import Control.Applicative
@@ -36,6 +40,7 @@ import Data.Default
 import Data.List as L
 import Data.Monoid
 import Data.Serialize as S
+import Data.Ratio
 import Data.Word
 
 
@@ -111,3 +116,14 @@ enqueuedProgress amount = left +~ fromIntegral amount
 dequeuedProgress :: Integer -> Progress -> Progress
 dequeuedProgress amount = left -~ fromIntegral amount
 {-# INLINE dequeuedProgress #-}
+
+ri2rw64 :: Ratio Int -> Ratio Word64
+ri2rw64 x = fromIntegral (numerator x) % fromIntegral (denominator x)
+
+-- | Check global /download/ limit by uploaded \/ downloaded ratio.
+canDownload :: Ratio Int -> Progress -> Bool
+canDownload limit Progress {..} = _uploaded % _downloaded > ri2rw64 limit
+
+-- | Check global /upload/ limit by downloaded \/ uploaded ratio.
+canUpload :: Ratio Int -> Progress -> Bool
+canUpload limit Progress {..} = _downloaded % _uploaded > ri2rw64 limit
diff --git a/src/Network/BitTorrent/Core/PeerAddr.hs b/src/Network/BitTorrent/Core/PeerAddr.hs
index 84b1e1f6..0eeaae9d 100644
--- a/src/Network/BitTorrent/Core/PeerAddr.hs
+++ b/src/Network/BitTorrent/Core/PeerAddr.hs
@@ -5,8 +5,8 @@
 --   Stability   :  experimental
 --   Portability :  portable
 --
---   * 'PeerAddr' is used to represent peer location. Currently it's
--- just peer IP and peer port but this might be changed later.
+--   'PeerAddr' is used to represent peer location. Currently it's
+--   just peer IP and peer port but this might be changed later.
 --
 {-# LANGUAGE TemplateHaskell            #-}
 {-# LANGUAGE StandaloneDeriving         #-}