refactor PeerID and associated functions.

[functorrent.git] / src / Tracker.hs

diff --git a/src/Tracker.hs b/src/Tracker.hs
index 6db0c3682a96430ed7f592724a66f99f422fa8c5..599c2938d8c0e8d8046768c2adcdb6d30b9b3275 100644 (file)
--- a/src/Tracker.hs
+++ b/src/Tracker.hs
@@ -1,16 +1,55 @@
 module Tracker where
 
+import qualified Data.ByteString.Char8 as BC
+import qualified Data.Map as M
+import qualified Data.List as List
+import qualified Network.HTTP as HTTP
+import qualified Network.HTTP.Base as HB
 import qualified Bencode as Benc
-import Data.Conduit
-import Network.HTTP
+import qualified Crypto.Hash.SHA1 as SHA1
+import qualified Data.ByteString.Base16 as B16
+import qualified Utils as U
+import Data.Char
+
 
 type Url = String
 
-requestUrl :: String
 
+-- | urlEncodeHash
+--
+-- >>> urlEncodeHash $ BC.pack "123456789abcdef123456789abcdef123456789a"
+-- "%124Vx%9a%bc%de%f1%23Eg%89%ab%cd%ef%124Vx%9a"
+urlEncodeHash :: BC.ByteString -> String
+urlEncodeHash bs = concatMap (encode . BC.unpack) (U.splitN 2 bs)
+  where encode b@[c1, c2] = let c =  chr (read ("0x" ++ b))
+                                  in
+                                   escape c c1 c2
+        encode _ = ""
+        escape i c1 c2 | i `elem` nonSpecialChars = [i]
+                       | otherwise = "%" ++ [c1] ++ [c2]
+          where nonSpecialChars = ['A'..'Z'] ++
+                                  ['a'..'z'] ++
+                                  ['0'..'9'] ++
+                                  "-_.~"
+
+infoHash :: M.Map Benc.BVal Benc.BVal -> BC.ByteString
+infoHash m = let info = m M.! Benc.Bstr (BC.pack "info")
+             in (B16.encode . SHA1.hash . BC.pack . Benc.encode) info
+
+prepareRequest :: Benc.BVal -> String -> Integer -> String
+prepareRequest (Benc.Bdict d) peer_id len =
+  let p = [("info_hash", urlEncodeHash (infoHash d)),
+           ("peer_id", HB.urlEncode peer_id),
+           ("port", "6881"),
+           ("uploaded", "0"),
+           ("downloaded", "0"),
+           ("left", show len),
+           ("compact", "1"),
+           ("event", "started")]
+  in
+   List.intercalate "&" [f ++ "=" ++ s | (f,s) <- p]
 
-connect :: Url -> String -> IO (Benc.BVal)
-connect url infoHash = case (parseUrl url) of
-                        Nothing -> putStrLn "invalid tracker URL"
-                        Just req -> let 
-              
+connect :: Url -> String -> IO String
+connect baseurl qstr = let url = baseurl ++ "?" ++ qstr
+                       in HTTP.simpleHTTP (HTTP.getRequest url) >>=
+                          HTTP.getResponseBody