121 lines
4.3 KiB
Python
121 lines
4.3 KiB
Python
# matrix-appservice-kakaotalk - A Matrix-KakaoTalk puppeting bridge.
|
|
# Copyright (C) 2022 Tulir Asokan, Andrew Ferrazzutti
|
|
#
|
|
# 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 __future__ import annotations
|
|
|
|
from typing import NamedTuple
|
|
|
|
from mautrix.types import Format, MessageEventContent, RelationType, RoomID
|
|
from mautrix.util.formatter import (
|
|
EntityString,
|
|
EntityType,
|
|
MarkdownString,
|
|
MatrixParser as BaseMatrixParser,
|
|
SimpleEntity,
|
|
)
|
|
from mautrix.util.logging import TraceLogger
|
|
|
|
from .. import puppet as pu, user as u
|
|
from ..db import Message as DBMessage
|
|
|
|
|
|
class SendParams(NamedTuple):
|
|
text: str
|
|
mentions: list[None]
|
|
reply_to: str
|
|
|
|
|
|
class FacebookFormatString(EntityString[SimpleEntity, EntityType], MarkdownString):
|
|
def format(self, entity_type: EntityType, **kwargs) -> FacebookFormatString:
|
|
prefix = suffix = ""
|
|
if entity_type == EntityType.USER_MENTION:
|
|
self.entities.append(
|
|
SimpleEntity(
|
|
type=entity_type,
|
|
offset=0,
|
|
length=len(self.text),
|
|
extra_info={"user_id": kwargs["user_id"]},
|
|
)
|
|
)
|
|
return self
|
|
elif entity_type == EntityType.BOLD:
|
|
prefix = suffix = "*"
|
|
elif entity_type == EntityType.ITALIC:
|
|
prefix = suffix = "_"
|
|
elif entity_type == EntityType.STRIKETHROUGH:
|
|
prefix = suffix = "~"
|
|
elif entity_type == EntityType.URL:
|
|
if kwargs["url"] != self.text:
|
|
suffix = f" ({kwargs['url']})"
|
|
elif entity_type == EntityType.PREFORMATTED:
|
|
prefix = f"```{kwargs['language']}\n"
|
|
suffix = "\n```"
|
|
elif entity_type == EntityType.INLINE_CODE:
|
|
prefix = suffix = "`"
|
|
elif entity_type == EntityType.BLOCKQUOTE:
|
|
children = self.trim().split("\n")
|
|
children = [child.prepend("> ") for child in children]
|
|
return self.join(children, "\n")
|
|
elif entity_type == EntityType.HEADER:
|
|
prefix = "#" * kwargs["size"] + " "
|
|
else:
|
|
return self
|
|
|
|
self._offset_entities(len(prefix))
|
|
self.text = f"{prefix}{self.text}{suffix}"
|
|
return self
|
|
|
|
|
|
class MatrixParser(BaseMatrixParser[FacebookFormatString]):
|
|
fs = FacebookFormatString
|
|
|
|
|
|
async def matrix_to_kakaotalk(
|
|
content: MessageEventContent, room_id: RoomID, log: TraceLogger
|
|
) -> SendParams:
|
|
mentions = []
|
|
reply_to = None
|
|
if content.relates_to.rel_type == RelationType.REPLY:
|
|
message = await DBMessage.get_by_mxid(content.relates_to.event_id, room_id)
|
|
if message:
|
|
content.trim_reply_fallback()
|
|
reply_to = message.ktid
|
|
else:
|
|
log.warning(
|
|
f"Couldn't find reply target {content.relates_to.event_id}"
|
|
" to bridge text message reply metadata to Facebook"
|
|
)
|
|
if content.get("format", None) == Format.HTML and content["formatted_body"]:
|
|
parsed = await MatrixParser().parse(content["formatted_body"])
|
|
text = parsed.text
|
|
mentions = []
|
|
for mention in parsed.entities:
|
|
mxid = mention.extra_info["user_id"]
|
|
user = await u.User.get_by_mxid(mxid, create=False)
|
|
if user and user.ktid:
|
|
ktid = user.ktid
|
|
else:
|
|
puppet = await pu.Puppet.get_by_mxid(mxid, create=False)
|
|
if puppet:
|
|
ktid = puppet.ktid
|
|
else:
|
|
continue
|
|
#mentions.append(
|
|
# Mention(user_id=str(ktid), offset=mention.offset, length=mention.length)
|
|
#)
|
|
else:
|
|
text = content.body
|
|
return SendParams(text=text, mentions=mentions, reply_to=reply_to)
|