iframe-proxy | Sunbelt Computer Software

History

148 lines (127 loc) · 5.09 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

# Pyrogram - Telegram MTProto API Client Library for Python

# 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

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 = 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)

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

markdown.py

Latest commit

History

markdown.py

File metadata and controls