matrix-appservice-kakaotalk/matrix_appservice_kakaotalk/formatter/from_matrix.py

152 lines
6.0 KiB
Python
Raw Normal View History

2022-02-25 02:22:50 -05:00
# 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
2022-04-05 15:44:02 -04:00
from mautrix.appservice import IntentAPI
2022-04-06 12:49:23 -04:00
from mautrix.types import Format, MessageEventContent, RelationType, RoomID, UserID
from mautrix.util.formatter import (
EntityString,
EntityType,
MarkdownString,
MatrixParser,
SimpleEntity,
)
2022-02-25 02:22:50 -05:00
from mautrix.util.logging import TraceLogger
2022-04-05 15:44:02 -04:00
from ..kt.types.bson import Long
2022-04-06 14:33:01 -04:00
from ..kt.types.chat import KnownChatType
2022-04-06 12:49:23 -04:00
from ..kt.types.chat.attachment import ReplyAttachment, MentionStruct
2022-04-05 15:44:02 -04:00
from ..kt.client.types import TO_MSGTYPE_MAP
2022-04-06 12:49:23 -04:00
from .. import puppet as pu, user as u
2022-02-25 02:22:50 -05:00
from ..db import Message as DBMessage
class SendParams(NamedTuple):
text: str
2022-04-06 12:49:23 -04:00
mentions: list[MentionStruct] | None
2022-04-05 15:44:02 -04:00
reply_to: ReplyAttachment
2022-02-25 02:22:50 -05:00
2022-04-06 12:49:23 -04:00
class KakaoTalkFormatString(EntityString[SimpleEntity, EntityType], MarkdownString):
def format(self, entity_type: EntityType, **kwargs) -> KakaoTalkFormatString:
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"]},
)
)
self.text = f"@{self.text}"
return self
class ToKakaoTalkParser(MatrixParser[KakaoTalkFormatString]):
fs = KakaoTalkFormatString
async def _get_id_from_mxid(mxid: UserID) -> Long | None:
user = await u.User.get_by_mxid(mxid, create=False)
if user and user.ktid:
return user.ktid
else:
puppet = await pu.Puppet.get_by_mxid(mxid, create=False)
return puppet.ktid if puppet else None
2022-02-25 02:22:50 -05:00
async def matrix_to_kakaotalk(
2022-04-06 14:11:15 -04:00
content: MessageEventContent,
room_id: RoomID,
log: TraceLogger,
intent: IntentAPI,
skip_reply: bool = False
2022-02-25 02:22:50 -05:00
) -> SendParams:
2022-04-05 15:44:02 -04:00
# NOTE By design, this *throws* if user intent can't be matched (i.e. if a reply can't be created)
2022-04-06 14:11:15 -04:00
if content.relates_to.rel_type == RelationType.REPLY and not skip_reply:
2022-02-25 02:22:50 -05:00
message = await DBMessage.get_by_mxid(content.relates_to.event_id, room_id)
2022-04-05 15:44:02 -04:00
if not message:
raise ValueError(
2022-02-25 02:22:50 -05:00
f"Couldn't find reply target {content.relates_to.event_id}"
2022-04-05 15:44:02 -04:00
" to bridge text message reply metadata to KakaoTalk"
)
try:
2022-04-06 14:11:15 -04:00
src_event = await intent.get_event(room_id, message.mxid)
2022-04-05 15:44:02 -04:00
except:
log.exception(f"Failed to find Matrix event for reply target {message.mxid}")
raise
2022-04-06 14:11:15 -04:00
src_kt_sender = await _get_id_from_mxid(src_event.sender)
if src_kt_sender is None:
2022-04-05 15:44:02 -04:00
raise ValueError(
2022-04-06 14:11:15 -04:00
f"Found no KakaoTalk user ID for reply target sender {src_event.sender}"
2022-02-25 02:22:50 -05:00
)
2022-04-05 15:44:02 -04:00
content.trim_reply_fallback()
2022-04-06 14:11:15 -04:00
src_converted = await matrix_to_kakaotalk(src_event.content, room_id, log, intent, skip_reply=True)
2022-04-06 14:33:01 -04:00
if src_event.content.relates_to.rel_type == RelationType.REPLY:
src_type = KnownChatType.REPLY
else:
src_type = TO_MSGTYPE_MAP[src_event.content.msgtype]
2022-04-05 15:44:02 -04:00
reply_to = ReplyAttachment(
2022-04-06 14:11:15 -04:00
# NOTE mentions will be merged into this later
# TODO Set this for emoticon reply, but must first support them
2022-04-05 15:44:02 -04:00
attach_only=False,
# TODO If replying with media works, must set type AND all attachment properties
# But then, the reply object must be an intersection of a ReplyAttachment and something else
#attach_type=TO_MSGTYPE_MAP.get(content.msgtype),
2022-04-06 14:33:01 -04:00
# TODO Confirm why official client sets this to 0, and whether this should be left as None instead
attach_type=0,
2022-04-05 15:44:02 -04:00
src_logId=message.ktid,
2022-04-06 14:11:15 -04:00
src_mentions=src_converted.mentions or [],
src_message=src_converted.text,
2022-04-06 14:33:01 -04:00
src_type=src_type,
2022-04-06 14:11:15 -04:00
src_userId=src_kt_sender,
2022-04-05 15:44:02 -04:00
)
2022-02-25 02:22:50 -05:00
else:
2022-04-05 15:44:02 -04:00
reply_to = None
2022-04-06 12:49:23 -04:00
if content.get("format", None) == Format.HTML and content["formatted_body"]:
parsed = await ToKakaoTalkParser().parse(content["formatted_body"])
text = parsed.text
mentions_by_user: dict[Long, MentionStruct] = {}
# Make sure to not create remote mentions for any remote user not in the room
if parsed.entities:
joined_members = set(await intent.get_room_members(room_id))
last_offset = 0
at = 0
for mention in sorted(parsed.entities, key=lambda entity: entity.offset):
mxid = mention.extra_info["user_id"]
if mxid not in joined_members:
continue
ktid = await _get_id_from_mxid(mxid)
if ktid is None:
continue
at += text[last_offset:mention.offset+1].count("@")
last_offset = mention.offset+1
mention = mentions_by_user.setdefault(ktid, MentionStruct(
at=[],
len=mention.length,
user_id=ktid,
))
mention.at.append(at)
2022-04-06 14:33:01 -04:00
mentions = list(mentions_by_user.values()) if mentions_by_user else None
2022-04-06 12:49:23 -04:00
else:
text = content.body
mentions = None
return SendParams(text=text, mentions=mentions, reply_to=reply_to)