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

use crate::tokenizer::Token;
use crate::QueryParams;

pub(crate) struct Params<'a> {
    index: usize,
    params: &'a QueryParams,
}

impl<'a> Params<'a> {
    pub fn new(params: &'a QueryParams) -> Self {
        Params { index: 0, params }
    }

    pub fn get(&mut self, token: &'a Token<'a>) -> &'a str {
        match self.params {
            QueryParams::Named(params) => token
                .key
                .as_ref()
                .and_then(|key| {
                    params
                        .iter()
                        .find(|param| param.0 == key.named())
                        .map(|param| param.1.as_str())
                })
                .unwrap_or(token.value),
            QueryParams::Indexed(params) => {
                if let Some(key) = token.key.as_ref().and_then(|key| key.indexed()) {
                    params
                        .get(key)
                        .map(|param| param.as_str())
                        .unwrap_or(token.value)
                } else {
                    let value = params
                        .get(self.index)
                        .map(|param| param.as_str())
                        .unwrap_or(token.value);
                    self.index += 1;
                    value
                }
            }
            QueryParams::None => token.value,
        }
    }
}