Flatten the synapse.rest.client package (#10600)

1 files changed, 47 insertions, 0 deletions

diff --git a/synapse/rest/client/initial_sync.py b/synapse/rest/client/initial_sync.py
new file mode 100644
index 0000000000..12ba0e91db
--- /dev/null
+++ b/synapse/rest/client/initial_sync.py
@@ -0,0 +1,47 @@
+# 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 synapse.http.servlet import RestServlet, parse_boolean
+from synapse.rest.client._base import client_patterns
+from synapse.streams.config import PaginationConfig
+
+
+# TODO: Needs unit testing
+class InitialSyncRestServlet(RestServlet):
+    PATTERNS = client_patterns("/initialSync$", v1=True)
+
+    def __init__(self, hs):
+        super().__init__()
+        self.initial_sync_handler = hs.get_initial_sync_handler()
+        self.auth = hs.get_auth()
+        self.store = hs.get_datastore()
+
+    async def on_GET(self, request):
+        requester = await self.auth.get_user_by_req(request)
+        as_client_event = b"raw" not in request.args
+        pagination_config = await PaginationConfig.from_request(self.store, request)
+        include_archived = parse_boolean(request, "archived", default=False)
+        content = await self.initial_sync_handler.snapshot_all_rooms(
+            user_id=requester.user.to_string(),
+            pagin_config=pagination_config,
+            as_client_event=as_client_event,
+            include_archived=include_archived,
+        )
+
+        return 200, content
+
+
+def register_servlets(hs, http_server):
+    InitialSyncRestServlet(hs).register(http_server)