81 lines
1.9 KiB
Python
81 lines
1.9 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import logging
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class RocketChatSessionManager:
|
||
|
|
def build_session_key(
|
||
|
|
self,
|
||
|
|
account_id: str,
|
||
|
|
chat_type: str,
|
||
|
|
room_id: str,
|
||
|
|
thread_id: str | None = None,
|
||
|
|
) -> str:
|
||
|
|
parts = [account_id, chat_type, room_id]
|
||
|
|
if thread_id:
|
||
|
|
parts.append(f"thread:{thread_id}")
|
||
|
|
return ":".join(parts)
|
||
|
|
|
||
|
|
def build_dm_session_key(
|
||
|
|
self,
|
||
|
|
account_id: str,
|
||
|
|
user_id: str,
|
||
|
|
) -> str:
|
||
|
|
return f"{account_id}:direct:{user_id}"
|
||
|
|
|
||
|
|
def build_group_session_key(
|
||
|
|
self,
|
||
|
|
account_id: str,
|
||
|
|
room_id: str,
|
||
|
|
) -> str:
|
||
|
|
return f"{account_id}:group:{room_id}"
|
||
|
|
|
||
|
|
def build_channel_session_key(
|
||
|
|
self,
|
||
|
|
account_id: str,
|
||
|
|
room_id: str,
|
||
|
|
) -> str:
|
||
|
|
return f"{account_id}:channel:{room_id}"
|
||
|
|
|
||
|
|
def build_thread_session_key(
|
||
|
|
self,
|
||
|
|
account_id: str,
|
||
|
|
room_id: str,
|
||
|
|
thread_root_id: str,
|
||
|
|
) -> str:
|
||
|
|
return f"{account_id}:thread:{room_id}:{thread_root_id}"
|
||
|
|
|
||
|
|
def extract_from_session_key(self, session_key: str) -> dict:
|
||
|
|
parts = session_key.split(":")
|
||
|
|
if len(parts) < 3:
|
||
|
|
return {"account_id": "", "chat_type": "", "room_id": ""}
|
||
|
|
|
||
|
|
result = {
|
||
|
|
"account_id": parts[0],
|
||
|
|
"chat_type": parts[1],
|
||
|
|
"room_id": parts[2],
|
||
|
|
"thread_id": None,
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(parts) >= 5 and parts[3] == "thread":
|
||
|
|
result["thread_id"] = parts[4]
|
||
|
|
|
||
|
|
return result
|
||
|
|
|
||
|
|
def resolve_parent_session_key(
|
||
|
|
self,
|
||
|
|
session_key: str,
|
||
|
|
chat_type: str,
|
||
|
|
) -> str | None:
|
||
|
|
if chat_type == "direct":
|
||
|
|
return None
|
||
|
|
|
||
|
|
parts = session_key.split(":")
|
||
|
|
if len(parts) >= 5:
|
||
|
|
non_thread = ":".join(parts[:3])
|
||
|
|
return non_thread
|
||
|
|
|
||
|
|
return None
|