from __future__ import annotations

from enum import Enum
from typing import Any

from yuxi.utils.logging_config import logger


class IntegrationStatus(str, Enum):
    NOT_CONFIGURED = "not_configured"
    CONFIGURED = "configured"
    CONNECTING = "connecting"
    CONNECTED = "connected"
    ERROR = "error"
    DISCONNECTED = "disconnected"


class WeChatIntegrationStatus:
    @staticmethod
    def resolve_status(config: dict[str, Any], connection_status: str) -> IntegrationStatus:
        has_config = any(config.get(k) for k in ("corp_id", "app_id", "bridge_url"))
        if not has_config:
            return IntegrationStatus.NOT_CONFIGURED
        status_map = {
            "connected": IntegrationStatus.CONNECTED,
            "connecting": IntegrationStatus.CONNECTING,
            "disconnected": IntegrationStatus.DISCONNECTED,
            "error": IntegrationStatus.ERROR,
        }
        return status_map.get(connection_status, IntegrationStatus.CONFIGURED)

    @staticmethod
    def get_status_summary(status: IntegrationStatus) -> dict[str, Any]:
        return {
            "status": status.value,
            "is_operational": status in (IntegrationStatus.CONNECTED,),
            "is_configured": status != IntegrationStatus.NOT_CONFIGURED,
        }

    @staticmethod
    def log_self_id(
        config: dict[str, Any],
        mode: str = "wecom",
        user_id: str = "",
        group_ids: list[str] | None = None,
    ) -> dict[str, Any]:
        identity_info: dict[str, Any] = {
            "mode": mode,
            "user_id": user_id or config.get("agent_id", config.get("app_id", "")),
            "group_ids": group_ids or [],
        }

        if mode == "wecom":
            identity_info.update(
                {
                    "corp_id": config.get("corp_id", "")[:8] + "***",
                    "agent_id": config.get("agent_id", ""),
                }
            )
        elif mode == "mp":
            identity_info.update(
                {
                    "app_id": config.get("app_id", "")[:8] + "***",
                }
            )
        elif mode == "personal":
            identity_info.update(
                {
                    "bridge_url": config.get("bridge_url", ""),
                }
            )

        logger.info(f"[WeChat/Status] Self identity: {identity_info}")
        return identity_info