#!/usr/bin/env python
import logging
import sys
from argparse import ArgumentParser
import sleekxmpp
if sys.version_info < (3, 0):
    reload(sys)
    sys.setdefaultencoding('utf8')
else:
    raw_input = input
class RollBot(sleekxmpp.ClientXMPP):
    def __init__(self, jid, password):
        sleekxmpp.ClientXMPP.__init__(self, jid, password)
        self.add_event_handler('session_start', self.start)
    def start(self, event):
        self.get_roster()
        self.send_presence()
def main():
    parser = ArgumentParser()
    parser.add_argument('-j', '--jid', dest='jid', help='JID to use')
    parser.add_argument('-p', '--password', dest='password', help='password to use')
    parser.add_argument('-d', '--debug', help='set logging to DEBUG', action='store_const', dest='loglevel', const=logging.DEBUG, default=logging.INFO)
    args = parser.parse_args()
    logging.basicConfig(level=args.loglevel, format='%(asctime)s %(levelname)-8s %(message)s')
    rollbot = RollBot(args.jid, args.password)
    rollbot.auto_authorize = True
    rollbot.auto_subscribe = True
    rollbot.register_plugin('xep_0030') # Service Discovery
    rollbot.register_plugin('xep_0092') # Software Version
    rollbot.register_plugin('xep_0199') # XMPP Ping
    rollbot.register_plugin('xep_0202') # Entity Time
    if rollbot.connect():
        logging.info('RollBot connected')
        rollbot.process(block=True)
        logging.info('RollBot disconnected')
    else:
        logging.fatal("RollBot couldn't connect")
if __name__ == '__main__':
    main()