Add support for XEP-0319, idle presence

1 files changed, 70 insertions, 0 deletions

diff --git a/sleekxmpp/plugins/xep_0319/idle.py b/sleekxmpp/plugins/xep_0319/idle.py
new file mode 100644
index 00000000..8f3d104b
--- /dev/null
+++ b/sleekxmpp/plugins/xep_0319/idle.py
@@ -0,0 +1,70 @@
+"""
+    SleekXMPP: The Sleek XMPP Library
+    Copyright (C) 2013 Nathanael C. Fritz, Lance J.T. Stout
+    This file is part of SleekXMPP.
+
+    See the file LICENSE for copying permission.
+"""
+
+from datetime import datetime, timedelta
+
+from sleekxmpp.stanza import Presence
+from sleekxmpp.plugins import BasePlugin
+from sleekxmpp.xmlstream import register_stanza_plugin
+from sleekxmpp.xmlstream.handler import Callback
+from sleekxmpp.xmlstream.matcher import StanzaPath
+from sleekxmpp.plugins.xep_0319 import stanza
+
+
+class XEP_0319(BasePlugin):
+    name = 'xep_0319'
+    description = 'XEP-0319: Last User Interaction in Presence'
+    dependencies = set()
+    stanza = stanza
+
+    def plugin_init(self):
+        self._idle_stamps = {}
+        register_stanza_plugin(Presence, stanza.Idle)
+        self.api.register(self._set_idle,
+                'set_idle',
+                default=True)
+        self.api.register(self._get_idle,
+                'get_idle',
+                default=True)
+        self.xmpp.register_handler(
+                Callback('Idle Presence',
+                    StanzaPath('presence/idle'),
+                    self._idle_presence))
+        self.xmpp.add_filter('out', self._stamp_idle_presence)
+
+    def session_bind(self, jid):
+        self.xmpp['xep_0030'].add_feature('urn:xmpp:idle:0')
+
+    def plugin_end(self):
+        self.xmpp['xep_0030'].del_feature(feature='urn:xmpp:idle:0')
+        self.xmpp.del_filter('out', self._stamp_idle_presence)
+        self.xmpp.remove_handler('Idle Presence')
+
+    def idle(self, jid=None, since=None):
+        if since is None:
+            since = datetime.now()
+        self.api['set_idle'](jid, None, None, since)
+
+    def active(self, jid=None):
+        self.api['set_idle'](jid, None, None, None)
+
+    def _set_idle(self, jid, node, ifrom, data):
+        self._idle_stamps[jid] = data
+
+    def _get_idle(self, jid, node, ifrom, data):
+        return self._idle_stamps.get(jid, None)
+
+    def _idle_presence(self, pres):
+        self.xmpp.event('presence_idle', pres)
+
+    def _stamp_idle_presence(self, stanza):
+        if isinstance(stanza, Presence):
+            since = self.api['get_idle'](stanza['from'] or self.xmpp.boundjid)
+            if since:
+                stanza['idle']['since'] = since
+        return stanza