Tulir Asokan
100 lines
3.8 KiB
Python
100 lines
3.8 KiB
Python
# mautrix-telegram - A Matrix-Telegram puppeting bridge
|
|
# Copyright (C) 2019 Tulir Asokan
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program 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 Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
from typing import Optional, Union, Any, List, Type, Dict
|
|
from enum import Enum
|
|
|
|
from telethon.tl.types import (MessageEntityMention as Mention, MessageEntityBotCommand as Command,
|
|
MessageEntityMentionName as MentionName, MessageEntityUrl as URL,
|
|
MessageEntityEmail as Email, MessageEntityTextUrl as TextURL,
|
|
MessageEntityBold as Bold, MessageEntityItalic as Italic,
|
|
MessageEntityCode as Code, MessageEntityPre as Pre,
|
|
MessageEntityStrike as Strike, MessageEntityUnderline as Underline,
|
|
MessageEntityBlockquote as Blockquote, TypeMessageEntity,
|
|
InputMessageEntityMentionName as InputMentionName)
|
|
|
|
from mautrix.util.formatter import EntityString, SemiAbstractEntity
|
|
|
|
|
|
class TelegramEntityType(Enum):
|
|
"""EntityType is a Matrix formatting entity type."""
|
|
BOLD = Bold
|
|
ITALIC = Italic
|
|
STRIKETHROUGH = Strike
|
|
UNDERLINE = Underline
|
|
URL = URL
|
|
INLINE_URL = TextURL
|
|
EMAIL = Email
|
|
PREFORMATTED = Pre
|
|
INLINE_CODE = Code
|
|
BLOCKQUOTE = Blockquote
|
|
MENTION = Mention
|
|
MENTION_NAME = MentionName
|
|
COMMAND = Command
|
|
|
|
USER_MENTION = 1
|
|
ROOM_MENTION = 2
|
|
HEADER = 3
|
|
|
|
|
|
class TelegramEntity(SemiAbstractEntity):
|
|
internal: TypeMessageEntity
|
|
|
|
def __init__(self, type: Union[TelegramEntityType, Type[TypeMessageEntity]],
|
|
offset: int, length: int, extra_info: Dict[str, Any]) -> None:
|
|
if isinstance(type, TelegramEntityType):
|
|
if isinstance(type.value, int):
|
|
raise ValueError(f"Can't create Entity with non-Telegram EntityType {type}")
|
|
type = type.value
|
|
self.internal = type(offset=offset, length=length, **extra_info)
|
|
|
|
def copy(self) -> Optional['TelegramEntity']:
|
|
extra_info = {}
|
|
if isinstance(self.internal, Pre):
|
|
extra_info["language"] = self.internal.language
|
|
elif isinstance(self.internal, TextURL):
|
|
extra_info["url"] = self.internal.url
|
|
elif isinstance(self.internal, (MentionName, InputMentionName)):
|
|
extra_info["user_id"] = self.internal.user_id
|
|
return TelegramEntity(type(self.internal), offset=self.internal.offset,
|
|
length=self.internal.length, extra_info=extra_info)
|
|
|
|
def __repr__(self) -> str:
|
|
return str(self.internal)
|
|
|
|
@property
|
|
def offset(self) -> int:
|
|
return self.internal.offset
|
|
|
|
@offset.setter
|
|
def offset(self, value: int) -> None:
|
|
self.internal.offset = value
|
|
|
|
@property
|
|
def length(self) -> int:
|
|
return self.internal.length
|
|
|
|
@length.setter
|
|
def length(self, value: int) -> None:
|
|
self.internal.length = value
|
|
|
|
|
|
class TelegramMessage(EntityString[TelegramEntity, TelegramEntityType]):
|
|
entity_class = TelegramEntity
|
|
|
|
@property
|
|
def telegram_entities(self) -> List[TypeMessageEntity]:
|
|
return [entity.internal for entity in self.entities]
|