use crate::query::match_ast;
use crate::syntax::builders::if_statement::build_if_node;
use crate::syntax::fields::javascript::if_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 condition_id = args.required_field_alt(n_id, if_statement::CONDITION)?;
if graph.label_type(condition_id) == Some("parenthesized_expression") {
if let Some(inner) = match_ast(graph, condition_id, &[], None)
.get("__1__")
.copied()
.flatten()
{
condition_id = inner;
}
}
let mut true_id = args.required_field_alt(n_id, if_statement::CONSEQUENCE)?;
if graph.label_type(true_id) == Some("expression_statement") {
if let Some(inner) = match_ast(graph, true_id, &[], None)
.get("__0__")
.copied()
.flatten()
{
true_id = inner;
}
}
let false_id = args.optional_field_alt(n_id, if_statement::ALTERNATIVE);
build_if_node(args, n_id, condition_id, Some(true_id), false_id, None).map(Some)
}