summaryrefslogtreecommitdiffhomepage
path: root/Role/Downloader.hs
blob: 243410b2e87c40e88043d5a2dcd38a45743d5e7f (plain)
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
module Role.Downloader where

import Types
import Log
import CmdLine
import SessionID
import Role.Developer (run', processSessionStart, getServerMessage, Output(..))

import Control.Concurrent.STM
import Control.Concurrent.STM.TMChan
import Data.Time.Clock.POSIX

run :: DownloadOpts -> IO ()
run = run' downloader . downloadUrl

downloader :: TMChan (Message Entered) -> TMChan LogMessage -> SessionID -> IO ()
downloader _ichan ochan sid = do
	let logfile = sessionLogFile "." sid
	putStrLn $ "Starting download to " ++ logfile
	putStrLn "(Will keep downloading until the debug-me session is done.)"
	withLogger logfile $ \logger -> do
		(st, _startoutput) <- processSessionStart ochan logger
		go logger st
  where
	go logger st = do
		ts <- getPOSIXTime
		v <- atomically $ getServerMessage ochan st ts
		case v of
			Nothing -> return ()
			Just (o, msg) -> do
				_ <- logger msg
				case o of
					ProtocolError e -> error ("Protocol error: " ++ e)
					_ -> go logger st