from __future__ import annotations

from datetime import datetime
from types import SimpleNamespace

from opc.plugins.office_ui.snapshot_builder import (
    _sanitize_ui_message_dict,
    _strip_trailing_verification_footer,
    _transcript_item_to_ui_message,
)


def test_strip_trailing_verification_footer_removes_only_footer() -> None:
    content = (
        "I can help with many executable tasks.\n\n"
        "Verification: not required because no code edits or risky runtime actions were detected."
    )

    body, footer = _strip_trailing_verification_footer(content)

    assert body == "I can help with many executable tasks."
    assert footer == "Verification: not required because no code edits or risky runtime actions were detected."


def test_strip_trailing_verification_footer_keeps_footer_only_message() -> None:
    content = "Verification: verified by verify."

    body, footer = _strip_trailing_verification_footer(content)

    assert body == content
    assert footer is None


def test_sanitize_ui_message_dict_strips_assistant_footer_and_preserves_metadata() -> None:
    message = {
        "message_id": "m1",
        "channel_id": "session:t1",
        "sender": "assistant",
        "sender_name": "OPC",
        "content": (
            "Completed the task.\n\n"
            "Verification: verified by verify."
        ),
        "created_at": 123.0,
        "metadata": {"source": "engine", "role": "assistant"},
    }

    sanitized = _sanitize_ui_message_dict(message)

    assert sanitized["content"] == "Completed the task."
    assert sanitized["metadata"]["verification_verdict"] == "Verification: verified by verify."
    assert sanitized["metadata"]["role"] == "assistant"


def test_transcript_item_to_ui_message_hides_runtime_internal_user_turns() -> None:
    item = {
        "message": SimpleNamespace(
            message_id="runtime-user-1",
            role="user",
            agent_id="",
            created_at=datetime.now(),
            summary_flag=False,
            metadata={"kind": "runtime_v2_user_turn"},
        ),
        "parts": [
            SimpleNamespace(
                part_type="text",
                payload={"text": "## Task\nWhat tasks can you help with?"},
            ),
        ],
    }

    assert _transcript_item_to_ui_message(item, channel_id="session:t1", task_id="t1") is None


def test_transcript_item_to_ui_message_strips_trailing_verification_footer() -> None:
    item = {
        "message": SimpleNamespace(
            message_id="reply-1",
            role="assistant",
            agent_id="",
            created_at=datetime.now(),
            summary_flag=False,
            metadata={"kind": "top_level_reply"},
        ),
        "parts": [
            SimpleNamespace(
                part_type="text",
                payload={
                    "text": (
                        "Here is the answer.\n\n"
                        "Verification: not required because no code edits or risky runtime actions were detected."
                    ),
                },
            ),
        ],
    }

    mapped = _transcript_item_to_ui_message(item, channel_id="session:t1", task_id="t1")

    assert mapped is not None
    assert mapped["content"] == "Here is the answer."
    assert mapped["metadata"]["verification_verdict"].startswith("Verification:")