use crate::syntax::fields::swift::assignment;
use crate::{
query::adj_ast,
syntax::{builders::assignment::build_assignment_node, SyntaxGraphArgs, SyntaxGraphError},
utilities::text_nodes::node_to_str,
CodeGraph, NodeId,
};
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let graph = args.ast_graph;
let mut variable_id = args.required_field_alt(n_id, assignment::TARGET)?;
if graph.label_type(variable_id) == Some("directly_assignable_expression") {
let childs = adj_ast(graph, variable_id, Some(1), &[]);
if childs.len() == 1 {
variable_id = childs
.first()
.copied()
.ok_or(SyntaxGraphError::UnexpectedAstShape)?;
}
}
let value_id = args.required_field_alt(n_id, assignment::RESULT)?;
let operator_id = args.required_field_alt(n_id, assignment::OPERATOR)?;
let operator = node_to_str(graph, operator_id);
build_assignment_node(args, n_id, variable_id, Some(value_id), Some(operator)).map(Some)
}