1 files changed, 83 insertions, 3 deletions

diff --git a/src/Network/BitTorrent/DHT.hs b/src/Network/BitTorrent/DHT.hs
index b0aac002..bdb76c76 100644
--- a/src/Network/BitTorrent/DHT.hs
+++ b/src/Network/BitTorrent/DHT.hs
@@ -1,6 +1,86 @@
 module Network.BitTorrent.DHT
-       ( newNodeSession
-       , dhtServer
+       ( dht
+       , ping
+       , Network.BitTorrent.DHT.bootstrap
+       , Network.BitTorrent.DHT.lookup
+       , Network.BitTorrent.DHT.insert
        ) where
 
-import Network.BitTorrent.DHT.Protocol
\ No newline at end of file
+import Control.Applicative
+import Control.Monad
+import Control.Monad.Reader
+import Data.List as L
+import Network.Socket (PortNumber)
+
+import Data.Torrent.InfoHash
+import Network.BitTorrent.Core
+import Network.BitTorrent.DHT.Message
+import Network.BitTorrent.DHT.Session
+
+
+{-----------------------------------------------------------------------
+--  Handlers
+-----------------------------------------------------------------------}
+
+pingH :: Address ip => NodeHandler ip
+pingH = nodeHandler $ \ _ Ping -> return Ping
+
+{-
+findNodeH :: (Eq ip, Serialize ip, Typeable ip) => Handler (DHT ip)
+findNodeH = dhtHandler $ \ _ (FindNode nid) ->
+  NodeFound <$> getClosest nid
+
+getPeersH :: (Eq ip, Serialize ip, Typeable ip) => Handler (DHT ip)
+getPeersH = dhtHandler $ \ addr (GetPeers ih) ->
+  GotPeers <$> getPeerList ih <*> grantToken addr
+
+announceH :: Handler (DHT ip)
+announceH = dhtHandler $ \ addr (Announce {..}) -> do
+  checkToken addr sessionToken
+  insertPeer topic undefined -- PeerAddr (add, port)
+  return Announced
+-}
+
+handlers :: Address ip => [NodeHandler ip]
+handlers = [pingH]
+
+{-----------------------------------------------------------------------
+--  Query
+-----------------------------------------------------------------------}
+
+-- | Run DHT on specified port. <add note about resources>
+dht :: Address ip => NodeAddr ip -> DHT ip a -> IO a
+dht addr = runDHT addr handlers
+
+ping :: Address ip => NodeAddr ip -> DHT ip ()
+ping addr = do
+  Ping <- Ping <@> addr
+  return ()
+
+-- | One good node may be sufficient. <note about 'Data.Torrent.tNodes'>
+bootstrap :: Address ip => [NodeAddr ip] -> DHT ip ()
+bootstrap = mapM_ insertClosest
+  where
+    insertClosest addr = do
+      nid <- getNodeId
+      NodeFound closest <- FindNode nid <@> addr
+      forM_ closest insertNode
+
+-- | Get list of peers which downloading
+lookup :: Address ip => InfoHash -> DHT ip [PeerAddr ip]
+lookup ih = getClosestHash ih >>= collect
+  where
+    collect nodes = L.concat <$> forM (nodeAddr <$> nodes) retrieve
+    retrieve addr = do
+      GotPeers {..} <- GetPeers ih <@> addr
+      either collect pure peers
+
+-- | Announce that /this/ peer may have some pieces of the specified
+-- torrent.
+insert :: Address ip => InfoHash -> PortNumber -> DHT ip ()
+insert ih port = do
+  nodes <- getClosestHash ih
+  forM_ (nodeAddr <$> nodes) $ \ addr -> do
+--    GotPeers {..} <- GetPeers ih <@> addr
+--    Announced     <- Announce False ih undefined grantedToken <@> addr
+    return ()