diff options
Diffstat (limited to 'validator_derive/src')
| -rw-r--r-- | validator_derive/src/lib.rs | 32 |
1 files changed, 19 insertions, 13 deletions
diff --git a/validator_derive/src/lib.rs b/validator_derive/src/lib.rs index 177f315..0ad26a9 100644 --- a/validator_derive/src/lib.rs +++ b/validator_derive/src/lib.rs @@ -71,10 +71,12 @@ fn expand_validation(ast: &syn::MacroInput) -> quote::Tokens { quote!(&self.#field_ident) }; // same but for the ident used in a if let block - let optional_validator_param = if field_name.starts_with("Option<&") { + let optional_validator_param = quote!(#field_ident); + // same but for the ident used in a if let Some variable + let optional_pattern_matched = if field_name.starts_with("Option<&") { quote!(#field_ident) } else { - quote!(&#field_ident) + quote!(ref #field_ident) }; for validator in &validators { @@ -87,14 +89,14 @@ fn expand_validation(ast: &syn::MacroInput) -> quote::Tokens { // wrap in if-let if we have an option if field_name.starts_with("Option<") { quote!( - if let Some(#field_ident) = self.#field_ident { + if let Some(#optional_pattern_matched) = self.#field_ident { if !::validator::validate_length( ::validator::Validator::Length { min: #min_tokens, max: #max_tokens, equal: #equal_tokens }, - #field_ident + #optional_validator_param ) { errors.add(#name, "length"); } @@ -143,8 +145,8 @@ fn expand_validation(ast: &syn::MacroInput) -> quote::Tokens { // wrap in if-let if we have an option if field_name.starts_with("Option<") { quote!( - if let Some(#field_ident) = self.#field_ident { - if !::validator::validate_email(#field_ident) { + if let Some(#optional_pattern_matched) = self.#field_ident { + if !::validator::validate_email(#optional_validator_param) { errors.add(#name, "email"); } } @@ -161,8 +163,8 @@ fn expand_validation(ast: &syn::MacroInput) -> quote::Tokens { // wrap in if-let if we have an option if field_name.starts_with("Option<") { quote!( - if let Some(#field_ident) = self.#field_ident { - if !::validator::validate_url(#field_ident) { + if let Some(#optional_pattern_matched) = self.#field_ident { + if !::validator::validate_url(#optional_validator_param) { errors.add(#name, "url"); } } @@ -188,8 +190,8 @@ fn expand_validation(ast: &syn::MacroInput) -> quote::Tokens { // wrap in if-let if we have an option if field_name.starts_with("Option<") { quote!( - if let Some(#field_ident) = self.#field_ident { - match #fn_ident(#field_ident) { + if let Some(#optional_pattern_matched) = self.#field_ident { + match #fn_ident(#optional_validator_param) { ::std::option::Option::Some(s) => { errors.add(#name, &s); }, @@ -212,7 +214,7 @@ fn expand_validation(ast: &syn::MacroInput) -> quote::Tokens { // wrap in if-let if we have an option if field_name.starts_with("Option<") { quote!( - if let Some(#field_ident) = self.#field_ident { + if let Some(#optional_pattern_matched) = self.#field_ident { if !::validator::validate_contains(#optional_validator_param, &#n) { errors.add(#name, "contains"); } @@ -231,8 +233,8 @@ fn expand_validation(ast: &syn::MacroInput) -> quote::Tokens { // wrap in if-let if we have an option if field_name.starts_with("Option<") { quote!( - if let Some(#field_ident) = self.#field_ident { - if !#re_ident.is_match(#field_ident) { + if let Some(#optional_pattern_matched) = self.#field_ident { + if !#re_ident.is_match(#optional_validator_param) { errors.add(#name, "regex"); } } @@ -539,6 +541,7 @@ fn find_validators_for_field(field: &syn::Field, field_types: &HashMap<String, S "email" => { if field_type != "String" && field_type != "&str" + && field_type != "Option<String>" && !(field_type.starts_with("Option<") && field_type.ends_with("str>")) { panic!("`email` validator can only be used on String or &str"); } @@ -547,6 +550,7 @@ fn find_validators_for_field(field: &syn::Field, field_types: &HashMap<String, S "url" => { if field_type != "String" && field_type != "&str" + && field_type != "Option<String>" && !(field_type.starts_with("Option<") && field_type.ends_with("str>")) { panic!("`url` validator can only be used on String or &str"); } @@ -600,6 +604,8 @@ fn find_validators_for_field(field: &syn::Field, field_types: &HashMap<String, S if name == "length" { if field_type != "String" && !field_type.starts_with("Vec<") + && !field_type.starts_with("Option<Vec<") + && field_type != "Option<String>" // a bit ugly && !(field_type.starts_with("Option<") && field_type.ends_with("str>")) && field_type != "&str" { |