use crate::{
query::match_ast_group_d,
syntax::{
builders::method_declaration::{
build_method_declaration_node, MethodDirectChildren, MethodListChildren,
},
fields::{swift::function_declaration, Field},
SyntaxGraphArgs, SyntaxGraphError,
},
utilities::text_nodes::node_to_str,
NodeId,
};
const BODY: Field<false> = Field::new("body_id");
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let name_id = args.required_field_alt(n_id, function_declaration::NAME)?;
let name = node_to_str(args.ast_graph, name_id);
let block_id = args.optional_field_alt(n_id, BODY);
let parameters_ids = match_ast_group_d(args.ast_graph, n_id, "parameter", Some(1));
let direct_children = MethodDirectChildren {
block_id,
..MethodDirectChildren::default()
};
let list_children = MethodListChildren {
parameters_ids,
..MethodListChildren::default()
};
build_method_declaration_node(
args,
n_id,
Some(name),
&direct_children,
&list_children,
None,
)
.map(Some)
}