Skip to content
This repository has been archived by the owner on Apr 26, 2024. It is now read-only.

Commit

Permalink
Move the presence handler out of the Handlers object
Browse files Browse the repository at this point in the history
  • Loading branch information
Mark Haines committed May 16, 2016
1 parent 2d98c96 commit 3b86ecf
Show file tree
Hide file tree
Showing 12 changed files with 29 additions and 18 deletions.
2 changes: 0 additions & 2 deletions synapse/handlers/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,6 @@
from .events import EventStreamHandler, EventHandler
from .federation import FederationHandler
from .profile import ProfileHandler
from .presence import PresenceHandler
from .directory import DirectoryHandler
from .typing import TypingNotificationHandler
from .admin import AdminHandler
Expand Down Expand Up @@ -53,7 +52,6 @@ def __init__(self, hs):
self.event_handler = EventHandler(hs)
self.federation_handler = FederationHandler(hs)
self.profile_handler = ProfileHandler(hs)
self.presence_handler = PresenceHandler(hs)
self.room_list_handler = RoomListHandler(hs)
self.directory_handler = DirectoryHandler(hs)
self.typing_notification_handler = TypingNotificationHandler(hs)
Expand Down
2 changes: 1 addition & 1 deletion synapse/handlers/events.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ def get_stream(self, auth_user_id, pagin_config, timeout=0,
If `only_keys` is not None, events from keys will be sent down.
"""
auth_user = UserID.from_string(auth_user_id)
presence_handler = self.hs.get_handlers().presence_handler
presence_handler = self.hs.get_presence_handler()

context = yield presence_handler.user_syncing(
auth_user_id, affect_presence=affect_presence,
Expand Down
4 changes: 2 additions & 2 deletions synapse/handlers/message.py
Original file line number Diff line number Diff line change
Expand Up @@ -236,7 +236,7 @@ def send_nonmember_event(self, requester, event, context, ratelimit=True):
)

if event.type == EventTypes.Message:
presence = self.hs.get_handlers().presence_handler
presence = self.hs.get_presence_handler()
yield presence.bump_presence_active_time(user)

def deduplicate_state_event(self, event, context):
Expand Down Expand Up @@ -674,7 +674,7 @@ def _room_initial_sync_joined(self, user_id, room_id, pagin_config,
and m.content["membership"] == Membership.JOIN
]

presence_handler = self.hs.get_handlers().presence_handler
presence_handler = self.hs.get_presence_handler()

@defer.inlineCallbacks
def get_presence():
Expand Down
2 changes: 1 addition & 1 deletion synapse/handlers/presence.py
Original file line number Diff line number Diff line change
Expand Up @@ -860,7 +860,7 @@ def get_new_events(self, user, from_key, room_ids=None, include_offline=True,
from_key = int(from_key)
room_ids = room_ids or []

presence = self.hs.get_handlers().presence_handler
presence = self.hs.get_presence_handler()
stream_change_cache = self.store.presence_stream_cache

if not room_ids:
Expand Down
2 changes: 1 addition & 1 deletion synapse/handlers/sync.py
Original file line number Diff line number Diff line change
Expand Up @@ -639,7 +639,7 @@ def incremental_sync_with_gap(self, sync_config, since_token):

# For each newly joined room, we want to send down presence of
# existing users.
presence_handler = self.hs.get_handlers().presence_handler
presence_handler = self.hs.get_presence_handler()
extra_presence_users = set()
for room_id in newly_joined_rooms:
users = yield self.store.get_users_in_room(event.room_id)
Expand Down
2 changes: 1 addition & 1 deletion synapse/replication/resource.py
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ def __init__(self, hs):
self.version_string = hs.version_string
self.store = hs.get_datastore()
self.sources = hs.get_event_sources()
self.presence_handler = hs.get_handlers().presence_handler
self.presence_handler = hs.get_presence_handler()
self.typing_handler = hs.get_handlers().typing_notification_handler
self.notifier = hs.notifier
self.clock = hs.get_clock()
Expand Down
20 changes: 14 additions & 6 deletions synapse/rest/client/v1/presence.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,20 +30,24 @@
class PresenceStatusRestServlet(ClientV1RestServlet):
PATTERNS = client_path_patterns("/presence/(?P<user_id>[^/]*)/status")

def __init__(self, hs):
super(PresenceStatusRestServlet, self).__init__(hs)
self.presence_handler = hs.get_presence_handler()

@defer.inlineCallbacks
def on_GET(self, request, user_id):
requester = yield self.auth.get_user_by_req(request)
user = UserID.from_string(user_id)

if requester.user != user:
allowed = yield self.handlers.presence_handler.is_visible(
allowed = yield self.presence_handler.is_visible(
observed_user=user, observer_user=requester.user,
)

if not allowed:
raise AuthError(403, "You are not allowed to see their presence.")

state = yield self.handlers.presence_handler.get_state(target_user=user)
state = yield self.presence_handler.get_state(target_user=user)

defer.returnValue((200, state))

Expand Down Expand Up @@ -74,7 +78,7 @@ def on_PUT(self, request, user_id):
except:
raise SynapseError(400, "Unable to parse state")

yield self.handlers.presence_handler.set_state(user, state)
yield self.presence_handler.set_state(user, state)

defer.returnValue((200, {}))

Expand All @@ -85,6 +89,10 @@ def on_OPTIONS(self, request):
class PresenceListRestServlet(ClientV1RestServlet):
PATTERNS = client_path_patterns("/presence/list/(?P<user_id>[^/]*)")

def __init__(self, hs):
super(PresenceListRestServlet, self).__init__(hs)
self.presence_handler = hs.get_presence_handler()

@defer.inlineCallbacks
def on_GET(self, request, user_id):
requester = yield self.auth.get_user_by_req(request)
Expand All @@ -96,7 +104,7 @@ def on_GET(self, request, user_id):
if requester.user != user:
raise SynapseError(400, "Cannot get another user's presence list")

presence = yield self.handlers.presence_handler.get_presence_list(
presence = yield self.presence_handler.get_presence_list(
observer_user=user, accepted=True
)

Expand All @@ -123,7 +131,7 @@ def on_POST(self, request, user_id):
if len(u) == 0:
continue
invited_user = UserID.from_string(u)
yield self.handlers.presence_handler.send_presence_invite(
yield self.presence_handler.send_presence_invite(
observer_user=user, observed_user=invited_user
)

Expand All @@ -134,7 +142,7 @@ def on_POST(self, request, user_id):
if len(u) == 0:
continue
dropped_user = UserID.from_string(u)
yield self.handlers.presence_handler.drop(
yield self.presence_handler.drop(
observer_user=user, observed_user=dropped_user
)

Expand Down
2 changes: 1 addition & 1 deletion synapse/rest/client/v1/room.py
Original file line number Diff line number Diff line change
Expand Up @@ -570,7 +570,7 @@ class RoomTypingRestServlet(ClientV1RestServlet):

def __init__(self, hs):
super(RoomTypingRestServlet, self).__init__(hs)
self.presence_handler = hs.get_handlers().presence_handler
self.presence_handler = hs.get_presence_handler()

@defer.inlineCallbacks
def on_PUT(self, request, room_id, user_id):
Expand Down
2 changes: 1 addition & 1 deletion synapse/rest/client/v2_alpha/receipts.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ def __init__(self, hs):
self.hs = hs
self.auth = hs.get_auth()
self.receipts_handler = hs.get_handlers().receipts_handler
self.presence_handler = hs.get_handlers().presence_handler
self.presence_handler = hs.get_presence_handler()

@defer.inlineCallbacks
def on_POST(self, request, room_id, receipt_type, event_id):
Expand Down
2 changes: 1 addition & 1 deletion synapse/rest/client/v2_alpha/sync.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ def __init__(self, hs):
self.sync_handler = hs.get_handlers().sync_handler
self.clock = hs.get_clock()
self.filtering = hs.get_filtering()
self.presence_handler = hs.get_handlers().presence_handler
self.presence_handler = hs.get_presence_handler()

@defer.inlineCallbacks
def on_GET(self, request):
Expand Down
5 changes: 5 additions & 0 deletions synapse/server.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@
from synapse.notifier import Notifier
from synapse.api.auth import Auth
from synapse.handlers import Handlers
from synapse.handlers.presence import PresenceHandler
from synapse.state import StateHandler
from synapse.storage import DataStore
from synapse.util import Clock
Expand Down Expand Up @@ -78,6 +79,7 @@ def build_DEPENDENCY(self)
'auth',
'rest_servlet_factory',
'state_handler',
'presence_handler',
'notifier',
'distributor',
'client_resource',
Expand Down Expand Up @@ -164,6 +166,9 @@ def build_v1auth(self):
def build_state_handler(self):
return StateHandler(self)

def build_presence_handler(self):
return PresenceHandler(self)

def build_event_sources(self):
return EventSources(self)

Expand Down
2 changes: 1 addition & 1 deletion tests/replication/test_resource.py
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@ def test_events_and_state(self):
@defer.inlineCallbacks
def test_presence(self):
get = self.get(presence="-1")
yield self.hs.get_handlers().presence_handler.set_state(
yield self.hs.get_presence_handler().set_state(
self.user, {"presence": "online"}
)
code, body = yield get
Expand Down

0 comments on commit 3b86ecf

Please sign in to comment.