use crate::query::adj_ast;
use crate::syntax::builders::for_statement::build_for_statement_node;
use crate::syntax::fields::javascript::for_statement;
use crate::syntax::{SyntaxGraphArgs, SyntaxGraphError};
use crate::{CodeGraph, NodeId};
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let graph = args.ast_graph;
let mut body_id = args.required_field_alt(n_id, for_statement::BODY)?;
if graph.label_type(body_id) == Some("expression_statement") {
body_id = *adj_ast(graph, body_id, None, &[])
.first()
.ok_or(SyntaxGraphError::UnexpectedAstShape)?;
}
if graph.label_type(body_id) == Some("parenthesized_expression") {
if let Some(&id) = adj_ast(graph, body_id, None, &[]).get(1) {
body_id = id;
}
}
let mut initializer_id = args.required_field_alt(n_id, for_statement::INITIALIZER)?;
if graph.label_type(initializer_id) == Some("expression_statement") {
initializer_id = *adj_ast(graph, initializer_id, None, &[])
.first()
.ok_or(SyntaxGraphError::UnexpectedAstShape)?;
}
let condition_id = args.required_field_alt(n_id, for_statement::CONDITION)?;
let increment_id = args.optional_field_alt(n_id, for_statement::INCREMENT);
build_for_statement_node(
args,
n_id,
Some(initializer_id),
Some(condition_id),
increment_id,
body_id,
)
.map(Some)
}