use crate::syntax::builders::assignment::build_assignment_node;
use crate::syntax::builders::variable_declaration::build_variable_declaration_node;
use crate::syntax::fields::python::assignment;
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 target_id = args.required_field_alt(n_id, assignment::LEFT)?;
let value_id = args.optional_field_alt(n_id, assignment::RIGHT);
let type_id = args.optional_field_alt(n_id, assignment::TYPE);
if graph.label_type(target_id) == Some("identifier") || value_id.is_none() {
let var_name = node_to_str(graph, target_id);
let var_type = type_id.map(|id| node_to_str(graph, id));
return build_variable_declaration_node(
args, n_id, var_name, var_type, value_id, None, None,
)
.map(Some);
}
let type_str = type_id.map(|id| node_to_str(graph, id));
build_assignment_node(args, n_id, target_id, value_id, type_str).map(Some)
}