From 05c7cba73a050f19cc52129b65b0183eaa832a42 Mon Sep 17 00:00:00 2001
From: "Paul \"LeoNerd\" Evans" <paul@matrix.org>
Date: Tue, 27 Jan 2015 14:28:56 +0000
Subject: [PATCH] Initial trivial implementation of an actual 'Filtering'
 object; move storage of user filters into there

---
 synapse/api/filtering.py               | 41 ++++++++++++++++++++++++++
 synapse/rest/client/v2_alpha/filter.py | 25 ++++++++--------
 synapse/server.py                      |  5 ++++
 3 files changed, 58 insertions(+), 13 deletions(-)
 create mode 100644 synapse/api/filtering.py

diff --git a/synapse/api/filtering.py b/synapse/api/filtering.py
new file mode 100644
index 000000000..922c40004
--- /dev/null
+++ b/synapse/api/filtering.py
@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+# Copyright 2015 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.
+
+
+# TODO(paul)
+_filters_for_user = {}
+
+
+class Filtering(object):
+
+    def __init__(self, hs):
+        super(Filtering, self).__init__()
+        self.hs = hs
+
+    def get_user_filter(self, user_localpart, filter_id):
+        filters = _filters_for_user.get(user_localpart, None)
+
+        if not filters or filter_id >= len(filters):
+            raise KeyError()
+
+        return filters[filter_id]
+
+    def add_user_filter(self, user_localpart, definition):
+        filters = _filters_for_user.setdefault(user_localpart, [])
+
+        filter_id = len(filters)
+        filters.append(definition)
+
+        return filter_id
diff --git a/synapse/rest/client/v2_alpha/filter.py b/synapse/rest/client/v2_alpha/filter.py
index a9a180ec0..585c8e02e 100644
--- a/synapse/rest/client/v2_alpha/filter.py
+++ b/synapse/rest/client/v2_alpha/filter.py
@@ -28,10 +28,6 @@ import logging
 logger = logging.getLogger(__name__)
 
 
-# TODO(paul)
-_filters_for_user = {}
-
-
 class GetFilterRestServlet(RestServlet):
     PATTERN = client_v2_pattern("/user/(?P<user_id>[^/]*)/filter/(?P<filter_id>[^/]*)")
 
@@ -39,6 +35,7 @@ class GetFilterRestServlet(RestServlet):
         super(GetFilterRestServlet, self).__init__()
         self.hs = hs
         self.auth = hs.get_auth()
+        self.filtering = hs.get_filtering()
 
     @defer.inlineCallbacks
     def on_GET(self, request, user_id, filter_id):
@@ -56,13 +53,14 @@ class GetFilterRestServlet(RestServlet):
         except:
             raise SynapseError(400, "Invalid filter_id")
 
-        filters = _filters_for_user.get(target_user.localpart, None)
-
-        if not filters or filter_id >= len(filters):
+        try:
+            defer.returnValue((200, self.filtering.get_user_filter(
+                user_localpart=target_user.localpart,
+                filter_id=filter_id,
+            )))
+        except KeyError:
             raise SynapseError(400, "No such filter")
 
-        defer.returnValue((200, filters[filter_id]))
-
 
 class CreateFilterRestServlet(RestServlet):
     PATTERN = client_v2_pattern("/user/(?P<user_id>[^/]*)/filter")
@@ -71,6 +69,7 @@ class CreateFilterRestServlet(RestServlet):
         super(CreateFilterRestServlet, self).__init__()
         self.hs = hs
         self.auth = hs.get_auth()
+        self.filtering = hs.get_filtering()
 
     @defer.inlineCallbacks
     def on_POST(self, request, user_id):
@@ -90,10 +89,10 @@ class CreateFilterRestServlet(RestServlet):
         except:
             raise SynapseError(400, "Invalid filter definition")
 
-        filters = _filters_for_user.setdefault(target_user.localpart, [])
-
-        filter_id = len(filters)
-        filters.append(content)
+        filter_id = self.filtering.add_user_filter(
+            user_localpart=target_user.localpart,
+            definition=content,
+        )
 
         defer.returnValue((200, {"filter_id": str(filter_id)}))
 
diff --git a/synapse/server.py b/synapse/server.py
index f09d5d581..9b42079e0 100644
--- a/synapse/server.py
+++ b/synapse/server.py
@@ -32,6 +32,7 @@ from synapse.streams.events import EventSources
 from synapse.api.ratelimiting import Ratelimiter
 from synapse.crypto.keyring import Keyring
 from synapse.events.builder import EventBuilderFactory
+from synapse.api.filtering import Filtering
 
 
 class BaseHomeServer(object):
@@ -79,6 +80,7 @@ class BaseHomeServer(object):
         'ratelimiter',
         'keyring',
         'event_builder_factory',
+        'filtering',
     ]
 
     def __init__(self, hostname, **kwargs):
@@ -197,3 +199,6 @@ class HomeServer(BaseHomeServer):
             clock=self.get_clock(),
             hostname=self.hostname,
         )
+
+    def build_filtering(self):
+        return Filtering(self)