mirror of https://github.com/watcha-fr/synapse
parent
b3de67234e
commit
2f9eafdd36
@ -0,0 +1,278 @@ |
||||
# -*- coding: utf-8 -*- |
||||
# Copyright 2017 Vector Creations 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 SynapseError |
||||
|
||||
import logging |
||||
|
||||
logger = logging.getLogger(__name__) |
||||
|
||||
|
||||
# TODO: Validate attestations |
||||
# TODO: Allow users to "knock" or simpkly join depending on rules |
||||
# TODO: is_priveged flag to users and is_public to users and rooms |
||||
# TODO: Roles |
||||
# TODO: Audit log for admins (profile updates, membership changes, users who tried |
||||
# to join but were rejected, etc) |
||||
# TODO: Flairs |
||||
# TODO: Add group memebership /sync |
||||
|
||||
|
||||
def _create_rerouter(name): |
||||
def f(self, group_id, *args, **kwargs): |
||||
if self.is_mine_id(group_id): |
||||
return getattr(self.groups_server_handler, name)( |
||||
group_id, *args, **kwargs |
||||
) |
||||
|
||||
repl_layer = self.hs.get_replication_layer() |
||||
return getattr(repl_layer, name)(group_id, *args, **kwargs) |
||||
return f |
||||
|
||||
|
||||
class GroupsLocalHandler(object): |
||||
def __init__(self, hs): |
||||
self.hs = hs |
||||
self.store = hs.get_datastore() |
||||
self.room_list_handler = hs.get_room_list_handler() |
||||
self.groups_server_handler = hs.get_groups_server_handler() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.keyring = hs.get_keyring() |
||||
self.is_mine_id = hs.is_mine_id |
||||
self.signing_key = hs.config.signing_key[0] |
||||
self.server_name = hs.hostname |
||||
self.attestations = hs.get_groups_attestation_signing() |
||||
|
||||
# Ensure attestations get renewed |
||||
hs.get_groups_attestation_renewer() |
||||
|
||||
get_group_profile = _create_rerouter("get_group_profile") |
||||
get_rooms_in_group = _create_rerouter("get_rooms_in_group") |
||||
|
||||
update_group_summary_room = _create_rerouter("update_group_summary_room") |
||||
delete_group_summary_room = _create_rerouter("delete_group_summary_room") |
||||
|
||||
update_group_category = _create_rerouter("update_group_category") |
||||
delete_group_category = _create_rerouter("delete_group_category") |
||||
get_group_category = _create_rerouter("get_group_category") |
||||
get_group_categories = _create_rerouter("get_group_categories") |
||||
|
||||
update_group_summary_user = _create_rerouter("update_group_summary_user") |
||||
delete_group_summary_user = _create_rerouter("delete_group_summary_user") |
||||
|
||||
update_group_role = _create_rerouter("update_group_role") |
||||
delete_group_role = _create_rerouter("delete_group_role") |
||||
get_group_role = _create_rerouter("get_group_role") |
||||
get_group_roles = _create_rerouter("get_group_roles") |
||||
|
||||
@defer.inlineCallbacks |
||||
def get_group_summary(self, group_id, requester_user_id): |
||||
if self.is_mine_id(group_id): |
||||
res = yield self.groups_server_handler.get_group_summary( |
||||
group_id, requester_user_id |
||||
) |
||||
defer.returnValue(res) |
||||
|
||||
repl_layer = self.hs.get_replication_layer() |
||||
res = yield repl_layer.get_group_summary(group_id, requester_user_id) |
||||
|
||||
chunk = res["users_section"]["users"] |
||||
valid_users = [] |
||||
for entry in chunk: |
||||
g_user_id = entry["user_id"] |
||||
attestation = entry.pop("attestation") |
||||
try: |
||||
yield self.attestations.verify_attestation( |
||||
attestation, |
||||
group_id=group_id, |
||||
user_id=g_user_id, |
||||
) |
||||
valid_users.append(entry) |
||||
except Exception as e: |
||||
logger.info("Failed to verify user is in group: %s", e) |
||||
|
||||
res["users_section"]["users"] = valid_users |
||||
|
||||
res["users_section"]["users"].sort(key=lambda e: e.get("order", 0)) |
||||
res["rooms_section"]["rooms"].sort(key=lambda e: e.get("order", 0)) |
||||
|
||||
defer.returnValue(res) |
||||
|
||||
def create_group(self, group_id, user_id, content): |
||||
logger.info("Asking to create group with ID: %r", group_id) |
||||
|
||||
if self.is_mine_id(group_id): |
||||
return self.groups_server_handler.create_group( |
||||
group_id, user_id, content |
||||
) |
||||
|
||||
repl_layer = self.hs.get_replication_layer() |
||||
return repl_layer.create_group(group_id, user_id, content) # TODO |
||||
|
||||
def add_room(self, group_id, user_id, room_id, content): |
||||
if self.is_mine_id(group_id): |
||||
return self.groups_server_handler.add_room( |
||||
group_id, user_id, room_id, content |
||||
) |
||||
|
||||
repl_layer = self.hs.get_replication_layer() |
||||
return repl_layer.add_room_to_group(group_id, user_id, room_id, content) # TODO |
||||
|
||||
@defer.inlineCallbacks |
||||
def get_users_in_group(self, group_id, requester_user_id): |
||||
if self.is_mine_id(group_id): |
||||
res = yield self.groups_server_handler.get_users_in_group( |
||||
group_id, requester_user_id |
||||
) |
||||
defer.returnValue(res) |
||||
|
||||
repl_layer = self.hs.get_replication_layer() |
||||
res = yield repl_layer.get_users_in_group(group_id, requester_user_id) # TODO |
||||
|
||||
chunk = res["chunk"] |
||||
valid_entries = [] |
||||
for entry in chunk: |
||||
g_user_id = entry["user_id"] |
||||
attestation = entry.pop("attestation") |
||||
try: |
||||
yield self.attestations.verify_attestation( |
||||
attestation, |
||||
group_id=group_id, |
||||
user_id=g_user_id, |
||||
) |
||||
valid_entries.append(entry) |
||||
except Exception as e: |
||||
logger.info("Failed to verify user is in group: %s", e) |
||||
|
||||
res["chunk"] = valid_entries |
||||
|
||||
defer.returnValue(res) |
||||
|
||||
@defer.inlineCallbacks |
||||
def join_group(self, group_id, user_id, content): |
||||
raise NotImplementedError() # TODO |
||||
|
||||
@defer.inlineCallbacks |
||||
def accept_invite(self, group_id, user_id, content): |
||||
if self.is_mine_id(group_id): |
||||
yield self.groups_server_handler.accept_invite( |
||||
group_id, user_id, content |
||||
) |
||||
local_attestation = None |
||||
remote_attestation = None |
||||
else: |
||||
local_attestation = self.attestations.create_attestation(group_id, user_id) |
||||
content["attestation"] = local_attestation |
||||
|
||||
repl_layer = self.hs.get_replication_layer() |
||||
res = yield repl_layer.accept_group_invite(group_id, user_id, content) |
||||
|
||||
remote_attestation = res["attestation"] |
||||
|
||||
yield self.attestations.verify_attestation( |
||||
remote_attestation, |
||||
group_id=group_id, |
||||
user_id=user_id, |
||||
) |
||||
|
||||
yield self.store.register_user_group_membership( |
||||
group_id, user_id, |
||||
membership="join", |
||||
is_admin=False, |
||||
local_attestation=local_attestation, |
||||
remote_attestation=remote_attestation, |
||||
) |
||||
|
||||
defer.returnValue({}) |
||||
|
||||
@defer.inlineCallbacks |
||||
def invite(self, group_id, user_id, requester_user_id, config): |
||||
content = { |
||||
"requester_user_id": requester_user_id, |
||||
"config": config, |
||||
} |
||||
if self.is_mine_id(group_id): |
||||
res = yield self.groups_server_handler.invite_to_group( |
||||
group_id, user_id, requester_user_id, content, |
||||
) |
||||
else: |
||||
repl_layer = self.hs.get_replication_layer() |
||||
res = yield repl_layer.invite_to_group( |
||||
group_id, user_id, content, |
||||
) |
||||
|
||||
defer.returnValue(res) |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_invite(self, group_id, user_id, content): |
||||
# TODO: Support auto join and rejection |
||||
|
||||
if not self.is_mine_id(user_id): |
||||
raise SynapseError(400, "User not on this server") |
||||
|
||||
local_profile = {} |
||||
if "profile" in content: |
||||
if "name" in content["profile"]: |
||||
local_profile["name"] = content["profile"]["name"] |
||||
if "avatar_url" in content["profile"]: |
||||
local_profile["avatar_url"] = content["profile"]["avatar_url"] |
||||
|
||||
yield self.store.register_user_group_membership( |
||||
group_id, user_id, |
||||
membership="invite", |
||||
content={"profile": local_profile, "inviter": content["inviter"]}, |
||||
) |
||||
|
||||
defer.returnValue({"state": "invite"}) |
||||
|
||||
@defer.inlineCallbacks |
||||
def remove_user_from_group(self, group_id, user_id, requester_user_id, content): |
||||
if user_id == requester_user_id: |
||||
yield self.store.register_user_group_membership( |
||||
group_id, user_id, |
||||
membership="leave", |
||||
) |
||||
|
||||
# TODO: Should probably remember that we tried to leave so that we can |
||||
# retry if the group server is currently down. |
||||
|
||||
if self.is_mine_id(group_id): |
||||
res = yield self.groups_server_handler.remove_user_from_group( |
||||
group_id, user_id, requester_user_id, content, |
||||
) |
||||
else: |
||||
content["requester_user_id"] = requester_user_id |
||||
repl_layer = self.hs.get_replication_layer() |
||||
res = yield repl_layer.remove_user_from_group( |
||||
group_id, user_id, content |
||||
) # TODO |
||||
|
||||
defer.returnValue(res) |
||||
|
||||
@defer.inlineCallbacks |
||||
def user_removed_from_group(self, group_id, user_id, content): |
||||
# TODO: Check if user in group |
||||
yield self.store.register_user_group_membership( |
||||
group_id, user_id, |
||||
membership="leave", |
||||
) |
||||
|
||||
@defer.inlineCallbacks |
||||
def get_joined_groups(self, user_id): |
||||
group_ids = yield self.store.get_joined_groups(user_id) |
||||
defer.returnValue({"groups": group_ids}) |
@ -0,0 +1,642 @@ |
||||
# -*- coding: utf-8 -*- |
||||
# Copyright 2017 Vector Creations 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.http.servlet import RestServlet, parse_json_object_from_request |
||||
from synapse.types import GroupID |
||||
|
||||
from ._base import client_v2_patterns |
||||
|
||||
import logging |
||||
|
||||
logger = logging.getLogger(__name__) |
||||
|
||||
|
||||
class GroupServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns("/groups/(?P<group_id>[^/]*)/profile$") |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_GET(self, request, group_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
group_description = yield self.groups_handler.get_group_profile(group_id, user_id) |
||||
|
||||
defer.returnValue((200, group_description)) |
||||
|
||||
|
||||
class GroupSummaryServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns("/groups/(?P<group_id>[^/]*)/summary$") |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupSummaryServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_GET(self, request, group_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
get_group_summary = yield self.groups_handler.get_group_summary(group_id, user_id) |
||||
|
||||
defer.returnValue((200, get_group_summary)) |
||||
|
||||
|
||||
class GroupSummaryRoomsServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns("/groups/(?P<group_id>[^/]*)/summary/rooms$") |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupSummaryServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_GET(self, request, group_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
get_group_summary = yield self.groups_handler.get_group_summary(group_id, user_id) |
||||
|
||||
defer.returnValue((200, get_group_summary)) |
||||
|
||||
|
||||
class GroupSummaryRoomsDefaultCatServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns( |
||||
"/groups/(?P<group_id>[^/]*)/summary/rooms/(?P<room_id>[^/]*)$" |
||||
) |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupSummaryRoomsDefaultCatServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_PUT(self, request, group_id, room_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
content = parse_json_object_from_request(request) |
||||
resp = yield self.groups_handler.update_group_summary_room( |
||||
group_id, user_id, |
||||
room_id=room_id, |
||||
category_id=None, |
||||
content=content, |
||||
) |
||||
|
||||
defer.returnValue((200, resp)) |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_DELETE(self, request, group_id, room_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
resp = yield self.groups_handler.delete_group_summary_room( |
||||
group_id, user_id, |
||||
room_id=room_id, |
||||
category_id=None, |
||||
) |
||||
|
||||
defer.returnValue((200, resp)) |
||||
|
||||
|
||||
class GroupSummaryRoomsCatServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns( |
||||
"/groups/(?P<group_id>[^/]*)/summary" |
||||
"/categories/(?P<category_id>[^/]+)/rooms/(?P<room_id>[^/]+)$" |
||||
) |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupSummaryRoomsCatServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_PUT(self, request, group_id, category_id, room_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
content = parse_json_object_from_request(request) |
||||
resp = yield self.groups_handler.update_group_summary_room( |
||||
group_id, user_id, |
||||
room_id=room_id, |
||||
category_id=category_id, |
||||
content=content, |
||||
) |
||||
|
||||
defer.returnValue((200, resp)) |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_DELETE(self, request, group_id, category_id, room_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
resp = yield self.groups_handler.delete_group_summary_room( |
||||
group_id, user_id, |
||||
room_id=room_id, |
||||
category_id=category_id, |
||||
) |
||||
|
||||
defer.returnValue((200, resp)) |
||||
|
||||
|
||||
class GroupCategoryServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns( |
||||
"/groups/(?P<group_id>[^/]*)/categories/(?P<category_id>[^/]+)$" |
||||
) |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupCategoryServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_GET(self, request, group_id, category_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
category = yield self.groups_handler.get_group_category( |
||||
group_id, user_id, |
||||
category_id=category_id, |
||||
) |
||||
|
||||
defer.returnValue((200, category)) |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_PUT(self, request, group_id, category_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
content = parse_json_object_from_request(request) |
||||
resp = yield self.groups_handler.update_group_category( |
||||
group_id, user_id, |
||||
category_id=category_id, |
||||
content=content, |
||||
) |
||||
|
||||
defer.returnValue((200, resp)) |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_DELETE(self, request, group_id, category_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
resp = yield self.groups_handler.delete_group_category( |
||||
group_id, user_id, |
||||
category_id=category_id, |
||||
) |
||||
|
||||
defer.returnValue((200, resp)) |
||||
|
||||
|
||||
class GroupCategoriesServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns( |
||||
"/groups/(?P<group_id>[^/]*)/categories/$" |
||||
) |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupCategoriesServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_GET(self, request, group_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
category = yield self.groups_handler.get_group_categories( |
||||
group_id, user_id, |
||||
) |
||||
|
||||
defer.returnValue((200, category)) |
||||
|
||||
|
||||
class GroupRoleServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns( |
||||
"/groups/(?P<group_id>[^/]*)/roles/(?P<role_id>[^/]+)$" |
||||
) |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupRoleServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_GET(self, request, group_id, role_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
category = yield self.groups_handler.get_group_role( |
||||
group_id, user_id, |
||||
role_id=role_id, |
||||
) |
||||
|
||||
defer.returnValue((200, category)) |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_PUT(self, request, group_id, role_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
content = parse_json_object_from_request(request) |
||||
resp = yield self.groups_handler.update_group_role( |
||||
group_id, user_id, |
||||
role_id=role_id, |
||||
content=content, |
||||
) |
||||
|
||||
defer.returnValue((200, resp)) |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_DELETE(self, request, group_id, role_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
resp = yield self.groups_handler.delete_group_role( |
||||
group_id, user_id, |
||||
role_id=role_id, |
||||
) |
||||
|
||||
defer.returnValue((200, resp)) |
||||
|
||||
|
||||
class GroupRolesServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns( |
||||
"/groups/(?P<group_id>[^/]*)/roles/$" |
||||
) |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupRolesServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_GET(self, request, group_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
category = yield self.groups_handler.get_group_roles( |
||||
group_id, user_id, |
||||
) |
||||
|
||||
defer.returnValue((200, category)) |
||||
|
||||
|
||||
class GroupSummaryUsersDefaultRoleServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns( |
||||
"/groups/(?P<group_id>[^/]*)/summary/users/(?P<user_id>[^/]*)$" |
||||
) |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupSummaryUsersDefaultRoleServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_PUT(self, request, group_id, user_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
requester_user_id = requester.user.to_string() |
||||
|
||||
content = parse_json_object_from_request(request) |
||||
resp = yield self.groups_handler.update_group_summary_user( |
||||
group_id, requester_user_id, |
||||
user_id=user_id, |
||||
role_id=None, |
||||
content=content, |
||||
) |
||||
|
||||
defer.returnValue((200, resp)) |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_DELETE(self, request, group_id, user_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
requester_user_id = requester.user.to_string() |
||||
|
||||
resp = yield self.groups_handler.delete_group_summary_user( |
||||
group_id, requester_user_id, |
||||
user_id=user_id, |
||||
role_id=None, |
||||
) |
||||
|
||||
defer.returnValue((200, resp)) |
||||
|
||||
|
||||
class GroupSummaryUsersRoleServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns( |
||||
"/groups/(?P<group_id>[^/]*)/summary" |
||||
"/roles/(?P<role_id>[^/]+)/users/(?P<user_id>[^/]+)$" |
||||
) |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupSummaryUsersRoleServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_PUT(self, request, group_id, role_id, user_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
requester_user_id = requester.user.to_string() |
||||
|
||||
content = parse_json_object_from_request(request) |
||||
resp = yield self.groups_handler.update_group_summary_user( |
||||
group_id, requester_user_id, |
||||
user_id=user_id, |
||||
role_id=role_id, |
||||
content=content, |
||||
) |
||||
|
||||
defer.returnValue((200, resp)) |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_DELETE(self, request, group_id, role_id, user_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
requester_user_id = requester.user.to_string() |
||||
|
||||
resp = yield self.groups_handler.delete_group_summary_user( |
||||
group_id, requester_user_id, |
||||
user_id=user_id, |
||||
role_id=role_id, |
||||
) |
||||
|
||||
defer.returnValue((200, resp)) |
||||
|
||||
|
||||
class GroupRoomServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns("/groups/(?P<group_id>[^/]*)/rooms$") |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupRoomServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_GET(self, request, group_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
result = yield self.groups_handler.get_rooms_in_group(group_id, user_id) |
||||
|
||||
defer.returnValue((200, result)) |
||||
|
||||
|
||||
class GroupUsersServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns("/groups/(?P<group_id>[^/]*)/users$") |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupUsersServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_GET(self, request, group_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
result = yield self.groups_handler.get_users_in_group(group_id, user_id) |
||||
|
||||
defer.returnValue((200, result)) |
||||
|
||||
|
||||
class GroupCreateServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns("/create_group$") |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupCreateServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
self.server_name = hs.hostname |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_POST(self, request): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
# TODO: Create group on remote server |
||||
content = parse_json_object_from_request(request) |
||||
localpart = content.pop("localpart") |
||||
group_id = GroupID.create(localpart, self.server_name).to_string() |
||||
|
||||
result = yield self.groups_handler.create_group(group_id, user_id, content) |
||||
|
||||
defer.returnValue((200, result)) |
||||
|
||||
|
||||
class GroupAdminRoomsServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns( |
||||
"/groups/(?P<group_id>[^/]*)/admin/rooms/(?P<room_id>[^/]*)$" |
||||
) |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupAdminRoomsServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_PUT(self, request, group_id, room_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
content = parse_json_object_from_request(request) |
||||
result = yield self.groups_handler.add_room(group_id, user_id, room_id, content) |
||||
|
||||
defer.returnValue((200, result)) |
||||
|
||||
|
||||
class GroupAdminUsersInviteServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns( |
||||
"/groups/(?P<group_id>[^/]*)/admin/users/invite/(?P<user_id>[^/]*)$" |
||||
) |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupAdminUsersInviteServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
self.store = hs.get_datastore() |
||||
self.is_mine_id = hs.is_mine_id |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_PUT(self, request, group_id, user_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
requester_user_id = requester.user.to_string() |
||||
|
||||
content = parse_json_object_from_request(request) |
||||
config = content.get("config", {}) |
||||
result = yield self.groups_handler.invite( |
||||
group_id, user_id, requester_user_id, config, |
||||
) |
||||
|
||||
defer.returnValue((200, result)) |
||||
|
||||
|
||||
class GroupAdminUsersKickServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns( |
||||
"/groups/(?P<group_id>[^/]*)/admin/users/remove/(?P<user_id>[^/]*)$" |
||||
) |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupAdminUsersKickServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_PUT(self, request, group_id, user_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
requester_user_id = requester.user.to_string() |
||||
|
||||
content = parse_json_object_from_request(request) |
||||
result = yield self.groups_handler.remove_user_from_group( |
||||
group_id, user_id, requester_user_id, content, |
||||
) |
||||
|
||||
defer.returnValue((200, result)) |
||||
|
||||
|
||||
class GroupSelfLeaveServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns( |
||||
"/groups/(?P<group_id>[^/]*)/self/leave$" |
||||
) |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupSelfLeaveServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_PUT(self, request, group_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
requester_user_id = requester.user.to_string() |
||||
|
||||
content = parse_json_object_from_request(request) |
||||
result = yield self.groups_handler.remove_user_from_group( |
||||
group_id, requester_user_id, requester_user_id, content, |
||||
) |
||||
|
||||
defer.returnValue((200, result)) |
||||
|
||||
|
||||
class GroupSelfJoinServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns( |
||||
"/groups/(?P<group_id>[^/]*)/self/join$" |
||||
) |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupSelfJoinServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_PUT(self, request, group_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
requester_user_id = requester.user.to_string() |
||||
|
||||
content = parse_json_object_from_request(request) |
||||
result = yield self.groups_handler.join_group( |
||||
group_id, requester_user_id, content, |
||||
) |
||||
|
||||
defer.returnValue((200, result)) |
||||
|
||||
|
||||
class GroupSelfAcceptInviteServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns( |
||||
"/groups/(?P<group_id>[^/]*)/self/accept_invite$" |
||||
) |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupSelfAcceptInviteServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_PUT(self, request, group_id): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
requester_user_id = requester.user.to_string() |
||||
|
||||
content = parse_json_object_from_request(request) |
||||
result = yield self.groups_handler.accept_invite( |
||||
group_id, requester_user_id, content, |
||||
) |
||||
|
||||
defer.returnValue((200, result)) |
||||
|
||||
|
||||
class GroupsForUserServlet(RestServlet): |
||||
PATTERNS = client_v2_patterns( |
||||
"/joined_groups$" |
||||
) |
||||
|
||||
def __init__(self, hs): |
||||
super(GroupsForUserServlet, self).__init__() |
||||
self.auth = hs.get_auth() |
||||
self.clock = hs.get_clock() |
||||
self.groups_handler = hs.get_groups_local_handler() |
||||
|
||||
@defer.inlineCallbacks |
||||
def on_GET(self, request): |
||||
requester = yield self.auth.get_user_by_req(request) |
||||
user_id = requester.user.to_string() |
||||
|
||||
result = yield self.groups_handler.get_joined_groups(user_id) |
||||
|
||||
defer.returnValue((200, result)) |
||||
|
||||
|
||||
def register_servlets(hs, http_server): |
||||
GroupServlet(hs).register(http_server) |
||||
GroupSummaryServlet(hs).register(http_server) |
||||
GroupUsersServlet(hs).register(http_server) |
||||
GroupRoomServlet(hs).register(http_server) |
||||
GroupCreateServlet(hs).register(http_server) |
||||
GroupAdminRoomsServlet(hs).register(http_server) |
||||
GroupAdminUsersInviteServlet(hs).register(http_server) |
||||
GroupAdminUsersKickServlet(hs).register(http_server) |
||||
GroupSelfLeaveServlet(hs).register(http_server) |
||||
GroupSelfJoinServlet(hs).register(http_server) |
||||
GroupSelfAcceptInviteServlet(hs).register(http_server) |
||||
GroupsForUserServlet(hs).register(http_server) |
||||
GroupSummaryRoomsDefaultCatServlet(hs).register(http_server) |
||||
GroupCategoryServlet(hs).register(http_server) |
||||
GroupCategoriesServlet(hs).register(http_server) |
||||
GroupSummaryRoomsCatServlet(hs).register(http_server) |
||||
GroupRoleServlet(hs).register(http_server) |
||||
GroupRolesServlet(hs).register(http_server) |
||||
GroupSummaryUsersDefaultRoleServlet(hs).register(http_server) |
||||
GroupSummaryUsersRoleServlet(hs).register(http_server) |
Loading…
Reference in new issue