module Main where
import Prelude hiding (length, readFile, writeFile)
-import Data.ByteString.Char8 (ByteString, length, readFile, writeFile, length)
+import Data.ByteString.Char8 (ByteString, readFile, writeFile, length, unpack)
import System.Environment (getArgs)
import System.Exit (exitSuccess)
+import System.Directory (doesFileExist)
import Text.ParserCombinators.Parsec (ParseError)
-import Bencode (decode, BVal(..))
-import Logger (initLogger, logMessage, logStop)
-import Metainfo (announce, lengthInBytes, mkMetaInfo, info, name)
-import Peer (getPeers, getPeerResponse, handShakeMsg)
-import Tracker (connect, prepareRequest)
+import FuncTorrent.Bencode (decode, BVal(..))
+import FuncTorrent.Logger (initLogger, logMessage, logStop)
+import FuncTorrent.Metainfo (announce, lengthInBytes, mkMetaInfo, info, name, getTrackers)
+import FuncTorrent.Peer (peers, mkPeerResp, handShakeMsg)
+import FuncTorrent.Tracker (connect, prepareRequest)
logError :: ParseError -> (String -> IO ()) -> IO ()
logError e logMsg = logMsg $ "parse error: \n" ++ show e
parse :: [String] -> IO ByteString
parse [] = usage >> exit
-parse [a] = readFile a
+parse [a] = do
+ fileExist <- doesFileExist a
+ if fileExist
+ then readFile a
+ else error "file does not exist"
parse _ = exit
main :: IO ()
let len = lengthInBytes $ info m
(Bdict d') = d
+ trackers = getTrackers m
logMsg "Trying to fetch peers: "
- body <- connect (announce m) (prepareRequest d' peerId len)
-
- -- TODO: Write to ~/.functorrent/caches
- writeFile (name (info m) ++ ".cache") body
-
- let peerResponse = show $ getPeers $ getPeerResponse body
- logMsg $ "Peers List : " ++ peerResponse
+ response <- connect (head trackers) (prepareRequest d' peerId len)
let hsMsgLen = show $ length $ handShakeMsg d' peerId
logMsg $ "Hand-shake message length : " ++ hsMsgLen
+ -- TODO: Write to ~/.functorrent/caches
+ writeFile (name (info m) ++ ".cache") response
+
+ case decode response of
+ Right trackerInfo ->
+ case mkPeerResp trackerInfo of
+ Right peerResp ->
+ logMsg $ "Peers List : " ++ (show . peers $ peerResp)
+ Left e -> logMsg $ "Error" ++ unpack e
+ Left e -> logError e logMsg
+
Left e -> logError e logMsg
logStop logR