ForcePilot/backend/package/yuxi/channels/adapters/dingding/stream_handler.py

from __future__ import annotations

import logging
from typing import TYPE_CHECKING

from yuxi.channels.models import EventType

logger = logging.getLogger(__name__)

if TYPE_CHECKING:
    from yuxi.channels.adapters.dingding.adapter import DingDingChannelAdapter

HAS_DINGTALK_STREAM = False
try:
    import dingtalk_stream

    HAS_DINGTALK_STREAM = True
except ImportError:
    dingtalk_stream = None


class DingDingChatbotHandler:
    def __init__(self, adapter: DingDingChannelAdapter):
        self._adapter = adapter

    async def process(self, callback: object) -> tuple[int, str]:
        if not HAS_DINGTALK_STREAM:
            return 200, "OK"

        from dingtalk_stream import AckMessage

        try:
            raw_data = getattr(callback, "data", {})
            session_webhook = getattr(callback, "sessionWebhook", None) or raw_data.get("sessionWebhook", "")

            if session_webhook:
                self._adapter._session_webhooks[raw_data.get("conversationId", "")] = session_webhook

            msg = self._adapter.normalize_inbound(raw_data)
            await self._adapter._handle_message(msg)
            return AckMessage.STATUS_OK, "OK"
        except Exception as e:
            logger.error(f"[DingDing] ChatbotHandler error: {e}")
            return AckMessage.STATUS_INTERNAL_SERVER_ERROR, str(e)


class DingDingCallbackHandler:
    def __init__(self, adapter: DingDingChannelAdapter):
        self._adapter = adapter

    async def process(self, callback: object) -> tuple[int, str]:
        if not HAS_DINGTALK_STREAM:
            return 200, "OK"

        from dingtalk_stream import AckMessage

        try:
            raw_data = getattr(callback, "data", {})
            msg = self._adapter.normalize_inbound(raw_data)
            msg.event_type = EventType.CARD_ACTION
            await self._adapter._handle_message(msg)
            return AckMessage.STATUS_OK, "OK"
        except Exception as e:
            logger.error(f"[DingDing] CallbackHandler error: {e}")
            return AckMessage.STATUS_INTERNAL_SERVER_ERROR, str(e)


class DingDingEventHandler:
    def __init__(self, adapter: DingDingChannelAdapter):
        self._adapter = adapter

    async def process(self, callback: object) -> tuple[int, str]:
        if not HAS_DINGTALK_STREAM:
            return 200, "OK"

        from dingtalk_stream import AckMessage

        try:
            raw_data = getattr(callback, "data", {})
            event_type_raw = raw_data.get("eventType", raw_data.get("event_type", ""))

            event_type_map = {
                "chat_update_title": EventType.CHANNEL_UPDATED,
                "chat_update_owner": EventType.CHANNEL_UPDATED,
                "chat_update_icon": EventType.CHANNEL_UPDATED,
                "chat_update_extension": EventType.CHANNEL_UPDATED,
                "chat_member_add": EventType.MEMBER_JOINED,
                "chat_member_remove": EventType.MEMBER_LEFT,
                "chat_member_admin_add": EventType.CHANNEL_UPDATED,
                "chat_member_admin_remove": EventType.CHANNEL_UPDATED,
                "chat_add": EventType.CHANNEL_CREATED,
                "chat_disband": EventType.CHANNEL_DELETED,
                "org_dept_create": EventType.SYSTEM_EVENT,
                "org_dept_modify": EventType.SYSTEM_EVENT,
                "org_dept_remove": EventType.SYSTEM_EVENT,
                "org_user_add": EventType.SYSTEM_EVENT,
                "org_user_modify": EventType.SYSTEM_EVENT,
                "org_user_leave": EventType.SYSTEM_EVENT,
                "bot_added_to_group": EventType.BOT_ADDED,
                "bot_removed_from_group": EventType.BOT_REMOVED,
                "check_url": EventType.SYSTEM_EVENT,
            }

            event_type = event_type_map.get(event_type_raw, EventType.SYSTEM_EVENT)
            msg = self._adapter.normalize_inbound(raw_data)
            msg.event_type = event_type

            await self._adapter._handle_message(msg)
            return AckMessage.STATUS_OK, "OK"
        except Exception as e:
            logger.error(f"[DingDing] EventHandler error: {e}")
            return AckMessage.STATUS_INTERNAL_SERVER_ERROR, str(e)