import re
from typing import Iterable, Optional

AUTO = "auto"
BASIC = "basic"
ADVANCED = "advanced"
VERSIONS = (AUTO, BASIC, ADVANCED)

CHOICES = [
    ("Auto: the model decides whether to use the Basic or Advanced pages", AUTO),
    ("Basic: the 5 reference pages only", BASIC),
    ("Advanced: also consult the Advanced pages", ADVANCED),
]

MARKER = re.compile(r"\[\[asp:reference-version=(basic|advanced)\]\]")

DIRECTIVES = {
    BASIC: (
        "[[asp:reference-version=basic]]\n"
        "Reference version selected by the user: BASIC. Use only the 5 reference images already "
        "attached to this conversation. The image viewer tools are disabled for this run."
    ),
    ADVANCED: (
        "[[asp:reference-version=advanced]]\n"
        "Reference version selected by the user: ADVANCED. In addition to the 5 attached reference "
        "images, you MUST consult the Advanced pages before writing your profile: call list_sections, "
        "then view_section for the sections relevant to this case, and use what you see in your analysis."
    ),
}


def normalize(version) -> str:
    return version if version in VERSIONS else AUTO


def label(version) -> str:
    names = {value: text.split(" - ")[0] for text, value in CHOICES}
    return names[normalize(version)]


def directive(version) -> Optional[str]:
    return DIRECTIVES.get(normalize(version))


def texts_of(content) -> Iterable[str]:
    if isinstance(content, str):
        yield content
    elif isinstance(content, (list, tuple)):
        for item in content:
            if isinstance(item, str):
                yield item


def version_from_messages(messages) -> str:
    for message in reversed(list(messages)):
        for part in getattr(message, "parts", ()):
            if getattr(part, "part_kind", None) != "user-prompt":
                continue
            for text in texts_of(getattr(part, "content", None)):
                found = MARKER.search(text)
                if found:
                    return found.group(1)
    return AUTO