123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141 |
- # Copyright 2018-2021 The Matrix.org Foundation C.I.C.
- #
- # Licensed under the Apache License, Version 2.0 (the "License");
- # you may not use this file except in compliance with the License.
- # You may obtain a copy of the License at
- #
- # http://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- import logging
- from typing import TYPE_CHECKING, List, Optional, Tuple
- from synapse.api.errors import SynapseError
- from synapse.handlers.room_member import RoomMemberHandler
- from synapse.replication.http.membership import (
- ReplicationRemoteJoinRestServlet as ReplRemoteJoin,
- ReplicationRemoteKnockRestServlet as ReplRemoteKnock,
- ReplicationRemoteRejectInviteRestServlet as ReplRejectInvite,
- ReplicationRemoteRescindKnockRestServlet as ReplRescindKnock,
- ReplicationUserJoinedLeftRoomRestServlet as ReplJoinedLeft,
- )
- from synapse.types import JsonDict, Requester, UserID
- if TYPE_CHECKING:
- from synapse.server import HomeServer
- logger = logging.getLogger(__name__)
- class RoomMemberWorkerHandler(RoomMemberHandler):
- def __init__(self, hs: "HomeServer"):
- super().__init__(hs)
- self._remote_join_client = ReplRemoteJoin.make_client(hs)
- self._remote_knock_client = ReplRemoteKnock.make_client(hs)
- self._remote_reject_client = ReplRejectInvite.make_client(hs)
- self._remote_rescind_client = ReplRescindKnock.make_client(hs)
- self._notify_change_client = ReplJoinedLeft.make_client(hs)
- async def _remote_join(
- self,
- requester: Requester,
- remote_room_hosts: List[str],
- room_id: str,
- user: UserID,
- content: dict,
- ) -> Tuple[str, int]:
- """Implements RoomMemberHandler._remote_join"""
- if len(remote_room_hosts) == 0:
- raise SynapseError(404, "No known servers")
- ret = await self._remote_join_client(
- requester=requester,
- remote_room_hosts=remote_room_hosts,
- room_id=room_id,
- user_id=user.to_string(),
- content=content,
- )
- return ret["event_id"], ret["stream_id"]
- async def remote_reject_invite(
- self,
- invite_event_id: str,
- txn_id: Optional[str],
- requester: Requester,
- content: dict,
- ) -> Tuple[str, int]:
- """
- Rejects an out-of-band invite received from a remote user
- Implements RoomMemberHandler.remote_reject_invite
- """
- ret = await self._remote_reject_client(
- invite_event_id=invite_event_id,
- txn_id=txn_id,
- requester=requester,
- content=content,
- )
- return ret["event_id"], ret["stream_id"]
- async def remote_rescind_knock(
- self,
- knock_event_id: str,
- txn_id: Optional[str],
- requester: Requester,
- content: JsonDict,
- ) -> Tuple[str, int]:
- """
- Rescinds a local knock made on a remote room
- Args:
- knock_event_id: the knock event
- txn_id: optional transaction ID supplied by the client
- requester: user making the request, according to the access token
- content: additional content to include in the leave event.
- Normally an empty dict.
- Returns:
- A tuple containing (event_id, stream_id of the leave event)
- """
- ret = await self._remote_rescind_client(
- knock_event_id=knock_event_id,
- txn_id=txn_id,
- requester=requester,
- content=content,
- )
- return ret["event_id"], ret["stream_id"]
- async def remote_knock(
- self,
- remote_room_hosts: List[str],
- room_id: str,
- user: UserID,
- content: dict,
- ) -> Tuple[str, int]:
- """Sends a knock to a room.
- Implements RoomMemberHandler.remote_knock
- """
- ret = await self._remote_knock_client(
- remote_room_hosts=remote_room_hosts,
- room_id=room_id,
- user=user,
- content=content,
- )
- return ret["event_id"], ret["stream_id"]
- async def _user_left_room(self, target: UserID, room_id: str) -> None:
- """Implements RoomMemberHandler._user_left_room"""
- await self._notify_change_client(
- user_id=target.to_string(), room_id=room_id, change="left"
- )
- async def forget(self, target: UserID, room_id: str) -> None:
- raise RuntimeError("Cannot forget rooms on workers.")
|