summaryrefslogtreecommitdiff
path: root/poezio/core/handlers.py
diff options
context:
space:
mode:
authorGeorg Lukas <georg@op-co.de>2019-10-08 11:08:18 +0200
committerMaxime “pep” Buquet <pep@bouah.net>2019-10-10 14:57:19 +0200
commitd4d05aec0371008557a73348c975861c6b0cb707 (patch)
tree1aed7d9e221e6d91bfa19460b5d0046cfd428160 /poezio/core/handlers.py
parent5032905f96f92485e09d1599aeb3266f0d043ca9 (diff)
downloadpoezio-d4d05aec0371008557a73348c975861c6b0cb707.tar.gz
poezio-d4d05aec0371008557a73348c975861c6b0cb707.tar.bz2
poezio-d4d05aec0371008557a73348c975861c6b0cb707.tar.xz
poezio-d4d05aec0371008557a73348c975861c6b0cb707.zip
MUC-PM: centralize and improve PM detection in all code paths
Diffstat (limited to 'poezio/core/handlers.py')
-rw-r--r--poezio/core/handlers.py52
1 files changed, 40 insertions, 12 deletions
diff --git a/poezio/core/handlers.py b/poezio/core/handlers.py
index 7edf060b..da94f093 100644
--- a/poezio/core/handlers.py
+++ b/poezio/core/handlers.py
@@ -102,6 +102,33 @@ class HandlerCore:
self.core.xmpp['xep_0030'].get_info_from_domain(),
)
+ def is_known_muc_pm(self, message, with_jid):
+ """
+ Try to determine whether a given message is a MUC-PM, without a roundtrip. Returns None when it's not clear
+ """
+ # first, look for the x (XEP-0045 version 1.28)
+ if message.xml.find(
+ '{http://jabber.org/protocol/muc#user}x'
+ ) is not None:
+ log.debug('MUC-PM from %s with <x>',
+ with_jid)
+ return True
+ # then, look in the roster
+ if with_jid.bare in roster and roster[with_jid.bare].subscription != 'none':
+ return False
+ # then, check bookmarks
+ if with_jid.bare in self.core.bookmarks:
+ log.debug('MUC-PM from %s in bookmarks', with_jid)
+ return True
+ # then, look whether we know the MUC JID
+ for tab in self.core.get_tabs(tabs.MucTab):
+ if tab.jid.bare == with_jid.bare:
+ if with_jid.resource:
+ log.debug('MUC-PM from %s in known MucTab',
+ with_jid)
+ return True
+ return None
+
def on_carbon_received(self, message):
"""
Carbon <received/> received
@@ -118,8 +145,11 @@ class HandlerCore:
self.on_normal_message(recv)
recv = message['carbon_received']
- if (recv['from'].bare not in roster
- or roster[recv['from'].bare].subscription == 'none'):
+ is_muc_pm = self.is_known_muc_pm(recv, recv['from'])
+ if is_muc_pm:
+ log.debug('%s sent a MUC-PM, ignoring carbon', recv['from'])
+ return
+ if is_muc_pm is None:
fixes.has_identity(
self.core.xmpp,
recv['from'].server,
@@ -143,10 +173,11 @@ class HandlerCore:
self.on_normal_message(sent)
sent = message['carbon_sent']
- # todo: implement proper MUC detection logic
- if (sent['to'].resource
- and (sent['to'].bare not in roster
- or roster[sent['to'].bare].subscription == 'none')):
+ is_muc_pm = self.is_known_muc_pm(sent, sent['to'])
+ if is_muc_pm:
+ groupchat_private_message(sent)
+ return
+ if is_muc_pm is None:
fixes.has_identity(
self.core.xmpp,
sent['to'].server,
@@ -233,12 +264,9 @@ class HandlerCore:
if message['type'] == 'groupchat':
return
# Differentiate both type of messages, and call the appropriate handler.
- jid_from = message['from']
- for tab in self.core.get_tabs(tabs.MucTab):
- if tab.jid.bare == jid_from.bare:
- if jid_from.resource:
- self.on_groupchat_private_message(message, sent=False)
- return
+ if self.is_known_muc_pm(message, message['from']):
+ self.on_groupchat_private_message(message, sent=False)
+ return
self.on_normal_message(message)
def on_error_message(self, message):