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

use alloc::vec::Vec;

use crate::query::match_ast_group_d;
use crate::syntax::builders::array_node::build_array_node;
use crate::syntax::fields::python::{
    dictionary_comprehension, list_comprehension, set_comprehension,
};
use crate::syntax::{SyntaxGraphArgs, SyntaxGraphError};
use crate::NodeId;

pub fn reader(
    args: &mut SyntaxGraphArgs<'_>,
    n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
    let body_id = args
        .required_field_alt(n_id, list_comprehension::BODY)
        .or_else(|_| args.required_field_alt(n_id, set_comprehension::BODY))
        .or_else(|_| args.required_field_alt(n_id, dictionary_comprehension::BODY))?;
    let graph = args.ast_graph;
    let mut c_ids: Vec<NodeId> = Vec::new();
    c_ids.push(body_id);
    c_ids.extend(match_ast_group_d(graph, n_id, "for_in_clause", None));
    c_ids.extend(match_ast_group_d(graph, n_id, "if_clause", None));
    build_array_node(args, n_id, &c_ids).map(Some)
}