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
|
pub mod token;
pub mod serialize;
#[cfg(test)]
pub mod tests;
mod pest_rst {
use pest_derive::Parser;
#[derive(Parser)]
#[grammar = "rst.pest"]
pub struct RstParser;
}
use self::pest_rst::Rule;
use std::io::Write;
use failure::Error;
use failure_derive::Fail;
use pest::Parser;
use crate::document_tree::{
HasChildren,
elements::{
Document,
Title,
},
element_categories::{
StructuralSubElement
},
};
#[derive(Debug, Fail)]
enum ConversionError {
#[fail(display = "unknown rule: {:?}", rule)]
UnknownRuleError {
rule: Rule,
},
}
fn convert_ssubel(pair: pest::iterators::Pair<Rule>) -> Result<StructuralSubElement, Error> {
match pair.as_rule() {
Rule::title => Ok(convert_title(pair).into()),
rule => Err(ConversionError::UnknownRuleError { rule }.into()),
}
}
fn convert_title(pair: pest::iterators::Pair<pest_rst::Rule>) -> Title {
let mut title: Option<&str> = None;
let mut adornment_char: Option<char> = None;
for p in pair.into_inner() {
match p.as_rule() {
Rule::line => title = Some(p.as_str()),
Rule::adornments => adornment_char = Some(p.as_str().chars().next().expect("Empty adornment?")),
rule => panic!("Unexpected rule in title: {:?}", rule),
};
}
// TODO adornment char
Title::with_children(vec![
title.expect("No text in title").into()
])
}
/// tokens to Document tree. resolves sections, but not references
pub fn parse(source: &str) -> Result<Document, Error> {
let pairs = pest_rst::RstParser::parse(pest_rst::Rule::document, source)?;
let structural_elems = pairs.map(convert_ssubel).collect::<Result<_, _>>()?;
Ok(Document::with_children(structural_elems))
}
/// only until we can serialize DocumentTrees
pub fn serialize_json<W>(source: &str, stream: W) -> Result<(), Error> where W: Write {
use self::pest_rst::{RstParser, Rule};
use self::serialize::PairsWrap;
let parsed = RstParser::parse(Rule::document, source)?;
serde_json::to_writer(stream, &PairsWrap(parsed))?;
Ok(())
}
|
