use crate::{
query::adj_ast,
syntax::{builders::if_statement::build_if_node, SyntaxGraphArgs, SyntaxGraphError},
CodeGraph, NodeId,
};
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let graph = args.ast_graph;
let child_ids = adj_ast(graph, n_id, Some(1), &[]);
let mut condition_id = child_ids
.get(2)
.copied()
.ok_or(SyntaxGraphError::MissingAstNode)?;
if graph.label_type(condition_id) == Some("parenthesized_expression") {
let childs = adj_ast(graph, condition_id, Some(1), &[]);
if childs.len() > 1 {
condition_id = childs
.get(1)
.copied()
.ok_or(SyntaxGraphError::UnexpectedAstShape)?;
}
}
let true_id = child_ids
.get(4)
.copied()
.ok_or(SyntaxGraphError::UnexpectedAstShape)?;
let false_id = if child_ids.len() == 7 {
child_ids.get(6).copied()
} else {
None
};
build_if_node(args, n_id, condition_id, Some(true_id), false_id, None).map(Some)
}