mirror of
https://github.com/zulip/zulip.git
synced 2025-10-31 12:03:46 +00:00
365fe0b3d5
Fixes #2665. Regenerated by tabbott with `lint --fix` after a rebase and change in parameters. Note from tabbott: In a few cases, this converts technical debt in the form of unsorted imports into different technical debt in the form of our largest files having very long, ugly import sequences at the start. I expect this change will increase pressure for us to split those files, which isn't a bad thing. Signed-off-by: Anders Kaseorg <anders@zulip.com>
104 lines
4.1 KiB
Python
104 lines
4.1 KiB
Python
import re
|
|
from typing import Any, Callable, Dict, Tuple
|
|
|
|
from django.conf import settings
|
|
from django.utils.text import slugify
|
|
|
|
from zerver.models import Stream
|
|
|
|
|
|
def default_option_handler_factory(address_option: str) -> Callable[[Dict[str, Any]], None]:
|
|
def option_setter(options_dict: Dict[str, Any]) -> None:
|
|
options_dict[address_option.replace('-', '_')] = True
|
|
|
|
return option_setter
|
|
|
|
optional_address_tokens = {
|
|
"show-sender": default_option_handler_factory("show-sender"),
|
|
"include-footer": default_option_handler_factory("include-footer"),
|
|
"include-quotes": default_option_handler_factory("include-quotes"),
|
|
"prefer-text": lambda options: options.update(prefer_text=True),
|
|
"prefer-html": lambda options: options.update(prefer_text=False),
|
|
}
|
|
|
|
class ZulipEmailForwardError(Exception):
|
|
pass
|
|
|
|
def get_email_gateway_message_string_from_address(address: str) -> str:
|
|
pattern_parts = [re.escape(part) for part in settings.EMAIL_GATEWAY_PATTERN.split('%s')]
|
|
if settings.EMAIL_GATEWAY_EXTRA_PATTERN_HACK:
|
|
# Accept mails delivered to any Zulip server
|
|
pattern_parts[-1] = settings.EMAIL_GATEWAY_EXTRA_PATTERN_HACK
|
|
match_email_re = re.compile("(.*?)".join(pattern_parts))
|
|
match = match_email_re.match(address)
|
|
|
|
if not match:
|
|
raise ZulipEmailForwardError('Address not recognized by gateway.')
|
|
msg_string = match.group(1)
|
|
|
|
return msg_string
|
|
|
|
def encode_email_address(stream: Stream, show_sender: bool=False) -> str:
|
|
return encode_email_address_helper(stream.name, stream.email_token, show_sender)
|
|
|
|
def encode_email_address_helper(name: str, email_token: str, show_sender: bool=False) -> str:
|
|
# Some deployments may not use the email gateway
|
|
if settings.EMAIL_GATEWAY_PATTERN == '':
|
|
return ''
|
|
|
|
# Given the fact that we have almost no restrictions on stream names and
|
|
# that what characters are allowed in e-mail addresses is complicated and
|
|
# dependent on context in the address, we opt for a simple scheme:
|
|
# 1. Replace all substrings of non-alphanumeric characters with a single hyphen.
|
|
# 2. Use Django's slugify to convert the resulting name to ascii.
|
|
# 3. If the resulting name is shorter than the name we got in step 1,
|
|
# it means some letters can't be reasonably turned to ascii and have to be dropped,
|
|
# which would mangle the name, so we just skip the name part of the address.
|
|
name = re.sub(r"\W+", '-', name)
|
|
slug_name = slugify(name)
|
|
encoded_name = slug_name if len(slug_name) == len(name) else ''
|
|
|
|
# If encoded_name ends up empty, we just skip this part of the address:
|
|
if encoded_name:
|
|
encoded_token = f"{encoded_name}.{email_token}"
|
|
else:
|
|
encoded_token = email_token
|
|
|
|
if show_sender:
|
|
encoded_token += ".show-sender"
|
|
|
|
return settings.EMAIL_GATEWAY_PATTERN % (encoded_token,)
|
|
|
|
def decode_email_address(email: str) -> Tuple[str, Dict[str, bool]]:
|
|
# Perform the reverse of encode_email_address. Returns a tuple of
|
|
# (email_token, options)
|
|
msg_string = get_email_gateway_message_string_from_address(email)
|
|
|
|
# Support both + and . as separators. For background, the `+` is
|
|
# more aesthetically pleasing, but because Google groups silently
|
|
# drops the use of `+` in email addresses, which would completely
|
|
# break the integration, we now favor `.` as the separator between
|
|
# tokens in the email addresses we generate.
|
|
#
|
|
# We need to keep supporting `+` indefinitely for backwards
|
|
# compatibility with older versions of Zulip that offered users
|
|
# email addresses prioritizing using `+` for better aesthetics.
|
|
msg_string = msg_string.replace('.', '+')
|
|
|
|
parts = msg_string.split('+')
|
|
options: Dict[str, bool] = {}
|
|
for part in parts:
|
|
if part in optional_address_tokens:
|
|
optional_address_tokens[part](options)
|
|
|
|
remaining_parts = [part for part in parts if part not in optional_address_tokens]
|
|
|
|
# There should be one or two parts left:
|
|
# [stream_name, email_token] or just [email_token]
|
|
if len(remaining_parts) == 1:
|
|
token = remaining_parts[0]
|
|
else:
|
|
token = remaining_parts[1]
|
|
|
|
return token, options
|