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)