from dataclasses import dataclass

from mypy.nodes import CallExpr, RefExpr

from refurb.checks.common import normalize_os_path
from refurb.checks.pathlib.util import is_pathlike
from refurb.error import Error


@dataclass
class ErrorInfo(Error):
    """
    When checking whether a file exists or not, try and use the more modern
    `pathlib` module instead of `os.path`.

    Bad:

    ```
    import os

    if os.path.exists("filename"):
        pass
    ```

    Good:

    ```
    from pathlib import Path

    if Path("filename").exists():
        pass
    ```
    """

    name = "use-pathlib-exists"
    code = 141
    categories = ("pathlib",)


def check(node: CallExpr, errors: list[Error]) -> None:
    match node:
        case CallExpr(
            callee=RefExpr() as expr,
            args=[arg],
        ) if normalize_os_path(expr.fullname or "") == "os.path.exists":
            replace = "x.exists()" if is_pathlike(arg) else "Path(x).exists()"

            errors.append(
                ErrorInfo.from_node(node, f"Replace `os.path.exists(x)` with `{replace}`")
            )