Implement scheduleBlock function

1 files changed, 3 insertions, 2 deletions

diff --git a/src/Network/BitTorrent/Exchange/Session/Metadata.hs b/src/Network/BitTorrent/Exchange/Session/Metadata.hs
index 174550fa..5d72172a 100644
--- a/src/Network/BitTorrent/Exchange/Session/Metadata.hs
+++ b/src/Network/BitTorrent/Exchange/Session/Metadata.hs
@@ -21,6 +21,7 @@ import Control.Monad.State
 import Data.ByteString as BS
 import Data.ByteString.Lazy as BL
 import Data.List as L
+import Data.Tuple
 
 import Data.BEncode as BE
 import Data.Torrent
@@ -45,7 +46,7 @@ nullStatus ps = Status [] (Block.empty ps)
 type Updates a = State Status a
 
 runUpdates :: MVar Status -> Updates a -> IO a
-runUpdates v m = undefined
+runUpdates v m = modifyMVar v (return . swap . runState m)
 
 scheduleBlock :: PeerAddr IP -> Updates (Maybe PieceIx)
 scheduleBlock addr = do
@@ -53,7 +54,7 @@ scheduleBlock addr = do
   case spans metadataPieceSize bkt of
     []              -> return Nothing
     ((off, _ ) : _) -> do
-      let pix = undefined
+      let pix = off `div` metadataPieceSize
       pending %= ((addr, pix) :)
       return (Just pix)