blob: 04dbec1b99686b98c9a18b7996afa1e7fe185af2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
-- |
-- Copyright : (c) Sam Truzjan 2013
-- License : BSD
-- Maintainer : pxqr.sta@gmail.com
-- Stability : experimental
-- Portability : portable
--
module Network.BitTorrent.Tracker.RPC
( Tracker
, Network.BitTorrent.Tracker.RPC.connect
, Network.BitTorrent.Tracker.RPC.announce
, Network.BitTorrent.Tracker.RPC.scrape
) where
import Control.Applicative
import Control.Exception
import Control.Monad.Trans.Resource
import Network.URI
import Network.BitTorrent.Tracker.Message
import Network.BitTorrent.Tracker.RPC.HTTP as HTTP
import Network.BitTorrent.Tracker.RPC.UDP as UDP
data Tracker
= HTracker Connection
| UTracker UDPTracker
connect :: URI -> IO Tracker
connect uri @ URI {..}
| uriScheme == "http:" = HTracker <$> runResourceT (HTTP.connect uri)
| uriScheme == "udp:" = UTracker <$> UDP.connect uri
| otherwise = throwIO $ userError msg
where
msg = "unknown tracker protocol scheme: " ++ show uriScheme
announce :: AnnounceQuery -> Tracker -> IO AnnounceInfo
announce q (HTracker t) = runResourceT $ HTTP.announce q t
announce q (UTracker t) = UDP.announce q t
scrape :: ScrapeQuery -> Tracker -> IO ScrapeInfo
scrape q (HTracker t) = runResourceT $ HTTP.scrape q t
scrape q (UTracker t) = UDP.scrape q t
|
