mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 22:13:26 +00:00
d5a36ac5e2
Earlier, MENTIONS_RE was: r"(?<![^\s\'\"\(,:<])@(?P<silent>_?)(?P<match>\*\*[^\*]+\*\*)" For the syntax: **foo**, this was unnecessarily capturing it as **foo** and adding extra operation for the extraction of `foo`. This is now changed as: r"(?<![^\s\'\"\(,:<])@(?P<silent>_?)(\*\*(?P<match>[^\*]+)\*\*)" and extraction of `foo` can be done just by using the named capture group `match`. This change also helps to simplify its related code path.
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
import re
|
|
from typing import Match, Optional, Set, Tuple
|
|
|
|
# Match multi-word string between @** ** or match any one-word
|
|
# sequences after @
|
|
MENTIONS_RE = re.compile(r"(?<![^\s\'\"\(,:<])@(?P<silent>_?)(\*\*(?P<match>[^\*]+)\*\*)")
|
|
USER_GROUP_MENTIONS_RE = r"(?<![^\s\'\"\(,:<])@(\*[^\*]+\*)"
|
|
|
|
wildcards = ["all", "everyone", "stream"]
|
|
|
|
|
|
def user_mention_matches_wildcard(mention: str) -> bool:
|
|
return mention in wildcards
|
|
|
|
|
|
def extract_mention_text(m: Match[str]) -> Tuple[Optional[str], bool]:
|
|
text = m.group("match")
|
|
if text in wildcards:
|
|
return None, True
|
|
return text, False
|
|
|
|
|
|
def possible_mentions(content: str) -> Tuple[Set[str], bool]:
|
|
# mention texts can either be names, or an extended name|id syntax.
|
|
texts = set()
|
|
message_has_wildcards = False
|
|
for m in MENTIONS_RE.finditer(content):
|
|
text, is_wildcard = extract_mention_text(m)
|
|
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_RE, content)
|
|
return {extract_user_group(match) for match in matches}
|