from __future__ import annotations

from datetime import timedelta

from cashews._typing import TTL


def ttl_to_seconds(ttl: TTL, *args, with_callable: bool = False, result=None, **kwargs) -> int | float | None:
    if ttl is None:
        return None
    _type = type(ttl)  # isinstance is slow
    if _type is str:
        return _ttl_from_str(ttl)  # type: ignore[arg-type]
    if _type is int:
        return ttl  # type: ignore[return-value]
    if _type is timedelta:
        return ttl.total_seconds()  # type: ignore[union-attr]

    if callable(ttl) and with_callable:
        try:
            ttl = ttl(*args, result=result, **kwargs)
        except TypeError:
            ttl = ttl(*args, **kwargs)  # type: ignore[operator, misc]
        return ttl_to_seconds(ttl)
    return ttl  # type: ignore[return-value]


_STR_TO_DELTA = {
    "h": timedelta(hours=1),
    "m": timedelta(minutes=1),
    "s": timedelta(seconds=1),
    "d": timedelta(days=1),
}


def _ttl_from_str(ttl: str) -> int:
    result = 0
    mul = ""
    for char in ttl.strip().lower():
        if char.isdigit():
            mul += char
        elif char in _STR_TO_DELTA:
            result += int(mul) * int(_STR_TO_DELTA[char].total_seconds())
            mul = ""
        else:
            raise ValueError(f"ttl '{ttl}' has wrong string representation")
    if mul != "" and not result:
        return int(mul)
    return result