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
/*
* parsing/rule/impls/header.rs
*
* ftml - Library to parse Wikidot text
* Copyright (C) 2019-2026 Wikijump Team
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
use super::prelude::*;
use std::convert::TryInto;
pub const RULE_HEADER: Rule = Rule {
name: "header",
position: LineRequirement::StartOfLine,
try_consume_fn,
};
fn try_consume_fn<'r, 't>(
parser: &mut Parser<'r, 't>,
) -> ParseResult<'r, 't, Elements<'t>> {
debug!("Trying to create header container");
macro_rules! step {
($token:expr) => {{
let current = parser.current();
if current.token != $token {
return Err(parser.make_err(ParseErrorKind::RuleFailed));
}
parser.step()?;
current
}};
}
// Get header depth
let heading = step!(Token::Heading)
.slice
.try_into()
.expect("Received invalid heading length token slice");
// Step over whitespace
step!(Token::Whitespace);
let (elements, mut all_errors, _) = collect_container(
parser,
RULE_HEADER,
ContainerType::Header(heading),
&[
ParseCondition::current(Token::InputEnd),
ParseCondition::current(Token::LineBreak),
ParseCondition::current(Token::ParagraphBreak),
],
&[],
None,
)?
.into();
// If this heading wants a table of contents (TOC) entry, then add one
if heading.has_toc {
// collect_container() always produces one Element::Container.
// We unwrap it so we can get the elements composing the name.
let elements = match elements {
Elements::Single(Element::Container(ref container)) => container.elements(),
_ => panic!("Collected heading produced a non-single non-container element"),
};
// Create table of contents entry with the given level and name.
parser.push_table_of_contents_entry(heading.level, elements);
}
// Recursively collect headings until we hit an error.
//
// We do this because the container consumes the newline,
// which we need to trigger the next header when using regular rules.
let mut all_elements: Vec<_> = elements.into_iter().collect();
if let Ok(success) = (try_consume_fn)(parser) {
let (elements, mut errors, _) = success.into();
all_elements.extend(elements);
all_errors.append(&mut errors);
}
// Build final Elements object
ok!(false; all_elements, all_errors)
}