mirror of https://github.com/zulip/zulip.git
109 lines
3.2 KiB
Python
109 lines
3.2 KiB
Python
import json
|
|
import re
|
|
from typing import Any, Optional, Tuple
|
|
|
|
from zerver.lib.message import SendMessageRequest
|
|
from zerver.models import Message, SubMessage
|
|
|
|
|
|
def get_widget_data(content: str) -> Tuple[Optional[str], Optional[str]]:
|
|
valid_widget_types = ["poll", "todo"]
|
|
tokens = re.split(r"\s+|\n+", content)
|
|
|
|
# tokens[0] will always exist
|
|
if tokens[0].startswith("/"):
|
|
widget_type = tokens[0][1:]
|
|
if widget_type in valid_widget_types:
|
|
remaining_content = content.replace(tokens[0], "", 1).strip()
|
|
extra_data = get_extra_data_from_widget_type(remaining_content, widget_type)
|
|
return widget_type, extra_data
|
|
|
|
return None, None
|
|
|
|
|
|
def get_extra_data_from_widget_type(content: str, widget_type: Optional[str]) -> Any:
|
|
if widget_type == "poll":
|
|
# This is used to extract the question from the poll command.
|
|
# The command '/poll question' will pre-set the question in the poll
|
|
lines = content.splitlines()
|
|
question = ""
|
|
options = []
|
|
if lines and lines[0]:
|
|
question = lines.pop(0).strip()
|
|
for line in lines:
|
|
# If someone is using the list syntax, we remove it
|
|
# before adding an option.
|
|
option = re.sub(r"(\s*[-*]?\s*)", "", line.strip(), count=1)
|
|
if len(option) > 0:
|
|
options.append(option)
|
|
extra_data = {
|
|
"question": question,
|
|
"options": options,
|
|
}
|
|
return extra_data
|
|
return None
|
|
|
|
|
|
def do_widget_post_save_actions(send_request: SendMessageRequest) -> None:
|
|
"""
|
|
This code works with the web app; mobile and other
|
|
clients should also start supporting this soon.
|
|
"""
|
|
message_content = send_request.message.content
|
|
sender_id = send_request.message.sender_id
|
|
message_id = send_request.message.id
|
|
|
|
widget_type = None
|
|
extra_data = None
|
|
|
|
widget_type, extra_data = get_widget_data(message_content)
|
|
widget_content = send_request.widget_content
|
|
if widget_content is not None:
|
|
# Note that we validate this data in check_message,
|
|
# so we can trust it here.
|
|
widget_type = widget_content["widget_type"]
|
|
extra_data = widget_content["extra_data"]
|
|
|
|
if widget_type:
|
|
content = dict(
|
|
widget_type=widget_type,
|
|
extra_data=extra_data,
|
|
)
|
|
submessage = SubMessage(
|
|
sender_id=sender_id,
|
|
message_id=message_id,
|
|
msg_type="widget",
|
|
content=json.dumps(content),
|
|
)
|
|
submessage.save()
|
|
send_request.submessages = SubMessage.get_raw_db_rows([message_id])
|
|
|
|
|
|
def get_widget_type(*, message_id: int) -> Optional[str]:
|
|
submessage = (
|
|
SubMessage.objects.filter(
|
|
message_id=message_id,
|
|
msg_type="widget",
|
|
)
|
|
.only("content")
|
|
.first()
|
|
)
|
|
|
|
if submessage is None:
|
|
return None
|
|
|
|
try:
|
|
data = json.loads(submessage.content)
|
|
except Exception:
|
|
return None
|
|
|
|
try:
|
|
return data["widget_type"]
|
|
except Exception:
|
|
return None
|
|
|
|
|
|
def is_widget_message(message: Message) -> bool:
|
|
# Right now all messages that are widgetized use submessage, and vice versa.
|
|
return message.submessage_set.exists()
|