serde-querystring 0.0.8

A query string deserializer based on serde
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
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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133

use std::collections::VecDeque;

use super::parser::parse_char;
use crate::error::{Error, Result};

#[derive(Debug)]
pub struct SubKey<'de> {
    index: usize,
    slice: &'de [u8],
}

impl<'de> SubKey<'de> {
    pub(crate) fn new(slice: &'de [u8]) -> Self {
        Self { index: 0, slice }
    }

    pub(crate) fn next_key(&mut self) -> Result<&'de [u8]> {
        while self.index < self.slice.len() {
            match self.slice[self.index] {
                b']' => {
                    let key = &self.slice[0..self.index];
                    self.slice = &self.slice[(self.index + 1)..];
                    self.index = 0;
                    return Ok(&key);
                }
                b'%' => {
                    if self.index + 2 < self.slice.len() {
                        match parse_char(&self.slice[(self.index + 1)..=(self.index + 2)]) {
                            Some(b']') => {
                                let key = &self.slice[0..self.index];
                                self.slice = &self.slice[(self.index + 3)..];
                                self.index = 0;
                                return Ok(&key);
                            }
                            _ => {
                                self.index += 1;
                            }
                        }
                    } else {
                        self.index += 1;
                    }
                }
                _ => {
                    self.index += 1;
                }
            }
        }
        Err(Error::InvalidMapKey)
    }

    pub(crate) fn next_subkey(mut self) -> Result<Option<Self>> {
        // Check if the starting character is either `[` or percent encoded version of it
        if self.slice.len() > self.index {
            if self.slice.len() > self.index + 1 && self.slice[self.index] == b'[' {
                self.slice = &self.slice[1..];
                Ok(Some(self))
            } else if self.slice.len() > self.index + 2
                && self.slice[self.index] == b'%'
                && parse_char(&self.slice[(self.index + 1)..=(self.index + 2)]) == Some(b'[')
            {
                self.slice = &self.slice[3..];
                Ok(Some(self))
            } else {
                // The subkey part didn't start with opening bracket
                Err(Error::InvalidMapKey)
            }
        } else {
            Ok(None)
        }
    }
}
#[derive(Debug)]
pub(crate) struct Pair<'de> {
    pub(crate) key: SubKey<'de>,
    pub(crate) value: &'de [u8],
}

pub(crate) struct Stash<'de> {
    pairs: VecDeque<(&'de [u8], VecDeque<Pair<'de>>)>,
    values: Option<VecDeque<Pair<'de>>>,
}

impl<'de> Stash<'de> {
    pub(crate) fn new() -> Self {
        Self {
            pairs: VecDeque::new(),
            values: None,
        }
    }

    pub(crate) fn add(&mut self, key: &'de [u8], subkey: &'de [u8], value: &'de [u8]) {
        if let Some((_, pairs)) = self.pairs.iter_mut().find(|item| item.0 == key) {
            pairs.push_front(Pair {
                key: SubKey::new(subkey),
                value,
            });
        } else {
            let mut pairs = VecDeque::new();
            pairs.push_front(Pair {
                key: SubKey::new(subkey),
                value,
            });
            self.pairs.push_front((key, pairs));
        }
    }

    pub(crate) fn add_subkey(&mut self, key: &'de [u8], subkey: SubKey<'de>, value: &'de [u8]) {
        if let Some((_, pairs)) = self.pairs.iter_mut().find(|item| item.0 == key) {
            pairs.push_front(Pair { key: subkey, value });
        } else {
            let mut pairs = VecDeque::new();
            pairs.push_front(Pair { key: subkey, value });
            self.pairs.push_front((key, pairs));
        }
    }

    pub(crate) fn next_key(&mut self) -> Option<&'de [u8]> {
        if let Some((key, pairs)) = self.pairs.pop_back() {
            self.values = Some(pairs);
            Some(key)
        } else {
            None
        }
    }

    pub(crate) fn next_value(&mut self) -> Result<VecDeque<Pair<'de>>> {
        // Calling next_value before next_key is an error
        match self.values.take() {
            None => Err(Error::EofReached),
            Some(pairs) => Ok(pairs),
        }
    }
}