fix downloader and watcher

1 files changed, 9 insertions, 9 deletions

diff --git a/Role/Developer.hs b/Role/Developer.hs
index 64ff094..1cc5a10 100644
--- a/Role/Developer.hs
+++ b/Role/Developer.hs
@@ -36,7 +36,8 @@ userMessages (Developer _) = Nothing
 
 developer :: TMChan (Message Entered) -> TMChan (Message Seen) -> SessionID -> IO ()
 developer ichan ochan _ = withLogger "debug-me-developer.log" $ \logger -> do
-	devstate <- processSessionStart ochan logger
+	(devstate, startoutput) <- processSessionStart ochan logger
+	emitOutput startoutput
 	ok <- authUser ichan ochan devstate logger
 	if ok
 		then inRawMode $ void $ 
@@ -115,7 +116,7 @@ authUser ichan ochan devstate logger = do
 	waitresp pk
   where
 	waitresp pk = do
-		(o, msg) <- fromMaybe (error "No response from server to our session key")
+		(o, msg) <- fromMaybe (error "Looks like that debug-me session is over.")
 			<$> atomically (getUserMessage ochan devstate)
 		logger $ User msg
 		emitOutput o
@@ -231,7 +232,7 @@ isLegalSeen act@(Activity (Seen (Val b)) (Just hp) _) ds
 
 -- | Start by reading the initial two messages from the user side,
 -- their session key and the startup message.
-processSessionStart :: TMChan (Message Seen) -> Logger -> IO (TVar DeveloperState)
+processSessionStart :: TMChan (Message Seen) -> Logger -> IO (TVar DeveloperState, Output)
 processSessionStart ochan logger = do
 	sessionmsg <- fromMaybe (error "Did not get session initialization message")
 		<$> atomically (readTMChan ochan)
@@ -246,15 +247,13 @@ processSessionStart ochan logger = do
 	startmsg <- fromMaybe (error "Did not get session startup message")
 		<$> atomically (readTMChan ochan)
 	logger $ User startmsg
-	starthash <- case startmsg of
+	let (starthash, output) = case startmsg of
 		ActivityMessage act@(Activity (Seen (Val b)) Nothing _)
-			| verifySigned sigverifier act -> do
-				B.hPut stdout b
-				hFlush stdout	
-				return (hash act)
+			| verifySigned sigverifier act ->
+				(hash act, TtyOutput b)
 		_ -> error $ "Unexpected startup message: " ++ show startmsg
 	sk <- genMySessionKey
-	newTVarIO $ DeveloperState
+	st <- newTVarIO $ DeveloperState
 		{ lastSeen = starthash
 		, sentSince = mempty
 		, enteredSince = mempty
@@ -262,3 +261,4 @@ processSessionStart ochan logger = do
 		, developerSessionKey = sk
 		, developerSigVerifier = sigverifier
 		}
+	return (st, output)