Skip to content
This repository has been archived by the owner on Apr 26, 2024. It is now read-only.

Commit

Permalink
kill off send_nonmember_event
Browse files Browse the repository at this point in the history
This is now redundant, and we can just call `handle_new_client_event` directly.
  • Loading branch information
richvdh committed Oct 5, 2020
1 parent fd02822 commit e775b5b
Show file tree
Hide file tree
Showing 4 changed files with 31 additions and 53 deletions.
74 changes: 25 additions & 49 deletions synapse/handlers/message.py
Original file line number Diff line number Diff line change
Expand Up @@ -635,47 +635,6 @@ async def assert_accepted_privacy_policy(self, requester: Requester) -> None:
msg = self._block_events_without_consent_error % {"consent_uri": consent_uri}
raise ConsentNotGivenError(msg=msg, consent_uri=consent_uri)

async def send_nonmember_event(
self,
requester: Requester,
event: EventBase,
context: EventContext,
ratelimit: bool = True,
ignore_shadow_ban: bool = False,
) -> int:
"""
Persists and notifies local clients and federation of an event.
Args:
requester: The requester sending the event.
event: The event to send.
context: The context of the event.
ratelimit: Whether to rate limit this send.
ignore_shadow_ban: True if shadow-banned users should be allowed to
send this event.
Return:
The stream_id of the persisted event.
"""
if event.type == EventTypes.Member:
raise SynapseError(
500, "Tried to send member event through non-member codepath"
)

ev = await self.handle_new_client_event(
requester=requester,
event=event,
context=context,
ratelimit=ratelimit,
ignore_shadow_ban=ignore_shadow_ban,
)

# we know it was persisted, so must have a stream ordering
assert ev.internal_metadata.stream_ordering
return ev.internal_metadata.stream_ordering

async def deduplicate_state_event(
self, event: EventBase, context: EventContext
) -> Optional[EventBase]:
Expand Down Expand Up @@ -716,7 +675,7 @@ async def create_and_send_nonmember_event(
"""
Creates an event, then sends it.
See self.create_event and self.send_nonmember_event.
See self.create_event and self.handle_new_client_event.
Args:
requester: The requester sending the event.
Expand All @@ -726,9 +685,19 @@ async def create_and_send_nonmember_event(
ignore_shadow_ban: True if shadow-banned users should be allowed to
send this event.
Returns:
The event, and its stream ordering (if state event deduplication happened,
the previous, duplicate event).
Raises:
ShadowBanError if the requester has been shadow-banned.
"""

if event_dict["type"] == EventTypes.Member:
raise SynapseError(
500, "Tried to send member event through non-member codepath"
)

if not ignore_shadow_ban and requester.shadow_banned:
# We randomly sleep a bit just to annoy the requester.
await self.clock.sleep(random.randint(1, 10))
Expand All @@ -754,14 +723,17 @@ async def create_and_send_nonmember_event(
spam_error = "Spam is not permitted here"
raise SynapseError(403, spam_error, Codes.FORBIDDEN)

stream_id = await self.send_nonmember_event(
requester,
event,
context,
ev = await self.handle_new_client_event(
requester=requester,
event=event,
context=context,
ratelimit=ratelimit,
ignore_shadow_ban=ignore_shadow_ban,
)
return event, stream_id

# we know it was persisted, so must have a stream ordering
assert ev.internal_metadata.stream_ordering
return ev, ev.internal_metadata.stream_ordering

@measure_func("create_new_client_event")
async def create_new_client_event(
Expand Down Expand Up @@ -1255,8 +1227,12 @@ async def _send_dummy_event_for_room(self, room_id: str) -> bool:

# Since this is a dummy-event it is OK if it is sent by a
# shadow-banned user.
await self.send_nonmember_event(
requester, event, context, ratelimit=False, ignore_shadow_ban=True,
await self.handle_new_client_event(
requester=requester,
event=event,
context=context,
ratelimit=False,
ignore_shadow_ban=True,
)
return True
except ConsentNotGivenError:
Expand Down
4 changes: 2 additions & 2 deletions synapse/handlers/room.py
Original file line number Diff line number Diff line change
Expand Up @@ -230,8 +230,8 @@ async def _upgrade_room(
)

# now send the tombstone
await self.event_creation_handler.send_nonmember_event(
requester, tombstone_event, tombstone_context
await self.event_creation_handler.handle_new_client_event(
requester=requester, event=tombstone_event, context=tombstone_context,
)

old_room_state = await tombstone_context.get_current_state_ids()
Expand Down
2 changes: 1 addition & 1 deletion tests/handlers/test_register.py
Original file line number Diff line number Diff line change
Expand Up @@ -413,7 +413,7 @@ def test_auto_create_auto_join_room_preset_invalid_permissions(self):
)
)
self.get_success(
event_creation_handler.send_nonmember_event(requester, event, context)
event_creation_handler.handle_new_client_event(requester, event, context)
)

# Register a second user, which won't be be in the room (or even have an invite)
Expand Down
4 changes: 3 additions & 1 deletion tests/unittest.py
Original file line number Diff line number Diff line change
Expand Up @@ -608,7 +608,9 @@ def create_and_send_event(
if soft_failed:
event.internal_metadata.soft_failed = True

self.get_success(event_creator.send_nonmember_event(requester, event, context))
self.get_success(
event_creator.handle_new_client_event(requester, event, context)
)

return event.event_id

Expand Down

0 comments on commit e775b5b

Please sign in to comment.