mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 22:13:26 +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>
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
import re
|
|
from typing import Optional, Set, Tuple
|
|
|
|
# Match multi-word string between @** ** or match any one-word
|
|
# sequences after @
|
|
find_mentions = r'(?<![^\s\'\"\(,:<])@(?P<silent>_?)(?P<match>\*\*[^\*]+\*\*|all|everyone|stream)'
|
|
user_group_mentions = r'(?<![^\s\'\"\(,:<])@(\*[^\*]+\*)'
|
|
|
|
wildcards = ['all', 'everyone', 'stream']
|
|
|
|
def user_mention_matches_wildcard(mention: str) -> bool:
|
|
return mention in wildcards
|
|
|
|
def extract_mention_text(m: Tuple[str, str]) -> Tuple[Optional[str], bool]:
|
|
# re.findall provides tuples of match elements; we want the second
|
|
# to get the main mention content.
|
|
s = m[1]
|
|
if s.startswith("**") and s.endswith("**"):
|
|
text = s[2:-2]
|
|
if text in wildcards:
|
|
return None, True
|
|
return text, False
|
|
return None, False
|
|
|
|
def possible_mentions(content: str) -> Tuple[Set[str], bool]:
|
|
matches = re.findall(find_mentions, content)
|
|
# mention texts can either be names, or an extended name|id syntax.
|
|
texts = set()
|
|
message_has_wildcards = False
|
|
for match in matches:
|
|
text, is_wildcard = extract_mention_text(match)
|
|
if text:
|
|
texts.add(text)
|
|
if is_wildcard:
|
|
message_has_wildcards = True
|
|
return texts, message_has_wildcards
|
|
|
|
def extract_user_group(matched_text: str) -> str:
|
|
return matched_text[1:-1]
|
|
|
|
def possible_user_group_mentions(content: str) -> Set[str]:
|
|
matches = re.findall(user_group_mentions, content)
|
|
return {extract_user_group(match) for match in matches}
|