prevent terminal echo before user grants write access

And improve some messages.

1 files changed, 23 insertions, 20 deletions

diff --git a/Role/Developer.hs b/Role/Developer.hs
index fc54c9f..0277bed 100644
--- a/Role/Developer.hs
+++ b/Role/Developer.hs
@@ -58,26 +58,29 @@ developer dsv ichan ochan sid = withSessionLogger sid $ \logger -> do
 		"Using gpg to sign the debug-me session key."
 	spk <- myPublicKey sk (GpgSign True)
 	displayInControlWindow controlinput
-		"Connecting to user ..."
-	(devstate, startoutput) <- processSessionStart sk ochan logger dsv
-	emitOutput startoutput
-	displayInControlWindow controlinput
-		"Waiting for the user to grant write access ..."
-	res <- authUser spk ichan ochan devstate logger
-	case res of
-		Authed -> inRawMode $ void $ do
-			displayInControlWindow controlinput
-				"Write access granted. You can now type into the user's shell."
-			displayInControlWindow controlinput
-				"(And, you can type in this window to chat with the user.)"
-			sendTtyInput ichan devstate logger
-				`race` sendTtyOutput ochan devstate controlinput logger
-				`race` sendControlOutput controloutput ichan devstate logger
-		AuthFailed -> do
-			displayInControlWindow controlinput
-				"User did not grant access to their terminal. Watching session in read-only mode."
-			watchSessionReadOnly ochan logger devstate
-		SessionEnded -> hPutStrLn stderr "\n** This debug-me session has already ended."
+		"Connecting to the user's session ..."
+	inRawMode $ do
+		(devstate, startoutput) <- processSessionStart sk ochan logger dsv
+		emitOutput startoutput
+		displayInControlWindow controlinput
+			"Waiting for the user to check your Gnupg key and grant access ..."
+		authUser spk ichan ochan devstate logger
+			>>= go controlinput controloutput logger devstate
+  where
+	go controlinput controloutput logger devstate Authed = void $ do
+		displayInControlWindow controlinput
+			"Write access granted. You can now type into the user's shell."
+		displayInControlWindow controlinput
+			"(And, you can type in this window to chat with the user.)"
+		sendTtyInput ichan devstate logger
+			`race` sendTtyOutput ochan devstate controlinput logger
+			`race` sendControlOutput controloutput ichan devstate logger
+	go controlinput _controloutput logger devstate AuthFailed = do
+		displayInControlWindow controlinput
+			"User did not grant write access to their terminal. Watching session in read-only mode."
+		watchSessionReadOnly ochan logger devstate
+	go _ _ _ _ SessionEnded =
+		hPutStrLn stderr "\n** This debug-me session has already ended."
 
 watchSessionReadOnly :: TMChan AnyMessage -> Logger -> TVar DeveloperState -> IO ()
 watchSessionReadOnly ochan logger st = loop