"""Content preparation for LLM input."""

import logging

logger = logging.getLogger('Beta')


def prepare_message_content(message) -> str:
    """Prepare message content for LLM by normalizing mentions and attachments."""
    if not message:
        return ""

    content = message.content

    # Replace channel mentions <#123> -> #general
    for channel in message.channel_mentions:
        content = content.replace(f'<#{channel.id}>', f'#{channel.name}')

    # Replace role mentions <@&123> -> @admin
    for role in message.role_mentions:
        content = content.replace(f'<@&{role.id}>', f'@{role.name}')

    # Replace user mentions <@123> or <@!123> -> @User
    for mention in message.mentions:
        content = content.replace(f'<@{mention.id}>', f'@{mention.display_name}')
        content = content.replace(f'<@!{mention.id}>', f'@{mention.display_name}')

    # Handle attachments
    if message.attachments:
        for attachment in message.attachments:
            if attachment.content_type and attachment.content_type.startswith('image/'):
                label = f"[Image: {attachment.filename}]"
            else:
                label = f"[Fichier: {attachment.filename}]"
            content = f"{content} {label}".strip() if content else label

    return content.strip()