"""Расчёт дат для напоминалок админов: парсинг ввода и следующее срабатывание.

Не зависит от bot.py/database.py — чистая логика дат, легко тестируется отдельно.
"""
import calendar
from datetime import datetime, timedelta
from typing import Optional

DT_FORMAT = "%Y-%m-%d %H:%M:%S"   # формат хранения в БД
INPUT_FORMAT = "%d.%m.%Y %H:%M"   # формат ввода админом

RECURRENCES = ("once", "weekly", "monthly", "yearly")


def parse_reminder_datetime(text: str) -> Optional[datetime]:
    """Парсит дату/время в формате ДД.ММ.ГГГГ ЧЧ:ММ. None, если формат неверный."""
    text = text.strip()
    try:
        return datetime.strptime(text, INPUT_FORMAT)
    except ValueError:
        return None


def _clamp_day(year: int, month: int, day: int) -> int:
    """День месяца, не превышающий число дней в этом месяце
    (31.01 -> 28/29.02, 31.03 и т.п. — без ползучего дрейфа, см. compute_next_occurrence)."""
    last_day = calendar.monthrange(year, month)[1]
    return min(day, last_day)


def compute_next_occurrence(anchor: datetime, recurrence: str, after: datetime) -> datetime:
    """Следующее срабатывание строго после `after`, отталкиваясь от якорной даты `anchor`
    (а не от предыдущего срабатывания) — так периодичность не «дрейфует» из-за клампинга
    коротких месяцев. Если пропущено несколько периодов (бот был выключен) — промежуточные
    пропускаются, возвращается ближайшее будущее относительно `after`.
    """
    if recurrence not in RECURRENCES:
        raise ValueError(f"Неизвестная периодичность: {recurrence}")

    if recurrence == "weekly":
        candidate = anchor
        while candidate <= after:
            candidate += timedelta(days=7)
        return candidate

    if recurrence == "monthly":
        year, month = anchor.year, anchor.month
        candidate = anchor
        while candidate <= after:
            month += 1
            if month > 12:
                month = 1
                year += 1
            day = _clamp_day(year, month, anchor.day)
            candidate = candidate.replace(year=year, month=month, day=day)
        return candidate

    # yearly
    year = anchor.year
    candidate = anchor
    while candidate <= after:
        year += 1
        day = _clamp_day(year, anchor.month, anchor.day)
        candidate = candidate.replace(year=year, day=day)
    return candidate