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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
{-# LANGUAGE GeneralizedNewtypeDeriving #-}
{-# LANGUAGE TemplateHaskell #-}
module Network.BitTorrent.Client
( -- * Options
Options (..)
-- * Client session
, Client
-- ** Session data
, clientPeerId
, clientListenerPort
, allowedExtensions
-- ** Session initialization
, newClient
, closeClient
, withClient
, simpleClient
-- * BitTorrent monad
, MonadBitTorrent (..)
, BitTorrent
, runBitTorrent
, getClient
-- * Handle
, openTorrent
, openMagnet
, closeHandle
) where
import Control.Exception
import Control.Concurrent
import Control.Monad.Trans.Resource
import Data.Default
import Data.HashMap.Strict as HM
import Data.Maybe
import Data.Text
import Network
import Network.BitTorrent.Client.Types
import Network.BitTorrent.Client.Handle
import Network.BitTorrent.Core
import Network.BitTorrent.DHT
import Network.BitTorrent.Tracker as Tracker hiding (Options)
import Network.BitTorrent.Exchange.Message
data Options = Options
{ optFingerprint :: Fingerprint
, optName :: Text
, optPort :: PortNumber
, optExtensions :: [Extension]
, optNodeAddr :: NodeAddr IPv4
, optBootNode :: Maybe (NodeAddr IPv4)
}
instance Default Options where
def = Options
{ optFingerprint = def
, optName = "hs-bittorrent"
, optPort = 6882
, optExtensions = []
, optNodeAddr = "0.0.0.0:6882"
, optBootNode = Nothing
}
newClient :: Options -> LogFun -> IO Client
newClient Options {..} logger = do
pid <- genPeerId
ts <- newMVar HM.empty
let peerInfo = PeerInfo pid Nothing optPort
mgr <- Tracker.newManager def peerInfo
node <- runResourceT $ do
node <- startNode handlers def optNodeAddr logger
runDHT node $ bootstrap (maybeToList optBootNode)
return node
return Client
{ clientPeerId = pid
, clientListenerPort = optPort
, allowedExtensions = toCaps optExtensions
, trackerManager = mgr
, clientNode = node
, clientTorrents = ts
, clientLogger = logger
}
closeClient :: Client -> IO ()
closeClient Client {..} = do
Tracker.closeManager trackerManager
return ()
-- closeNode clientNode
withClient :: Options -> LogFun -> (Client -> IO a) -> IO a
withClient opts lf action = bracket (newClient opts lf) closeClient action
-- | Run bittorrent client with default options and log to @stderr@.
--
-- For testing purposes only.
--
simpleClient :: BitTorrent () -> IO ()
simpleClient m = withClient def logger (`runBitTorrent` m)
where
logger _ _ _ _ = return ()
|
