rust-embed-for-web-impl 11.4.1

The proc-macro implementation of rust-embed-for-web.
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
52
53

use rust_embed_for_web_utils::Config;
use syn::{Attribute, Expr, ExprLit, Lit, Meta, MetaNameValue};

fn parse_str(attribute: &Attribute) -> Option<String> {
    if let Meta::NameValue(MetaNameValue {
        value: Expr::Lit(ExprLit {
            lit: Lit::Str(value),
            ..
        }),
        ..
    }) = &attribute.meta
    {
        return Some(value.value());
    }
    None
}

fn parse_bool(attribute: &Attribute) -> Option<bool> {
    if let Meta::NameValue(MetaNameValue {
        value: Expr::Lit(ExprLit {
            lit: Lit::Bool(value),
            ..
        }),
        ..
    }) = &attribute.meta
    {
        return Some(value.value());
    }
    None
}

pub(crate) fn read_attribute_config(ast: &syn::DeriveInput) -> Config {
    let mut config = Config::default();

    for attribute in &ast.attrs {
        if let Some(ident) = attribute.path().get_ident() {
            let ident = ident.to_string();
            match ident.as_str() {
                #[cfg(feature = "include-exclude")]
                "include" => parse_str(attribute).map(|v| config.add_include(v)),
                #[cfg(feature = "include-exclude")]
                "exclude" => parse_str(attribute).map(|v| config.add_exclude(v)),
                "gzip" => parse_bool(attribute).map(|v| config.set_gzip(v)),
                "br" => parse_bool(attribute).map(|v| config.set_br(v)),
                "zstd" => parse_bool(attribute).map(|v| config.set_zstd(v)),
                "allow_missing" => parse_bool(attribute).map(|v| config.set_allow_missing(v)),
                _ => None,
            };
        }
    }

    config
}