wagon_codegen/nodes/
term.rs

1use crate::SpannableIdent;
2
3use super::{Rc, ToTokensState};
4use proc_macro2::{TokenStream, Ident};
5use quote::quote;
6
7use wagon_parser::parser::term::{Term, TermP};
8
9fn to_tokens_termp<U>(left: &TokenStream, termp: &TermP, state: &mut U, label: Rc<Ident>, attr_fun: fn(&mut U, Rc<Ident>, SpannableIdent)) -> TokenStream {
10    let op = &termp.op;
11    let right = termp.right.to_tokens(state, label.clone(), attr_fun);
12    termp.cont.as_ref().map_or_else(|| quote!(#op (#left, #right)?), |cont| {
13        let cont_stream = to_tokens_termp(&right, cont, state, label, attr_fun);
14        quote!(#op (#left, #cont_stream)?)
15    })
16}
17
18impl<U> ToTokensState<U> for Term {
19    fn to_tokens(&self, state: &mut U, label: Rc<Ident>, attr_fun: fn(&mut U, Rc<Ident>, SpannableIdent)) -> TokenStream {
20        let left = self.left.to_tokens(state, label.clone(), attr_fun);
21        if let Some(termp) = &self.cont {
22            to_tokens_termp(&left, termp, state, label, attr_fun)
23        } else {
24            left
25        }
26    }
27}