webhooks/to_room.py

# coding: utf-8

import sys
import xmpp
from contextlib import contextmanager
import uuid


@contextmanager
def XMPPBot(password, room="dev"):
    jid = xmpp.protocol.JID("gitbot@im.yunohost.org")

    client = xmpp.Client(jid.getDomain(), debug=[])

    # hack to connect only if I need to send messages
    client.connected = False

    def connect():
        client.connect()

        # yes, this is how this stupid lib tells you that the connection
        # succeed, it return the string "sasl", this doesn't make any sens and
        # it documented nowhere, because xmpp is THE FUTUR
        if client.auth(jid.getNode(), password) != "sasl":
            print("Failed to connect, bad login/password combination")
            sys.exit(1)

        client.sendInitPresence(requestRoster=0)

        presence = xmpp.Presence(
            attrs={"id": uuid.uuid4()}, to="%s@conference.yunohost.org" % room
        )
        presence.setTag("x", namespace="http://jabber.org/protocol/muc")

        client.send(presence)

        client.send(
            xmpp.Presence(
                attrs={"id": uuid.uuid4()},
                to="%s@conference.yunohost.org/GitBot" % room,
            )
        )

    def sendToChatRoom(message):
        if not client.connected:
            connect()
            client.connected = True

        client.send(
            xmpp.protocol.Message(
                "%s@conference.yunohost.org" % room,
                message,
                typ="groupchat",
                attrs={"id": uuid.uuid4()},
            )
        )

    client.sendToChatRoom = sendToChatRoom

    yield client

    if client.connected:
        client.disconnect()


if __name__ == "__main__":
    if len(sys.argv[1:]) < 2:
        print("Usage : python to_room.py <password> <message> [<room name>]")
        sys.exit(1)

    password, message = sys.argv[1:3]

    if len(sys.argv[1:]) > 2:
        room = sys.argv[1:][2]
    else:
        room = "dev"

    with XMPPBot(password, room=room) as bot:
        bot.sendToChatRoom(message)