from __future__ import annotations
from collections.abc import Iterable
from typing import Callable

from .pseudo_traceback import _build_pseudo_trace_str, _build_single_line_trace_str

should_use_color: bool = False
COLOR_RED = "\x1b[31m"
COLOR_RESET = "\x1b[0m"
_failures: list[str] = []
_stop_on_fail = False

_default_max_fail = None
_default_max_report = None
_default_max_tb: int = 1
_default_max_tb_line: int | None = None

_max_fail: int | None = _default_max_fail
_max_report: int | None = _default_max_report
_max_tb: int = _default_max_tb
_max_tb_line: int | None = _default_max_tb_line
_num_failures = 0
_fail_function: Callable[[str], None] | None = None

_showlocals: bool = False

# Track checks with xfail reasons
_xfailed_failure: str | None = None


def clear_failures() -> None:
    # gets called at the beginning of each test function
    global _failures, _num_failures
    global _max_fail, _max_report, _max_tb, _max_tb_line
    global _xfailed_failure, _fail_function
    _failures = []
    _num_failures = 0
    _max_fail = _default_max_fail
    _max_report = _default_max_report
    _max_tb = _default_max_tb
    _max_tb_line = _default_max_tb_line
    _fail_function = None
    _xfailed_failure = None


def any_failures() -> bool:
    return bool(get_failures())


def get_failures() -> list[str]:
    return _failures


def log_failure(
    msg: object = "",
    check_str: str = "",
    tb: Iterable[str] | None = None,
    xfail: str | None = None,
) -> None:
    global _num_failures
    global _xfailed_failure
    __tracebackhide__ = True
    _num_failures += 1

    msg = str(msg).strip()

    if check_str:
        msg = f"{msg}: {check_str}"

    if (_max_report is None) or (_num_failures <= _max_report):
        if _num_failures <= _max_tb:
            pseudo_trace_str = _build_pseudo_trace_str(
                _showlocals, tb, should_use_color
            )
            msg = f"{msg}\n{pseudo_trace_str}"
        elif _max_tb_line is not None and _num_failures <= _max_tb_line:
            pseudo_trace_str = _build_single_line_trace_str(tb, should_use_color)
            if pseudo_trace_str:
                msg = f"{msg}, {pseudo_trace_str}"

        if should_use_color:
            msg = f"{COLOR_RED}FAILURE: {COLOR_RESET}{msg}"
        else:
            msg = f"FAILURE: {msg}"
        _failures.append(msg)

        if xfail and _xfailed_failure is None:
            _xfailed_failure = xfail

        if _fail_function:
            _fail_function(str(msg))

    if _max_fail and (_num_failures >= _max_fail):
        assert_msg = f"pytest-check max fail of {_num_failures} reached"
        assert _num_failures < _max_fail, assert_msg

    if _stop_on_fail:
        assert False, "Stopping on first failure"


def get_xfailed_failure() -> str | None:
    """Return the xfail reason for the first check that failed with xfail."""
    return _xfailed_failure
