fluidattacks-blends-domain 0.3.0

Blends functional core: pure AST graph to syntax graph (no_std)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

use crate::query::adj_ast;
use crate::syntax::builders::member_access::build_member_access_node;
use crate::syntax::fields::python::attribute;
use crate::syntax::{SyntaxGraphArgs, SyntaxGraphError};
use crate::utilities::text_nodes::node_to_str;
use crate::{CodeGraph, NodeId};

pub fn reader(
    args: &mut SyntaxGraphArgs<'_>,
    n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
    let graph = args.ast_graph;
    let attr_id = args.required_field_alt(n_id, attribute::ATTRIBUTE)?;
    let mut obj_id = args.required_field_alt(n_id, attribute::OBJECT)?;

    // Unwrap a parenthesized receiver `(expr).attr` down to `expr`.
    if graph.label_type(obj_id) == Some("parenthesized_expression") {
        let childs = adj_ast(graph, obj_id, Some(1), &[]);
        if childs.len() == 3 {
            obj_id = childs.get(1).copied().unwrap_or(obj_id);
        }
    }

    let member = node_to_str(graph, attr_id);
    let expression = node_to_str(graph, obj_id);
    build_member_access_node(args, n_id, member, expression, obj_id).map(Some)
}