diff --git a/synapse/push/baserules.py b/synapse/push/baserules.py
index a17b35a605..4c7278b5a1 100644
--- a/synapse/push/baserules.py
+++ b/synapse/push/baserules.py
@@ -139,6 +139,7 @@ BASE_APPEND_CONTENT_RULES: List[Dict[str, Any]] = [
{
"kind": "event_match",
"key": "content.body",
+ # Match the localpart of the requester's MXID.
"pattern_type": "user_localpart",
}
],
@@ -191,6 +192,7 @@ BASE_APPEND_OVERRIDE_RULES: List[Dict[str, Any]] = [
"pattern": "invite",
"_cache_key": "_invite_member",
},
+ # Match the requester's MXID.
{"kind": "event_match", "key": "state_key", "pattern_type": "user_id"},
],
"actions": [
@@ -351,6 +353,18 @@ BASE_APPEND_UNDERRIDE_RULES: List[Dict[str, Any]] = [
],
},
{
+ "rule_id": "global/underride/.org.matrix.msc3772.thread_reply",
+ "conditions": [
+ {
+ "kind": "org.matrix.msc3772.relation_match",
+ "rel_type": "m.thread",
+ # Match the requester's MXID.
+ "sender_type": "user_id",
+ }
+ ],
+ "actions": ["notify", {"set_tweak": "highlight", "value": False}],
+ },
+ {
"rule_id": "global/underride/.m.rule.message",
"conditions": [
{
diff --git a/synapse/push/bulk_push_rule_evaluator.py b/synapse/push/bulk_push_rule_evaluator.py
index 4cc8a2ecca..1a8e7ef3dc 100644
--- a/synapse/push/bulk_push_rule_evaluator.py
+++ b/synapse/push/bulk_push_rule_evaluator.py
@@ -13,8 +13,9 @@
# See the License for the specific language governing permissions and
# limitations under the License.
+import itertools
import logging
-from typing import TYPE_CHECKING, Any, Dict, List, Optional, Set, Tuple, Union
+from typing import TYPE_CHECKING, Any, Dict, Iterable, List, Optional, Set, Tuple, Union
import attr
from prometheus_client import Counter
@@ -121,6 +122,9 @@ class BulkPushRuleEvaluator:
resizable=False,
)
+ # Whether to support MSC3772 is supported.
+ self._relations_match_enabled = self.hs.config.experimental.msc3772_enabled
+
async def _get_rules_for_event(
self, event: EventBase, context: EventContext
) -> Dict[str, List[Dict[str, Any]]]:
@@ -192,6 +196,60 @@ class BulkPushRuleEvaluator:
return pl_event.content if pl_event else {}, sender_level
+ async def _get_mutual_relations(
+ self, event: EventBase, rules: Iterable[Dict[str, Any]]
+ ) -> Dict[str, Set[Tuple[str, str]]]:
+ """
+ Fetch event metadata for events which related to the same event as the given event.
+
+ If the given event has no relation information, returns an empty dictionary.
+
+ Args:
+ event_id: The event ID which is targeted by relations.
+ rules: The push rules which will be processed for this event.
+
+ Returns:
+ A dictionary of relation type to:
+ A set of tuples of:
+ The sender
+ The event type
+ """
+
+ # If the experimental feature is not enabled, skip fetching relations.
+ if not self._relations_match_enabled:
+ return {}
+
+ # If the event does not have a relation, then cannot have any mutual
+ # relations.
+ relation = relation_from_event(event)
+ if not relation:
+ return {}
+
+ # Pre-filter to figure out which relation types are interesting.
+ rel_types = set()
+ for rule in rules:
+ # Skip disabled rules.
+ if "enabled" in rule and not rule["enabled"]:
+ continue
+
+ for condition in rule["conditions"]:
+ if condition["kind"] != "org.matrix.msc3772.relation_match":
+ continue
+
+ # rel_type is required.
+ rel_type = condition.get("rel_type")
+ if rel_type:
+ rel_types.add(rel_type)
+
+ # If no valid rules were found, no mutual relations.
+ if not rel_types:
+ return {}
+
+ # If any valid rules were found, fetch the mutual relations.
+ return await self.store.get_mutual_event_relations(
+ relation.parent_id, rel_types
+ )
+
@measure_func("action_for_event_by_user")
async def action_for_event_by_user(
self, event: EventBase, context: EventContext
@@ -216,8 +274,17 @@ class BulkPushRuleEvaluator:
sender_power_level,
) = await self._get_power_levels_and_sender_level(event, context)
+ relations = await self._get_mutual_relations(
+ event, itertools.chain(*rules_by_user.values())
+ )
+
evaluator = PushRuleEvaluatorForEvent(
- event, len(room_members), sender_power_level, power_levels
+ event,
+ len(room_members),
+ sender_power_level,
+ power_levels,
+ relations,
+ self._relations_match_enabled,
)
# If the event is not a state event check if any users ignore the sender.
diff --git a/synapse/push/clientformat.py b/synapse/push/clientformat.py
index 63b22d50ae..5117ef6854 100644
--- a/synapse/push/clientformat.py
+++ b/synapse/push/clientformat.py
@@ -48,6 +48,10 @@ def format_push_rules_for_user(
elif pattern_type == "user_localpart":
c["pattern"] = user.localpart
+ sender_type = c.pop("sender_type", None)
+ if sender_type == "user_id":
+ c["sender"] = user.to_string()
+
rulearray = rules["global"][template_name]
template_rule = _rule_to_template(r)
diff --git a/synapse/push/push_rule_evaluator.py b/synapse/push/push_rule_evaluator.py
index 54db6b5612..2e8a017add 100644
--- a/synapse/push/push_rule_evaluator.py
+++ b/synapse/push/push_rule_evaluator.py
@@ -15,7 +15,7 @@
import logging
import re
-from typing import Any, Dict, List, Mapping, Optional, Pattern, Tuple, Union
+from typing import Any, Dict, List, Mapping, Optional, Pattern, Set, Tuple, Union
from matrix_common.regex import glob_to_regex, to_word_pattern
@@ -120,11 +120,15 @@ class PushRuleEvaluatorForEvent:
room_member_count: int,
sender_power_level: int,
power_levels: Dict[str, Union[int, Dict[str, int]]],
+ relations: Dict[str, Set[Tuple[str, str]]],
+ relations_match_enabled: bool,
):
self._event = event
self._room_member_count = room_member_count
self._sender_power_level = sender_power_level
self._power_levels = power_levels
+ self._relations = relations
+ self._relations_match_enabled = relations_match_enabled
# Maps strings of e.g. 'content.body' -> event["content"]["body"]
self._value_cache = _flatten_dict(event)
@@ -188,7 +192,16 @@ class PushRuleEvaluatorForEvent:
return _sender_notification_permission(
self._event, condition, self._sender_power_level, self._power_levels
)
+ elif (
+ condition["kind"] == "org.matrix.msc3772.relation_match"
+ and self._relations_match_enabled
+ ):
+ return self._relation_match(condition, user_id)
else:
+ # XXX This looks incorrect -- we have reached an unknown condition
+ # kind and are unconditionally returning that it matches. Note
+ # that it seems possible to provide a condition to the /pushrules
+ # endpoint with an unknown kind, see _rule_tuple_from_request_object.
return True
def _event_match(self, condition: dict, user_id: str) -> bool:
@@ -256,6 +269,41 @@ class PushRuleEvaluatorForEvent:
return bool(r.search(body))
+ def _relation_match(self, condition: dict, user_id: str) -> bool:
+ """
+ Check an "relation_match" push rule condition.
+
+ Args:
+ condition: The "event_match" push rule condition to match.
+ user_id: The user's MXID.
+
+ Returns:
+ True if the condition matches the event, False otherwise.
+ """
+ rel_type = condition.get("rel_type")
+ if not rel_type:
+ logger.warning("relation_match condition missing rel_type")
+ return False
+
+ sender_pattern = condition.get("sender")
+ if sender_pattern is None:
+ sender_type = condition.get("sender_type")
+ if sender_type == "user_id":
+ sender_pattern = user_id
+ type_pattern = condition.get("type")
+
+ # If any other relations matches, return True.
+ for sender, event_type in self._relations.get(rel_type, ()):
+ if sender_pattern and not _glob_matches(sender_pattern, sender):
+ continue
+ if type_pattern and not _glob_matches(type_pattern, event_type):
+ continue
+ # All values must have matched.
+ return True
+
+ # No relations matched.
+ return False
+
# Caches (string, is_glob, word_boundary) -> regex for push. See _glob_matches
regex_cache: LruCache[Tuple[str, bool, bool], Pattern] = LruCache(
|
