Files
zulip/zerver/lib/addressee.py
Prakhar Pratyush 9a46104684 addressee: Rename "(no topic)" to empty string topic.
This commit renames "(no topic)" to "" when used as
topic name, while sending or scheduling messages.

This is particularly helpful for older clients where
sending messages with empty topic input box resulted
in messages being sent to "(no topic)" topic.

Now, it will be sent in empty string topic and will
appear in the "general chat" topic.

This helps older clients to somewhat align with the UX
of the clients supporting empty string topic.
2025-03-19 11:39:28 -07:00

219 lines
7.6 KiB
Python

from collections.abc import Iterable, Sequence
from typing import cast
from django.utils.translation import gettext as _
from zerver.lib.exceptions import JsonableError
from zerver.lib.string_validation import check_stream_topic
from zerver.lib.topic import (
maybe_rename_general_chat_to_empty_topic,
maybe_rename_no_topic_to_empty_topic,
)
from zerver.models import Realm, Stream, UserProfile
from zerver.models.users import (
get_user_by_id_in_realm_including_cross_realm,
get_user_including_cross_realm,
)
def get_user_profiles(emails: Iterable[str], realm: Realm) -> list[UserProfile]:
user_profiles: list[UserProfile] = []
for email in emails:
try:
user_profile = get_user_including_cross_realm(email, realm)
except UserProfile.DoesNotExist:
raise JsonableError(_("Invalid email '{email}'").format(email=email))
user_profiles.append(user_profile)
return user_profiles
def get_user_profiles_by_ids(user_ids: Iterable[int], realm: Realm) -> list[UserProfile]:
user_profiles: list[UserProfile] = []
for user_id in user_ids:
try:
user_profile = get_user_by_id_in_realm_including_cross_realm(user_id, realm)
except UserProfile.DoesNotExist:
raise JsonableError(_("Invalid user ID {user_id}").format(user_id=user_id))
user_profiles.append(user_profile)
return user_profiles
class Addressee:
# This is really just a holder for vars that tended to be passed
# around in a non-type-safe way before this class was introduced.
#
# It also avoids some nonsense where you have to think about whether
# topic should be None or '' for a direct message, or you have to
# make an array of one stream.
#
# Eventually we can use this to cache Stream and UserProfile objects
# in memory.
#
# This should be treated as an immutable class.
def __init__(
self,
msg_type: str,
user_profiles: Sequence[UserProfile] | None = None,
stream: Stream | None = None,
stream_name: str | None = None,
stream_id: int | None = None,
topic_name: str | None = None,
) -> None:
assert msg_type in ["stream", "private"]
if msg_type == "stream" and topic_name is None:
raise JsonableError(_("Missing topic"))
self._msg_type = msg_type
self._user_profiles = user_profiles
self._stream = stream
self._stream_name = stream_name
self._stream_id = stream_id
self._topic_name = topic_name
def is_stream(self) -> bool:
return self._msg_type == "stream"
def is_private(self) -> bool:
return self._msg_type == "private"
def user_profiles(self) -> Sequence[UserProfile]:
assert self.is_private()
assert self._user_profiles is not None
return self._user_profiles
def stream(self) -> Stream | None:
assert self.is_stream()
return self._stream
def stream_name(self) -> str | None:
assert self.is_stream()
return self._stream_name
def stream_id(self) -> int | None:
assert self.is_stream()
return self._stream_id
def topic_name(self) -> str:
assert self.is_stream()
assert self._topic_name is not None
return self._topic_name
@staticmethod
def legacy_build(
sender: UserProfile,
recipient_type_name: str,
message_to: Sequence[int] | Sequence[str],
topic_name: str | None,
realm: Realm | None = None,
) -> "Addressee":
# For legacy reason message_to used to be either a list of
# emails or a list of streams. We haven't fixed all of our
# callers yet.
if realm is None:
realm = sender.realm
if recipient_type_name == "stream":
if len(message_to) > 1:
raise JsonableError(_("Cannot send to multiple channels"))
if message_to:
stream_name_or_id = message_to[0]
else:
# This is a hack to deal with the fact that we still support
# default streams (and the None will be converted later in the
# call path).
if sender.default_sending_stream_id:
# Use the user's default stream
stream_name_or_id = sender.default_sending_stream_id
else:
raise JsonableError(_("Missing channel"))
if topic_name is None:
raise JsonableError(_("Missing topic"))
if isinstance(stream_name_or_id, int):
return Addressee.for_stream_id(stream_name_or_id, topic_name)
return Addressee.for_stream_name(stream_name_or_id, topic_name)
elif recipient_type_name == "private":
if not message_to:
raise JsonableError(_("Message must have recipients"))
if isinstance(message_to[0], str):
emails = cast(Sequence[str], message_to)
return Addressee.for_private(emails, realm)
elif isinstance(message_to[0], int):
user_ids = cast(Sequence[int], message_to)
return Addressee.for_user_ids(user_ids=user_ids, realm=realm)
else:
raise JsonableError(_("Invalid message type"))
@staticmethod
def for_stream(stream: Stream, topic_name: str) -> "Addressee":
topic_name = topic_name.strip()
topic_name = maybe_rename_general_chat_to_empty_topic(topic_name)
topic_name = maybe_rename_no_topic_to_empty_topic(topic_name)
check_stream_topic(topic_name)
return Addressee(
msg_type="stream",
stream=stream,
topic_name=topic_name,
)
@staticmethod
def for_stream_name(stream_name: str, topic_name: str) -> "Addressee":
topic_name = topic_name.strip()
topic_name = maybe_rename_general_chat_to_empty_topic(topic_name)
topic_name = maybe_rename_no_topic_to_empty_topic(topic_name)
check_stream_topic(topic_name)
return Addressee(
msg_type="stream",
stream_name=stream_name,
topic_name=topic_name,
)
@staticmethod
def for_stream_id(stream_id: int, topic_name: str) -> "Addressee":
topic_name = topic_name.strip()
topic_name = maybe_rename_general_chat_to_empty_topic(topic_name)
topic_name = maybe_rename_no_topic_to_empty_topic(topic_name)
check_stream_topic(topic_name)
return Addressee(
msg_type="stream",
stream_id=stream_id,
topic_name=topic_name,
)
@staticmethod
def for_private(emails: Sequence[str], realm: Realm) -> "Addressee":
assert len(emails) > 0
user_profiles = get_user_profiles(emails, realm)
return Addressee(
msg_type="private",
user_profiles=user_profiles,
)
@staticmethod
def for_user_ids(user_ids: Sequence[int], realm: Realm) -> "Addressee":
assert len(user_ids) > 0
user_profiles = get_user_profiles_by_ids(user_ids, realm)
return Addressee(
msg_type="private",
user_profiles=user_profiles,
)
@staticmethod
def for_user_profile(user_profile: UserProfile) -> "Addressee":
user_profiles = [user_profile]
return Addressee(
msg_type="private",
user_profiles=user_profiles,
)
@staticmethod
def for_user_profiles(user_profiles: Sequence[UserProfile]) -> "Addressee":
assert len(user_profiles) > 0
return Addressee(
msg_type="private",
user_profiles=user_profiles,
)