fluidattacks-blends-domain 0.2.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
28
29
30
31
32
33
34
35
36
37
38

use alloc::string::String;

use crate::query::adj_ast;
use crate::syntax::builders::binary_operation::build_binary_operation_node;
use crate::syntax::builders::string_literal::build_string_literal_node;
use crate::syntax::fields::python::{binary_operator, comparison_operator};
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 operator_id = args
        .required_field_alt(n_id, comparison_operator::OPERATORS)
        .ok()
        .or_else(|| {
            args.required_field_alt(n_id, binary_operator::OPERATOR)
                .ok()
        });
    let operator = operator_id.map_or_else(String::new, |id| node_to_str(graph, id));

    let childs = adj_ast(graph, n_id, None, &[]);
    let all_strings = childs
        .iter()
        .filter(|id| Some(**id) != operator_id)
        .all(|id| graph.label_type(*id) == Some("string"));
    if all_strings {
        let text = node_to_str(graph, n_id);
        return build_string_literal_node(args, n_id, text, &[]).map(Some);
    }

    let left = childs.first().copied();
    let right = childs.last().copied();
    build_binary_operation_node(args, n_id, operator, left, right).map(Some)
}