From 32051dc3518d769860e7baa7d8c88c4de4e75b43 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Reynir=20Bj=C3=B6rnsson?= <reynir@reynir.dk>
Date: Wed, 3 May 2023 16:53:04 +0200
Subject: [PATCH] Tweak messages

---
 unikernel.ml | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/unikernel.ml b/unikernel.ml
index 6dd8f4f..e993214 100644
--- a/unikernel.ml
+++ b/unikernel.ml
@@ -37,7 +37,7 @@ module Main (_ : Mirage_random.S) (T : Mirage_time.S) (M : Mirage_clock.MCLOCK)
     | Message { sender; message=_ } when String.equal sender me ->
       write me oc
     | Message { sender; message } ->
-      let* () = oc (Printf.ksprintf Cstruct.of_string "%s: %s\r\n" sender message) in
+      let* () = oc (Printf.ksprintf Cstruct.of_string "\x07%s: %s\r\n" sender message) in
       write me oc
     | Join username ->
       let* () = oc (Printf.ksprintf Cstruct.of_string "--> %s joined!\r\n" username) in
@@ -49,6 +49,7 @@ module Main (_ : Mirage_random.S) (T : Mirage_time.S) (M : Mirage_clock.MCLOCK)
 
   let chat flow stop username ic oc =
     Lwt_condition.broadcast c (Join username);
+    let* () = oc (Cstruct.of_string (Printf.sprintf "Hello, %s!\r\n" username)) in
     let* () =
       Lwt.pick [
         read username ic;
@@ -63,10 +64,10 @@ module Main (_ : Mirage_random.S) (T : Mirage_time.S) (M : Mirage_clock.MCLOCK)
     | Ssh.Pty_req _ | Ssh.Pty_set _ | Ssh.Set_env _ ->
       Lwt.return_unit
     | Ssh.Channel { cmd; ic; oc; ec; } ->
-      let* () = oc (Cstruct.of_string (Printf.sprintf "Hello, %s!\r\n" username)) in
       chat flow stop username ic oc
     | Ssh.Shell { ic; oc; ec; } ->
-      let* () = oc (Cstruct.of_string (Printf.sprintf "Hello, %s!\r\n" username)) in
+      let* () = ec (Cstruct.of_string "Banawá-chat does currently not work well with TTY.\r\n\
+                                       Consider reconnecting without tty (ssh -T)\r\n") in
       chat flow stop username ic oc
 
   let start _random _time _mtime stack =