1 files changed, 54 insertions, 0 deletions
diff --git a/server/component.py b/server/component.py
new file mode 100644
index 0000000..198b159
--- /dev/null
+++ b/server/component.py
@@ -0,0 +1,54 @@
+import logging
+logger = logging.getLogger(__name__)
+from sleekxmpp.componentxmpp import ComponentXMPP
+from sleekxmpp.xmlstream.xmlstream import XMLStream
+
+from user import UserHandler
+
+class ObjectComponent(ComponentXMPP):
+
+    def __init__(self, jid, secret, server, port, root):
+        ComponentXMPP.__init__(self, jid, secret, server, port)
+        self.register_plugin('xep_0030')
+        self.register_plugin('xep_0077', pconfig = {'root': root})
+        self.plugin['xep_0077']
+        self.register_plugin("AliasPlugin", module = "alias_plugin", pconfig = {'root': root})
+        self.add_event_handler("session_start", self.start)
+        self.add_event_handler("presence_probe", self.presence_probe)
+        self.add_event_handler("message", self.message)
+        self.add_event_handler("changed_subscription", self.presence_subscription)
+        self.userHandler = UserHandler(root)
+
+    def start(self, event):
+        for user in self.userHandler.get_user_list():
+            self.send_presence(pto = user)
+
+    def disconnect(self, reconnect = False):
+        for user in self.userHandler.get_user_list():
+            self.send_presence(pto = user, ptype = "unavailable")
+        XMLStream.disconnect(self, reconnect)
+        logger.info('Component {} disconnected'.format(self.boundjid.bare))
+
+    def message(self, msg):
+        msg.reply("Thanks for sending\n{[body]}".format(msg)).send()
+
+    def presence_subscription(self, subscription):
+        if subscription["type"] == "subscribe":
+            userJID = subscription["from"].full
+            if not self.userHandler.registered(userJID):
+                self.userHandler.register(userJID)
+                logger.info('registering user {}'.format(userJID))
+            subscription.reply().send()
+            self.send_presence(pto = userJID)
+            self.send_presence_subscription(pto = userJID, ptype = "subscribe")
+        if subscription["type"] == "unsubscribe":
+            userJID = subscription["from"].full
+            if self.userHandler.registered(userJID):
+                self.userHandler.unregister(userJID)
+                logger.info('unregistering user {}'.format(userJID))
+
+    def presence_probe(self, event):
+        self.send_presence(pto = event["from"].full)
+
+
+