149 lines
5.1 KiB
Python
149 lines
5.1 KiB
Python
# Pyrogram - Telegram MTProto API Client Library for Python
|
|
# Copyright (C) 2017-2019 Dan Tès <https://github.com/delivrance>
|
|
#
|
|
# This file is part of Pyrogram.
|
|
#
|
|
# Pyrogram is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Lesser General Public License as published
|
|
# by the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Pyrogram is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public License
|
|
# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import re
|
|
from collections import OrderedDict
|
|
|
|
import pyrogram
|
|
from pyrogram.api.types import (
|
|
MessageEntityBold as Bold,
|
|
MessageEntityItalic as Italic,
|
|
MessageEntityCode as Code,
|
|
MessageEntityTextUrl as Url,
|
|
MessageEntityPre as Pre,
|
|
MessageEntityMentionName as MentionInvalid,
|
|
InputMessageEntityMentionName as Mention
|
|
)
|
|
from pyrogram.errors import PeerIdInvalid
|
|
from . import utils
|
|
|
|
|
|
class Markdown:
|
|
BOLD_DELIMITER = "**"
|
|
ITALIC_DELIMITER = "__"
|
|
CODE_DELIMITER = "`"
|
|
PRE_DELIMITER = "```"
|
|
|
|
MARKDOWN_RE = re.compile(r"({d})([\w\W]*?)\1|\[([^[]+?)\]\(([^(]+?)\)".format(
|
|
d="|".join(
|
|
["".join(i) for i in [
|
|
["\{}".format(j) for j in i]
|
|
for i in [
|
|
PRE_DELIMITER,
|
|
CODE_DELIMITER,
|
|
ITALIC_DELIMITER,
|
|
BOLD_DELIMITER
|
|
]
|
|
]]
|
|
)
|
|
))
|
|
MENTION_RE = re.compile(r"tg://user\?id=(\d+)")
|
|
|
|
def __init__(self, client: "pyrogram.BaseClient" = None):
|
|
self.client = client
|
|
|
|
async def parse(self, message: str):
|
|
message = utils.add_surrogates(str(message or "")).strip()
|
|
entities = []
|
|
offset = 0
|
|
|
|
for match in self.MARKDOWN_RE.finditer(message):
|
|
start = match.start() - offset
|
|
style, body, text, url = match.groups()
|
|
|
|
if url:
|
|
mention = self.MENTION_RE.match(url)
|
|
|
|
if mention:
|
|
user_id = int(mention.group(1))
|
|
|
|
try:
|
|
input_user = await self.client.resolve_peer(user_id)
|
|
except PeerIdInvalid:
|
|
input_user = None
|
|
|
|
entity = (
|
|
Mention(offset=start, length=len(text), user_id=input_user)
|
|
if input_user else MentionInvalid(offset=start, length=len(text), user_id=user_id)
|
|
)
|
|
else:
|
|
entity = Url(offset=start, length=len(text), url=url)
|
|
|
|
body = text
|
|
offset += len(url) + 4
|
|
else:
|
|
if style == self.BOLD_DELIMITER:
|
|
entity = Bold(offset=start, length=len(body))
|
|
elif style == self.ITALIC_DELIMITER:
|
|
entity = Italic(offset=start, length=len(body))
|
|
elif style == self.CODE_DELIMITER:
|
|
entity = Code(offset=start, length=len(body))
|
|
elif style == self.PRE_DELIMITER:
|
|
entity = Pre(offset=start, length=len(body), language="")
|
|
else:
|
|
continue
|
|
|
|
offset += len(style) * 2
|
|
|
|
entities.append(entity)
|
|
message = message.replace(match.group(), body)
|
|
|
|
# TODO: OrderedDict to be removed in Python3.6
|
|
return OrderedDict([
|
|
("message", utils.remove_surrogates(message)),
|
|
("entities", entities)
|
|
])
|
|
|
|
def unparse(self, message: str, entities: list):
|
|
message = utils.add_surrogates(message).strip()
|
|
offset = 0
|
|
|
|
for entity in entities:
|
|
start = entity.offset + offset
|
|
type = entity.type
|
|
url = entity.url
|
|
user = entity.user
|
|
sub = message[start: start + entity.length]
|
|
|
|
if type == "bold":
|
|
style = self.BOLD_DELIMITER
|
|
elif type == "italic":
|
|
style = self.ITALIC_DELIMITER
|
|
elif type == "code":
|
|
style = self.CODE_DELIMITER
|
|
elif type == "pre":
|
|
style = self.PRE_DELIMITER
|
|
elif type == "text_link":
|
|
offset += 4 + len(url)
|
|
message = message[:start] + message[start:].replace(
|
|
sub, "[{}]({})".format(sub, url), 1)
|
|
continue
|
|
elif type == "text_mention":
|
|
offset += 17 + len(str(user.id))
|
|
message = message[:start] + message[start:].replace(
|
|
sub, "[{}](tg://user?id={})".format(sub, user.id), 1)
|
|
continue
|
|
else:
|
|
continue
|
|
|
|
offset += len(style) * 2
|
|
message = message[:start] + message[start:].replace(
|
|
sub, "{0}{1}{0}".format(style, sub), 1)
|
|
|
|
return utils.remove_surrogates(message)
|