use alloc::string::String;
use crate::query::match_ast_d;
use crate::syntax::builders::parameter::build_parameter_node;
use crate::syntax::fields::python::{default_parameter, typed_default_parameter, typed_parameter};
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 (type_id, name_id, value_id) = match graph.label_type(n_id) {
Some("typed_parameter") => (
args.required_field_alt(n_id, typed_parameter::TYPE).ok(),
None,
None,
),
Some("default_parameter") => (
None,
args.required_field_alt(n_id, default_parameter::NAME).ok(),
args.required_field_alt(n_id, default_parameter::VALUE).ok(),
),
Some("typed_default_parameter") => (
args.required_field_alt(n_id, typed_default_parameter::TYPE)
.ok(),
args.required_field_alt(n_id, typed_default_parameter::NAME)
.ok(),
args.required_field_alt(n_id, typed_default_parameter::VALUE)
.ok(),
),
_ => (None, None, None),
};
let var_type = type_id.map(|type_id| node_to_str(graph, type_id));
let identifier_id = name_id.or_else(|| match_ast_d(graph, n_id, "identifier", Some(1)));
let var_name =
identifier_id.map_or_else(|| String::from("UnnamedVar"), |id| node_to_str(graph, id));
build_parameter_node(
args,
n_id,
Some(var_name),
var_type,
value_id,
&[],
None,
None,
)
.map(Some)
}