26 lines
1.1 KiB
Python
26 lines
1.1 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from yuxi.channel.extensions.imessage.targets import build_session_key
|
||
|
|
from yuxi.channel.message.models import PeerKind
|
||
|
|
from yuxi.channel.protocols import SessionResolution
|
||
|
|
|
||
|
|
|
||
|
|
class IMessageSessionAdapter:
|
||
|
|
def resolve_session(self, msg: object) -> SessionResolution:
|
||
|
|
if hasattr(msg, "sender") and hasattr(msg.sender, "kind"):
|
||
|
|
if msg.sender.kind == PeerKind.DIRECT:
|
||
|
|
sender_id = msg.sender.id if hasattr(msg.sender, "id") else "unknown"
|
||
|
|
return SessionResolution(kind="direct", conversation_id=sender_id)
|
||
|
|
|
||
|
|
gid = "unknown"
|
||
|
|
if hasattr(msg, "group") and msg.group:
|
||
|
|
gid = msg.group.id or "unknown"
|
||
|
|
|
||
|
|
return SessionResolution(kind="group", conversation_id=gid)
|
||
|
|
|
||
|
|
def build_dm_session_key(self, account_id: str, handle: str) -> str:
|
||
|
|
return build_session_key("imessage", account_id, "direct", handle)
|
||
|
|
|
||
|
|
def build_group_session_key(self, account_id: str, chat_guid: str) -> str:
|
||
|
|
return build_session_key("imessage", account_id, "group", chat_guid)
|