from __future__ import annotations

from yuxi.channels.models import ChannelResponse, MessageType


def format_outbound(response: ChannelResponse) -> dict:
    chat_id = response.identity.channel_chat_id
    if not chat_id:
        raise ValueError(
            "channel_chat_id is empty, cannot format outbound message. "
            "Ensure the ChannelIdentity has a valid channel_chat_id."
        )
    metadata = response.metadata or {}

    payload: dict = {
        "msg_type": _map_message_type(response.message_type, response.content),
    }

    if metadata.get("group_open_id"):
        payload["group_open_id"] = metadata["group_open_id"]
    elif metadata.get("channel_id"):
        payload["channel_id"] = metadata["channel_id"]
    else:
        payload["open_id"] = chat_id

    if response.reply_to_message_id:
        payload["reply_to_msg_id"] = response.reply_to_message_id

    if response.message_type == MessageType.IMAGE or response.message_type == MessageType.STICKER:
        if response.attachments:
            payload["media_url"] = response.attachments[0].url
            payload["content"] = response.content or ""
        else:
            payload["content"] = response.content
    elif response.message_type == MessageType.FILE:
        if response.attachments:
            payload["media_url"] = response.attachments[0].url
            payload["filename"] = response.attachments[0].filename or "file"
        payload["content"] = response.content or ""
    else:
        payload["content"] = response.content

    if metadata.get("extra"):
        payload["extra"] = metadata["extra"]

    buttons = metadata.get("buttons")
    if buttons and isinstance(buttons, list):
        payload["buttons"] = buttons

    card = metadata.get("card")
    if card and isinstance(card, dict):
        payload["card"] = card

    return payload


def _map_message_type(message_type: MessageType, content: str) -> str:
    _type_map = {
        MessageType.TEXT: "text",
        MessageType.IMAGE: "image",
        MessageType.FILE: "file",
        MessageType.AUDIO: "audio",
        MessageType.VIDEO: "video",
        MessageType.STICKER: "sticker",
        MessageType.CARD: "card",
        MessageType.COMMAND: "text",
    }
    return _type_map.get(message_type, "text")