      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
     81
     82
     83
     84
     85
     86
     87
     88
     89
     90
     91
     92
     93
     94
     95
     96
     97
     98
     99
    100
    101
    102
    103
    104
    105
    106
    107
    108
    109
    110
    111

    use std::str;

    use actix_web::web::BytesMut;

    enum State {
        YieldStr,
        YieldQuote,
    }

    struct Quoted<'a> {
        inner: ::std::iter::Peekable<str::Split<'a, char>>,
        state: State,
    }

    impl<'a> Quoted<'a> {
        pub fn new(s: &'a str) -> Quoted<'_> {
            Quoted {
                inner: s.split('"').peekable(),
                state: State::YieldStr,
            }
        }
    }

    impl<'a> Iterator for Quoted<'a> {
        type Item = &'a str;

        fn next(&mut self) -> Option<Self::Item> {
            match self.state {
                State::YieldStr => match self.inner.next() {
                    Some(s) => {
                        self.state = State::YieldQuote;
                        Some(s)
                    }
                    None => None,
                },
                State::YieldQuote => match self.inner.peek() {
                    Some(_) => {
                        self.state = State::YieldStr;
                        Some("\\\"")
                    }
                    None => None,
                },
            }
        }
    }

    /// Tries to quote the quotes in the passed `value`
    pub fn put_quoted(buf: &mut BytesMut, value: &str) {
        for part in Quoted::new(value) {
            buf.extend_from_slice(part.as_bytes());
        }
    }

    #[cfg(test)]
    mod tests {
        use std::str;

        use actix_web::web::BytesMut;

        use super::put_quoted;

        #[test]
        fn test_quote_str() {
            let input = "a \"quoted\" string";
            let mut output = BytesMut::new();
            put_quoted(&mut output, input);
            let result = str::from_utf8(&output).unwrap();

            assert_eq!(result, "a \\\"quoted\\\" string");
        }

        #[test]
        fn test_without_quotes() {
            let input = "non-quoted string";
            let mut output = BytesMut::new();
            put_quoted(&mut output, input);
            let result = str::from_utf8(&output).unwrap();

            assert_eq!(result, "non-quoted string");
        }

        #[test]
        fn test_starts_with_quote() {
            let input = "\"first-quoted string";
            let mut output = BytesMut::new();
            put_quoted(&mut output, input);
            let result = str::from_utf8(&output).unwrap();

            assert_eq!(result, "\\\"first-quoted string");
        }

        #[test]
        fn test_ends_with_quote() {
            let input = "last-quoted string\"";
            let mut output = BytesMut::new();
            put_quoted(&mut output, input);
            let result = str::from_utf8(&output).unwrap();

            assert_eq!(result, "last-quoted string\\\"");
        }

        #[test]
        fn test_double_quote() {
            let input = "quote\"\"string";
            let mut output = BytesMut::new();
            put_quoted(&mut output, input);
            let result = str::from_utf8(&output).unwrap();

            assert_eq!(result, "quote\\\"\\\"string");
        }
    }