WIP: peer handshake

[functorrent.git] / src / Main.hs

diff --git a/src/Main.hs b/src/Main.hs
index 7ab471258fdb0f70f9d5db6b32da77d5f69ce6c7..fa79ac30e8046ab9fe125fb6c45db098a825d2e9 100644 (file)
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -1,12 +1,74 @@
+{-# 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 qualified Data.ByteString.Char8 as BC
-import qualified Bencode as Benc
+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 ByteString
+parse [] = usage >> exit
+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 <- BC.readFile (head args)
-  let metadata = Benc.decode torrentStr
-  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