diff options
Diffstat (limited to 'Role/Downloader.hs')
-rw-r--r-- | Role/Downloader.hs | 18 |
1 files changed, 11 insertions, 7 deletions
diff --git a/Role/Downloader.hs b/Role/Downloader.hs index d327c8c..55d7b63 100644 --- a/Role/Downloader.hs +++ b/Role/Downloader.hs @@ -6,12 +6,13 @@ import CmdLine import SessionID import Control.Concurrent.STM +import Control.Concurrent.STM.TMChan import Role.Developer (run', processSessionStart, getUserMessage, Output(..)) run :: DownloadOpts -> IO () run = run' downloader . downloadUrl -downloader :: TChan (Message Entered) -> TChan (Message Seen) -> SessionID -> IO () +downloader :: TMChan (Message Entered) -> TMChan (Message Seen) -> SessionID -> IO () downloader _ichan ochan sid = do let logfile = sessionLogFile "." sid putStrLn $ "Starting download to " ++ logfile @@ -21,9 +22,12 @@ downloader _ichan ochan sid = do go logger st where go logger st = do - (o, msg) <- atomically $ getUserMessage ochan st - _ <- logger $ User msg - case o of - ProtocolError e -> error ("Protocol error: " ++ e) - _ -> return () - go logger st + v <- atomically $ getUserMessage ochan st + case v of + Nothing -> return () + Just (o, msg) -> do + _ <- logger $ User msg + case o of + ProtocolError e -> error ("Protocol error: " ++ e) + _ -> return () + go logger st |