use crate::syntax::builders::assignment::build_assignment_node;
use crate::syntax::fields::go::assignment_statement;
use crate::syntax::readers::go::common::unwrap_single_expression_list;
use crate::syntax::{SyntaxGraphArgs, SyntaxGraphError};
use crate::utilities::text_nodes::node_to_str;
use crate::{CodeGraph, NodeId};
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let graph = args.ast_graph;
let variable_id = unwrap_single_expression_list(
graph,
args.required_field_alt(n_id, assignment_statement::LEFT)?,
);
let value_id = unwrap_single_expression_list(
graph,
args.required_field_alt(n_id, assignment_statement::RIGHT)?,
);
let operator = if graph.label_type(n_id) == Some("assignment_statement") {
Some(node_to_str(
graph,
args.required_field_alt(n_id, assignment_statement::OPERATOR)?,
))
} else {
None
};
build_assignment_node(args, n_id, variable_id, Some(value_id), operator).map(Some)
}