# Copyright (c) 2026 Krzysztof Rudnicki. MIT licence.
"""Command wiring, the daily report, and the listen loop."""

from __future__ import annotations

import datetime as dt
from typing import TYPE_CHECKING

import httpx

from signal_bot.bot import Bot, build_bot, now_local
from signal_bot.liveness import Liveness
from signal_bot.stats.events import MessageEvent
from signal_bot.stats.periods import WARSAW
from tests.conftest import (
    WATCHED_GROUP,
    envelope,
    mock_client,
)

if TYPE_CHECKING:
    from signal_bot.config import Config
    from signal_bot.discovery import GroupDirectory
    from signal_bot.store import Store
    from signal_bot.transport_fake import FakeTransport


# The report covers the previous full day, so the tests freeze "now" and
# write their events into the day before it.
NOW = dt.datetime(2026, 9, 2, 21, 37, tzinfo=WARSAW)
YESTERDAY = "2026-09-01"


def _bot(config: Config, store: Store, transport: FakeTransport, directory: GroupDirectory) -> Bot:
    client = mock_client(lambda r: httpx.Response(200))
    return build_bot(
        config, client, store, transport, directory, Liveness(config, client), clock=lambda: NOW
    )


def wrote(
    store: Store, group_id: str, uuid: str, name: str, times: int, *, day: str = YESTERDAY
) -> None:
    """Put ``times`` messages from one person into one group on one day."""
    for i in range(times):
        store.events.record(
            MessageEvent(
                group_id=group_id,
                uuid=uuid,
                display_name=name,
                sent_at_ms=i + 1,
                day=day,
                local_hour=12,
                char_len=5,
            )
        )
    store.events.note_person(group_id, uuid, name, "", day)


def test_now_local_is_timezone_aware() -> None:
    assert now_local().tzinfo is not None


def test_build_bot_registers_the_animal_commands(
    config: Config, store: Store, transport: FakeTransport, directory: GroupDirectory
) -> None:
    bot = _bot(config, store, transport, directory)
    assert bot.dispatcher.registry.resolve("!cat") is not None
    assert bot.dispatcher.registry.resolve("!pies") is not None


async def test_listen_dispatches_every_frame(
    config: Config, store: Store, transport: FakeTransport, directory: GroupDirectory
) -> None:
    transport.queue([envelope(text="hello", group_id=WATCHED_GROUP)])
    await _bot(config, store, transport, directory).listen()
    row = store.events.connection.execute(
        "SELECT COUNT(*) AS n FROM message_events WHERE group_id = ?", (WATCHED_GROUP,)
    ).fetchone()
    assert row["n"] == 1