266 lines
9.4 KiB
Python
266 lines
9.4 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 TYPE_CHECKING, Union
|
|
import time
|
|
|
|
from mautrix.bridge import BaseMatrixHandler
|
|
from mautrix.errors import MatrixError
|
|
from mautrix.types import (
|
|
Event,
|
|
EventID,
|
|
EventType,
|
|
MessageType,
|
|
PresenceEvent,
|
|
ReactionEvent,
|
|
ReactionEventContent,
|
|
ReceiptEvent,
|
|
RedactionEvent,
|
|
RelationType,
|
|
RoomID,
|
|
SingleReceiptEventContent,
|
|
TextMessageEventContent,
|
|
TypingEvent,
|
|
UserID,
|
|
)
|
|
|
|
from . import portal as po, puppet as pu, user as u
|
|
from .db import Message as DBMessage
|
|
|
|
if TYPE_CHECKING:
|
|
from .__main__ import KakaoTalkBridge
|
|
|
|
|
|
class MatrixHandler(BaseMatrixHandler):
|
|
def __init__(self, bridge: KakaoTalkBridge) -> None:
|
|
prefix, suffix = bridge.config["bridge.username_template"].format(userid=":").split(":")
|
|
homeserver = bridge.config["homeserver.domain"]
|
|
self.user_id_prefix = f"@{prefix}"
|
|
self.user_id_suffix = f"{suffix}:{homeserver}"
|
|
super().__init__(bridge=bridge)
|
|
|
|
async def send_welcome_message(self, room_id: RoomID, inviter: u.User) -> None:
|
|
await super().send_welcome_message(room_id, inviter)
|
|
if not inviter.notice_room:
|
|
inviter.notice_room = room_id
|
|
await inviter.save()
|
|
await self.az.intent.send_notice(
|
|
room_id, "This room has been marked as your KakaoTalk bridge notice room."
|
|
)
|
|
|
|
async def handle_puppet_invite(
|
|
self, room_id: RoomID, puppet: pu.Puppet, invited_by: u.User, event_id: EventID
|
|
) -> None:
|
|
intent = puppet.default_mxid_intent
|
|
self.log.debug(f"{invited_by.mxid} invited puppet for {puppet.ktid} to {room_id}")
|
|
if not await invited_by.is_logged_in():
|
|
await intent.error_and_leave(
|
|
room_id,
|
|
text="Please log in before inviting KakaoTalk puppets to private chats.",
|
|
)
|
|
return
|
|
|
|
portal = await po.Portal.get_by_mxid(room_id)
|
|
if portal:
|
|
if portal.is_direct:
|
|
await intent.error_and_leave(
|
|
room_id, text="You can not invite additional users to private chats."
|
|
)
|
|
return
|
|
# TODO add KakaoTalk inviting
|
|
# await portal.invite_kakaotalk(inviter, puppet)
|
|
# await intent.join_room(room_id)
|
|
return
|
|
await intent.join_room(room_id)
|
|
try:
|
|
members = await intent.get_room_members(room_id)
|
|
except MatrixError:
|
|
self.log.exception(f"Failed to get member list after joining {room_id}")
|
|
await intent.leave_room(room_id)
|
|
return
|
|
if len(members) > 2:
|
|
# TODO add KakaoTalk group creating
|
|
await intent.send_notice(
|
|
room_id, "You can not invite KakaoTalk puppets to multi-user rooms."
|
|
)
|
|
await intent.leave_room(room_id)
|
|
return
|
|
portal = await po.Portal.get_by_ktid(
|
|
puppet.ktid, fb_receiver=invited_by.ktid # TODO kt_type=??
|
|
)
|
|
if portal.mxid:
|
|
try:
|
|
await intent.invite_user(portal.mxid, invited_by.mxid, check_cache=False)
|
|
await intent.send_notice(
|
|
room_id,
|
|
text=f"You already have a private chat with me in room {portal.mxid}",
|
|
html=(
|
|
"You already have a private chat with me: "
|
|
f"<a href='https://matrix.to/#/{portal.mxid}'>Link to room</a>"
|
|
),
|
|
)
|
|
await intent.leave_room(room_id)
|
|
return
|
|
except MatrixError:
|
|
pass
|
|
portal.mxid = room_id
|
|
e2be_ok = await portal.check_dm_encryption()
|
|
await portal.save()
|
|
if e2be_ok is True:
|
|
evt_type, content = await self.e2ee.encrypt(
|
|
room_id,
|
|
EventType.ROOM_MESSAGE,
|
|
TextMessageEventContent(
|
|
msgtype=MessageType.NOTICE,
|
|
body="Portal to private chat created and end-to-bridge encryption enabled.",
|
|
),
|
|
)
|
|
await intent.send_message_event(room_id, evt_type, content)
|
|
else:
|
|
message = "Portal to private chat created."
|
|
if e2be_ok is False:
|
|
message += "\n\nWarning: Failed to enable end-to-bridge encryption"
|
|
await intent.send_notice(room_id, message)
|
|
|
|
async def handle_invite(
|
|
self, room_id: RoomID, user_id: UserID, invited_by: u.User, event_id: EventID
|
|
) -> None:
|
|
# TODO handle puppet and user invites for group chats
|
|
# The rest can probably be ignored
|
|
pass
|
|
|
|
async def handle_join(self, room_id: RoomID, user_id: UserID, event_id: EventID) -> None:
|
|
user = await u.User.get_by_mxid(user_id)
|
|
|
|
portal = await po.Portal.get_by_mxid(room_id)
|
|
if not portal:
|
|
return
|
|
|
|
if not user.relay_whitelisted:
|
|
await portal.main_intent.kick_user(
|
|
room_id, user.mxid, "You are not whitelisted on this KakaoTalk bridge."
|
|
)
|
|
return
|
|
elif (
|
|
not await user.is_logged_in()
|
|
and not portal.has_relay
|
|
and not self.config["bridge.allow_invites"]
|
|
):
|
|
await portal.main_intent.kick_user(
|
|
room_id, user.mxid, "You are not logged in to this KakaoTalk bridge."
|
|
)
|
|
return
|
|
|
|
self.log.debug(f"{user.mxid} joined {room_id}")
|
|
# await portal.join_matrix(user, event_id)
|
|
|
|
async def handle_leave(self, room_id: RoomID, user_id: UserID, event_id: EventID) -> None:
|
|
portal = await po.Portal.get_by_mxid(room_id)
|
|
if not portal:
|
|
return
|
|
|
|
user = await u.User.get_by_mxid(user_id, create=False)
|
|
if not user:
|
|
return
|
|
|
|
await portal.handle_matrix_leave(user)
|
|
|
|
@staticmethod
|
|
async def handle_redaction(
|
|
room_id: RoomID, user_id: UserID, event_id: EventID, redaction_event_id: EventID
|
|
) -> None:
|
|
user = await u.User.get_by_mxid(user_id)
|
|
if not user:
|
|
return
|
|
|
|
portal = await po.Portal.get_by_mxid(room_id)
|
|
if not portal:
|
|
return
|
|
|
|
await portal.handle_matrix_redaction(user, event_id, redaction_event_id)
|
|
|
|
@classmethod
|
|
async def handle_reaction(
|
|
cls,
|
|
room_id: RoomID,
|
|
user_id: UserID,
|
|
event_id: EventID,
|
|
content: ReactionEventContent,
|
|
) -> None:
|
|
if content.relates_to.rel_type != RelationType.ANNOTATION:
|
|
cls.log.debug(
|
|
f"Ignoring m.reaction event in {room_id} from {user_id} with unexpected "
|
|
f"relation type {content.relates_to.rel_type}"
|
|
)
|
|
return
|
|
user = await u.User.get_by_mxid(user_id)
|
|
if not user:
|
|
return
|
|
|
|
portal = await po.Portal.get_by_mxid(room_id)
|
|
if not portal:
|
|
return
|
|
|
|
await portal.handle_matrix_reaction(
|
|
user, event_id, content.relates_to.event_id, content.relates_to.key
|
|
)
|
|
|
|
@staticmethod
|
|
async def handle_typing(room_id: RoomID, typing: list[UserID]) -> None:
|
|
portal = await po.Portal.get_by_mxid(room_id)
|
|
if not portal or not portal.is_direct:
|
|
return
|
|
|
|
await portal.handle_matrix_typing(set(typing))
|
|
|
|
async def handle_read_receipt(
|
|
self,
|
|
user: u.User,
|
|
portal: po.Portal,
|
|
event_id: EventID,
|
|
data: SingleReceiptEventContent,
|
|
) -> None:
|
|
self.log.info("TODO")
|
|
"""
|
|
if not user.mqtt:
|
|
return
|
|
timestamp = data.get("ts", int(time.time() * 1000))
|
|
message = await DBMessage.get_by_mxid(event_id, portal.mxid)
|
|
await user.mqtt.mark_read(
|
|
portal.ktid,
|
|
True, # TODO
|
|
#portal.fb_type != ThreadType.USER,
|
|
read_to=message.timestamp if message else timestamp,
|
|
)
|
|
"""
|
|
|
|
async def handle_ephemeral_event(
|
|
self, evt: Union[ReceiptEvent, PresenceEvent, TypingEvent]
|
|
) -> None:
|
|
if evt.type == EventType.TYPING:
|
|
await self.handle_typing(evt.room_id, evt.content.user_ids)
|
|
elif evt.type == EventType.RECEIPT:
|
|
await self.handle_receipt(evt)
|
|
|
|
async def handle_event(self, evt: Event) -> None:
|
|
if evt.type == EventType.ROOM_REDACTION:
|
|
evt: RedactionEvent
|
|
await self.handle_redaction(evt.room_id, evt.sender, evt.redacts, evt.event_id)
|
|
elif evt.type == EventType.REACTION:
|
|
evt: ReactionEvent
|
|
await self.handle_reaction(evt.room_id, evt.sender, evt.event_id, evt.content)
|