Pass in current state to push action handler

This commit is contained in:
Erik Johnston 2016-02-09 16:01:40 +00:00
parent feb294d552
commit f28cc45183
3 changed files with 20 additions and 27 deletions

View file

@ -36,7 +36,7 @@ class ActionGenerator:
# tag (ie. we just need all the users).
@defer.inlineCallbacks
def handle_push_actions_for_event(self, event, handler):
def handle_push_actions_for_event(self, event, handler, current_state):
if event.type == EventTypes.Redaction and event.redacts is not None:
yield self.store.remove_push_actions_for_event_id(
event.room_id, event.redacts
@ -46,7 +46,9 @@ class ActionGenerator:
event.room_id, self.hs, self.store
)
actions_by_user = yield bulk_evaluator.action_for_event_by_user(event, handler)
actions_by_user = yield bulk_evaluator.action_for_event_by_user(
event, handler, current_state
)
yield self.store.set_push_actions_for_event_and_users(
event,