133 lines
4.4 KiB
Python
133 lines
4.4 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
SleekXMPP: The Sleek XMPP Library
|
|
Copyright (C) 2010 Nathanael C. Fritz
|
|
This file is part of SleekXMPP.
|
|
|
|
See the file LICENSE for copying permission.
|
|
"""
|
|
|
|
import sys
|
|
import logging
|
|
from optparse import OptionParser
|
|
|
|
#import sleekxmpp
|
|
#from sleekxmpp.componentxmpp import ComponentXMPP
|
|
|
|
#import slixmpp
|
|
from slixmpp.componentxmpp import ComponentXMPP
|
|
|
|
# Python versions before 3.0 do not use UTF-8 encoding
|
|
# by default. To ensure that Unicode is handled properly
|
|
# throughout SleekXMPP, we will set the default encoding
|
|
# ourselves to UTF-8.
|
|
if sys.version_info < (3, 0):
|
|
reload(sys)
|
|
sys.setdefaultencoding('utf8')
|
|
else:
|
|
raw_input = input
|
|
|
|
|
|
class EchoComponent(ComponentXMPP):
|
|
|
|
"""
|
|
A simple SleekXMPP component that echoes messages.
|
|
"""
|
|
|
|
def __init__(self, jid, secret, server, port):
|
|
ComponentXMPP.__init__(self, jid, secret, server, port)
|
|
|
|
# You don't need a session_start handler, but that is
|
|
# where you would broadcast initial presence.
|
|
|
|
# The message event is triggered whenever a message
|
|
# stanza is received. Be aware that that includes
|
|
# MUC messages and error messages.
|
|
self.add_event_handler("message", self.message)
|
|
|
|
def message(self, msg):
|
|
"""
|
|
Process incoming message stanzas. Be aware that this also
|
|
includes MUC messages and error messages. It is usually
|
|
a good idea to check the messages's type before processing
|
|
or sending replies.
|
|
|
|
Since a component may send messages from any number of JIDs,
|
|
it is best to always include a from JID.
|
|
|
|
Arguments:
|
|
msg -- The received message stanza. See the documentation
|
|
for stanza objects and the Message stanza to see
|
|
how it may be used.
|
|
"""
|
|
# The reply method will use the messages 'to' JID as the
|
|
# outgoing reply's 'from' JID.
|
|
|
|
if msg.find('{https://code.moparisthebest.com/moparisthebest/echo-self}echo') is not None:
|
|
to = msg['to']
|
|
msg['to'] = msg['from']
|
|
msg['from'] = to
|
|
|
|
# this might be cleaner, except sleekxmpp isn't sending body-less messages here at all, just silently swallowing them...
|
|
#to = msg['to']
|
|
#ffrom = msg['from']
|
|
#msg = msg.find('{urn:xmpp:forward:0}forwarded')
|
|
#msg['to'] = ffrom
|
|
#msg['from'] = to
|
|
|
|
msg.send()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# Setup the command line arguments.
|
|
optp = OptionParser()
|
|
|
|
# Output verbosity options.
|
|
optp.add_option('-q', '--quiet', help='set logging to ERROR',
|
|
action='store_const', dest='loglevel',
|
|
const=logging.ERROR, default=logging.INFO)
|
|
optp.add_option('-d', '--debug', help='set logging to DEBUG',
|
|
action='store_const', dest='loglevel',
|
|
const=logging.DEBUG, default=logging.INFO)
|
|
optp.add_option('-v', '--verbose', help='set logging to COMM',
|
|
action='store_const', dest='loglevel',
|
|
const=5, default=logging.INFO)
|
|
|
|
# JID and password options.
|
|
optp.add_option("-j", "--jid", dest="jid",
|
|
help="JID to use")
|
|
optp.add_option("-p", "--password", dest="password",
|
|
help="password to use")
|
|
optp.add_option("-s", "--server", dest="server",
|
|
help="server to connect to")
|
|
optp.add_option("-P", "--port", dest="port",
|
|
help="port to connect to")
|
|
|
|
opts, args = optp.parse_args()
|
|
|
|
if opts.jid is None or opts.server is None or opts.port is None:
|
|
print("must define jid, server, port and password")
|
|
exit(1)
|
|
if opts.password is None:
|
|
opts.password = ''
|
|
|
|
# Setup logging.
|
|
logging.basicConfig(level=opts.loglevel,
|
|
format='%(levelname)-8s %(message)s')
|
|
|
|
# Setup the EchoComponent and register plugins. Note that while plugins
|
|
# may have interdependencies, the order in which you register them does
|
|
# not matter.
|
|
xmpp = EchoComponent(opts.jid, opts.password, opts.server, opts.port)
|
|
xmpp.registerPlugin('xep_0030') # Service Discovery
|
|
xmpp.registerPlugin('xep_0199') # XMPP Ping
|
|
|
|
# Connect to the XMPP server and start processing XMPP stanzas.
|
|
if xmpp.connect():
|
|
xmpp.process(block=True)
|
|
print("Done")
|
|
else:
|
|
print("Unable to connect.")
|