kumify/msgio/gateways/matrix.py

45 lines
1.4 KiB
Python

from django.views.generic import View
from django.dispatch import receiver
import json
import asyncio
from nio import AsyncClient
from dbsettings.functions import dbsettings
from ..signals import send_message
from ..models import GatewayUser
from ..helpers import run_filters
class MatrixDispatcher:
def __init__(self, username=None, password=None, homeserver=None):
self.username = username or dbsettings.MATRIX_USERNAME
self.password = password or dbsettings.MATRIX_PASSWORD
self.homeserver = homeserver or dbsettings.MATRIX_HOMESERVER
async def send(self, message, room_id):
client = AsyncClient(self.homeserver, self.username)
await client.login(self.password)
await client.join(room_id)
await client.room_send(
room_id=room_id,
message_type="m.room.message",
content={
"msgtype": "m.text",
"body": message
}
)
@receiver(send_message)
def matrix_sender(sender, **kwargs):
if kwargs["dispatcher"] == "matrix":
notification = kwargs["notification"]
settings = GatewayUser.objects.get(user=notification.recipient, gateway="matrix")
room_id = settings.gatewayusersetting_set.get(key="room_id").value
text = run_filters(notification)
asyncio.get_event_loop().run_until_complete(MatrixDispatcher().send(text, room_id))