diff --git a/synapse/storage/appservice.py b/synapse/storage/appservice.py
index cdf26ee434..48bc7e0fe6 100644
--- a/synapse/storage/appservice.py
+++ b/synapse/storage/appservice.py
@@ -13,119 +13,16 @@
# See the License for the specific language governing permissions and
# limitations under the License.
import logging
-import re
from twisted.internet import defer
-from synapse.api.constants import EventTypes
from synapse.api.errors import StoreError
+from synapse.appservice import ApplicationService
from ._base import SQLBaseStore
logger = logging.getLogger(__name__)
-# XXX: This feels like it should belong in a "models" module, not storage.
-class ApplicationService(object):
- """Defines an application service.
-
- Provides methods to check if this service is "interested" in events.
- """
- NS_USERS = "users"
- NS_ALIASES = "aliases"
- NS_ROOMS = "rooms"
- # The ordering here is important as it is used to map database values (which
- # are stored as ints representing the position in this list) to namespace
- # values.
- NS_LIST = [NS_USERS, NS_ALIASES, NS_ROOMS]
-
- def __init__(self, token, url=None, namespaces=None):
- self.token = token
- self.url = url
- self.namespaces = self._check_namespaces(namespaces)
-
- def _check_namespaces(self, namespaces):
- # Sanity check that it is of the form:
- # {
- # users: ["regex",...],
- # aliases: ["regex",...],
- # rooms: ["regex",...],
- # }
- if not namespaces:
- return None
-
- for ns in ApplicationService.NS_LIST:
- if type(namespaces[ns]) != list:
- raise ValueError("Bad namespace value for '%s'", ns)
- for regex in namespaces[ns]:
- if not isinstance(regex, basestring):
- raise ValueError("Expected string regex for ns '%s'", ns)
- return namespaces
-
- def _matches_regex(self, test_string, namespace_key):
- for regex in self.namespaces[namespace_key]:
- if re.match(regex, test_string):
- return True
- return False
-
- def _matches_user(self, event):
- if (hasattr(event, "user_id") and
- self._matches_regex(
- event.user_id, ApplicationService.NS_USERS)):
- return True
- # also check m.room.member state key
- if (hasattr(event, "type") and event.type == EventTypes.Member
- and hasattr(event, "state_key")
- and self._matches_regex(
- event.state_key, ApplicationService.NS_USERS)):
- return True
- return False
-
- def _matches_room_id(self, event):
- if hasattr(event, "room_id"):
- return self._matches_regex(
- event.room_id, ApplicationService.NS_ROOMS
- )
- return False
-
- def _matches_aliases(self, event, alias_list):
- for alias in alias_list:
- if self._matches_regex(alias, ApplicationService.NS_ALIASES):
- return True
- return False
-
- def is_interested(self, event, restrict_to=None, aliases_for_event=None):
- """Check if this service is interested in this event.
-
- Args:
- event(Event): The event to check.
- restrict_to(str): The namespace to restrict regex tests to.
- aliases_for_event(list): A list of all the known room aliases for
- this event.
- Returns:
- bool: True if this service would like to know about this event.
- """
- if aliases_for_event is None:
- aliases_for_event = []
- if restrict_to and restrict_to not in ApplicationService.NS_LIST:
- # this is a programming error, so fail early and raise a general
- # exception
- raise Exception("Unexpected restrict_to value: %s". restrict_to)
-
- if not restrict_to:
- return (self._matches_user(event)
- or self._matches_aliases(event, aliases_for_event)
- or self._matches_room_id(event))
- elif restrict_to == ApplicationService.NS_ALIASES:
- return self._matches_aliases(event, aliases_for_event)
- elif restrict_to == ApplicationService.NS_ROOMS:
- return self._matches_room_id(event)
- elif restrict_to == ApplicationService.NS_USERS:
- return self._matches_user(event)
-
- def __str__(self):
- return "ApplicationService: %s" % (self.__dict__,)
-
-
class ApplicationServiceCache(object):
"""Caches ApplicationServices and provides utility functions on top.
|
