mirror of
https://github.com/YunoHost/webhooks.git
synced 2024-09-03 19:56:54 +02:00
65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
# 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)
|