use crate::syntax::builders::if_statement::build_if_node;
use crate::syntax::fields::python::{elif_clause, if_statement};
use crate::syntax::{SyntaxGraphArgs, SyntaxGraphError};
use crate::NodeId;
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let condition_id = args
.required_field_alt(n_id, if_statement::CONDITION)
.or_else(|_| args.required_field_alt(n_id, elif_clause::CONDITION))?;
let true_id = args
.required_field_alt(n_id, if_statement::CONSEQUENCE)
.or_else(|_| args.required_field_alt(n_id, elif_clause::CONSEQUENCE))?;
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)
}