from __future__ import annotations

import types
from typing import Any

from pybind11_stubgen.parser.errors import InvalidIdentifierError
from pybind11_stubgen.parser.interface import IParser
from pybind11_stubgen.structs import (
    Alias,
    Attribute,
    Class,
    Docstring,
    Field,
    Function,
    Identifier,
    Import,
    Method,
    Module,
    Property,
    QualifiedName,
)


class FilterClassMembers(IParser):
    __attribute_blacklist: set[Identifier] = {
        *map(
            Identifier,
            (
                "__file__",
                "__loader__",
                "__name__",
                "__package__",
                "__spec__",
                "__path__",
                "__cached__",
                "__builtins__",
            ),
        )
    }
    __class_member_blacklist: set[Identifier] = {
        *map(
            Identifier,
            (
                "__annotations__",
                "__class__",
                "__dict__",
                "__module__",
                "__qualname__",
                "__weakref__",
            ),
        )
    }

    __method_blacklist: set[Identifier] = {*map(Identifier, ("__dir__", "__sizeof__"))}

    def handle_class_member(
        self, path: QualifiedName, class_: type, member: Any
    ) -> Docstring | Alias | Class | list[Method] | Field | Property | None:
        name = path[-1]
        if name in self.__class_member_blacklist:
            return None
        if not hasattr(class_, "__dict__") or name not in class_.__dict__:
            # Skip members inherited from base classes
            return None
        return super().handle_class_member(path, class_, member)

    def handle_method(self, path: QualifiedName, value: Any) -> list[Method]:
        if path[-1] in self.__method_blacklist:
            return []
        return super().handle_method(path, value)

    def handle_attribute(self, path: QualifiedName, attr: Any) -> Attribute | None:
        if path[-1] in self.__attribute_blacklist:
            return None
        return super().handle_attribute(path, attr)


class FilterPybindInternals(IParser):
    __attribute_blacklist: set[Identifier] = {*map(Identifier, ("__entries",))}

    __class_blacklist: set[Identifier] = {*map(Identifier, ("pybind11_type",))}

    def handle_attribute(self, path: QualifiedName, value: Any) -> Attribute | None:
        if path[-1] in self.__attribute_blacklist:
            return None
        return super().handle_attribute(path, value)

    def handle_class_member(
        self, path: QualifiedName, class_: type, member: Any
    ) -> Docstring | Alias | Class | list[Method] | Field | Property | None:
        name = path[-1]
        if name in self.__class_blacklist:
            return None
        if name.startswith("__pybind11_module"):
            return None
        return super().handle_class_member(path, class_, member)


class FilterInvalidIdentifiers(IParser):
    def handle_module_member(
        self, path: QualifiedName, module: types.ModuleType, obj: Any
    ) -> (
        Docstring | Import | Alias | Class | list[Function] | Attribute | Module | None
    ):
        if not path[-1].isidentifier():
            self.report_error(InvalidIdentifierError(path[-1], path.parent))
            return None
        return super().handle_module_member(path, module, obj)

    def handle_class_member(
        self, path: QualifiedName, class_: type, obj: Any
    ) -> Docstring | Alias | Class | list[Method] | Field | Property | None:
        if not path[-1].isidentifier():
            self.report_error(InvalidIdentifierError(path[-1], path.parent))
            return None
        return super().handle_class_member(path, class_, obj)
