1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
# Copyright (C) 2018 Philipp Hörist <philipp AT hoerist.com>
#
# This file is part of nbxmpp.
#
# SPDX-License-Identifier: GPL-3.0-or-later
from __future__ import annotations
from typing import TYPE_CHECKING
from nbxmpp.modules.base import BaseModule
from nbxmpp.namespaces import Namespace
from nbxmpp.protocol import Message
from nbxmpp.protocol import Presence
from nbxmpp.structs import MessageProperties
from nbxmpp.structs import PresenceProperties
from nbxmpp.structs import StanzaHandler
if TYPE_CHECKING:
from nbxmpp.client import Client
class PGPLegacy(BaseModule):
def __init__(self, client: Client) -> None:
BaseModule.__init__(self, client)
self._client = client
self.handlers = [
StanzaHandler(
name="message",
callback=self._process_pgplegacy_message,
ns=Namespace.ENCRYPTED,
priority=7,
),
StanzaHandler(
name="presence",
callback=self._process_signed,
ns=Namespace.SIGNED,
priority=15,
),
]
@staticmethod
def _process_signed(
_client: Client, stanza: Presence, properties: PresenceProperties
) -> None:
signed = stanza.getTag("x", namespace=Namespace.SIGNED)
if signed is None:
return
properties.signed = signed.getData()
def _process_pgplegacy_message(
self, _client: Client, stanza: Message, properties: MessageProperties
) -> None:
pgplegacy = stanza.getTag("x", namespace=Namespace.ENCRYPTED)
if pgplegacy is None:
self._log.warning("No x node found")
self._log.warning(stanza)
return
data = pgplegacy.getData()
if not data:
self._log.warning("No data in x node found")
self._log.warning(stanza)
return
self._log.info("Encrypted message received")
properties.pgp_legacy = data
|
