diff --git a/idlebot.py b/idlebot.py index aabac67..fd8b84f 100755 --- a/idlebot.py +++ b/idlebot.py @@ -20,6 +20,8 @@ except ImportError: import logging from sleekxmpp import ClientXMPP +got_hangup = False + class bot(ClientXMPP): def __init__(self, jid, password, rooms, nick): ClientXMPP.__init__(self, jid, password) @@ -28,19 +30,38 @@ class bot(ClientXMPP): self.nick = nick self.add_event_handler('session_start', self.session_start) + self.add_event_handler('groupchat_message', self.muc_message) def session_start(self, event): self.get_roster() self.send_presence() for room in self.rooms: - print('joining %s' % room) + logger('info', 'joining %s' % room) self.plugin['xep_0045'].joinMUC( room, self.nick, wait=True ) + def muc_message(self, msg_obj): + global got_hangup + + # don't talk to yourself + if msg_obj['mucnick'] == self.nick: + return + + if 'groupchat' != msg_obj['type']: + return + + if msg_obj['body'].startswith(conf('bot_user')) and 'hangup' in msg_obj['body']: + logger('warn', "got 'hangup' from '%s': '%s'" % ( + msg_obj['mucnick'], msg_obj['body'] + )) + got_hangup = True + sys.exit(1) + + if '__main__' == __name__: print(sys.argv[0] + ' ' + VERSION) @@ -63,6 +84,9 @@ if '__main__' == __name__: while 1: try: # do nothing here, just idle + if got_hangup: + xmpp.disconnect() + sys.exit(1) time.sleep(delay) except KeyboardInterrupt: diff --git a/urlbot.py b/urlbot.py index 04ab8c0..c7362cf 100755 --- a/urlbot.py +++ b/urlbot.py @@ -285,6 +285,10 @@ class bot(ClientXMPP): print('msg from %s: %s' % (msg_obj['from'].bare, msg_obj)) +# def set_presence(self, msg): +# for room in self.rooms: +# self.send_presence(pto=room, pstatus=msg) + if '__main__' == __name__: import plugins