use crate::query::{adj_ast, match_ast, match_ast_d};
use crate::syntax::builders::array_node::build_array_node;
use crate::syntax::fields::java::array_creation_expression;
use crate::syntax::{SyntaxGraphArgs, SyntaxGraphError};
use crate::NodeId;
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let graph = args.ast_graph;
let mut array_value = match args.optional_field_alt(n_id, array_creation_expression::VALUE) {
Some(value_id) => value_id,
None => adj_ast(graph, n_id, None, &[])
.last()
.copied()
.ok_or(SyntaxGraphError::UnexpectedAstShape)?,
};
if let Some(dimensions_id) = match_ast_d(graph, n_id, "dimensions_expr", None) {
if let Some(inner) = match_ast(graph, dimensions_id, &["[", "]"], None)
.get("__0__")
.copied()
.flatten()
{
array_value = inner;
}
}
build_array_node(args, n_id, &[array_value]).map(Some)
}