import logging
from pathlib import Path

log = logging.getLogger(__name__)
from dataclasses import dataclass

from markdown2 import markdown
from nio import (
    AsyncClient,
    JoinedMembersResponse,
    JoinResponse,
    ProfileSetAvatarResponse,
    RoomMember,
    RoomSendResponse,
    UploadResponse,
)


async def upload_image(client: AsyncClient, image_path: str) -> str:
    with open(image_path, "rb") as image_file:
        response: UploadResponse
        response, _ = await client.upload(image_file, content_type="image/png")
        if not response.transport_response.ok:
            raise Exception(f"Failed to upload image {response}")
        return response.content_uri  # This is the MXC URL


async def set_avatar(client: AsyncClient, mxc_url: str) -> None:
    response: ProfileSetAvatarResponse
    response = await client.set_avatar(mxc_url)
    if not response.transport_response.ok:
        raise Exception(f"Failed to set avatar {response}")


async def get_room_members(client: AsyncClient, room: JoinResponse) -> list[RoomMember]:
    users: JoinedMembersResponse = await client.joined_members(room.room_id)

    if not users.transport_response.ok:
        raise Exception(f"Failed to get users {users}")
    return users.members


async def send_message(
    client: AsyncClient,
    room: JoinResponse,
    message: str,
    user_ids: list[str] | None = None,
) -> None:
    """
    Send a message in a Matrix room, optionally mentioning users.
    """

    # If user_ids are provided, format the message to mention them
    formatted_message = markdown(message)
    if user_ids:
        mention_list = ", ".join(
            [
                f"<a href='https://matrix.to/#/{user_id}'>{user_id}</a>"
                for user_id in user_ids
            ]
        )
        formatted_message = f"{mention_list}: {formatted_message}"

    content = {
        "msgtype": "m.text" if user_ids else "m.notice",
        "format": "org.matrix.custom.html",
        "body": message,
        "formatted_body": formatted_message,
    }

    res: RoomSendResponse = await client.room_send(
        room_id=room.room_id, message_type="m.room.message", content=content
    )

    if not res.transport_response.ok:
        raise Exception(f"Failed to send message {res}")


@dataclass
class MatrixData:
    server: str
    user: str
    avatar: Path
    password: str
    changelog_room: str
    review_room: str
    changelog_frequency: int
    publish_day: str