wip at implementing MSC 7075
1 files changed, 84 insertions, 0 deletions
diff --git a/synapse/server_notices/resource_limits_server_notices.py b/synapse/server_notices/resource_limits_server_notices.py
new file mode 100644
index 0000000000..33b1d80cd9
--- /dev/null
+++ b/synapse/server_notices/resource_limits_server_notices.py
@@ -0,0 +1,84 @@
+# -*- coding: utf-8 -*-
+# Copyright 2018 New Vector 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.
+import logging
+
+from six import iteritems, string_types
+
+from twisted.internet import defer
+
+from synapse.api.errors import SynapseError
+from synapse.api.urls import ConsentURIBuilder
+from synapse.config import ConfigError
+from synapse.types import get_localpart_from_id
+
+logger = logging.getLogger(__name__)
+
+
+class ResourceLimitsServerNotices(object):
+ """
+ """
+ def __init__(self, hs):
+ """
+
+ Args:
+ hs (synapse.server.HomeServer):
+ """
+ self._server_notices_manager = hs.get_server_notices_manager()
+ self._store = hs.get_datastore()
+ self._api = hs.get_api()
+ self._server_notice_content = hs.config.user_consent_server_notice_content
+ self._limit_usage_by_mau = config.limit_usage_by_mau = False
+ self._hs_disabled.config.hs_disabled = False
+
+ self._notified = set()
+ self._resouce_limited = False
+ # Config checks?
+
+ @defer.inlineCallbacks
+ def maybe_send_server_notice_to_user(self, user_id):
+ """Check if we need to send a notice to this user, and does so if so
+
+ Args:
+ user_id (str): user to check
+
+ Returns:
+ Deferred
+ """
+ if self._limit_usage_by_mau is False and self._hs_disabled is False:
+ # not enabled
+ return
+
+ timestamp = yield self.store.user_last_seen_monthly_active(user_id)
+ if timestamp is None:
+ # This user will be blocked from receiving the notice anyway
+ return
+ try:
+ yield self.api.check_auth_blocking()
+ if self._resouce_limited:
+ # Need to start removing notices
+ pass
+ except AuthError as e:
+ # Need to start notifying of blocking
+ if not self._resouce_limited:
+ pass
+
+ # need to send a message.
+ try:
+ yield self._server_notices_manager.send_notice(
+ user_id, content,
+ )
+
+ except SynapseError as e:
+ logger.error("Error sending server notice about resource limits: %s", e)
|
