123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224 |
- # -*- coding: utf-8 -*-
- # Copyright 2014-2016 OpenMarket Ltd
- #
- # 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.
- from twisted.internet import defer
- from synapse.api.errors import AuthError, SynapseError, Codes
- from synapse.types import RoomAlias
- from synapse.http.servlet import parse_json_object_from_request
- from .base import ClientV1RestServlet, client_path_patterns
- import logging
- logger = logging.getLogger(__name__)
- def register_servlets(hs, http_server):
- ClientDirectoryServer(hs).register(http_server)
- ClientDirectoryListServer(hs).register(http_server)
- ClientAppserviceDirectoryListServer(hs).register(http_server)
- class ClientDirectoryServer(ClientV1RestServlet):
- PATTERNS = client_path_patterns("/directory/room/(?P<room_alias>[^/]*)$")
- def __init__(self, hs):
- super(ClientDirectoryServer, self).__init__(hs)
- self.store = hs.get_datastore()
- self.handlers = hs.get_handlers()
- @defer.inlineCallbacks
- def on_GET(self, request, room_alias):
- room_alias = RoomAlias.from_string(room_alias)
- dir_handler = self.handlers.directory_handler
- res = yield dir_handler.get_association(room_alias)
- defer.returnValue((200, res))
- @defer.inlineCallbacks
- def on_PUT(self, request, room_alias):
- content = parse_json_object_from_request(request)
- if "room_id" not in content:
- raise SynapseError(400, "Missing room_id key",
- errcode=Codes.BAD_JSON)
- logger.debug("Got content: %s", content)
- room_alias = RoomAlias.from_string(room_alias)
- logger.debug("Got room name: %s", room_alias.to_string())
- room_id = content["room_id"]
- servers = content["servers"] if "servers" in content else None
- logger.debug("Got room_id: %s", room_id)
- logger.debug("Got servers: %s", servers)
- # TODO(erikj): Check types.
- room = yield self.store.get_room(room_id)
- if room is None:
- raise SynapseError(400, "Room does not exist")
- dir_handler = self.handlers.directory_handler
- try:
- # try to auth as a user
- requester = yield self.auth.get_user_by_req(request)
- try:
- user_id = requester.user.to_string()
- yield dir_handler.create_association(
- user_id, room_alias, room_id, servers
- )
- yield dir_handler.send_room_alias_update_event(
- requester,
- user_id,
- room_id
- )
- except SynapseError as e:
- raise e
- except Exception:
- logger.exception("Failed to create association")
- raise
- except AuthError:
- # try to auth as an application service
- service = yield self.auth.get_appservice_by_req(request)
- yield dir_handler.create_appservice_association(
- service, room_alias, room_id, servers
- )
- logger.info(
- "Application service at %s created alias %s pointing to %s",
- service.url,
- room_alias.to_string(),
- room_id
- )
- defer.returnValue((200, {}))
- @defer.inlineCallbacks
- def on_DELETE(self, request, room_alias):
- dir_handler = self.handlers.directory_handler
- try:
- service = yield self.auth.get_appservice_by_req(request)
- room_alias = RoomAlias.from_string(room_alias)
- yield dir_handler.delete_appservice_association(
- service, room_alias
- )
- logger.info(
- "Application service at %s deleted alias %s",
- service.url,
- room_alias.to_string()
- )
- defer.returnValue((200, {}))
- except AuthError:
- # fallback to default user behaviour if they aren't an AS
- pass
- requester = yield self.auth.get_user_by_req(request)
- user = requester.user
- room_alias = RoomAlias.from_string(room_alias)
- yield dir_handler.delete_association(
- requester, user.to_string(), room_alias
- )
- logger.info(
- "User %s deleted alias %s",
- user.to_string(),
- room_alias.to_string()
- )
- defer.returnValue((200, {}))
- class ClientDirectoryListServer(ClientV1RestServlet):
- PATTERNS = client_path_patterns("/directory/list/room/(?P<room_id>[^/]*)$")
- def __init__(self, hs):
- super(ClientDirectoryListServer, self).__init__(hs)
- self.store = hs.get_datastore()
- self.handlers = hs.get_handlers()
- @defer.inlineCallbacks
- def on_GET(self, request, room_id):
- room = yield self.store.get_room(room_id)
- if room is None:
- raise SynapseError(400, "Unknown room")
- defer.returnValue((200, {
- "visibility": "public" if room["is_public"] else "private"
- }))
- @defer.inlineCallbacks
- def on_PUT(self, request, room_id):
- requester = yield self.auth.get_user_by_req(request)
- content = parse_json_object_from_request(request)
- visibility = content.get("visibility", "public")
- yield self.handlers.directory_handler.edit_published_room_list(
- requester, room_id, visibility,
- )
- defer.returnValue((200, {}))
- @defer.inlineCallbacks
- def on_DELETE(self, request, room_id):
- requester = yield self.auth.get_user_by_req(request)
- yield self.handlers.directory_handler.edit_published_room_list(
- requester, room_id, "private",
- )
- defer.returnValue((200, {}))
- class ClientAppserviceDirectoryListServer(ClientV1RestServlet):
- PATTERNS = client_path_patterns(
- "/directory/list/appservice/(?P<network_id>[^/]*)/(?P<room_id>[^/]*)$"
- )
- def __init__(self, hs):
- super(ClientAppserviceDirectoryListServer, self).__init__(hs)
- self.store = hs.get_datastore()
- self.handlers = hs.get_handlers()
- def on_PUT(self, request, network_id, room_id):
- content = parse_json_object_from_request(request)
- visibility = content.get("visibility", "public")
- return self._edit(request, network_id, room_id, visibility)
- def on_DELETE(self, request, network_id, room_id):
- return self._edit(request, network_id, room_id, "private")
- @defer.inlineCallbacks
- def _edit(self, request, network_id, room_id, visibility):
- requester = yield self.auth.get_user_by_req(request)
- if not requester.app_service:
- raise AuthError(
- 403, "Only appservices can edit the appservice published room list"
- )
- yield self.handlers.directory_handler.edit_published_appservice_room_list(
- requester.app_service.id, network_id, room_id, visibility,
- )
- defer.returnValue((200, {}))
|