1 files changed, 102 insertions, 0 deletions

diff --git a/src/Network/BitTorrent/Tracker/Wai.hs b/src/Network/BitTorrent/Tracker/Wai.hs
new file mode 100644
index 00000000..f290c380
--- /dev/null
+++ b/src/Network/BitTorrent/Tracker/Wai.hs
@@ -0,0 +1,102 @@
+-- supported extensions:
+--
+--   no_peer_id - do not send peer id if no_peer_id=1 specified
+--   http://www.bittorrent.org/beps/bep_0023.html
+--
+--   compact - compact=1 or compact=0
+--   http://permalink.gmane.org/gmane.network.bit-torrent.general/4030
+--
+--
+{-# LANGUAGE RecordWildCards #-}
+module Network.BitTorrent.Tracker.Wai
+       ( tracker
+       ) where
+
+import Control.Monad.Trans.Resource
+import Data.BEncode as BE
+import Data.ByteString
+import Data.Default
+import Data.List as L
+import Network.HTTP.Types
+import Network.Wai
+
+import Data.Torrent.Progress
+import Network.BitTorrent.Core.PeerId
+import Network.BitTorrent.Core.PeerAddr
+import Network.BitTorrent.Tracker.RPC.Message
+
+
+data TrackerSettings = TrackerSettings
+  { -- | If peer did not specified the "numwant" then this value is
+    -- used.
+    defNumWant            :: {-# UNPACK #-} !Int
+
+    -- | If peer specified to big numwant value.
+  , maxNumWant            :: {-# UNPACK #-} !Int
+
+    -- | Recommended time interval to wait between regular announce
+    -- requests.
+  , reannounceInterval    :: {-# UNPACK #-} !Int
+
+    -- | Minimum time interval to wait between regular announce
+    -- requests.
+  , reannounceMinInterval :: !(Maybe Int)
+
+    -- | Whether to send count of seeders.
+  , completePeers         :: !Bool
+
+    -- | Whether to send count of leechers.
+  , incompletePeers       :: !Bool
+
+    -- | Do not send peer id in response. Peer can override this value
+    -- by setting "no_peer_id" to 0 or 1.
+  , noPeerId              :: !Bool
+
+    -- | Whether to send compact peer list. Peer can override this
+    -- value by setting "compact" to 0 or 1.
+  , compactPeerList       :: !Bool
+  }
+
+instance Default TrackerSettings where
+  def = TrackerSettings
+    { defNumWant            = defaultNumWant
+    , maxNumWant            = defaultMaxNumWant
+    , reannounceInterval    = defaultReannounceInterval
+    , reannounceMinInterval = Nothing
+    , compactPeerList       = False
+    , completePeers         = False
+    , incompletePeers       = False
+    , noPeerId              = False
+    }
+
+getAnnounceR :: AnnounceRequest -> ResourceT IO AnnounceInfo
+getAnnounceR = undefined
+
+getScrapeR :: ScrapeQuery -> ResourceT IO ScrapeInfo
+getScrapeR = undefined
+
+-- content-type: "text/plain" ?
+tracker :: Application
+tracker Request {..}
+  | requestMethod /= methodGet
+  = return $ responseLBS methodNotAllowed405 [] ""
+
+  | otherwise = do
+    case pathInfo of
+      ["announce"] ->
+        case parseAnnounceRequest $ queryToSimpleQuery queryString of
+          Right query -> do
+            info <- getAnnounceR query
+            return $ responseLBS ok200 [] $ BE.encode info
+          Left msg ->
+            return $ responseLBS (parseFailureStatus msg) [] ""
+
+      ["scrape"]   ->
+        case Right $ parseScrapeQuery $ queryToSimpleQuery queryString of -- TODO
+          Right query -> do
+            info <- getScrapeR query
+            return $ responseLBS ok200 [] $ BE.encode info
+          Left _ ->
+            return $ responseLBS badRequest400 [] ""
+
+      _            -> undefined --badPath