Add is_interested_in_presence func

hs/super-wip-edus-down-sync
Will Hunt 2020-09-21 16:21:22 +01:00
parent 42090bcc7c
commit 3bf1b79d3c
1 changed files with 14 additions and 1 deletions

View File

@ -19,7 +19,7 @@ from typing import TYPE_CHECKING
from synapse.api.constants import EventTypes
from synapse.appservice.api import ApplicationServiceApi
from synapse.types import GroupID, get_domain_from_id
from synapse.util.caches.descriptors import cached
from synapse.util.caches.descriptors import cached, cachedList
if TYPE_CHECKING:
from synapse.storage.databases.main import DataStore
@ -241,6 +241,19 @@ class ApplicationService:
return False
@cached(num_args=1, cache_context=True)
async def is_interested_in_presence(self, user_id, store, cache_context):
# Find all the rooms the sender is in
if self.is_interested_in_user(user_id.to_string()):
return True
room_ids = await store.get_rooms_for_user(user_id.to_string())
# Then find out if the appservice is interested in any of those rooms
for room_id in room_ids:
if await self.matches_user_in_member_list(room_id, store, cache_context):
return True
return False
def is_interested_in_user(self, user_id):
return (
self._matches_regex(user_id, ApplicationService.NS_USERS)