use crate::{
query::{label_text, match_ast_d},
syntax::{
builders::import_statement::{build_import_node, ImportElement},
SyntaxGraphArgs, SyntaxGraphError,
},
utilities::text_nodes::node_to_str,
NodeId,
};
use alloc::borrow::ToOwned;
use alloc::string::String;
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let graph = args.ast_graph;
let mut expression = match_ast_d(graph, n_id, "qualified_identifier", Some(1))
.or_else(|| match_ast_d(graph, n_id, "identifier", Some(1)))
.map_or_else(String::new, |identifier| node_to_str(graph, identifier));
let alias = match_ast_d(graph, n_id, "import_alias", Some(1))
.and_then(|alias_n_id| match_ast_d(graph, alias_n_id, "type_identifier", Some(1)))
.and_then(|type_id| label_text(graph, type_id))
.map(str::to_owned);
if match_ast_d(graph, n_id, ".*", Some(1)).is_some() {
expression.push_str(".*");
}
build_import_node(
args,
n_id,
&ImportElement {
expression: Some(expression),
alias,
..ImportElement::default()
},
)
.map(Some)
}