slixmpp-omemo/stanza.py

"""
    Slixmpp: The Slick XMPP Library
    Copyright (C) 2018 Maxime “pep” Buquet <pep@bouah.net>
    This file is part of Slixmpp.

    See the file LICENSE for copying permission.
"""

from slixmpp import Message
from slixmpp.plugins.xep_0060.stanza.pubsub import Item
from slixmpp.plugins.xep_0060.stanza.pubsub_event import EventItem
from slixmpp.xmlstream import register_stanza_plugin, ElementBase

OMEMO_BASE_NS = 'eu.siacs.conversations.axolotl'
OMEMO_DEVICES_NS = OMEMO_BASE_NS + '.devicelist'
OMEMO_BUNDLES_NS = OMEMO_BASE_NS + '.bundles'


class Devices(ElementBase):
    namespace = OMEMO_BASE_NS
    name = 'list'
    plugin_attrib = 'devices'
    interfaces = set()


class Device(ElementBase):
    namespace = OMEMO_BASE_NS
    name = 'device'
    plugin_attrib = name
    interfaces = {'id'}


class Encrypted(ElementBase):
    namespace = OMEMO_BASE_NS
    name = 'encrypted'
    plugin_attrib = 'omemo_encrypted'
    interfaces = set()


class Header(ElementBase):
    namespace = OMEMO_BASE_NS
    name = 'header'
    plugin_attrib = name
    interfaces = {'sid'}


class Key(ElementBase):
    namespace = OMEMO_BASE_NS
    name = 'key'
    plugin_attrib = name
    interfaces = {'rid', 'prekey'}


class IV(ElementBase):
    namespace = OMEMO_BASE_NS
    name = 'iv'
    plugin_attrib = name
    interfaces = set()


class Payload(ElementBase):
    namespace = OMEMO_BASE_NS
    name = 'payload'
    plugin_attrib = name
    interfaces = set()


class Bundle(ElementBase):
    namespace = OMEMO_BASE_NS
    name = 'bundle'
    plugin_attrib = name
    interfaces = set()


class SignedPreKeyPublic(ElementBase):
    namespace = OMEMO_BASE_NS
    name = 'signedPreKeyPublic'
    plugin_attrib = name
    interfaces = {'signedPreKeyId', 'value'}

    def get_value(self):
        return self.xml.text

    def set_value(self, value):
        self.xml.text = value

    def del_value(self):
        self.xml.text = ''


class SignedPreKeySignature(ElementBase):
    namespace = OMEMO_BASE_NS
    name = 'signedPreKeySignature'
    plugin_attrib = name
    interfaces = {'value'}

    def get_value(self):
        return self.xml.text

    def set_value(self, value):
        self.xml.text = value

    def del_value(self):
        self.xml.text = ''


class IdentityKey(ElementBase):
    namespace = OMEMO_BASE_NS
    name = 'identityKey'
    plugin_attrib = name
    interfaces = {'value'}

    def get_value(self):
        return self.xml.text

    def set_value(self, value):
        self.xml.text = value

    def del_value(self):
        self.xml.text = ''


class PreKeys(ElementBase):
    namespace = OMEMO_BASE_NS
    name = 'prekeys'
    plugin_attrib = name
    interfaces = set()


class PreKeyPublic(ElementBase):
    namespace = OMEMO_BASE_NS
    name = 'preKeyPublic'
    plugin_attrib = name
    interfaces = {'preKeyId'}


register_stanza_plugin(Message, Encrypted)
register_stanza_plugin(Encrypted, Header)
register_stanza_plugin(Header, Key)
register_stanza_plugin(Header, IV)
register_stanza_plugin(Encrypted, Payload)

register_stanza_plugin(EventItem, Devices)
register_stanza_plugin(Devices, Device, iterable=True)

register_stanza_plugin(Item, Bundle)
register_stanza_plugin(Bundle, SignedPreKeyPublic)
register_stanza_plugin(Bundle, SignedPreKeySignature)
register_stanza_plugin(Bundle, IdentityKey)
register_stanza_plugin(Bundle, PreKeys)
register_stanza_plugin(PreKeys, PreKeyPublic, iterable=True)