summary refs log tree commit diff
path: root/synapse/rest/client
diff options
context:
space:
mode:
authorPaul "LeoNerd" Evans <paul@matrix.org>2015-01-27 14:28:56 +0000
committerPaul "LeoNerd" Evans <paul@matrix.org>2015-01-27 14:28:56 +0000
commit05c7cba73a050f19cc52129b65b0183eaa832a42 (patch)
treee0ad0e2a239a409a11fcc987802408683ec5bd55 /synapse/rest/client
parentUse new V2AlphaRestTestCase (diff)
downloadsynapse-05c7cba73a050f19cc52129b65b0183eaa832a42.tar.xz
Initial trivial implementation of an actual 'Filtering' object; move storage of user filters into there
Diffstat (limited to 'synapse/rest/client')
-rw-r--r--synapse/rest/client/v2_alpha/filter.py25
1 files changed, 12 insertions, 13 deletions
diff --git a/synapse/rest/client/v2_alpha/filter.py b/synapse/rest/client/v2_alpha/filter.py
index a9a180ec04..585c8e02e8 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)}))