stark 0.1.1

Stark is a blazingly fast static site generator
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

pub use yaml_rust::Yaml;

use yaml_rust::scanner::ScanError;
use yaml_rust::YamlLoader;

fn find_yaml_block(text: &str) -> Option<(usize, usize, usize)> {
    match text.starts_with("---\n") {
        true => {
            let slice_after_marker = &text[4..];
            let fm_end = slice_after_marker.find("---\n");
            if fm_end.is_none() {
                return None;
            };

            let fm_end = fm_end.unwrap();
            Some((4, fm_end + 4, fm_end + 2 * 4))
        }
        false => None,
    }
}

pub fn parse_and_find_content(text: &str) -> Result<(Option<Yaml>, &str), ScanError> {
    match find_yaml_block(text) {
        Some((fm_start, fm_end, content_start)) => {
            let yaml_str = &text[fm_start..fm_end];
            let mut documents = YamlLoader::load_from_str(yaml_str)?;

            let rest_of_text = &text[content_start..];

            Ok((documents.pop(), rest_of_text))
        }
        None => Ok((None, text)),
    }
}

pub fn parse(text: &str) -> Result<Option<Yaml>, ScanError> {
    let (matter, _) = parse_and_find_content(text)?;
    Ok(matter)
}

#[test]
fn test_valid() {
    let test_string = "---\ntitle: Valid Yaml Test\n---\nsomething that's not yaml";

    println!("Original String{:?}", test_string);
    let matter = parse(test_string);
    println!("Matter: {:?}", matter);
    assert!(matter.is_ok());
    let matter = matter.unwrap();
    assert!(matter.is_some());
    let matter = matter.unwrap();
    assert!(matter.as_hash().is_some());
}

#[test]
fn test_valid_find_content() {
    let test_string = "---\ntitle: Valid Yaml Test\n---\nsomething that's not yaml";

    println!("Original String{:?}", test_string);
    let result = parse_and_find_content(test_string);
    assert!(result.is_ok());
    let (matter, stripped_string) = result.unwrap();
    println!("Stripped String{:?}", stripped_string);
    println!("Matter: {:?}", matter);
    assert!(matter.is_some());
    let matter = matter.unwrap();
    assert!(matter.as_hash().is_some());
    assert!(stripped_string.to_string() == test_string[31..].to_string());
}

#[test]
fn test_none() {
    let test_string = "something that's not yaml even if it has\n---\nsome: yaml\n--";

    println!("Original String{:?}", test_string);
    let matter = parse(test_string);
    println!("Matter: {:?}", matter);
    assert!(matter.is_ok());
    let matter = matter.unwrap();
    assert!(matter.is_none());
}

#[test]
fn test_none_find_content() {
    let test_string = "something that's not yaml even if it has\n---\nsome: yaml\n--";

    println!("Original String{:?}", test_string);
    let result = parse_and_find_content(test_string);
    assert!(result.is_ok());
    let (matter, stripped_string) = result.unwrap();
    println!("Stripped String{:?}", stripped_string);
    println!("Matter: {:?}", matter);
    assert!(matter.is_none());
    assert!(stripped_string.to_string() == test_string);
}

#[test]
fn test_empty() {
    let test_string = "---\n---\nsomething that's not yaml";

    println!("Original String{:?}", test_string);
    let matter = parse(test_string);
    println!("Matter: {:?}", matter);
    assert!(matter.is_ok());
    let matter = matter.unwrap();
    assert!(matter.is_none());
}

#[test]
fn test_empty_find_content() {
    let test_string = "---\n---\nsomething that's not yaml";

    println!("Original String{:?}", test_string);
    let result = parse_and_find_content(test_string);
    assert!(result.is_ok());
    let (matter, stripped_string) = result.unwrap();
    println!("Stripped String{:?}", stripped_string);
    println!("Matter: {:?}", matter);
    assert!(matter.is_none());
    assert!(stripped_string.to_string() == test_string[8..]);
}

#[test]
fn test_tests() {
    assert_eq!(2 + 2, 4);
}