Method _extract_classes

code_review_graph/parser.py:4243–4350

Extract a class definition node and its inheritance edges. Returns True if the child was handled (class with a name found).

(
        self,
        child,
        source: bytes,
        language: str,
        file_path: str,
        nodes: list[NodeInfo],
        edges: list[EdgeInfo],
        enclosing_class: Optional[str],
        import_map: Optional[dict[str, str]],
        defined_names: Optional[set[str]],
        _depth: int,
    )

                    ))

    def _extract_classes(
        self,
        child,
        source: bytes,
        language: str,
        file_path: str,
        nodes: list[NodeInfo],
        edges: list[EdgeInfo],
        enclosing_class: Optional[str],
        import_map: Optional[dict[str, str]],
        defined_names: Optional[set[str]],
        _depth: int,
    ) -> bool:
        """Extract a class definition node and its inheritance edges.

        Returns True if the child was handled (class with a name found).
        """
        name = self._get_name(child, language, "class")
        if not name:
            return False

        # Swift: detect the actual type keyword (class/struct/enum/actor/extension)
        # and store it in extra["swift_kind"] for richer downstream analysis.
        # Tree-sitter maps struct/enum/actor/extension all to class_declaration;
        # protocol uses its own protocol_declaration node type.
        extra: dict = {}
        if language == "swift":
            if child.type == "class_declaration":
                _swift_keywords = {"class", "struct", "enum", "actor", "extension"}
                for kw_child in child.children:
                    kw_text = kw_child.text.decode("utf-8", errors="replace")
                    if kw_text in _swift_keywords:
                        extra["swift_kind"] = kw_text
                        break
            elif child.type == "protocol_declaration":
                extra["swift_kind"] = "protocol"

        # Java: detect Spring stereotype annotations and store as metadata
        class_annotations: list[str] = []
        if language == "java":
            class_annotations = self._get_java_annotations(child)
            spring_stereotypes = [
                a for a in class_annotations if a in _SPRING_STEREOTYPE_ANNOTATIONS
            ]
            if spring_stereotypes:
                extra["spring_stereotype"] = spring_stereotypes[0]
            if class_annotations:
                extra["spring_annotations"] = class_annotations
            temporal_roles = [
                a for a in class_annotations if a in _TEMPORAL_INTERFACE_ANNOTATIONS
            ]
            if temporal_roles:
                is_wf = "WorkflowInterface" in temporal_roles
                role = "workflow_interface" if is_wf else "activity_interface"
                extra["temporal_role"] = role

        node = NodeInfo(
            kind="Class",