use crate::query::{adj_ast, match_ast};
use crate::syntax::builders::await_expression::build_await_expression_node;
use crate::syntax::{SyntaxGraphArgs, SyntaxGraphError};
use crate::NodeId;
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let graph = args.ast_graph;
let expr_id = if let Some(id) = match_ast(graph, n_id, &["await"], None)
.get("__0__")
.copied()
.flatten()
{
id
} else {
let c_ids = adj_ast(graph, n_id, None, &[]);
*c_ids.last().ok_or(SyntaxGraphError::UnexpectedAstShape)?
};
build_await_expression_node(args, n_id, expr_id).map(Some)
}