sanitizer_macros 0.2.1

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

use crate::arg::ArgBuilder;
use crate::codegen::sanitizers::PathOrList;
use crate::sanitizer::SanitizerError;
use proc_macro2::TokenStream;
use quote::quote;

#[macro_use]
macro_rules! sanitizer_with_arg {
    ( $sanitizer : expr, $method_name : ident, $arg : expr, $func_call : ident ) => {
        if $sanitizer.has_args() {
            if $sanitizer.get_args().len() == 1 {
                let arg_one = ArgBuilder::$method_name($arg);
                Ok(quote! {
                    $func_call(#arg_one)
                })
            } else {
                Err(SanitizerError::WrongArguments)
            }
        } else {
            Err(SanitizerError::Only64BitInt)
        }
    };
}

pub fn get_string_sanitizers(sanitizer: &PathOrList) -> Result<TokenStream, SanitizerError> {
    match sanitizer.to_string().as_str() {
        "trim" => Ok(quote! { trim() }),
        "numeric" => Ok(quote! { numeric() }),
        "alphanumeric" => Ok(quote! { alphanumeric() }),
        "lower_case" => Ok(quote! { to_lowercase() }),
        "upper_case" => Ok(quote! { to_uppercase() }),
        "camel_case" => Ok(quote! { to_camel_case() }),
        "snake_case" => Ok(quote! { to_snake_case() }),
        "kebab_case" => Ok(quote! { to_kebab_case() }),
        "screaming_kebab_case" => Ok(quote! { to_screaming_kebab_case() }),
        "screaming_snake_case" => Ok(quote! { to_screaming_snakecase() }),
        "e164" => Ok(quote! { e164() }),
        "clamp" => {
            sanitizer_with_arg!(sanitizer, int, &sanitizer.get_args().args[0], cut)
        }
        "custom" => {
            sanitizer_with_arg!(
                sanitizer,
                ident,
                sanitizer.get_args().args[0].as_str(),
                call
            )
        }
        _ => Err(SanitizerError::InvalidSanitizer),
    }
}