use crate::query::{adj_ast, match_ast_d};
use crate::syntax::builders::assignment::build_assignment_node;
use crate::syntax::fields::python::generator_expression;
use crate::syntax::{SyntaxGraphArgs, SyntaxGraphError};
use crate::NodeId;
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let variable_id = args.required_field_alt(n_id, generator_expression::BODY)?;
let graph = args.ast_graph;
let value_id = match_ast_d(graph, n_id, "for_in_clause", None)
.or_else(|| match_ast_d(graph, n_id, "if_clause", None))
.or_else(|| adj_ast(graph, n_id, None, &[]).last().copied())
.ok_or(SyntaxGraphError::UnexpectedAstShape)?;
build_assignment_node(args, n_id, variable_id, Some(value_id), None).map(Some)
}