libhaystack 3.1.5

Rust implementation of the Haystack 4 data types, defs, filter, units, and encodings
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

// Copyright (C) 2020 - 2022, J2 Innovations

//! Zinc Uri scalar decoding

use super::super::scalar::str::parse_str_unicode_escape;
use super::super::scanner::Scanner;
use crate::haystack::val::Uri;
use std::io::{Error, Read};

// Parse Uri
pub(crate) fn parse_uri<R: Read>(scanner: &mut Scanner<R>) -> Result<Uri, Error> {
    let start = scanner.pos;
    scanner.expect_and_consume(b'`')?;

    let mut str = Vec::new();

    while scanner.cur != b'`' {
        if scanner.is_eof {
            return scanner.make_generic_err("Expected '`'");
        }

        if scanner.cur == b'\\' {
            let next = scanner.peek()?;
            match next {
                b':' | b'/' | b'?' | b'#' | b'\\' => {
                    str.push(scanner.cur);
                    str.push(next);
                    scanner.read()?;
                }
                b'[' | b']' | b'@' | b'`' | b'&' | b'=' | b';' => {
                    str.push(next);
                    scanner.read()?;
                }
                _ => {
                    let unicode = parse_str_unicode_escape(scanner)?;
                    str.extend_from_slice(unicode.as_bytes());
                }
            };
        } else {
            str.push(scanner.cur);
        }
        scanner.advance()?
    }

    if start == scanner.pos {
        return scanner.make_generic_err("Unterminated Uri");
    }

    scanner.advance()?;

    Ok(Uri {
        value: String::from_utf8_lossy(&str).to_string(),
    })
}

#[cfg(test)]
mod test {
    use super::parse_uri;
    use crate::haystack::val::Uri;
    use std::io::Cursor;

    #[test]
    fn test_zinc_parse_uri() {
        let mut input = Cursor::new("`/foo/bar/baz`".as_bytes());
        let mut scanner = super::Scanner::make(&mut input).expect("Scanner");

        let str = parse_uri(&mut scanner);
        assert_eq!(str.ok(), Some(Uri::make("/foo/bar/baz")));
        {
            let mut input = Cursor::new("`/\\@\\#`".as_bytes());
            let mut scanner = super::Scanner::make(&mut input).expect("Scanner");

            let str = parse_uri(&mut scanner);
            assert_eq!(str.ok(), Some(Uri::make("/@\\#")));
        }
    }

    #[test]
    fn test_zinc_parse_uri_bad() {
        let mut input = Cursor::new("`".as_bytes());
        let mut scanner = super::Scanner::make(&mut input).expect("Scanner");

        assert!(parse_uri(&mut scanner).is_err());

        let mut input = Cursor::new("`/foo/bar/baz".as_bytes());
        let mut scanner = super::Scanner::make(&mut input).expect("Scanner");

        let str = parse_uri(&mut scanner);
        assert!(str.is_err());
        {
            let mut input = Cursor::new("/foo/bar/baz".as_bytes());
            let mut scanner = super::Scanner::make(&mut input).expect("Scanner");

            let str = parse_uri(&mut scanner);
            assert!(str.is_err());
        }
    }
}