Slixfeed/slixfeed/xmpp/profile.py

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

"""

NOTE

The VCard XML fields that can be set are as follows:
    ‘FN’, ‘NICKNAME’, ‘URL’, ‘BDAY’, ‘ROLE’, ‘NOTE’, ‘MAILER’,
    ‘TZ’, ‘REV’, ‘UID’, ‘DESC’, ‘TITLE’, ‘PRODID’, ‘SORT-STRING’,
    ‘N’, ‘ADR’, ‘TEL’, ‘EMAIL’, ‘JABBERID’, ‘ORG’, ‘CATEGORIES’,
    ‘NOTE’, ‘PRODID’, ‘REV’, ‘SORT-STRING’, ‘SOUND’, ‘UID’, ‘URL’,
    ‘CLASS’, ‘KEY’, ‘MAILER’, ‘GEO’, ‘TITLE’, ‘ROLE’,
    ‘LOGO’, ‘AGENT’

TODO

1) Test XEP-0084.

2) Make sure to support all type of servers.
    
3) Catch IqError
    ERROR:slixmpp.basexmpp:internal-server-error: Database failure
    WARNING:slixmpp.basexmpp:You should catch IqError exceptions

"""

import glob
from slixfeed.config import get_value, get_default_config_directory
# from slixmpp.exceptions import IqTimeout, IqError
# import logging
import os


async def update(self):
    """
    Update profile.
    """
    await set_vcard(self)
    await set_avatar(self)


async def set_avatar(self):
    config_dir = get_default_config_directory()
    if not os.path.isdir(config_dir):
        config_dir = '/usr/share/slixfeed/'
    filename = glob.glob(config_dir + '/image.*')
    if not filename and os.path.isdir('/usr/share/slixfeed/'):
        # filename = '/usr/share/slixfeed/image.svg'
        filename = glob.glob('/usr/share/slixfeed/image.*')
    if not filename:
        config_dir = os.path.dirname(__file__)
        config_dir = config_dir.split('/')
        config_dir.pop()
        config_dir = '/'.join(config_dir)
        filename = glob.glob(config_dir + '/assets/image.*')
    filename = filename[0]
    image_file = os.path.join(config_dir, filename)
    with open(image_file, 'rb') as avatar_file:
        avatar = avatar_file.read()
        # await self.plugin['xep_0084'].publish_avatar(avatar)
        await self.plugin['xep_0153'].set_avatar(avatar=avatar)


def set_identity(self, category):
    """
    Identify for Service Descovery.

    Parameters
    ----------
    category : str
        "client" or "service".

    Returns
    -------
    None.

    """
    self['xep_0030'].add_identity(
        category=category,
        itype='news',
        name='slixfeed',
        node=None,
        jid=self.boundjid.full,
    )


async def set_vcard(self):
    vcard = self.plugin['xep_0054'].make_vcard()
    fields = {
        'BDAY': 'birthday',
        'DESC': 'description',
        'FN': 'name',
        'NICKNAME': 'nickname',
        'NOTE': 'note',
        'ORG': 'organization',
        'ROLE': 'role',
        'TITLE': 'title',
        'URL': 'url',
        }
    for key in fields:
        vcard[key] = get_value('accounts', 'XMPP Profile', fields[key])
    await self.plugin['xep_0054'].publish_vcard(vcard)