fluidattacks-blends-domain 0.34.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
52
53
54
55
56
57
58
59

use alloc::borrow::ToOwned;
use alloc::format;
use alloc::string::String;

use crate::query::{label_text, match_ast_d, pred_ast};
use crate::syntax::builders::import_statement::{build_import_node, ImportElement};
use crate::syntax::fields::python::{aliased_import, import_from_statement};
use crate::syntax::{SyntaxGraphArgs, SyntaxGraphError};
use crate::utilities::text_nodes::node_to_str;
use crate::NodeId;

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

    for p_id in pred_ast(graph, n_id, Some(-1)) {
        if let Ok(alias_id) = args.required_field_alt(p_id, aliased_import::ALIAS) {
            alias = label_text(graph, alias_id).map(ToOwned::to_owned);
        }
        if let Ok(module_name_id) =
            args.required_field_alt(p_id, import_from_statement::MODULE_NAME)
        {
            let name = node_to_str(graph, module_name_id);
            expression = if match_ast_d(graph, p_id, "wildcard_import", None).is_none() {
                format!("{name}.{expression}")
            } else {
                String::from("*")
            };
            module_name = Some(name);
        }
    }

    let bound_name = (module_name.is_none() && alias.is_none()).then(|| top_package(&expression));
    build_import_node(
        args,
        n_id,
        &ImportElement {
            expression: Some(expression),
            alias,
            module_name,
            bound_name,
            ..ImportElement::default()
        },
    )
    .map(Some)
}

fn top_package(expression: &str) -> String {
    String::from(
        expression
            .split_once('.')
            .map_or(expression, |(package, _)| package),
    )
}