use REPL to blast out chat messages.

1 files changed, 10 insertions, 4 deletions

diff --git a/lib/6-chit-chat/publish.rb b/lib/6-chit-chat/publish.rb
index e21cb3f..e899fdc 100644
--- a/lib/6-chit-chat/publish.rb
+++ b/lib/6-chit-chat/publish.rb
@@ -9,9 +9,15 @@ connection.start
 channel = connection.create_channel
 exchange = channel.topic("chitchat")
 username = `'whoami'`.chomp!
-message = ARGV.empty? ? "Hi!" : ARGV.join(" ")
 
-exchange.publish(message, routing_key: username)
-puts " [x] Sent #{username}:#{message}"
+begin
+  print "> "
+  while message = gets do
+    exchange.publish(message, routing_key: username)
+    puts " [x] Sent #{username}:#{message}"
+    print "> "
+  end
+rescue
+  connection.close
+end
 
-connection.close