sqlite3_ext_macro 0.1.0

Derive macros for sqlite3_ext
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

use super::kw;
use syn::{
    parse::{Parse, ParseStream},
    *,
};

pub enum FnAttr {
    NumArgs(LitInt),
    RiskLevel(FnAttrRiskLevel),
    Deterministic,
}

impl Parse for FnAttr {
    fn parse(input: ParseStream) -> Result<Self> {
        let lookahead = input.lookahead1();
        if lookahead.peek(kw::n_args) {
            input.parse::<kw::n_args>()?;
            input.parse::<Token![=]>()?;
            input.parse().map(FnAttr::NumArgs)
        } else if lookahead.peek(kw::risk_level) {
            input.parse::<kw::risk_level>()?;
            input.parse::<Token![=]>()?;
            input.parse().map(FnAttr::RiskLevel)
        } else if lookahead.peek(kw::deterministic) {
            input.parse::<kw::deterministic>()?;
            Ok(FnAttr::Deterministic)
        } else {
            Err(lookahead.error())
        }
    }
}

pub enum FnAttrRiskLevel {
    Innocuous,
    DirectOnly,
}

impl Parse for FnAttrRiskLevel {
    fn parse(input: ParseStream) -> Result<Self> {
        let lookahead = input.lookahead1();
        if lookahead.peek(kw::Innocuous) {
            input.parse::<kw::Innocuous>()?;
            Ok(FnAttrRiskLevel::Innocuous)
        } else if lookahead.peek(kw::DirectOnly) {
            input.parse::<kw::DirectOnly>()?;
            Ok(FnAttrRiskLevel::DirectOnly)
        } else {
            Err(lookahead.error())
        }
    }
}