use crate::query::match_ast;
use crate::syntax::builders::method_declaration::{
build_method_declaration_node, MethodDirectChildren, MethodListChildren,
};
use crate::syntax::fields::python::function_definition;
use crate::syntax::{SyntaxGraphArgs, SyntaxGraphError};
use crate::utilities::text_nodes::node_to_str;
use crate::NodeId;
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let graph = args.ast_graph;
let name = node_to_str(
graph,
args.required_field_alt(n_id, function_definition::NAME)?,
);
let block_id = args.required_field_alt(n_id, function_definition::BODY)?;
let parameters_id = args.required_field_alt(n_id, function_definition::PARAMETERS)?;
let include_params = match_ast(graph, parameters_id, &["(", ")"], None).contains_key("__0__");
let direct = MethodDirectChildren {
parameters_id: include_params.then_some(parameters_id),
block_id: Some(block_id),
..MethodDirectChildren::default()
};
build_method_declaration_node(
args,
n_id,
Some(name),
&direct,
&MethodListChildren::default(),
None,
)
.map(Some)
}