Browse IMAP folders in service discorvery (IMAP connection not yet implemented)

darcs-hash:20071011204051-86b55-b2e67101f0396919e907fe98c42b9034bc3142d7.gz

src/jmc/jabber/disco.py

@@ -20,7 +20,18 @@
 ## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 ##
 
+import logging
+
+from pyxmpp.jid import JID
+from pyxmpp.jabber.disco import DiscoItems, DiscoItem
+
+from jcl import jabber
+from jcl.model import account
 from jcl.jabber.command import CommandRootDiscoGetInfoHandler
+from jcl.jabber.disco import AccountTypeDiscoGetInfoHandler, \
+     AccountDiscoGetInfoHandler, DiscoHandler
+
+from jmc.model.account import IMAPAccount
 
 class MailRootDiscoGetInfoHandler(CommandRootDiscoGetInfoHandler):
     def handle(self, stanza, lang_class, node, disco_obj, data):
@@ -30,3 +41,89 @@ class MailRootDiscoGetInfoHandler(CommandRootDiscoGetInfoHandler):
         disco_infos[0].add_feature("jabber:iq:gateway")
         disco_infos[0].add_identity(self.component.name, "headline", "newmail")
         return disco_infos
+
+class MailAccountTypeDiscoGetInfoHandler(AccountTypeDiscoGetInfoHandler):
+    def handle(self, stanza, lang_class, node, disco_obj, data):
+        disco_infos = AccountTypeDiscoGetInfoHandler.handle(self, stanza,
+                                                            lang_class,
+                                                            node, disco_obj,
+                                                            data)
+        if node == "IMAP" and account.get_accounts(stanza.get_from().bare(),
+                                                   IMAPAccount).count() > 0:
+            disco_infos[0].add_feature("http://jabber.org/protocol/disco#info")
+            disco_infos[0].add_feature("http://jabber.org/protocol/disco#items")
+        return disco_infos
+
+class IMAPAccountDiscoGetInfoHandler(AccountDiscoGetInfoHandler):
+    def handle(self, stanza, lang_class, node, disco_obj, data):
+        disco_infos = AccountDiscoGetInfoHandler.handle(self, stanza,
+                                                        lang_class,
+                                                        node, disco_obj,
+                                                        data)
+        splitted_node = node.split("/")
+        splitted_node_len = len(splitted_node)
+        if splitted_node_len > 1 and \
+            splitted_node[0] == "IMAP":
+            _account = account.get_account(stanza.get_from().bare(),
+                                           splitted_node[1],
+                                           IMAPAccount)
+            if _account is not None:
+                disco_infos[0].add_feature("http://jabber.org/protocol/disco#info")
+                if splitted_node_len > 2:
+                    imap_dir = "/".join(splitted_node[2:])
+                    if len(_account.ls_dir(imap_dir)) > 0:
+                        disco_infos[0].add_feature("http://jabber.org/protocol/disco#items")
+                else:
+                    disco_infos[0].add_feature("http://jabber.org/protocol/disco#items")
+        return disco_infos
+
+class IMAPAccountDiscoGetItemsHandler(DiscoHandler):
+    def __init__(self, component):
+        DiscoHandler.__init__(self, component)
+        self.__logger = logging.getLogger("jcl.jabber.IMAPAccountDiscoGetItemsHandler")
+
+    def filter(self, stanza, lang_class, node=None):
+        if node is None:
+            return None
+        splitted_node = node.split("/")
+        splitted_node_len = len(splitted_node)
+        if splitted_node_len < 2:
+            return None
+        account_type = splitted_node[0]
+        if account_type != "IMAP":
+            return None
+        account_name = splitted_node[1]
+        if splitted_node_len == 2:
+            imap_dir = None
+        else:
+            imap_dir = "/".join(splitted_node[2:])
+        return (account_type, account_name, imap_dir)
+
+    def handle(self, stanza, lang_class, node, disco_obj, data):
+        """Discovery get_items on an IMAP account"""
+        account_type, account_name, imap_dir = data
+        bare_from_jid = stanza.get_from().bare()
+        self.__logger.debug("Listing " + str(imap_dir) + " IMAP directory")
+        account_class = self.component.account_manager.get_account_class(account_type)
+        _account = account.get_account(bare_from_jid, account_name,
+                                       account_class=account_class)
+        if account_class is not None and _account is not None:
+            disco_items = DiscoItems()
+            if imap_dir is None:
+                # TODO : test if INBOX really exist, is it possible to retrieve default dir from IMAP ?
+                DiscoItem(disco_items,
+                          JID(unicode(_account.jid) + "/" + account_type
+                              + "/INBOX"),
+                          account_type + "/" + account_name + "/INBOX",
+                          "INBOX")
+            else:
+                for subdir in _account.ls_dir(imap_dir):
+                    DiscoItem(disco_items,
+                              JID(unicode(_account.jid) + "/" + account_type + \
+                                  "/" + imap_dir + "/" + subdir),
+                              account_type + "/" + account_name + "/" + imap_dir
+                              + "/" + subdir,
+                              subdir)
+            return [disco_items]
+        return []
+# TODO : implement get_info on imap_Dir