mrDoctorWho/vk4xmpp

# coding: utf-8
# This file is a part of VK4XMPP transport
# © simpleApps, 2015 (27.08.15 09:31AM GMT)

"""
Implements XEP-0172: User Nickname (partially, case 4.2)
"""

GLOBAL_USER_SETTINGS["add_nicknames_prs"] = {"value": 0, "label": "Add nickname to presence stanzas"}


def add_username(stanza, user, uid):
    if uid == TransportID:
        name = IDENTIFIER["name"]
    else:
        key = "name"
        if user.settings.use_nicknames:
            key = "screen_name"
        name = user.vk.getData(uid).get(key, "Unknown User: %s" % uid)
    stanza.setTag("nick", namespace=xmpp.NS_NICK)
    stanza.setTagData("nick", name)


def add_nickname_msg03(msg, destination, source):
    if destination in Users and source != TransportID:  # That would be strange if user wasn't in Transport
        user = Users[destination]
        uid = vk2xmpp(source)
        strangers = getattr(user, "strangers", set([]))
        if uid not in strangers and uid not in user.friends:
            add_username(msg, user, uid)
            strangers.add(uid)
        user.strangers = strangers


def add_nickname_prs02(prs, destination, source):
    if destination in Users and not prs.getType():
        user = Users[destination]
        uid = vk2xmpp(source)
        if uid in user.friends and user.settings.add_nicknames_prs:
            add_username(prs, user, uid)


registerHandler("msg03", add_nickname_msg03)
registerHandler("prs02", add_nickname_prs02)