use crate::query::adj_ast;
use crate::syntax::builders::if_statement::build_if_node;
use crate::syntax::fields::c_sharp::{conditional_expression, 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 (condition_id, consequence_id, alternative) = match graph.label_type(n_id) {
Some("conditional_expression") => (
args.required_field_alt(n_id, conditional_expression::CONDITION)?,
args.required_field_alt(n_id, conditional_expression::CONSEQUENCE)?,
Some(args.required_field_alt(n_id, conditional_expression::ALTERNATIVE)?),
),
Some("if_statement") => (
args.required_field_alt(n_id, if_statement::CONDITION)?,
args.required_field_alt(n_id, if_statement::CONSEQUENCE)?,
args.optional_field_alt(n_id, if_statement::ALTERNATIVE),
),
_ => return Err(SyntaxGraphError::UnexpectedAstShape),
};
let mut true_id = consequence_id;
if graph.label_type(true_id) == Some("expression_statement") {
true_id = adj_ast(graph, true_id, None, &[])
.into_iter()
.next()
.ok_or(SyntaxGraphError::UnexpectedAstShape)?;
}
let false_id = alternative.map(|alt| {
if graph.label_type(alt) == Some("expression_statement") {
adj_ast(graph, alt, None, &[])
.into_iter()
.next()
.unwrap_or(alt)
} else {
alt
}
});
build_if_node(args, n_id, condition_id, Some(true_id), false_id, None).map(Some)
}