From 0de8d9f03637106ed6002bcfb60ab2b203c643a3 Mon Sep 17 00:00:00 2001
From: Jean-Marie Mineau <jean-marie.mineau@protonmail.com>
Date: Fri, 8 Oct 2021 16:53:50 +0200
Subject: [PATCH] add pong callback

---
 src/kassandra/bot.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/kassandra/bot.py b/src/kassandra/bot.py
index a58dcf7..66983c8 100644
--- a/src/kassandra/bot.py
+++ b/src/kassandra/bot.py
@@ -7,11 +7,25 @@ from typing import (
     Any,
     NoReturn
 )
+import nio
 from matrix_bot.client import Client
 from matrix_bot.invite_policy import WhiteList
+from matrix_bot.utils import ignore_client_message
 from .config import Config
 from .format import Message
 
+@ignore_client_message
+async def __pong(
+    client: Client,
+    room: nio.rooms.MatrixRoom,
+    message: nio.events.room_events.RoomMessageText
+):
+    """
+        If the bot is pinged, send "Pong"
+    """
+    if (client.user_name + ':') in message.body:
+        await client.send_message(room.room_id, "Pong")
+
 async def __send_messsages(
     message_queue: asyncio.Queue[Message],
     bot: Client,
@@ -45,6 +59,7 @@ async def send_messages(
     )
     invite_policy = await WhiteList(bot, config.alert_rooms)
     bot.set_invite_policy(invite_policy)
+    bot.add_message_callback(__pong)
     await asyncio.gather(
         bot.run(),
         __send_messsages(