datalit-macros-internals 0.1.0-alpha.4

Internals for the datalit crate
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

mod directives;
mod functions;

use syn::parse::ParseStream;

use crate::state::{EntryState, StateOperation};

use self::{directives::DirectiveEntry, functions::CallExprEntry};

#[derive(derive_syn_parse::Parse)]
pub enum CallEntry {
    #[peek_with(CallExprEntry::peek, name = "call expression")]
    CallExpr(CallExprEntry),
    #[peek_with(DirectiveEntry::peek, name = "directive")]
    Directive(DirectiveEntry),
}

impl CallEntry {
    pub fn peek(input: ParseStream) -> bool {
        CallExprEntry::peek(input) || DirectiveEntry::peek(input)
    }
}

impl StateOperation for CallEntry {
    fn apply_to(&self, state: &mut EntryState) -> syn::Result<()> {
        match self {
            CallEntry::CallExpr(call_expr) => call_expr.apply_to(state),
            CallEntry::Directive(directive) => directive.apply_to(state),
        }
    }
}