46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
|
from yuxi.channel.extensions.xmpp.types import InboundXmppMessage, ResolvedXmppAccount
|
||
|
|
from yuxi.channel.message.models import GroupContext, MessageType, PeerInfo, UnifiedMessage
|
||
|
|
from yuxi.channel.routing.models import PeerKind
|
||
|
|
|
||
|
|
|
||
|
|
def xmpp_stanza_to_unified(
|
||
|
|
inbound: InboundXmppMessage,
|
||
|
|
account: ResolvedXmppAccount,
|
||
|
|
) -> UnifiedMessage:
|
||
|
|
is_group = inbound.msg_type == "groupchat"
|
||
|
|
peer_kind = PeerKind.GROUP if is_group else PeerKind.DIRECT
|
||
|
|
|
||
|
|
sender = PeerInfo(
|
||
|
|
kind=peer_kind,
|
||
|
|
id=inbound.from_bare,
|
||
|
|
display_name=inbound.muc_nick or inbound.from_bare,
|
||
|
|
)
|
||
|
|
|
||
|
|
group = None
|
||
|
|
if is_group and inbound.room_jid:
|
||
|
|
group = GroupContext(id=inbound.room_jid, name=inbound.room_jid)
|
||
|
|
|
||
|
|
target_id = inbound.room_jid if is_group else inbound.from_bare
|
||
|
|
|
||
|
|
return UnifiedMessage(
|
||
|
|
msg_id=inbound.stanza_id or inbound.msg_id,
|
||
|
|
channel_type="xmpp",
|
||
|
|
account_id=account.account_id,
|
||
|
|
content=inbound.body,
|
||
|
|
sender=sender,
|
||
|
|
message_type=MessageType.TEXT,
|
||
|
|
group=group,
|
||
|
|
timestamp=inbound.timestamp,
|
||
|
|
raw_payload=inbound.raw,
|
||
|
|
metadata={
|
||
|
|
"room_jid": inbound.room_jid,
|
||
|
|
"muc_nick": inbound.muc_nick,
|
||
|
|
"thread_id": inbound.thread_id,
|
||
|
|
"was_mentioned": inbound.mentions_bot,
|
||
|
|
"target_id": target_id,
|
||
|
|
},
|
||
|
|
conversation_label=inbound.muc_nick if is_group else inbound.from_bare,
|
||
|
|
surface="xmpp",
|
||
|
|
originating_channel="xmpp",
|
||
|
|
)
|