use crate::{
query::adj_ast,
syntax::{
builders::await_expression::build_await_expression_node, SyntaxGraphArgs, SyntaxGraphError,
},
CodeGraph, NodeId,
};
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let graph = args.ast_graph;
let child_ids = adj_ast(graph, n_id, Some(1), &[]);
let expr_id = child_ids
.iter()
.copied()
.find(|&c_id| graph.label_type(c_id) != Some("await"))
.or_else(|| child_ids.last().copied())
.ok_or(SyntaxGraphError::UnexpectedAstShape)?;
build_await_expression_node(args, n_id, expr_id).map(Some)
}