aboutsummaryrefslogtreecommitdiffstats
path: root/validator_derive/src/asserts.rs
blob: 65db5cf3f77d2b35e63d5eceae5965f87dfb0517 (plain)
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
54

pub static NUMBER_TYPES: [&'static str; 24] = [
    "usize", "u8", "u16", "u32", "u64",
    "isize", "i8", "i16", "i32", "i64",
    "f32", "f64",

    "Option<usize>", "Option<u8>", "Option<u16>", "Option<u32>", "Option<u64>",
    "Option<isize>", "Option<i8>", "Option<i16>", "Option<i32>", "Option<i64>",
    "Option<f32>", "Option<f64>",
];


pub fn assert_string_type(name: &str, field_type: &String) {
    if field_type != "String"
        && field_type != "&str"
        && field_type != "Option<String>"
        && !(field_type.starts_with("Option<") && field_type.ends_with("str>")) {
        panic!("`{}` validator can only be used on String, &str or an Option of those", name);
    }
}

pub fn assert_type_matches(field_name: String, field_type: &String, field_type2: Option<&String>) {
    if let Some(t2) = field_type2 {
        if field_type != t2 {
            panic!("Invalid argument for `must_match` validator of field `{}`: types of field can't match", field_name);
        }
    } else {
        panic!("Invalid argument for `must_match` validator of field `{}`: the other field doesn't exist in struct", field_name);
    }
}

pub fn assert_has_len(field_name: String, field_type: &String) {
    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" {
            panic!(
                "Validator `length` can only be used on types `String`, `&str` or `Vec` but found `{}` for field `{}`",
                field_type, field_name
            );
    }
}

pub fn assert_has_range(field_name: String, field_type: &String) {
    if !NUMBER_TYPES.contains(&field_type.as_ref()) {
        panic!(
            "Validator `range` can only be used on number types but found `{}` for field `{}`",
            field_type, field_name
        );
    }
}