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
134
135
136
137
138
139
140
141
142
143
144
145

use crate::{
    error::{IndexError, KeyError},
    types::{State, Tokenizer},
};

///
/// [DefaultTokenizer](DefaultTokenizer) have a format query likes :
/// ```
/// // [0].test.[1]
/// // test.test.[1]
/// ```
pub struct DefaultTokenizer;

impl Tokenizer for DefaultTokenizer {
    /// Parse index array.
    ///
    /// - should starts and end with '[' and ']'. No space allowed.
    ///
    /// ```rust
    /// use querable::{types::Tokenizer, default::DefaultTokenizer, error::{IndexError}};
    ///
    /// assert_eq!(DefaultTokenizer::index_parse("[]"), Err(IndexError::ParseError(String::from("[]"))));
    /// ```
    ///
    /// - index should be an integer, specificially, in the range of usize.
    ///
    /// ```rust
    /// use querable::{types::Tokenizer, default::DefaultTokenizer};
    ///
    /// assert!(DefaultTokenizer::index_parse("[x]").is_err());
    /// ```
    ///
    fn index_parse(key: &str) -> Result<usize, IndexError> {
        if key.starts_with('[') && key.ends_with(']') && key.len() > 2 {
            let index = &key[1..key.len() - 1];
            index.parse::<usize>().map_err(IndexError::IntError)
        } else {
            Err(IndexError::ParseError(String::from(key)))
        }
    }

    /// Parse key/path index.
    ///
    /// - shouldn't be an empty string or being prefixed & suffixed with empty string.
    ///
    /// ```rust
    /// use querable::{types::Tokenizer, default::DefaultTokenizer, error::{KeyError}};
    ///
    /// assert_eq!(DefaultTokenizer::dict_parse("   .test"), Err(KeyError::ParseError(String::from("   "))));
    /// assert_eq!(DefaultTokenizer::dict_parse(""), Err(KeyError::EmptyKey));
    /// ```
    ///
    #[inline]
    fn dict_parse(key: &str) -> Result<State, KeyError> {
        if key.is_empty() {
            Err(KeyError::EmptyKey)
        } else {
            let size = key.len();

            match key.find('.') {
                Some(0) => Err(KeyError::EmptyKey),
                Some(idx) => {
                    let current = &key[0..idx];

                    match current.find(char::is_whitespace) {
                        Some(_) => Err(KeyError::ParseError(String::from(current))),
                        _ => {
                            let pivot = idx + 1;
                            Ok((Some(current), Some(&key[pivot..size])))
                        }
                    }
                }
                _ => Ok((Some(&key[0..size]), None)),
            }
        }
    }
}

///
/// [SlashTokenizer](SlashTokenizer) have a format query likes :
/// ```
/// // /0/1/2/3
/// // /test/test/1/test/test/2
/// ```
pub struct SlashTokenizer;

impl Tokenizer for SlashTokenizer {
    #[inline]
    fn index_parse(key: &str) -> Result<usize, IndexError> {
        key.parse::<usize>().map_err(IndexError::IntError)
    }

    /// Parse dict key/path query.
    ///
    /// The query should :
    ///
    /// - starts with '/'
    ///
    /// ```rust
    /// use querable::{types::Tokenizer, default::SlashTokenizer, error::{KeyError, IndexError}};
    /// assert_eq!(SlashTokenizer::dict_parse("test."), Err(KeyError::ParseError(String::from("test."))));
    /// ```
    ///
    /// - have no empty path
    ///
    /// ```rust
    /// use querable::{types::Tokenizer, default::SlashTokenizer, error::{KeyError, IndexError}};
    /// assert_eq!(SlashTokenizer::dict_parse("//"), Err(KeyError::EmptyKey));
    /// ```
    ///
    /// - not an empty string
    ///
    /// ```rust
    /// use querable::{types::Tokenizer, default::SlashTokenizer, error::{KeyError, IndexError}};
    /// assert_eq!(SlashTokenizer::dict_parse(""), Err(KeyError::EmptyKey))
    /// ```
    fn dict_parse(key: &str) -> Result<State, KeyError> {
        if key.is_empty() {
            Err(KeyError::EmptyKey)
        } else if !key.starts_with('/') {
            // key should always prefixed with slash
            Err(KeyError::ParseError(String::from(key)))
        } else {
            let size = key.len();
            // /1/2
            // 1/2
            match key[1..size].find('/') {
                // since path is empty (case "//")
                Some(0) => Err(KeyError::EmptyKey),
                // if there is '/', then there will be next
                Some(idx) => {
                    let pivot = idx + 1;
                    let current = &key[1..pivot];
                    // check whether current have a whitespace or not
                    // key shouldn't have a whitespace
                    match current.find(char::is_whitespace) {
                        Some(_) => Err(KeyError::ParseError(String::from(current))),
                        _ => Ok((Some(current), Some(&key[pivot..size]))),
                    }
                }
                _ => Ok((Some(&key[1..size]), None)),
            }
        }
    }
}