slixmpp/examples/http_over_xmpp.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
    SleekXMPP: The Sleek XMPP Library
    Implementation of HTTP over XMPP transport
    http://xmpp.org/extensions/xep-0332.html
    Copyright (C) 2015 Riptide IO, sangeeth@riptideio.com
    This file is part of SleekXMPP.

    See the file LICENSE for copying permission.
"""

from sleekxmpp import ClientXMPP

from optparse import OptionParser
import logging
import getpass


class HTTPOverXMPPClient(ClientXMPP):
    def __init__(self, jid, password):
        ClientXMPP.__init__(self, jid, password)
        self.register_plugin('xep_0332')    # HTTP over XMPP Transport
        self.add_event_handler(
            'session_start', self.session_start, threaded=True
        )
        self.add_event_handler('http_request', self.http_request_received)
        self.add_event_handler('http_response', self.http_response_received)

    def http_request_received(self, iq):
        pass

    def http_response_received(self, iq):
        print 'HTTP Response Received : ', iq
        print 'From    : ', iq['from']
        print 'To      : ', iq['to']
        print 'Type    : ', iq['type']
        print 'Headers : ', iq['resp']['headers']
        print 'Code    : ', iq['resp']['code']
        print 'Message : ', iq['resp']['message']
        print 'Data    : ', iq['resp']['data']

    def session_start(self, event):
        # TODO: Fill in the blanks
        self['xep_0332'].send_request(
            to='?', method='?', resource='?', headers={}
        )
        self.disconnect()


if __name__ == '__main__':

    parser = OptionParser()

    # Output verbosity options.
    parser.add_option(
        '-v', '--verbose', help='set logging to DEBUG', action='store_const',
        dest='loglevel', const=logging.DEBUG, default=logging.ERROR
    )

    # JID and password options.
    parser.add_option('-J', '--jid', dest='jid', help='JID')
    parser.add_option('-P', '--password', dest='password', help='Password')

    # XMPP server ip and port options.
    parser.add_option(
        '-i', '--ipaddr', dest='ipaddr',
        help='IP Address of the XMPP server', default=None
    )
    parser.add_option(
        '-p', '--port', dest='port',
        help='Port of the XMPP server', default=None
    )

    opts, args = parser.parse_args()

    # Setup logging.
    logging.basicConfig(level=opts.loglevel,
                        format='%(levelname)-8s %(message)s')

    if opts.jid is None:
        opts.jid = raw_input('Username: ')
    if opts.password is None:
        opts.password = getpass.getpass('Password: ')

    xmpp = HTTPOverXMPPClient(opts.jid, opts.password)
    if xmpp.connect((opts.ipaddr, int(opts.port))):
        print 'Connected!'
        xmpp.process(block=True)
    else:
        print 'Not connected!'
    print 'Goodbye....'