From b39031d53dd6a9ab710255dd7cc258ed10292e89 Mon Sep 17 00:00:00 2001
From: Jean-Marie Mineau <jean-marie.mineau@protonmail.com>
Date: Mon, 4 Oct 2021 17:52:36 +0200
Subject: [PATCH] this is a failure :'(

---
 src/matrix_bot/client.py | 56 ++++++++++++++++++++++++++++++++++++++++
 tests/test.py            |  8 ++++++
 2 files changed, 64 insertions(+)

diff --git a/src/matrix_bot/client.py b/src/matrix_bot/client.py
index ae64e2c..05b93f7 100644
--- a/src/matrix_bot/client.py
+++ b/src/matrix_bot/client.py
@@ -144,6 +144,62 @@ class Client(Aobject):
         }
         await self.__send_message(room, msg)
 
+    async def send_firework_message(
+        self,
+        room: Union[RoomAlias, RoomId],
+        message: str,
+    ):
+        """
+            Send message with the firework effect.
+        """
+        msg = {
+            "body": message,
+            "msgtype": "nic.custom.fireworks"
+        }
+        await self.__send_message(room, msg)
+
+    async def send_confetti_message(
+        self,
+        room: Union[RoomAlias, RoomId],
+        message: str,
+    ):
+        """
+            Send message with the confetti effect.
+        """
+        msg = {
+            "body": message,
+            "msgtype": "nic.custom.confetti"
+        }
+        await self.__send_message(room, msg)
+
+    async def send_snow_message(
+        self,
+        room: Union[RoomAlias, RoomId],
+        message: str,
+    ):
+        """
+            Send message with the snowfall effect.
+        """
+        msg = {
+            "body": message,
+            "msgtype": "io.element.effect.snowfall"
+        }
+        await self.__send_message(room, msg)
+
+    async def send_space_invader_message(
+        self,
+        room: Union[RoomAlias, RoomId],
+        message: str,
+    ):
+        """
+            Send message with the space invader effect.
+        """
+        msg = {
+            "body": message,
+            "msgtype": "io.element.effects.space_invaders"
+        }
+        await self.__send_message(room, msg)
+
     async def __save_sync_tocken(self)->NoReturn:
         """
             Save the sync token.
diff --git a/tests/test.py b/tests/test.py
index 335c136..d16ccd4 100644
--- a/tests/test.py
+++ b/tests/test.py
@@ -28,6 +28,14 @@ async def main():
     room_name = withelisted_room_names[0]
     client.set_invite_policy(whitelist_policy)
     await client.send_message(room_name, "Beware of Greeks bearing gifts")
+    await asyncio.sleep(5)
+    await client.send_firework_message(room_name, "fire")
+    await asyncio.sleep(5)
+    await client.send_confetti_message(room_name, "confetti")
+    await asyncio.sleep(5)
+    await client.send_snow_message(room_name, "snow")
+    await asyncio.sleep(5)
+    await client.send_space_invader_message(room_name, "space")
     await client.run()