use crate::query::match_ast;
use crate::syntax::builders::method_declaration::{
build_method_declaration_node, MethodDirectChildren, MethodListChildren,
};
use crate::syntax::fields::java::{constructor_declaration, method_declaration};
use crate::syntax::metadata::java::add_method_to_metadata;
use crate::syntax::readers::java::common::extract_modifiers;
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 (name_id, block_id, parameters_id) = match args.ast_graph.label_type(n_id) {
Some("constructor_declaration") => (
args.required_field_alt(n_id, constructor_declaration::NAME)?,
Some(args.required_field_alt(n_id, constructor_declaration::BODY)?),
args.required_field_alt(n_id, constructor_declaration::PARAMETERS)?,
),
Some("method_declaration") => (
args.required_field_alt(n_id, method_declaration::NAME)?,
args.optional_field_alt(n_id, method_declaration::BODY),
args.required_field_alt(n_id, method_declaration::PARAMETERS)?,
),
_ => return Err(SyntaxGraphError::UnexpectedAstShape),
};
let name = node_to_str(args.ast_graph, name_id);
let include_params =
match_ast(args.ast_graph, parameters_id, &["(", ")"], None).contains_key("__0__");
let (modifiers_id, access_modifiers) = extract_modifiers(args.ast_graph, n_id);
if args.syntax_graph.nodes.contains_key(&NodeId(0)) {
add_method_to_metadata(args, n_id, &name)?;
}
let direct = MethodDirectChildren {
block_id,
modifiers_id,
parameters_id: include_params.then_some(parameters_id),
};
build_method_declaration_node(
args,
n_id,
Some(name),
&direct,
&MethodListChildren::default(),
access_modifiers,
)
.map(Some)
}