from datetime import UTC, date, datetime
from pathlib import Path

from beaver_gateway.conversations.texts import UserSaid
from beaver_gateway.conversations.envelope import RecallContext

from beaver_agent.hands.calendars import Calendars, Event, Feed
from beaver_agent.memory.recall import (
    LABEL,
    Recall,
    ReplyLog,
    frontmatter_aliases,
    mentions,
    notes_head,
    routine_names,
)


def _card(path: Path, aliases: list[str] | None = None) -> None:
    path.parent.mkdir(parents=True, exist_ok=True)
    fm = "aliases:\n" + "".join(f"  - {a}\n" for a in aliases or [])
    path.write_text(
        f"---\n{fm}tags:\n  - люди/друг\n---\n# 👤 {path.stem}\n", encoding="utf-8"
    )


def _vault(tmp_path: Path, calendar: Calendars | None = None) -> Recall:
    people = tmp_path / "👤 люди"
    _card(people / "личное" / "Зина.md", ["Зина"])
    _card(people / "личное" / "Прохор.md", ["Прохор Тестов"])
    _card(
        people / "профессиональное" / "программирование" / "Фёдор Плюшкин.md",
        ["Фёдор Плюшкин", "Фёдор"],
    )
    _card(people / "личное" / "архив" / "Глафира.md")
    _card(people / "личное" / "рандомы" / "Устинья Пример.md")
    diary = tmp_path / "📅 дни"
    diary.mkdir()
    (diary / "2026-08-31.md").write_text("с [[Зина|Зиной]] гуляли\n", encoding="utf-8")
    (diary / "2026-08-26.md").write_text(
        "[[Зина]] звонила, [[Фёдор Плюшкин|Фёдор]] рассказывал\n", encoding="utf-8"
    )
    (diary / "2026-08-13.md").write_text("обед с [[Зина|Зины]]\n", encoding="utf-8")
    (diary / "2026-08-01.md").write_text("[[Зина]] ещё раз\n", encoding="utf-8")
    notes = tmp_path / "мета" / "бобер" / "наблюдения"
    notes.mkdir(parents=True)
    (notes / "Зина - наблюдения.md").write_text(
        "# Зина - наблюдения\n\n## сейчас\n- просила перезвонить вечером 26.08\n"
        "- ждёт ответа про поездку до четверга\n\n## паттерны\n"
        "- переносит встречи: [[2026-08-10]], [[2026-08-26]]\n\n## факты\n\n## открытое\n\n"
        "## лента\n- 2026-08-26 · перенесла встречу на час · [[2026-08-26]]\n"
        "- 2026-08-31 · прислала список покупок · транскрипт\n",
        encoding="utf-8",
    )
    (notes / "Прохор - наблюдения.md").write_text(
        "# Прохор - наблюдения\n\n## сейчас\n\n## лента\n- 2026-08-07 · приехал · [[2026-08-07]]\n",
        encoding="utf-8",
    )
    boards = tmp_path / "📆 доски"
    boards.mkdir()
    (boards / "организация.md").write_text(
        "## повседнев\n\n- [ ] оплатить счёт за свет 📅 2026-09-01\n- [ ] купить лампочки 📅 2026-09-02\n"
        "- [ ] далеко 📅 2026-09-20\n- [x] сделано 📅 2026-08-30\n\n***\n\n## Archive\n\n- [ ] старое 📅 2025-01-01\n",
        encoding="utf-8",
    )
    return Recall(
        vault=tmp_path,
        people_dir=people,
        notes_dir=notes,
        diary_dir=diary,
        boards_dir=boards,
        tz="Europe/Warsaw",
        calendar=calendar,
    )


def test_frontmatter_aliases_handles_empty_and_lists() -> None:
    assert frontmatter_aliases("---\naliases:\ntags:\n---\n# x") == ()
    assert frontmatter_aliases(
        "---\naliases:\n  - Фёдор Плюшкин\n  - Фёдор\ntags:\n---\n"
    ) == ("Фёдор Плюшкин", "Фёдор")
    assert frontmatter_aliases("нет фронтматтера") == ()


def test_mentions_by_case_and_link_but_archive_only_by_link(tmp_path: Path) -> None:
    recall = _vault(tmp_path)
    people = recall.people.all()
    names = [p.name for p in mentions("созвон с Фёдором, потом к Зине", people)]
    assert names == ["Фёдор Плюшкин", "Зина"]
    assert [p.name for p in mentions("Глафира опять написала", people)] == []
    assert [
        p.name for p in mentions("глянь [[Глафира]] и [[Устинья Пример]]", people)
    ] == ["Глафира", "Устинья Пример"]
    assert [p.name for p in mentions("Прохором обсуждали", people)] == ["Прохор"]
    # "Ван"+2 буквы ловит "ванна" - цена падежей; на голое имя короче 4 букв не смотрим
    assert [
        p.name for p in mentions("машина без людей, эля тут ни при чём", people)
    ] == []


def test_block_points_to_card_notes_diary_and_deadlines_once(tmp_path: Path) -> None:
    recall = _vault(tmp_path)
    now = datetime(2026, 9, 1, 12, 0, tzinfo=UTC)
    block = recall.block(RecallContext(text="что там у Зины", kind="master", now=now))
    assert block is not None
    lines = block.splitlines()
    assert lines[0] == LABEL
    assert lines[1] == (
        "👤 Зина - карточка `👤 люди/личное/Зина.md`; записки "
        "`мета/бобер/наблюдения/Зина - наблюдения.md`: лента 2 стр., паттернов 1; "
        "дневник: [[2026-08-31]], [[2026-08-26]], [[2026-08-13]]; сейчас:"
    )
    assert lines[2:4] == [
        "  - просила перезвонить вечером 26.08",
        "  - ждёт ответа про поездку до четверга",
    ]
    assert lines[4:] == [
        "📆 09-01 организация: оплатить счёт за свет",
        "📆 09-02 организация: купить лампочки",
    ]
    artem = recall.person_line(
        next(p for p in recall.people.all() if p.name == "Прохор")
    )
    assert "лента 1 стр., последняя: 2026-08-07 · приехал" in artem
    assert "сейчас:" not in artem
    again = recall.block(RecallContext(text="и ещё про Зину", kind="master", now=now))
    assert again is not None and "📆" not in again
    branch = recall.block(RecallContext(text="Фёдор звонил", kind="branch", now=now))
    assert branch is not None
    assert (
        "записок нет (`мета/бобер/наблюдения/Фёдор Плюшкин - наблюдения.md`)" in branch
    )
    assert "дневник: [[2026-08-26]]" in branch
    assert (
        recall.block(RecallContext(text="просто привет", kind="branch", now=now))
        is None
    )


def test_reply_log_appends_one_line_per_message_and_strips_seed(tmp_path: Path) -> None:
    log = ReplyLog(tmp_path / "реплики", tz="Europe/Warsaw")
    at = datetime(2026, 9, 1, 12, 4, tzinfo=UTC)
    log.write(
        UserSaid(
            conversation_id="c1",
            kind="master",
            title=None,
            text="привет\n\nты меня любишь",
            at=at,
        )
    )
    log.write(
        UserSaid(
            conversation_id="c2",
            kind="branch",
            title="крипто-карта",
            text="[сид: clean] branch, 2026-09-01.\n\nсравни Trustee и RedotPay",
            at=at,
        )
    )
    log.write(
        UserSaid(
            conversation_id="c3",
            kind="branch",
            title="x",
            text="[сид: brief] только сид",
            at=at,
        )
    )
    text = (tmp_path / "реплики" / "2026-09.md").read_text(encoding="utf-8")
    assert text.startswith("# реплики Бобра, 2026-09\n")
    assert text.endswith(
        "- 2026-09-01 14:04 · мастер · привет ⏎ ты меня любишь\n"
        "- 2026-09-01 14:04 · ветка «крипто-карта» · сравни Trustee и RedotPay\n"
    )
    assert date.fromisoformat("2026-09-01")  # tz shift stays inside the month here


def test_notes_head_picks_sections_for_the_seed(tmp_path: Path) -> None:
    recall = _vault(tmp_path)
    path = recall.notes_dir / "Зина - наблюдения.md"
    head = notes_head(path, ("сейчас", "паттерны"))
    assert head == (
        "## сейчас\n- просила перезвонить вечером 26.08\n"
        "- ждёт ответа про поездку до четверга\n"
        "## паттерны\n- переносит встречи: [[2026-08-10]], [[2026-08-26]]"
    )
    assert notes_head(path, ("факты",)) is None
    assert notes_head(recall.notes_dir / "нет.md", ("сейчас",)) is None
    short = notes_head(path, ("сейчас", "паттерны"), cap=2)
    assert short is not None and short.endswith("… ещё 3 строк в файле")


def _ics(*events: str) -> str:
    body = "".join(f"BEGIN:VEVENT\n{e}\nEND:VEVENT\n" for e in events)
    return f"BEGIN:VCALENDAR\nVERSION:2.0\n{body}END:VCALENDAR\n"


def _calendar(text: str | None) -> Calendars:
    """Фид, который отдаёт готовый ical; None - фид, который падает."""

    class Fake(Calendars):
        def fetch(self, url: str) -> str:
            if text is None:
                raise OSError("фид не ответил")
            return text

    return Fake((Feed("личный", "https://cal.example/x.ics"),), tz="Europe/Warsaw")


def _daily(title: str, clock: str) -> str:
    return f"SUMMARY:{title}\nDTSTART;TZID=Europe/Warsaw:20260902T{clock}\nRRULE:FREQ=DAILY"


SINGLES = (
    "SUMMARY:Созвон\nDTSTART:20260902T160000Z",
    "SUMMARY:Врач\nDTSTART:20260903T090000Z",
    "SUMMARY:Analiza matematyczna II - Egzamin\nDTSTART;VALUE=DATE-TIME:20260905T110000\n"
    "DESCRIPTION:Sala: 3\\nA23 - CW\\n\nLOCATION:ul. Piotrowo 2\\, Poznań",
    "SUMMARY:за окном\nDTSTART:20260925T080000Z",
)
ROUTINE = (
    _daily("Завтрак", "080000"),
    _daily("Обед", "150000"),
    _daily("Ужин", "200000"),
    _daily("Перекус", "180000"),
    _daily("Wind down", "223000"),
)
BUSY = _ics(*SINGLES, *ROUTINE)


def _section(recall: Recall, now: datetime, text: str = "привет") -> list[str]:
    block = recall.block(RecallContext(text=text, kind="master", now=now))
    assert block is not None
    return [ln for ln in block.splitlines() if "календар" in ln or ln.startswith("  -")]


def test_calendar_shows_every_single_event_and_folds_the_daily_routine(
    tmp_path: Path,
) -> None:
    recall = _vault(tmp_path, _calendar(BUSY))
    now = datetime(2026, 9, 1, 12, 0, tzinfo=UTC)
    lines = _section(recall, now)
    assert lines == [
        "🗓 календарь (7 дней), разовых 3, повседневных 31, данные на 2026-09-01 14:00:",
        "  - 09-02 18:00 Созвон",
        "  - 09-03 11:00 Врач",
        "  - 09-05 11:00 Analiza matematyczna II - Egzamin · Sala: 3, ul. Piotrowo 2, Poznań",
        "  - повседневных повторов: 31 (Завтрак, Обед, Перекус, Ужин, Wind down)",
    ]
    # секция дневная, как и сроки досок: второй конверт за те же сутки её не несёт
    again = recall.block(RecallContext(text="и ещё", kind="master", now=now))
    assert again is None or "календарь" not in again


def test_no_amount_of_routine_can_push_out_a_single_event(tmp_path: Path) -> None:
    noise = tuple(_daily(f"быт {i}", f"{i:02d}0000") for i in range(9, 24))
    recall = _vault(tmp_path, _calendar(_ics(*SINGLES, *noise)))
    now = datetime(2026, 9, 1, 12, 0, tzinfo=UTC)
    lines = _section(recall, now)
    assert lines[0].startswith("🗓 календарь (7 дней), разовых 3, повседневных 96")
    assert [ln for ln in lines if "Egzamin" in ln]
    assert (
        len([ln for ln in lines if ln.startswith("  - ")]) == 4
    )  # 3 разовых + свёртка
    assert lines[-1].startswith("  - повседневных повторов: 96")
    assert "и др." in lines[-1]


def test_routine_line_counts_and_names_up_to_five(tmp_path: Path) -> None:
    events = [
        Event(
            start=datetime(2026, 9, 2, 8, tzinfo=UTC),
            title=t,
            place="",
            all_day=False,
            routine=True,
        )
        for t in ("Завтрак", "Обед", "Завтрак", "Ужин")
    ]
    assert routine_names(events) == "Завтрак, Обед, Ужин"
    assert routine_names(events, limit=2) == "Завтрак, Обед и др."


def test_calendar_section_is_printed_even_when_the_week_is_empty(
    tmp_path: Path,
) -> None:
    recall = _vault(
        tmp_path, _calendar(_ics("SUMMARY:потом\nDTSTART:20261101T080000Z"))
    )
    now = datetime(2026, 9, 1, 12, 0, tzinfo=UTC)
    block = recall.block(RecallContext(text="просто привет", kind="master", now=now))
    assert block is not None
    assert (
        "🗓 календарь (7 дней): событий нет, данные на 2026-09-01 14:00"
        in block.splitlines()
    )


def test_calendar_failure_shows_up_in_the_envelope_instead_of_disappearing(
    tmp_path: Path,
) -> None:
    recall = _vault(tmp_path, _calendar(None))
    now = datetime(2026, 9, 1, 12, 0, tzinfo=UTC)
    block = recall.block(RecallContext(text="просто привет", kind="master", now=now))
    assert block is not None
    assert "🗓 календарь: ОШИБКА личный: OSError: фид не ответил" in block.splitlines()
    # ветке справка календарь не носит, как и сроки досок
    branch = recall.block(RecallContext(text="просто привет", kind="branch", now=now))
    assert branch is None


def test_without_configured_feeds_there_is_no_calendar_section(tmp_path: Path) -> None:
    recall = _vault(tmp_path)
    now = datetime(2026, 9, 1, 12, 0, tzinfo=UTC)
    block = recall.block(RecallContext(text="просто привет", kind="master", now=now))
    assert block is not None and "календарь" not in block