use crate::syntax::fields::php::foreach_statement;
use crate::{
query::adj_ast,
syntax::{
builders::for_each_statement::build_for_each_statement_node, SyntaxGraphArgs,
SyntaxGraphError,
},
NodeId,
};
const MIN_CHILDREN_FOR_FOREACH_WITH_ITERATOR: usize = 5;
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let graph = args.ast_graph;
let c_ids = adj_ast(graph, n_id, Some(1), &[]);
let (iter_item, var_node) = if c_ids.len() >= MIN_CHILDREN_FOR_FOREACH_WITH_ITERATOR {
(
c_ids
.get(2)
.copied()
.ok_or(SyntaxGraphError::UnexpectedAstShape)?,
c_ids
.get(4)
.copied()
.ok_or(SyntaxGraphError::UnexpectedAstShape)?,
)
} else {
(
c_ids
.first()
.copied()
.ok_or(SyntaxGraphError::UnexpectedAstShape)?,
c_ids
.last()
.copied()
.ok_or(SyntaxGraphError::UnexpectedAstShape)?,
)
};
let body_id = args.optional_field_alt(n_id, foreach_statement::BODY);
build_for_each_statement_node(args, n_id, var_node, iter_item, body_id).map(Some)
}