Check for message type in example/bot.rb

atipugin · Oct 28, 2022 · 81d98a5 · 81d98a5
1 parent 3a6d619
commit 81d98a5
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/examples/bot.rb b/examples/bot.rb
@@ -7,13 +7,16 @@
 
 Telegram::Bot::Client.run(token) do |bot|
   bot.listen do |message|
-    case message.text
-    when '/start'
-      bot.api.send_message(chat_id: message.chat.id, text: "Hello, #{message.from.first_name}!")
-    when '/end'
-      bot.api.send_message(chat_id: message.chat.id, text: "Bye, #{message.from.first_name}!")
-    else
-      bot.api.send_message(chat_id: message.chat.id, text: "I don't understand you :(")
+    case message
+    when Telegram::Bot::Types::Message
+      case message.text
+      when '/start'
+        bot.api.send_message(chat_id: message.chat.id, text: "Hello, #{message.from.first_name}!")
+      when '/end'
+        bot.api.send_message(chat_id: message.chat.id, text: "Bye, #{message.from.first_name}!")
+      else
+        bot.api.send_message(chat_id: message.chat.id, text: "I don't understand you :(")
+      end
     end
   end
 end