+{-# LANGUAGE OverloadedStrings #-}
module Main where
+import Prelude hiding (log, length, readFile, writeFile)
+import Data.ByteString.Char8 (ByteString, readFile, writeFile, unpack)
import System.Environment (getArgs)
-import System.Exit
-import qualified Data.ByteString.Char8 as BC
-import qualified Bencode as Benc
-import qualified Metainfo as MInfo
-import qualified Tracker as T
-import qualified Text.ParserCombinators.Parsec as Parsec
-import qualified Peer as P
-import Data.Functor
-
-printError :: Parsec.ParseError -> IO ()
-printError e = putStrLn $ "parse error: " ++ show e
-
-genPeerId :: String
-genPeerId = "-HS0001-20150215"
-
-exit :: IO BC.ByteString
+import System.Exit (exitSuccess)
+import System.Directory (doesFileExist)
+import Text.ParserCombinators.Parsec (ParseError)
+
+import FuncTorrent.Bencode (decode)
+import FuncTorrent.Logger (initLogger, logMessage, logStop)
+import FuncTorrent.Metainfo (Info(..), Metainfo(..), mkMetaInfo)
+import FuncTorrent.Peer (Peer(..), handShake)
+import FuncTorrent.Tracker (tracker, peers, mkTrackerResponse)
+
+logError :: ParseError -> (String -> IO ()) -> IO ()
+logError e logMsg = logMsg $ "parse error: \n" ++ show e
+
+peerId :: String
+peerId = "-HS0001-*-*-20150215"
+
+exit :: IO ByteString
exit = exitSuccess
usage :: IO ()
usage = putStrLn "usage: functorrent torrent-file"
-parse :: [String] -> IO BC.ByteString
+parse :: [String] -> IO ByteString
parse [] = usage >> exit
-parse [a] = BC.readFile a
+parse [a] = do
+ fileExist <- doesFileExist a
+ if fileExist
+ then readFile a
+ else error "file does not exist"
parse _ = exit
main :: IO ()
main = do
- args <- getArgs
- torrentStr <- parse args
- case Benc.decode torrentStr of
- Right d -> case MInfo.mkMetaInfo d of
- Nothing -> putStrLn "parse error"
- Just m -> do
- body <- BC.pack <$> T.connect (MInfo.announce m) (T.prepareRequest d genPeerId)
- print (P.getPeers (P.getPeerResponse body))
- Left e -> printError e
- putStrLn "done"
+ args <- getArgs
+ logR <- initLogger
+ let log = logMessage logR
+ log "Starting up functorrent"
+ log $ "Parsing input file " ++ concat args
+ torrentStr <- parse args
+ case decode torrentStr of
+ Right d ->
+ case mkMetaInfo d of
+ Nothing -> log "Unable to make meta info file"
+ Just m -> do
+ log "Input File OK"
+ log $ "Downloading file : " ++ name (info m)
+ log "Trying to fetch peers"
+
+ log $ "Trackers: " ++ head (announceList m)
+ response <- tracker m peerId
+
+ -- TODO: Write to ~/.functorrent/caches
+ writeFile (name (info m) ++ ".cache") response
+
+ case decode response of
+ Right trackerInfo ->
+ case mkTrackerResponse trackerInfo of
+ Right peerResp -> do
+ log $ "Peers List : " ++ (show . peers $ peerResp)
+ let p1 = head (peers peerResp)
+ msg <- handShake (Peer "" "95.188.88.59" 27000) (infoHash m) peerId
+ log $ "handshake: " ++ (show msg)
+ return ()
+ Left e -> log $ "Error" ++ unpack e
+ Left e -> logError e log
+
+ Left e -> logError e log
+ logStop logR