Updated Example..

1 files changed, 81 insertions, 57 deletions

diff --git a/examples/http_over_xmpp.py b/examples/http_over_xmpp.py
index 5de8713e..5630f778 100644
--- a/examples/http_over_xmpp.py
+++ b/examples/http_over_xmpp.py
@@ -1,70 +1,94 @@
-from sleekxmpp import ClientXMPP
-import logging
+#!/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
 
-# def tracefunc(frame, event, arg, indent=[0]):
-#     prefix = "/Users/sangeeth/code/SleekXMPP/sleekxmpp/"
-#     if not frame.f_code.co_filename.startswith(prefix):
-#         return tracefunc
-#     if event == "call":
-#         indent[0] += 2
-#         cn = getattr(
-#             getattr(frame.f_locals.get("self"), "__class__", None),
-#             "__name__", None
-#         )
-#         print "{}{} {} {}".format(
-#             "." * indent[0], frame.f_code.co_filename[len(prefix):],
-#             cn, frame.f_code.co_name
-#         )
-#     elif event == "return":
-#         indent[0] -= 2
-#     return tracefunc
-#
-#
-# import sys
-# sys.settrace(tracefunc)
+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()
+
 
-        self.register_plugin('xep_0332')    # HTTP over XMPP
-    #     self.register_plugin('xep_0030')  # Service Discovery
-    #     self.register_plugin('xep_0004')  # Data Forms
-    #     self.register_plugin('xep_0060')  # PubSub
-    #     self.register_plugin('xep_0199')  # XMPP Ping
-    #
-    #     self.add_event_handler("session_start", self.session_start)
-    #     self.add_event_handler("connected", self.connected)
-    #
-    # def session_start(self, event):
-    #     print "Client::session_start()"
-    #     self.send_presence()
-    #     print self.get_roster()
-    #
-    # def connected(self, event):
-    #     print "Client::connected()"
-
-
-def get_cred(filename="/tmp/.cred"):
-    with open(filename, "r") as f:
-        return f.readline().split()
-
-
-if __name__ == "__main__":
-
-    logging.basicConfig(
-        level=logging.DEBUG, format='%(levelname)-8s %(message)s'
+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')
 
-    jid, password = get_cred()
-    xmpp = HTTPOverXMPPClient(jid, password)
-    if xmpp.connect(("talk.l.google.com", 5222)):
-        print "Connected!"
+    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...."
-
+        print 'Not connected!'
+    print 'Goodbye....'