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::ast::AstGraph;
use crate::query::adj_ast;
use crate::syntax::builders::assignment::build_assignment_node;
use crate::syntax::fields::go::assignment_statement;
use crate::syntax::{SyntaxGraphArgs, SyntaxGraphError};
use crate::utilities::text_nodes::node_to_str;
use crate::{CodeGraph, NodeId};

fn unwrap_single_expression_list(graph: &AstGraph, id: NodeId) -> NodeId {
    if graph.label_type(id) == Some("expression_list") {
        if let [only] = *adj_ast(graph, id, None, &[]).as_slice() {
            return only;
        }
    }
    id
}

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

    let variable_id = unwrap_single_expression_list(
        graph,
        args.required_field_alt(n_id, assignment_statement::LEFT)?,
    );
    let value_id = unwrap_single_expression_list(
        graph,
        args.required_field_alt(n_id, assignment_statement::RIGHT)?,
    );

    let operator = if graph.label_type(n_id) == Some("assignment_statement") {
        Some(node_to_str(
            graph,
            args.required_field_alt(n_id, assignment_statement::OPERATOR)?,
        ))
    } else {
        None
    };

    build_assignment_node(args, n_id, variable_id, Some(value_id), operator).map(Some)
}