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

#![allow(dead_code)]
extern crate yaml_rust;

use yaml_rust::{Yaml, YamlLoader};
use yaml_rust::scanner::ScanError;

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 = try!(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, _) = try!(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);
}