mirror of
https://mau.dev/maunium/synapse.git
synced 2024-08-28 13:11:56 +00:00
Merge branch 'develop' of github.com:matrix-org/synapse into erikj/search
This commit is contained in:
commit
323d3e506d
@ -54,7 +54,7 @@ class Filtering(object):
|
|||||||
]
|
]
|
||||||
|
|
||||||
room_level_definitions = [
|
room_level_definitions = [
|
||||||
"state", "events", "ephemeral"
|
"state", "timeline", "ephemeral"
|
||||||
]
|
]
|
||||||
|
|
||||||
for key in top_level_definitions:
|
for key in top_level_definitions:
|
||||||
@ -135,17 +135,23 @@ class Filter(object):
|
|||||||
def __init__(self, filter_json):
|
def __init__(self, filter_json):
|
||||||
self.filter_json = filter_json
|
self.filter_json = filter_json
|
||||||
|
|
||||||
def filter_public_user_data(self, events):
|
def timeline_limit(self):
|
||||||
return self._filter_on_key(events, ["public_user_data"])
|
return self.filter_json.get("room", {}).get("timeline", {}).get("limit", 10)
|
||||||
|
|
||||||
def filter_private_user_data(self, events):
|
def presence_limit(self):
|
||||||
return self._filter_on_key(events, ["private_user_data"])
|
return self.filter_json.get("presence", {}).get("limit", 10)
|
||||||
|
|
||||||
|
def ephemeral_limit(self):
|
||||||
|
return self.filter_json.get("room", {}).get("ephemeral", {}).get("limit", 10)
|
||||||
|
|
||||||
|
def filter_presence(self, events):
|
||||||
|
return self._filter_on_key(events, ["presence"])
|
||||||
|
|
||||||
def filter_room_state(self, events):
|
def filter_room_state(self, events):
|
||||||
return self._filter_on_key(events, ["room", "state"])
|
return self._filter_on_key(events, ["room", "state"])
|
||||||
|
|
||||||
def filter_room_events(self, events):
|
def filter_room_timeline(self, events):
|
||||||
return self._filter_on_key(events, ["room", "events"])
|
return self._filter_on_key(events, ["room", "timeline"])
|
||||||
|
|
||||||
def filter_room_ephemeral(self, events):
|
def filter_room_ephemeral(self, events):
|
||||||
return self._filter_on_key(events, ["room", "ephemeral"])
|
return self._filter_on_key(events, ["room", "ephemeral"])
|
||||||
@ -169,11 +175,34 @@ class Filter(object):
|
|||||||
return [e for e in events if self._passes_definition(definition, e)]
|
return [e for e in events if self._passes_definition(definition, e)]
|
||||||
|
|
||||||
def _passes_definition(self, definition, event):
|
def _passes_definition(self, definition, event):
|
||||||
|
"""Check if the event passes the filter definition
|
||||||
|
Args:
|
||||||
|
definition(dict): The filter definition to check against
|
||||||
|
event(dict or Event): The event to check
|
||||||
|
Returns:
|
||||||
|
True if the event passes the filter in the definition
|
||||||
|
"""
|
||||||
|
if type(event) is dict:
|
||||||
|
room_id = event.get("room_id")
|
||||||
|
sender = event.get("sender")
|
||||||
|
event_type = event["type"]
|
||||||
|
else:
|
||||||
|
room_id = getattr(event, "room_id", None)
|
||||||
|
sender = getattr(event, "sender", None)
|
||||||
|
event_type = event.type
|
||||||
|
return self._event_passes_definition(
|
||||||
|
definition, room_id, sender, event_type
|
||||||
|
)
|
||||||
|
|
||||||
|
def _event_passes_definition(self, definition, room_id, sender,
|
||||||
|
event_type):
|
||||||
"""Check if the event passes through the given definition.
|
"""Check if the event passes through the given definition.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
definition(dict): The definition to check against.
|
definition(dict): The definition to check against.
|
||||||
event(Event): The event to check.
|
room_id(str): The id of the room this event is in or None.
|
||||||
|
sender(str): The sender of the event
|
||||||
|
event_type(str): The type of the event.
|
||||||
Returns:
|
Returns:
|
||||||
True if the event passes through the filter.
|
True if the event passes through the filter.
|
||||||
"""
|
"""
|
||||||
@ -185,8 +214,7 @@ class Filter(object):
|
|||||||
# and 'not_types' then it is treated as only being in 'not_types')
|
# and 'not_types' then it is treated as only being in 'not_types')
|
||||||
|
|
||||||
# room checks
|
# room checks
|
||||||
if hasattr(event, "room_id"):
|
if room_id is not None:
|
||||||
room_id = event.room_id
|
|
||||||
allow_rooms = definition.get("rooms", None)
|
allow_rooms = definition.get("rooms", None)
|
||||||
reject_rooms = definition.get("not_rooms", None)
|
reject_rooms = definition.get("not_rooms", None)
|
||||||
if reject_rooms and room_id in reject_rooms:
|
if reject_rooms and room_id in reject_rooms:
|
||||||
@ -195,9 +223,7 @@ class Filter(object):
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
# sender checks
|
# sender checks
|
||||||
if hasattr(event, "sender"):
|
if sender is not None:
|
||||||
# Should we be including event.state_key for some event types?
|
|
||||||
sender = event.sender
|
|
||||||
allow_senders = definition.get("senders", None)
|
allow_senders = definition.get("senders", None)
|
||||||
reject_senders = definition.get("not_senders", None)
|
reject_senders = definition.get("not_senders", None)
|
||||||
if reject_senders and sender in reject_senders:
|
if reject_senders and sender in reject_senders:
|
||||||
@ -208,12 +234,12 @@ class Filter(object):
|
|||||||
# type checks
|
# type checks
|
||||||
if "not_types" in definition:
|
if "not_types" in definition:
|
||||||
for def_type in definition["not_types"]:
|
for def_type in definition["not_types"]:
|
||||||
if self._event_matches_type(event, def_type):
|
if self._event_matches_type(event_type, def_type):
|
||||||
return False
|
return False
|
||||||
if "types" in definition:
|
if "types" in definition:
|
||||||
included = False
|
included = False
|
||||||
for def_type in definition["types"]:
|
for def_type in definition["types"]:
|
||||||
if self._event_matches_type(event, def_type):
|
if self._event_matches_type(event_type, def_type):
|
||||||
included = True
|
included = True
|
||||||
break
|
break
|
||||||
if not included:
|
if not included:
|
||||||
@ -221,9 +247,9 @@ class Filter(object):
|
|||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
def _event_matches_type(self, event, def_type):
|
def _event_matches_type(self, event_type, def_type):
|
||||||
if def_type.endswith("*"):
|
if def_type.endswith("*"):
|
||||||
type_prefix = def_type[:-1]
|
type_prefix = def_type[:-1]
|
||||||
return event.type.startswith(type_prefix)
|
return event_type.startswith(type_prefix)
|
||||||
else:
|
else:
|
||||||
return event.type == def_type
|
return event_type == def_type
|
||||||
|
@ -345,9 +345,9 @@ class FilteringTestCase(unittest.TestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def test_filter_public_user_data_match(self):
|
def test_filter_presence_match(self):
|
||||||
user_filter_json = {
|
user_filter_json = {
|
||||||
"public_user_data": {
|
"presence": {
|
||||||
"types": ["m.*"]
|
"types": ["m.*"]
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -368,13 +368,13 @@ class FilteringTestCase(unittest.TestCase):
|
|||||||
filter_id=filter_id,
|
filter_id=filter_id,
|
||||||
)
|
)
|
||||||
|
|
||||||
results = user_filter.filter_public_user_data(events=events)
|
results = user_filter.filter_presence(events=events)
|
||||||
self.assertEquals(events, results)
|
self.assertEquals(events, results)
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def test_filter_public_user_data_no_match(self):
|
def test_filter_presence_no_match(self):
|
||||||
user_filter_json = {
|
user_filter_json = {
|
||||||
"public_user_data": {
|
"presence": {
|
||||||
"types": ["m.*"]
|
"types": ["m.*"]
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -395,7 +395,7 @@ class FilteringTestCase(unittest.TestCase):
|
|||||||
filter_id=filter_id,
|
filter_id=filter_id,
|
||||||
)
|
)
|
||||||
|
|
||||||
results = user_filter.filter_public_user_data(events=events)
|
results = user_filter.filter_presence(events=events)
|
||||||
self.assertEquals([], results)
|
self.assertEquals([], results)
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
|
Loading…
Reference in New Issue
Block a user