use crate::{
syntax::{
builders::{class_declaration::build_class_node, declaration::build_declaration_node},
fields::{swift::class_declaration, Field},
SyntaxGraphArgs, SyntaxGraphError,
},
utilities::text_nodes::node_to_str,
NodeId,
};
use alloc::string::String;
const DECLARATION_KIND: Field<false> = Field::new("declaration_kind_id");
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let graph = args.ast_graph;
let name_id = args.required_field_alt(n_id, class_declaration::NAME)?;
let name = node_to_str(graph, name_id);
let block_id = args.required_field_alt(n_id, class_declaration::BODY)?;
let kind = args
.optional_field_alt(n_id, DECLARATION_KIND)
.map_or_else(String::new, |kind_id| node_to_str(graph, kind_id));
if matches!(kind.as_str(), "struct" | "enum") {
return build_declaration_node(args, n_id, name, Some(block_id), None).map(Some);
}
build_class_node(args, n_id, name, Some(block_id), &[], None, None, None).map(Some)
}