use alloc::vec::Vec;
use crate::query::adj_ast;
use crate::syntax::builders::array_node::build_array_node;
use crate::syntax::{SyntaxGraphArgs, SyntaxGraphError};
use crate::{CodeGraph, NodeId};
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let invalid_types = ["[", "]", "{", "}", ",", ";"];
let graph = args.ast_graph;
let valid_seqs: Vec<NodeId> = adj_ast(graph, n_id, None, &[])
.into_iter()
.filter(|c_id| {
!graph
.label_type(*c_id)
.is_some_and(|kind| invalid_types.contains(&kind))
})
.collect();
let mut usable_ids: Vec<NodeId> = Vec::new();
for item_id in valid_seqs {
let childs = adj_ast(graph, item_id, None, &[]);
let Some(current_id) = childs.get(1).copied() else {
continue;
};
if graph.label_type(current_id) != Some("block_node") {
usable_ids.push(current_id);
continue;
}
let current_id_childs = adj_ast(graph, current_id, None, &[]);
let Some(first_id) = current_id_childs.first().copied() else {
continue;
};
let chosen = if graph.label_type(first_id) == Some("tag") {
current_id_childs.last().copied().unwrap_or(first_id)
} else {
first_id
};
usable_ids.push(chosen);
}
build_array_node(args, n_id, &usable_ids).map(Some)
}