Fixed RoomEncryptedFilter

src/mab/context.py

@@ -31,6 +31,9 @@ CTX_CMD_VERB = ContextDataKey[str]("CTX_CMD_VERB")
 CTX_CMD_ARGS = ContextDataKey[list[str]]("CTX_CMD_ARGS")
 """Arguments that were passed with the command"""
 
+CTX_ROOM_ENCRYPTED = ContextDataKey[bool]("CTX_ROOM_ENCRYPTED")
+"""True if the room is encrypted"""
+
 
 #
 # EventContext implementation

src/mab/filters/room.py

@@ -1,7 +1,6 @@
 from .base import BaseEventFilter
 
-from nio import AsyncClient
+from ..context import EventContext, CTX_ROOM_ENCRYPTED
-from nio import MatrixRoom, Event
 
 class RoomEncryptedFilter(BaseEventFilter):
     """
@@ -10,8 +9,11 @@ class RoomEncryptedFilter(BaseEventFilter):
     def __init__(self):
         super().__init__()
 
-    async def __call__(self, room: MatrixRoom, event: Event, client: AsyncClient) -> bool:
+    async def __call__(self, context: EventContext) -> bool:
+        if not await super().__call__(context):
+            return False
         try:
-            return room.encrypted
+            context[CTX_ROOM_ENCRYPTED] = context.room.encrypted
+            return context.room.encrypted
         except:
             return False