from typing import Any, Callable, Literal
from buildbot.reporters.message import MessageFormatterBase
import dataclasses

@dataclasses.dataclass
class CallbackPayloadBuild:
    # buddy i have no idea what the fuck is in this
    build: dict[str, Any]

@dataclasses.dataclass
class CallbackPayloadBuildSet:
    buildset: dict[str, Any]
    # i have no idea what the fuck is in this honestly
    builds: Any

@dataclasses.dataclass
class CallbackReturn:
    body: str | None = None
    subject: str | None = None
    type: Literal['plain'] | Literal['html'] | Literal['json'] = 'plain'
    extra_info: dict[str, Any] | None = None
    # FIXME: support other template types, if they actually become necessary
    template_type: Literal['plain'] = 'plain'

class ReasonableMessageFormatter(MessageFormatterBase):
    """
    Message formatter which uses strongly typed data classes to reduce suffering slightly.
    """
    CallbackFunc = Callable[[CallbackPayloadBuild | CallbackPayloadBuildSet], CallbackReturn]
    def __init__(self, function: CallbackFunc, template_type: str, **kwargs):
        super().__init__(**kwargs)
        self.template_type = template_type
        self._function = function

    def format_message_for_build(self, master, build, **kwargs):
        return dataclasses.asdict(self._function(CallbackPayloadBuild(build=build)))

    def format_message_for_buildset(self, master, buildset, builds, **kwargs):
        return dataclasses.asdict(self._function(CallbackPayloadBuildSet(buildset=buildset, builds=builds)))

    # These only exist as callbacks, the only one actually used is render_message_dict
    def render_message_body(self, context):
        return None

    def render_message_subject(self, context):
        return None