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
39
40
41
42
43
44
45
46
47
48
49
50
51

//! Cfg builder that descends into an invocation's callbacks, chained calls
//! and blocks.

use crate::query::match_ast_d;
use crate::syntax::cfg::{SyntaxCfgArgs, SyntaxCfgError};
use crate::syntax::SyntaxNode;
use crate::CodeGraph;
use crate::NodeId;

pub fn build(
    args: &mut SyntaxCfgArgs<'_>,
    n_id: NodeId,
    nxt_id: Option<NodeId>,
) -> Result<NodeId, SyntaxCfgError> {
    let Some(&SyntaxNode::MethodInvocation {
        arguments_id,
        expression_id,
        block_id,
        ..
    }) = args.graph.nodes.get(&n_id)
    else {
        return Ok(n_id);
    };

    if let Some(al_id) = arguments_id
        .filter(|al_id| match_ast_d(args.graph, *al_id, "MethodDeclaration", None).is_some())
    {
        let built = args.generic(al_id, nxt_id)?;
        args.graph.add_cfg_edge(n_id, built);
    }

    if let Some(mi_id) = expression_id
        .filter(|el_id| args.graph.label_type(*el_id) == Some("MemberAccess"))
        .and_then(|el_id| match_ast_d(args.graph, el_id, "MethodInvocation", Some(1)))
    {
        let built = args.generic(mi_id, nxt_id)?;
        args.graph.add_cfg_edge(n_id, built);
    }

    if let Some(bl_id) = block_id.or_else(|| match_ast_d(args.graph, n_id, "ExecutionBlock", None))
    {
        let built = args.generic(bl_id, nxt_id)?;
        args.graph.add_cfg_edge(n_id, built);
    }

    if let Some(nxt_id) = nxt_id {
        args.graph.add_cfg_edge(n_id, nxt_id);
    }

    Ok(n_id)
}