webhooks/to_room.py
2019-02-06 01:25:40 +01:00

64 lines
1.7 KiB
Python

# coding: utf-8
import sys
import xmpp
from contextlib import contextmanager
@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(to="%s@conference.yunohost.org" % room)
presence.setTag('x', namespace='http://jabber.org/protocol/muc')
client.send(presence)
client.send(xmpp.Presence(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"))
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)