hitt_formatter/
lib.rs

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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
use json::format_json;

mod json;

#[derive(Copy, Clone, Default, PartialEq, Eq)]
pub enum ContentType {
    Json,
    #[default]
    Unknown,
}

impl From<&str> for ContentType {
    #[inline]
    fn from(value: &str) -> Self {
        value
            .parse::<mime::Mime>()
            .map_or(Self::Unknown, |m| match (m.type_(), m.subtype()) {
                (_, mime::JSON) => Self::Json,
                _ => Self::Unknown,
            })
    }
}

#[cfg(test)]
mod test_from_str_to_content_type {
    use crate::ContentType;

    #[test]
    fn it_should_parse_unknown_text_as_unknown() {
        for x in u8::MIN..u8::MAX {
            assert!(ContentType::Unknown == ContentType::from(x.to_string().as_str()));
            assert!(
                ContentType::Unknown == ContentType::from(format!("application/masd-{x}").as_str())
            );
            assert!(ContentType::Unknown == ContentType::from(format!("text/masd-{x}").as_str()));
        }
    }

    #[test]
    fn it_should_parse_application_json_as_json() {
        let input = "application/JSON";

        assert!(ContentType::Json == ContentType::from(input));
        assert!(ContentType::Json == ContentType::from(input.to_lowercase().as_str()));
        assert!(ContentType::Json == ContentType::from(input.to_uppercase().as_str()));

        assert!(ContentType::Json == ContentType::from("application/json; charset=utf-8"));
    }
}

#[inline]
pub fn format(input: &str, content_type: ContentType) -> Option<String> {
    match content_type {
        ContentType::Json => Some(format_json(input)),
        ContentType::Unknown => None,
    }
}

#[cfg(test)]
mod test_format {
    use crate::ContentType;

    #[test]
    fn it_should_ignore_unknown_content_types() {
        let input = "this is an unknown content type";

        assert!(crate::format(input, ContentType::Unknown).is_none());
    }

    #[test]
    fn it_should_format_json() {
        let input = "{ \"key\": \"value\" }";
        let content_type = ContentType::Json;

        assert_eq!(
            crate::format(input, content_type),
            Some(crate::json::format_json(input))
        );
    }
}