fluidattacks-blends-domain 0.17.1

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
39
40
41
42
43

use crate::{
    query::{adj_ast, match_ast},
    syntax::{
        builders::for_statement::build_for_statement_node, SyntaxGraphArgs, SyntaxGraphError,
    },
    CodeGraph, NodeId,
};

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

    let c_ids = match_ast(
        graph,
        n_id,
        &[
            "assignment_expression",
            "binary_expression",
            "update_expression",
        ],
        Some(1),
    );

    let initializer_id = c_ids.get("assignment_expression").copied().flatten();
    let condition_id = c_ids.get("binary_expression").copied().flatten();
    let update_id = c_ids.get("update_expression").copied().flatten();

    let mut body_id = adj_ast(graph, n_id, Some(1), &[])
        .last()
        .copied()
        .ok_or(SyntaxGraphError::UnexpectedAstShape)?;

    if graph.label_type(body_id) == Some("expression_statement") {
        body_id = adj_ast(graph, body_id, Some(1), &[])
            .first()
            .copied()
            .ok_or(SyntaxGraphError::UnexpectedAstShape)?;
    }

    build_for_statement_node(args, n_id, initializer_id, condition_id, update_id, body_id).map(Some)
}