use crate::query::adj_ast;
use crate::syntax::builders::assignment::build_assignment_node;
use crate::syntax::fields::python::as_pattern;
use crate::syntax::{SyntaxGraphArgs, SyntaxGraphError};
use crate::NodeId;
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let graph = args.ast_graph;
let children = adj_ast(graph, n_id, None, &[]);
let value_id = children
.first()
.copied()
.ok_or(SyntaxGraphError::UnexpectedAstShape)?;
let var_id = match args.optional_field_alt(n_id, as_pattern::ALIAS) {
Some(alias_id) => adj_ast(graph, alias_id, None, &[])
.first()
.copied()
.ok_or(SyntaxGraphError::UnexpectedAstShape)?,
None => children
.last()
.copied()
.ok_or(SyntaxGraphError::UnexpectedAstShape)?,
};
build_assignment_node(args, n_id, var_id, Some(value_id), None).map(Some)
}