use crate::query::adj_ast;
use crate::syntax::builders::return_statement::build_return_node;
use crate::syntax::{SyntaxGraphArgs, SyntaxGraphError};
use crate::{CodeGraph, NodeId};
const YIELD_KEYWORDS: [&str; 2] = ["yield", "*"];
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let graph = args.ast_graph;
let c_ids = adj_ast(graph, n_id, None, &[]);
let value_id = c_ids.into_iter().find(|c_id| {
!graph
.label_type(*c_id)
.is_some_and(|kind| YIELD_KEYWORDS.contains(&kind))
});
build_return_node(args, n_id, value_id).map(Some)
}