mirror of https://github.com/zulip/zulip.git
145 lines
5.3 KiB
Python
145 lines
5.3 KiB
Python
from dataclasses import dataclass
|
|
from typing import Collection, Dict, List, Optional, Set
|
|
|
|
from zerver.lib.mention import MentionData
|
|
from zerver.models import NotificationTriggers
|
|
|
|
|
|
@dataclass
|
|
class UserMessageNotificationsData:
|
|
user_id: int
|
|
flags: Collection[str]
|
|
mentioned: bool
|
|
online_push_enabled: bool
|
|
stream_push_notify: bool
|
|
stream_email_notify: bool
|
|
wildcard_mention_notify: bool
|
|
sender_is_muted: bool
|
|
|
|
def __post_init__(self) -> None:
|
|
if self.mentioned:
|
|
assert "mentioned" in self.flags
|
|
if self.wildcard_mention_notify:
|
|
assert "wildcard_mentioned" in self.flags
|
|
|
|
@classmethod
|
|
def from_user_id_sets(
|
|
cls,
|
|
user_id: int,
|
|
flags: Collection[str],
|
|
online_push_user_ids: Set[int],
|
|
stream_push_user_ids: Set[int],
|
|
stream_email_user_ids: Set[int],
|
|
wildcard_mention_user_ids: Set[int],
|
|
muted_sender_user_ids: Set[int],
|
|
) -> "UserMessageNotificationsData":
|
|
wildcard_mention_notify = (
|
|
user_id in wildcard_mention_user_ids and "wildcard_mentioned" in flags
|
|
)
|
|
return cls(
|
|
user_id=user_id,
|
|
flags=flags,
|
|
mentioned=("mentioned" in flags),
|
|
online_push_enabled=(user_id in online_push_user_ids),
|
|
stream_push_notify=(user_id in stream_push_user_ids),
|
|
stream_email_notify=(user_id in stream_email_user_ids),
|
|
wildcard_mention_notify=wildcard_mention_notify,
|
|
sender_is_muted=(user_id in muted_sender_user_ids),
|
|
)
|
|
|
|
# TODO: The following functions should also look at the `enable_offline_push_notifications` and
|
|
# `enable_offline_email_notifications` settings (for PMs and mentions), but currently they
|
|
# don't.
|
|
|
|
# For these functions, acting_user_id is the user sent a message
|
|
# (or edited a message) triggering the event for which we need to
|
|
# determine notifiability.
|
|
def is_notifiable(self, private_message: bool, acting_user_id: int, idle: bool) -> bool:
|
|
return self.is_email_notifiable(
|
|
private_message, acting_user_id, idle
|
|
) or self.is_push_notifiable(private_message, acting_user_id, idle)
|
|
|
|
def is_push_notifiable(self, private_message: bool, acting_user_id: int, idle: bool) -> bool:
|
|
return self.get_push_notification_trigger(private_message, acting_user_id, idle) is not None
|
|
|
|
def get_push_notification_trigger(
|
|
self, private_message: bool, acting_user_id: int, idle: bool
|
|
) -> Optional[str]:
|
|
if not idle and not self.online_push_enabled:
|
|
return None
|
|
|
|
if self.user_id == acting_user_id:
|
|
return None
|
|
|
|
if self.sender_is_muted:
|
|
return None
|
|
|
|
if private_message:
|
|
return NotificationTriggers.PRIVATE_MESSAGE
|
|
elif self.mentioned:
|
|
return NotificationTriggers.MENTION
|
|
elif self.wildcard_mention_notify:
|
|
return NotificationTriggers.WILDCARD_MENTION
|
|
elif self.stream_push_notify:
|
|
return NotificationTriggers.STREAM_PUSH
|
|
else:
|
|
return None
|
|
|
|
def is_email_notifiable(self, private_message: bool, acting_user_id: int, idle: bool) -> bool:
|
|
return (
|
|
self.get_email_notification_trigger(private_message, acting_user_id, idle) is not None
|
|
)
|
|
|
|
def get_email_notification_trigger(
|
|
self, private_message: bool, acting_user_id: int, idle: bool
|
|
) -> Optional[str]:
|
|
if not idle:
|
|
return None
|
|
|
|
if self.user_id == acting_user_id:
|
|
return None
|
|
|
|
if self.sender_is_muted:
|
|
return None
|
|
|
|
if private_message:
|
|
return NotificationTriggers.PRIVATE_MESSAGE
|
|
elif self.mentioned:
|
|
return NotificationTriggers.MENTION
|
|
elif self.wildcard_mention_notify:
|
|
return NotificationTriggers.WILDCARD_MENTION
|
|
elif self.stream_email_notify:
|
|
return NotificationTriggers.STREAM_EMAIL
|
|
else:
|
|
return None
|
|
|
|
|
|
def get_user_group_mentions_data(
|
|
mentioned_user_ids: Set[int], mentioned_user_group_ids: List[int], mention_data: MentionData
|
|
) -> Dict[int, int]:
|
|
# Maps user_id -> mentioned user_group_id
|
|
mentioned_user_groups_map: Dict[int, int] = dict()
|
|
|
|
# Add members of the mentioned user groups into `mentions_user_ids`.
|
|
for group_id in mentioned_user_group_ids:
|
|
member_ids = mention_data.get_group_members(group_id)
|
|
for member_id in member_ids:
|
|
if member_id in mentioned_user_ids:
|
|
# If a user is also mentioned personally, we use that as a trigger
|
|
# for notifications.
|
|
continue
|
|
|
|
if member_id in mentioned_user_groups_map:
|
|
# If multiple user groups are mentioned, we prefer the
|
|
# user group with the least members for email/mobile
|
|
# notifications.
|
|
previous_group_id = mentioned_user_groups_map[member_id]
|
|
previous_group_member_ids = mention_data.get_group_members(previous_group_id)
|
|
|
|
if len(previous_group_member_ids) > len(member_ids):
|
|
mentioned_user_groups_map[member_id] = group_id
|
|
else:
|
|
mentioned_user_groups_map[member_id] = group_id
|
|
|
|
return mentioned_user_groups_map
|