Method visit_data_class

src/extraction/kotlin_extractor.rs:465–523

Extract a data class.

(state: &mut ExtractionState, node: TsNode<'_>)

    /// Extract a data class.
    fn visit_data_class(state: &mut ExtractionState, node: TsNode<'_>) {
        let name = Self::extract_class_name(state, node);
        let visibility = Self::extract_visibility(node, state);
        let docstring = Self::extract_kdoc(state, node);
        let signature = Some(Self::extract_declaration_signature(state, node));
        let start_line = node.start_position().row as u32;
        let end_line = node.end_position().row as u32;
        let start_column = node.start_position().column as u32;
        let end_column = node.end_position().column as u32;
        let qualified_name = format!("{}::{}", state.qualified_prefix(), name);
        let id = generate_node_id(&state.file_path, &NodeKind::DataClass, &name, start_line);

        let graph_node = Node {
            id: id.clone(),
            kind: NodeKind::DataClass,
            name: name.clone(),
            qualified_name,
            file_path: state.file_path.clone(),
            start_line,
            attrs_start_line: start_line,
            end_line,
            start_column,
            end_column,
            signature,
            docstring,
            visibility,
            is_async: false,
            branches: 0,
            loops: 0,
            returns: 0,
            max_nesting: 0,
            unsafe_blocks: 0,
            unchecked_calls: 0,
            assertions: 0,
            updated_at: state.timestamp,
            parent_id: None,
        };
        state.nodes.push(graph_node);

        if let Some(parent_id) = state.parent_node_id() {
            state.edges.push(Edge {
                source: parent_id.to_string(),
                target: id.clone(),
                kind: EdgeKind::Contains,
                line: Some(start_line),
            });
        }

        Self::extract_annotations_from_modifiers(state, node, &id);
        Self::extract_delegation_specifiers(state, node, &id);

        state.node_stack.push((name, id));
        state.class_depth += 1;
        if let Some(body) = find_direct_child_by_kind(node, "class_body") {
            Self::visit_children(state, body);
        }
        state.class_depth -= 1;
        state.node_stack.pop();