mirror of https://github.com/zulip/zulip.git
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
|