use crate::{
query::{adj_ast, match_ast_d, match_ast_group_d, pred_ast},
syntax::{
builders::if_statement::build_if_node,
fields::{php::else_clause, Field},
SyntaxGraphArgs, SyntaxGraphError,
},
CodeGraph, NodeId,
};
const CONDITION: Field<true> = Field::new("condition_id");
const BODY: Field<true> = Field::new("body_id");
const ALTERNATIVE: Field<false> = Field::new("alternative_id");
fn get_false_id(
args: &SyntaxGraphArgs<'_>,
if_n_id: NodeId,
current_n_id: NodeId,
) -> Option<NodeId> {
let graph = args.ast_graph;
for elif_n_id in match_ast_group_d(graph, if_n_id, "else_if_clause", Some(1)) {
if elif_n_id > current_n_id {
return Some(elif_n_id);
}
}
let else_n_id = match_ast_d(graph, if_n_id, "else_clause", None)?;
args.required_field_alt(else_n_id, else_clause::BODY).ok()
}
pub fn reader(
args: &mut SyntaxGraphArgs<'_>,
n_id: NodeId,
) -> Result<Option<NodeId>, SyntaxGraphError> {
let graph = args.ast_graph;
let mut condition_id = args.required_field_alt(n_id, CONDITION)?;
if graph.label_type(condition_id) == Some("parenthesized_expression") {
if let Some(&inner) = adj_ast(graph, condition_id, Some(1), &[]).get(1) {
condition_id = inner;
}
}
let true_id = args.required_field_alt(n_id, BODY)?;
let mut false_id = args.optional_field_alt(n_id, ALTERNATIVE);
if false_id.is_none() {
for anc in pred_ast(graph, n_id, Some(-1)) {
if graph.label_type(anc) == Some("if_statement")
&& adj_ast(graph, anc, Some(1), &[]).contains(&n_id)
{
false_id = get_false_id(args, anc, n_id);
}
}
}
if let Some(fid) = false_id {
if graph.label_type(fid) == Some("else_clause") {
false_id = Some(args.required_field_alt(fid, else_clause::BODY)?);
}
}
build_if_node(args, n_id, condition_id, Some(true_id), false_id, None).map(Some)
}