use alloc::borrow::ToOwned;
use alloc::string::String;
use crate::syntax::builders::import_statement::{build_import_node, ImportElement};
use crate::syntax::fields::go::import_spec;
use crate::syntax::readers::go::common::package_name;
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 path_id = args.required_field_alt(n_id, import_spec::PATH)?;
let path = node_to_str(graph, path_id);
let expression = path.trim_matches(['"', '`']).to_owned();
let name: Option<String> = args
.optional_field_alt(n_id, import_spec::NAME)
.map(|name_id| node_to_str(graph, name_id))
.filter(|name| !name.is_empty());
let (alias, bound_name) = match name {
Some(name) if name == "_" || name == "." => (None, String::new()),
Some(name) => (Some(name.clone()), name),
None => (None, String::from(package_name(&expression))),
};
build_import_node(
args,
n_id,
&ImportElement {
expression: Some(expression),
alias,
bound_name: Some(bound_name),
..ImportElement::default()
},
)
.map(Some)
}