use alloc::vec::Vec;
use crate::query::adj_ast;
use crate::syntax::builders::file::build_file_node;
use crate::syntax::{SyntaxGraphArgs, SyntaxGraphError};
use crate::{CodeGraph, NodeId};
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let graph = args.ast_graph;
let mut c_ids: Vec<NodeId> = Vec::new();
for child in adj_ast(graph, n_id, None, &[]) {
match graph.label_type(child) {
Some("empty_statement") => {}
Some("expression_statement") => {
let inner = adj_ast(graph, child, None, &[])
.into_iter()
.next()
.ok_or(SyntaxGraphError::UnexpectedAstShape)?;
c_ids.push(inner);
}
_ => c_ids.push(child),
}
}
build_file_node(args, n_id, &c_ids).map(Some)
}