refactor: cleanup parser

This commit is contained in:
PoiScript 2019-04-05 22:54:50 +08:00
parent 1bb5286dd3
commit 8bb7ae41d3
6 changed files with 128 additions and 99 deletions

View file

@ -13,19 +13,19 @@ pub fn parse(src: &str) -> Option<(&str, Option<&str>, usize, usize, usize)> {
let name = memchr2(b' ', b'\n', src.as_bytes()) let name = memchr2(b' ', b'\n', src.as_bytes())
.filter(|&i| src.as_bytes()[8..i].iter().all(u8::is_ascii_alphabetic))?; .filter(|&i| src.as_bytes()[8..i].iter().all(u8::is_ascii_alphabetic))?;
let mut lines = Lines::new(src); let mut lines = Lines::new(src);
let (pre_cont_end, cont_beg, _) = lines.next()?; let (pre_limit, begin, _) = lines.next()?;
let args = if pre_cont_end == name { let args = if pre_limit == name {
None None
} else { } else {
Some(&src[name..pre_cont_end]) Some(&src[name..pre_limit])
}; };
let name = &src[8..name]; let name = &src[8..name];
let end_line = format!(r"#+END_{}", name.to_uppercase()); let end_line = format!(r"#+END_{}", name.to_uppercase());
let mut pre_end = cont_beg; let mut pre_end = begin;
for (_, end, line) in lines { for (_, end, line) in lines {
if line.trim() == end_line { if line.trim() == end_line {
return Some((name, args, cont_beg, pre_end, end)); return Some((name, args, begin, pre_end, end));
} else { } else {
pre_end = end; pre_end = end;
} }

View file

@ -11,9 +11,9 @@ pub fn parse(src: &str) -> Option<(&str, Option<&str>, usize, usize, usize)> {
} }
let mut lines = Lines::new(src); let mut lines = Lines::new(src);
let (mut pre_cont_end, _, _) = lines.next()?; let (mut pre_limit, _, _) = lines.next()?;
for (cont_end, end, line) in lines { for (limit, end, line) in lines {
if line.trim().eq_ignore_ascii_case("#+END:") { if line.trim().eq_ignore_ascii_case("#+END:") {
let bytes = src.as_bytes(); let bytes = src.as_bytes();
@ -23,21 +23,15 @@ pub fn parse(src: &str) -> Option<(&str, Option<&str>, usize, usize, usize)> {
let name = &src[8..i].trim(); let name = &src[8..i].trim();
return Some(if bytes[i] == b'\n' { return Some(if bytes[i] == b'\n' {
(name, None, i, pre_cont_end, end) (name, None, i, pre_limit, end)
} else { } else {
let cont_beg = memchr(b'\n', bytes) let begin = memchr(b'\n', bytes)
.map(|i| i + 1) .map(|i| i + 1)
.unwrap_or_else(|| src.len()); .unwrap_or_else(|| src.len());
( (name, Some(&src[i..begin].trim()), begin, pre_limit, end)
name,
Some(&src[i..cont_beg].trim()),
cont_beg,
pre_cont_end,
end,
)
}); });
} }
pre_cont_end = cont_end; pre_limit = limit;
} }
None None

View file

@ -1,39 +1,46 @@
use crate::lines::Lines; use crate::lines::Lines;
#[inline] #[inline]
pub fn is_item(src: &str) -> (bool, bool) { pub fn is_item(text: &str) -> Option<bool> {
if src.is_empty() { if text.is_empty() {
return (false, false); return None;
} }
let bytes = src.as_bytes();
let (i, ordered) = match bytes[0] { let bytes = text.as_bytes();
b'*' | b'-' | b'+' => (1, false), match bytes[0] {
b'*' | b'-' | b'+' => {
if text.len() > 1 && (bytes[1] == b' ' || bytes[1] == b'\n') {
Some(false)
} else {
None
}
}
b'0'...b'9' => { b'0'...b'9' => {
let i = bytes let i = bytes
.iter() .iter()
.position(|&c| !c.is_ascii_digit()) .position(|&c| !c.is_ascii_digit())
.unwrap_or_else(|| src.len() - 1); .unwrap_or_else(|| text.len() - 1);
let c = bytes[i]; if (bytes[i] == b'.' || bytes[i] == b')')
if !(c == b'.' || c == b')') { && i + 1 < text.len()
return (false, false); && (bytes[i + 1] == b' ' || bytes[i + 1] == b'\n')
{
Some(true)
} else {
None
} }
(i + 1, true)
} }
_ => return (false, false), _ => None,
};
if i < src.len() {
// bullet is follwed by a space or line ending
(bytes[i] == b' ' || bytes[i] == b'\n', ordered)
} else {
(false, false)
} }
} }
// returns (bullets, contents begin, contents end, end, has more) // returns (bullets, contents begin, contents end, end, has more)
#[inline] #[inline]
pub fn parse(src: &str, ident: usize) -> (&str, usize, usize, usize, bool) { pub fn parse(src: &str, ident: usize) -> (&str, usize, usize, usize, bool) {
debug_assert!(is_item(&src[ident..]).0); debug_assert!(
is_item(&src[ident..]).is_some(),
"{:?} is not a list item",
src
);
debug_assert!( debug_assert!(
src[..ident].chars().all(|c| c == ' ' || c == '\t'), src[..ident].chars().all(|c| c == ' ' || c == '\t'),
"{:?} doesn't starts with indentation {}", "{:?} doesn't starts with indentation {}",
@ -42,60 +49,60 @@ pub fn parse(src: &str, ident: usize) -> (&str, usize, usize, usize, bool) {
); );
let mut lines = Lines::new(src); let mut lines = Lines::new(src);
let (mut pre_cont_end, mut pre_end, first_line) = lines.next().unwrap(); let (mut pre_limit, mut pre_end, first_line) = lines.next().unwrap();
let beg = match memchr::memchr(b' ', &first_line.as_bytes()[ident..]) { let begin = match memchr::memchr(b' ', &first_line.as_bytes()[ident..]) {
Some(i) => i + ident + 1, Some(i) => i + ident + 1,
None => { None => {
let len = first_line.len(); let len = first_line.len();
return ( return (
&first_line, first_line,
len, len,
len, len,
len, len,
is_item(lines.next().unwrap().2).0, is_item(lines.next().unwrap().2).is_some(),
); );
} }
}; };
let bullet = &src[0..beg]; let bullet = &src[0..begin];
while let Some((mut cont_end, mut end, mut line)) = lines.next() { while let Some((mut limit, mut end, mut line)) = lines.next() {
// this line is emtpy // this line is emtpy
if line.is_empty() { if line.is_empty() {
if let Some((next_cont_end, next_end, next_line)) = lines.next() { if let Some((next_limit, next_end, next_line)) = lines.next() {
// next line is emtpy, too // next line is emtpy, too
if next_line.is_empty() { if next_line.is_empty() {
return (bullet, beg, pre_cont_end, next_end, false); return (bullet, begin, pre_limit, next_end, false);
} else { } else {
// move to next line // move to next line
pre_end = end; pre_end = end;
cont_end = next_cont_end; limit = next_limit;
end = next_end; end = next_end;
line = next_line; line = next_line;
} }
} else { } else {
return (bullet, beg, pre_cont_end, end, false); return (bullet, begin, pre_limit, end, false);
} }
} }
let line_ident = count_ident(line); let line_ident = count_ident(line);
if line_ident < ident { if line_ident < ident {
return (bullet, beg, pre_cont_end, pre_end, false); return (bullet, begin, pre_limit, pre_end, false);
} else if line_ident == ident { } else if line_ident == ident {
return ( return (
bullet, bullet,
beg, begin,
pre_cont_end, pre_limit,
pre_end, pre_end,
is_item(&line[ident..]).0, is_item(&line[ident..]).is_some(),
); );
} }
pre_end = end; pre_end = end;
pre_cont_end = cont_end; pre_limit = limit;
} }
(bullet, beg, src.len(), src.len(), false) (bullet, begin, src.len(), src.len(), false)
} }
#[inline] #[inline]
@ -112,18 +119,18 @@ mod tests {
fn is_item() { fn is_item() {
use super::is_item; use super::is_item;
assert_eq!(is_item("+ item"), (true, false)); assert_eq!(is_item("+ item"), Some(false));
assert_eq!(is_item("- item"), (true, false)); assert_eq!(is_item("- item"), Some(false));
assert_eq!(is_item("10. item"), (true, true)); assert_eq!(is_item("10. item"), Some(true));
assert_eq!(is_item("10) item"), (true, true)); assert_eq!(is_item("10) item"), Some(true));
assert_eq!(is_item("1. item"), (true, true)); assert_eq!(is_item("1. item"), Some(true));
assert_eq!(is_item("1) item"), (true, true)); assert_eq!(is_item("1) item"), Some(true));
assert_eq!(is_item("10. "), (true, true)); assert_eq!(is_item("10. "), Some(true));
assert_eq!(is_item("10.\n"), (true, true)); assert_eq!(is_item("10.\n"), Some(true));
assert_eq!(is_item("10."), (false, false)); assert_eq!(is_item("10."), None);
assert_eq!(is_item("+"), (false, false)); assert_eq!(is_item("+"), None);
assert_eq!(is_item("-item"), (false, false)); assert_eq!(is_item("-item"), None);
assert_eq!(is_item("+item"), (false, false)); assert_eq!(is_item("+item"), None);
} }
#[test] #[test]

View file

@ -23,13 +23,13 @@ impl<'a> Iterator for Lines<'a> {
#[inline] #[inline]
fn next(&mut self) -> Option<(usize, usize, &'a str)> { fn next(&mut self) -> Option<(usize, usize, &'a str)> {
self.iter.next().map(|i| { self.iter.next().map(|i| {
let (line, cont_end) = if i != self.src.len() && self.src.as_bytes()[i - 1] == b'\r' { let (line, limit) = if i != self.src.len() && self.src.as_bytes()[i - 1] == b'\r' {
(&self.src[self.start..i - 1], i - 1) (&self.src[self.start..i - 1], i - 1)
} else { } else {
(&self.src[self.start..i], i) (&self.src[self.start..i], i)
}; };
self.start = if i != self.src.len() { i + 1 } else { i }; self.start = if i != self.src.len() { i + 1 } else { i };
(cont_end, self.start, line) (limit, self.start, line)
}) })
} }

View file

@ -151,7 +151,10 @@ pub fn parse_inactive(text: &str) -> Option<(Timestamp<'_>, usize)> {
} }
fn parse_datetime(bytes: &[u8]) -> Option<(Datetime, Option<Datetime>)> { fn parse_datetime(bytes: &[u8]) -> Option<(Datetime, Option<Datetime>)> {
if !bytes[0].is_ascii_digit() || !bytes[bytes.len() - 1].is_ascii_alphanumeric() { if bytes.is_empty()
|| !bytes[0].is_ascii_digit()
|| !bytes[bytes.len() - 1].is_ascii_alphanumeric()
{
return None; return None;
} }

View file

@ -213,35 +213,56 @@ impl<'a> Parser<'a> {
} }
fn next_ele(&mut self, text: &'a str) -> Event<'a> { fn next_ele(&mut self, text: &'a str) -> Event<'a> {
fn skip_empty_lines(text: &str) -> usize {
let mut i = 0;
for pos in memchr_iter(b'\n', text.as_bytes()) {
if text.as_bytes()[i..pos].iter().all(u8::is_ascii_whitespace) {
i = pos + 1;
} else {
return i;
}
}
if text.as_bytes()[i..].iter().all(u8::is_ascii_whitespace) {
text.len()
} else {
i
}
}
let start = skip_empty_lines(text);
if start == text.len() {
self.off += text.len();
return self.end();
};
let tail = &text[start..];
let (ele, off, limit, end) = self let (ele, off, limit, end) = self
.ele_buf .ele_buf
.take() .take()
.or_else(|| self.real_next_ele(text)) .or_else(|| self.real_next_ele(tail))
.unwrap_or_else(|| { .unwrap_or_else(|| {
let len = text.len(); let mut pos = 0;
let start = text.find(|c| c != '\n').unwrap_or(0); for off in memchr_iter(b'\n', tail.as_bytes()) {
if start == len - 1 { if tail.as_bytes()[pos + 1..off]
(self.end(), len, 0, 0) .iter()
} else { .all(u8::is_ascii_whitespace)
let mut pos = start; {
for off in memchr_iter(b'\n', &text.as_bytes()[start..]) { return (Event::ParagraphBeg, 0, pos + start, off + start);
if text[pos..off + start].trim().is_empty() { } else {
return (Event::ParagraphBeg, start, pos, off + start); if let Some(buf) = self.real_next_ele(&tail[pos + 1..]) {
} else { self.ele_buf = Some(buf);
pos = off + start; return (Event::ParagraphBeg, 0, pos + start, pos + start);
if let Some(buf) = self.real_next_ele(&text[pos + 1..]) {
self.ele_buf = Some(buf);
return (Event::ParagraphBeg, start, pos, pos);
}
} }
} }
( pos = off;
Event::ParagraphBeg,
start,
if text.ends_with('\n') { len - 1 } else { len },
len,
)
} }
let len = text.len();
(
Event::ParagraphBeg,
0,
if text.ends_with('\n') { len - 1 } else { len },
len,
)
}); });
debug_assert!(off <= text.len() && limit <= text.len() && end <= text.len()); debug_assert!(off <= text.len() && limit <= text.len() && end <= text.len());
@ -259,13 +280,15 @@ impl<'a> Parser<'a> {
_ => (), _ => (),
} }
self.off += off; self.off += off + start;
ele ele
} }
// returns (event, offset, container limit, container end) // returns (event, offset, container limit, container end)
fn real_next_ele(&mut self, text: &'a str) -> Option<(Event<'a>, usize, usize, usize)> { fn real_next_ele(&self, text: &'a str) -> Option<(Event<'a>, usize, usize, usize)> {
debug_assert!(!text.starts_with('\n'));
if text.starts_with("[fn:") { if text.starts_with("[fn:") {
if let Some((label, cont, off)) = fn_def::parse(text) { if let Some((label, cont, off)) = fn_def::parse(text) {
return Some((Event::FnDef { label, cont }, off + 1, 0, 0)); return Some((Event::FnDef { label, cont }, off + 1, 0, 0));
@ -277,8 +300,7 @@ impl<'a> Parser<'a> {
.map(|off| (&text[off..], off)) .map(|off| (&text[off..], off))
.unwrap_or((text, 0)); .unwrap_or((text, 0));
let (is_item, ordered) = list::is_item(tail); if let Some(ordered) = list::is_item(tail) {
if is_item {
return Some((Event::ListBeg { ordered }, 0, line_begin, text.len())); return Some((Event::ListBeg { ordered }, 0, line_begin, text.len()));
} }
@ -567,7 +589,6 @@ fn parse() {
}), }),
SectionBeg, SectionBeg,
ParagraphBeg, ParagraphBeg,
Text("test "),
BoldBeg, BoldBeg,
Text("Section 1"), Text("Section 1"),
BoldEnd, BoldEnd,
@ -621,10 +642,14 @@ fn parse() {
assert_eq!( assert_eq!(
Parser::new( Parser::new(
r#"#+OPTIONS: H:3 num:nil toc:t \n:nil ::t |:t ^:t -:t f:t *:t tex:t d:(HIDE) tags:not-in-toc r#"* Title 1
*Section 1*
* Definitions ** Title 2
"# _Section 2_
* Title 3
/Section 3/
* Title 4
=Section 4="#
) )
.collect::<Vec<_>>(), .collect::<Vec<_>>(),
expected expected