Slixfeed/slixfeed/xmpp/bookmark.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""

TODO

1) Save groupchat name instead of jid in field name.

"""

from slixmpp.plugins.xep_0048.stanza import Bookmarks


class XmppBookmark:


    async def get(self):
        result = await self.plugin['xep_0048'].get_bookmarks()
        conferences = result['private']['bookmarks']['conferences']
        return conferences


    async def properties(self, jid):
        result = await self.plugin['xep_0048'].get_bookmarks()
        groupchats = result['private']['bookmarks']['conferences']
        for groupchat in groupchats:
            if jid == groupchat['jid']:
                properties = {'password': groupchat['password'],
                              'jid': groupchat['jid'],
                              'name': groupchat['name'],
                              'nick': groupchat['nick'],
                              'autojoin': groupchat['autojoin'],
                              'lang': groupchat['lang']}
                break
        return properties


    async def add(self, jid=None, properties=None):
        result = await self.plugin['xep_0048'].get_bookmarks()
        conferences = result['private']['bookmarks']['conferences']
        groupchats = []
        for conference in conferences:
            groupchats.extend([conference])
        if properties:
            properties['jid'] = properties['room'] + '@' + properties['host']
            if not properties['alias']: properties['alias'] = self.alias
                
        else:
            properties = {
                'jid' : jid,
                'alias' : self.alias,
                'name' : jid.split('@')[0],
                'autojoin' : True,
                'password' : None,
                }
        if jid not in groupchats:
            bookmarks = Bookmarks()
            for groupchat in groupchats:
                # if groupchat['jid'] == groupchat['name']:
                #     groupchat['name'] = groupchat['name'].split('@')[0]
                bookmarks.add_conference(groupchat['jid'],
                                         groupchat['alias'],
                                         name=groupchat['name'],
                                         autojoin=groupchat['autojoin'],
                                         password=groupchat['password'])
            bookmarks.add_conference(properties['jid'],
                                     properties['alias'],
                                     name=properties['name'],
                                     autojoin=properties['autojoin'],
                                     password=properties['password'])
            await self.plugin['xep_0048'].set_bookmarks(bookmarks)
        # bookmarks = Bookmarks()
        # await self.plugin['xep_0048'].set_bookmarks(bookmarks)
        # print(await self.plugin['xep_0048'].get_bookmarks())

        # bm = BookmarkStorage()
        # bm.conferences.append(Conference(muc_jid, autojoin=True, nick=self.alias))
        # await self['xep_0402'].publish(bm)


    async def remove(self, jid):
        result = await self.plugin['xep_0048'].get_bookmarks()
        conferences = result['private']['bookmarks']['conferences']
        groupchats = []
        for conference in conferences:
            if not conference['jid'] == jid:
                groupchats.extend([conference])
        bookmarks = Bookmarks()
        for groupchat in groupchats:
            bookmarks.add_conference(groupchat['jid'],
                                     groupchat['nick'],
                                     name=groupchat['name'],
                                     autojoin=groupchat['autojoin'],
                                     password=groupchat['password'])
        await self.plugin['xep_0048'].set_bookmarks(bookmarks)