from __future__ import annotations

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from collections.abc import Callable

    from crawlee._types import RequestHandlerRunResult


def create_default_comparator(
    result_checker: Callable[[RequestHandlerRunResult], bool] | None,
) -> Callable[[RequestHandlerRunResult, RequestHandlerRunResult], bool]:
    """Create a default comparator function for evaluating request handler results."""
    if result_checker:
        # Fallback comparator if only user-specific checker is defined.
        return lambda result_1, result_2: result_checker(result_1) and result_checker(result_2)
    # Fallback default comparator.
    return push_data_only_comparator


def full_result_comparator(result_1: RequestHandlerRunResult, result_2: RequestHandlerRunResult) -> bool:
    """Compare results by comparing all their parts.

    Comparison of `add_requests_calls` will consider same url requests with different parameters as different
    For example following two request will be considered as different requests:
    https://sdk.apify.com/docs/guides/getting-started
    https://sdk.apify.com/docs/guides/getting-started?__hsfp=1136113150&__hssc=7591405.1.173549427712
    """
    return (
        (result_1.push_data_calls == result_2.push_data_calls)
        and (result_1.add_requests_calls == result_2.add_requests_calls)
        and (result_1.key_value_store_changes == result_2.key_value_store_changes)
    )


def push_data_only_comparator(result_1: RequestHandlerRunResult, result_2: RequestHandlerRunResult) -> bool:
    """Compare results by comparing their push data calls. Ignore other parts of results in comparison."""
    return result_1.push_data_calls == result_2.push_data_calls