use std::cmp::{max, min};
use std::collections::{HashMap, VecDeque};
use std::ops::{Index, Range};
use unicase::UniCase;
use crate::linklabel::{scan_link_label_rest, LinkLabel, ReferenceLabel};
use crate::scanners::*;
use crate::strings::CowStr;
use crate::tree::{Tree, TreeIndex};
const LINK_MAX_NESTED_PARENS: usize = 5;
#[derive(Clone, Debug, PartialEq)]
pub enum CodeBlockKind<'a> {
Indented,
Fenced(CowStr<'a>),
}
impl<'a> CodeBlockKind<'a> {
pub fn is_indented(&self) -> bool {
match *self {
CodeBlockKind::Indented => true,
_ => false,
}
}
pub fn is_fenced(&self) -> bool {
match *self {
CodeBlockKind::Fenced(_) => true,
_ => false,
}
}
}
#[derive(Clone, Debug, PartialEq)]
pub enum Tag<'a> {
Paragraph,
Heading(u32),
BlockQuote,
CodeBlock(CodeBlockKind<'a>),
List(Option<u64>),
Item,
FootnoteDefinition(CowStr<'a>),
Table(Vec<Alignment>),
TableHead,
TableRow,
TableCell,
Emphasis,
Strong,
Strikethrough,
Link(LinkType, CowStr<'a>, CowStr<'a>),
Image(LinkType, CowStr<'a>, CowStr<'a>),
}
#[derive(Clone, Debug, PartialEq, Copy)]
pub enum LinkType {
Inline,
Reference,
ReferenceUnknown,
Collapsed,
CollapsedUnknown,
Shortcut,
ShortcutUnknown,
Autolink,
Email,
}
impl LinkType {
fn to_unknown(self) -> Self {
match self {
LinkType::Reference => LinkType::ReferenceUnknown,
LinkType::Collapsed => LinkType::CollapsedUnknown,
LinkType::Shortcut => LinkType::ShortcutUnknown,
_ => unreachable!(),
}
}
}
#[derive(Clone, Debug, PartialEq)]
pub enum Event<'a> {
Start(Tag<'a>),
End(Tag<'a>),
Text(CowStr<'a>),
Code(CowStr<'a>),
Html(CowStr<'a>),
FootnoteReference(CowStr<'a>),
SoftBreak,
HardBreak,
Rule,
TaskListMarker(bool),
}
#[derive(Copy, Clone, Debug, PartialEq)]
pub enum Alignment {
None,
Left,
Center,
Right,
}
bitflags! {
pub struct Options: u32 {
const ENABLE_TABLES = 1 << 1;
const ENABLE_FOOTNOTES = 1 << 2;
const ENABLE_STRIKETHROUGH = 1 << 3;
const ENABLE_TASKLISTS = 1 << 4;
const ENABLE_SMART_PUNCTUATION = 1 << 5;
}
}
#[derive(Debug, Default, Clone, Copy)]
struct Item {
start: usize,
end: usize,
body: ItemBody,
}
#[derive(Debug, PartialEq, Clone, Copy)]
enum ItemBody {
Paragraph,
Text,
SoftBreak,
HardBreak,
MaybeEmphasis(usize, bool, bool),
MaybeSmartQuote(u8, bool, bool),
MaybeCode(usize, bool),
MaybeHtml,
MaybeLinkOpen,
MaybeLinkClose(bool),
MaybeImage,
Emphasis,
Strong,
Strikethrough,
Code(CowIndex),
Link(LinkIndex),
Image(LinkIndex),
FootnoteReference(CowIndex),
TaskListMarker(bool),
Rule,
Heading(u32),
FencedCodeBlock(CowIndex),
IndentCodeBlock,
Html,
OwnedHtml(CowIndex),
BlockQuote,
List(bool, u8, u64),
ListItem(usize),
SynthesizeText(CowIndex),
SynthesizeChar(char),
FootnoteDefinition(CowIndex),
Table(AlignmentIndex),
TableHead,
TableRow,
TableCell,
Root,
}
impl<'a> ItemBody {
fn is_inline(&self) -> bool {
match *self {
ItemBody::MaybeEmphasis(..)
| ItemBody::MaybeSmartQuote(..)
| ItemBody::MaybeHtml
| ItemBody::MaybeCode(..)
| ItemBody::MaybeLinkOpen
| ItemBody::MaybeLinkClose(..)
| ItemBody::MaybeImage => true,
_ => false,
}
}
}
impl<'a> Default for ItemBody {
fn default() -> Self {
ItemBody::Root
}
}
#[derive(PartialEq, Eq, Copy, Clone)]
enum TableParseMode {
Scan,
Active,
Disabled,
}
pub struct BrokenLink<'a> {
pub span: std::ops::Range<usize>,
pub link_type: LinkType,
pub reference: &'a str,
}
struct FirstPass<'a, 'b> {
text: &'a str,
tree: Tree<Item>,
begin_list_item: bool,
last_line_blank: bool,
allocs: Allocations<'a>,
options: Options,
list_nesting: usize,
lookup_table: &'b LookupTable,
}
impl<'a, 'b> FirstPass<'a, 'b> {
fn new(text: &'a str, options: Options, lookup_table: &'b LookupTable) -> FirstPass<'a, 'b> {
let start_capacity = max(128, text.len() / 32);
let tree = Tree::with_capacity(start_capacity);
FirstPass {
text,
tree,
begin_list_item: false,
last_line_blank: false,
allocs: Allocations::new(),
options,
list_nesting: 0,
lookup_table,
}
}
fn run(mut self) -> (Tree<Item>, Allocations<'a>) {
let mut ix = 0;
while ix < self.text.len() {
ix = self.parse_block(ix);
}
for _ in 0..self.tree.spine_len() {
self.pop(ix);
}
(self.tree, self.allocs)
}
fn parse_block(&mut self, mut start_ix: usize) -> usize {
let bytes = self.text.as_bytes();
let mut line_start = LineStart::new(&bytes[start_ix..]);
let i = scan_containers(&self.tree, &mut line_start);
for _ in i..self.tree.spine_len() {
self.pop(start_ix);
}
if self.options.contains(Options::ENABLE_FOOTNOTES) {
if let Some(node_ix) = self.tree.peek_up() {
if let ItemBody::FootnoteDefinition(..) = self.tree[node_ix].item.body {
if self.last_line_blank {
self.pop(start_ix);
}
}
}
let container_start = start_ix + line_start.bytes_scanned();
if let Some(bytecount) = self.parse_footnote(container_start) {
start_ix = container_start + bytecount;
start_ix += scan_blank_line(&bytes[start_ix..]).unwrap_or(0);
line_start = LineStart::new(&bytes[start_ix..]);
}
}
loop {
let container_start = start_ix + line_start.bytes_scanned();
if let Some((ch, index, indent)) = line_start.scan_list_marker() {
let after_marker_index = start_ix + line_start.bytes_scanned();
self.continue_list(container_start, ch, index);
self.tree.append(Item {
start: container_start,
end: after_marker_index,
body: ItemBody::ListItem(indent),
});
self.tree.push();
if let Some(n) = scan_blank_line(&bytes[after_marker_index..]) {
self.begin_list_item = true;
return after_marker_index + n;
}
if self.options.contains(Options::ENABLE_TASKLISTS) {
if let Some(is_checked) = line_start.scan_task_list_marker() {
self.tree.append(Item {
start: after_marker_index,
end: start_ix + line_start.bytes_scanned(),
body: ItemBody::TaskListMarker(is_checked),
});
}
}
} else if line_start.scan_blockquote_marker() {
self.finish_list(start_ix);
self.tree.append(Item {
start: container_start,
end: 0,
body: ItemBody::BlockQuote,
});
self.tree.push();
} else {
break;
}
}
let ix = start_ix + line_start.bytes_scanned();
if let Some(n) = scan_blank_line(&bytes[ix..]) {
if let Some(node_ix) = self.tree.peek_up() {
match self.tree[node_ix].item.body {
ItemBody::BlockQuote => (),
_ => {
if self.begin_list_item {
self.pop(start_ix);
}
self.last_line_blank = true;
}
}
}
return ix + n;
}
self.begin_list_item = false;
self.finish_list(start_ix);
let remaining_space = line_start.remaining_space();
let indent = line_start.scan_space_upto(4);
if indent == 4 {
let ix = start_ix + line_start.bytes_scanned();
let remaining_space = line_start.remaining_space();
return self.parse_indented_code_block(ix, remaining_space);
}
let ix = start_ix + line_start.bytes_scanned();
if bytes[ix] == b'<' {
if let Some(html_end_tag) = get_html_end_tag(&bytes[(ix + 1)..]) {
return self.parse_html_block_type_1_to_5(ix, html_end_tag, remaining_space);
}
let possible_tag = scan_html_block_tag(&bytes[(ix + 1)..]).1;
if is_html_tag(possible_tag) {
return self.parse_html_block_type_6_or_7(ix, remaining_space);
}
if let Some(_html_bytes) = scan_html_type_7(&bytes[ix..]) {
return self.parse_html_block_type_6_or_7(ix, remaining_space);
}
}
if let Ok(n) = scan_hrule(&bytes[ix..]) {
return self.parse_hrule(n, ix);
}
if let Some(atx_size) = scan_atx_heading(&bytes[ix..]) {
return self.parse_atx_heading(ix, atx_size);
}
if let Some((bytecount, label, link_def)) = self.parse_refdef_total(ix) {
self.allocs.refdefs.entry(label).or_insert(link_def);
let ix = ix + bytecount;
return ix + scan_blank_line(&bytes[ix..]).unwrap_or(0);
}
if let Some((n, fence_ch)) = scan_code_fence(&bytes[ix..]) {
return self.parse_fenced_code_block(ix, indent, fence_ch, n);
}
self.parse_paragraph(ix)
}
fn parse_table(&mut self, table_cols: usize, head_start: usize, body_start: usize) -> usize {
let (_sep_start, thead_ix) = self.parse_table_row_inner(head_start, table_cols);
self.tree[thead_ix].item.body = ItemBody::TableHead;
let mut ix = body_start;
while let Some((next_ix, _row_ix)) = self.parse_table_row(ix, table_cols) {
ix = next_ix;
}
self.pop(ix);
ix
}
fn parse_table_row_inner(&mut self, mut ix: usize, row_cells: usize) -> (usize, TreeIndex) {
let bytes = self.text.as_bytes();
let mut cells = 0;
let mut final_cell_ix = None;
let row_ix = self.tree.append(Item {
start: ix,
end: 0,
body: ItemBody::TableRow,
});
self.tree.push();
loop {
ix += scan_ch(&bytes[ix..], b'|');
ix += scan_whitespace_no_nl(&bytes[ix..]);
if let Some(eol_bytes) = scan_eol(&bytes[ix..]) {
ix += eol_bytes;
break;
}
let cell_ix = self.tree.append(Item {
start: ix,
end: ix,
body: ItemBody::TableCell,
});
self.tree.push();
let (next_ix, _brk) = self.parse_line(ix, TableParseMode::Active);
let trailing_whitespace = scan_rev_while(&bytes[..next_ix], is_ascii_whitespace);
if let Some(cur_ix) = self.tree.cur() {
self.tree[cur_ix].item.end -= trailing_whitespace;
}
self.tree[cell_ix].item.end = next_ix - trailing_whitespace;
self.tree.pop();
ix = next_ix;
cells += 1;
if cells == row_cells {
final_cell_ix = Some(cell_ix);
}
}
for _ in cells..row_cells {
self.tree.append(Item {
start: ix,
end: ix,
body: ItemBody::TableCell,
});
}
if let Some(cell_ix) = final_cell_ix {
self.tree[cell_ix].next = None;
}
self.pop(ix);
(ix, row_ix)
}
fn parse_table_row(&mut self, mut ix: usize, row_cells: usize) -> Option<(usize, TreeIndex)> {
let bytes = self.text.as_bytes();
let mut line_start = LineStart::new(&bytes[ix..]);
let containers = scan_containers(&self.tree, &mut line_start);
if containers != self.tree.spine_len() {
return None;
}
line_start.scan_all_space();
ix += line_start.bytes_scanned();
if scan_paragraph_interrupt(&bytes[ix..]) {
return None;
}
let (ix, row_ix) = self.parse_table_row_inner(ix, row_cells);
Some((ix, row_ix))
}
fn parse_paragraph(&mut self, start_ix: usize) -> usize {
let node_ix = self.tree.append(Item {
start: start_ix,
end: 0,
body: ItemBody::Paragraph,
});
self.tree.push();
let bytes = self.text.as_bytes();
let mut ix = start_ix;
loop {
let scan_mode = if self.options.contains(Options::ENABLE_TABLES) && ix == start_ix {
TableParseMode::Scan
} else {
TableParseMode::Disabled
};
let (next_ix, brk) = self.parse_line(ix, scan_mode);
if let Some(Item {
body: ItemBody::Table(alignment_ix),
..
}) = brk
{
let table_cols = self.allocs[alignment_ix].len();
self.tree[node_ix].item.body = ItemBody::Table(alignment_ix);
self.tree[node_ix].child = None;
self.tree.pop();
self.tree.push();
return self.parse_table(table_cols, ix, next_ix);
}
ix = next_ix;
let mut line_start = LineStart::new(&bytes[ix..]);
let n_containers = scan_containers(&self.tree, &mut line_start);
if !line_start.scan_space(4) {
let ix_new = ix + line_start.bytes_scanned();
if n_containers == self.tree.spine_len() {
if let Some(ix_setext) = self.parse_setext_heading(ix_new, node_ix) {
if let Some(Item {
start,
body: ItemBody::HardBreak,
..
}) = brk
{
if bytes[start] == b'\\' {
self.tree.append_text(start, start + 1);
}
}
ix = ix_setext;
break;
}
}
let suffix = &bytes[ix_new..];
if self.interrupt_paragraph_by_list(suffix) || scan_paragraph_interrupt(suffix) {
break;
}
}
line_start.scan_all_space();
if line_start.is_at_eol() {
break;
}
ix = next_ix + line_start.bytes_scanned();
if let Some(item) = brk {
self.tree.append(item);
}
}
self.pop(ix);
ix
}
fn parse_setext_heading(&mut self, ix: usize, node_ix: TreeIndex) -> Option<usize> {
let bytes = self.text.as_bytes();
let (n, level) = scan_setext_heading(&bytes[ix..])?;
self.tree[node_ix].item.body = ItemBody::Heading(level);
if let Some(cur_ix) = self.tree.cur() {
self.tree[cur_ix].item.end -= scan_rev_while(
&bytes[..self.tree[cur_ix].item.end],
is_ascii_whitespace_no_nl,
);
}
Some(ix + n)
}
fn parse_line(&mut self, start: usize, mode: TableParseMode) -> (usize, Option<Item>) {
let bytes = &self.text.as_bytes();
let mut pipes = 0;
let mut last_pipe_ix = start;
let mut begin_text = start;
let (final_ix, brk) =
iterate_special_bytes(&self.lookup_table, bytes, start, |ix, byte| {
match byte {
b'\n' | b'\r' => {
if let TableParseMode::Active = mode {
return LoopInstruction::BreakAtWith(ix, None);
}
let mut i = ix;
let eol_bytes = scan_eol(&bytes[ix..]).unwrap();
if mode == TableParseMode::Scan && pipes > 0 {
let next_line_ix = ix + eol_bytes;
let mut line_start = LineStart::new(&bytes[next_line_ix..]);
if scan_containers(&self.tree, &mut line_start) == self.tree.spine_len()
{
let table_head_ix = next_line_ix + line_start.bytes_scanned();
let (table_head_bytes, alignment) =
scan_table_head(&bytes[table_head_ix..]);
if table_head_bytes > 0 {
let header_count =
count_header_cols(bytes, pipes, start, last_pipe_ix);
if alignment.len() == header_count {
let alignment_ix =
self.allocs.allocate_alignment(alignment);
let end_ix = table_head_ix + table_head_bytes;
return LoopInstruction::BreakAtWith(
end_ix,
Some(Item {
start: i,
end: end_ix,
body: ItemBody::Table(alignment_ix),
}),
);
}
}
}
}
let end_ix = ix + eol_bytes;
let trailing_backslashes = scan_rev_while(&bytes[..ix], |b| b == b'\\');
if trailing_backslashes % 2 == 1 && end_ix < self.text.len() {
i -= 1;
self.tree.append_text(begin_text, i);
return LoopInstruction::BreakAtWith(
end_ix,
Some(Item {
start: i,
end: end_ix,
body: ItemBody::HardBreak,
}),
);
}
let trailing_whitespace =
scan_rev_while(&bytes[..ix], is_ascii_whitespace_no_nl);
if trailing_whitespace >= 2 {
i -= trailing_whitespace;
self.tree.append_text(begin_text, i);
return LoopInstruction::BreakAtWith(
end_ix,
Some(Item {
start: i,
end: end_ix,
body: ItemBody::HardBreak,
}),
);
}
self.tree.append_text(begin_text, ix);
LoopInstruction::BreakAtWith(
end_ix,
Some(Item {
start: i,
end: end_ix,
body: ItemBody::SoftBreak,
}),
)
}
b'\\' => {
if ix + 1 < self.text.len() && is_ascii_punctuation(bytes[ix + 1]) {
self.tree.append_text(begin_text, ix);
if bytes[ix + 1] == b'`' {
let count = 1 + scan_ch_repeat(&bytes[(ix + 2)..], b'`');
self.tree.append(Item {
start: ix + 1,
end: ix + count + 1,
body: ItemBody::MaybeCode(count, true),
});
begin_text = ix + 1 + count;
LoopInstruction::ContinueAndSkip(count)
} else {
begin_text = ix + 1;
LoopInstruction::ContinueAndSkip(1)
}
} else {
LoopInstruction::ContinueAndSkip(0)
}
}
c @ b'*' | c @ b'_' | c @ b'~' => {
let string_suffix = &self.text[ix..];
let count = 1 + scan_ch_repeat(&string_suffix.as_bytes()[1..], c);
let can_open = delim_run_can_open(self.text, string_suffix, count, ix);
let can_close = delim_run_can_close(self.text, string_suffix, count, ix);
let is_valid_seq = c != b'~' || count == 2;
if (can_open || can_close) && is_valid_seq {
self.tree.append_text(begin_text, ix);
for i in 0..count {
self.tree.append(Item {
start: ix + i,
end: ix + i + 1,
body: ItemBody::MaybeEmphasis(count - i, can_open, can_close),
});
}
begin_text = ix + count;
}
LoopInstruction::ContinueAndSkip(count - 1)
}
b'`' => {
self.tree.append_text(begin_text, ix);
let count = 1 + scan_ch_repeat(&bytes[(ix + 1)..], b'`');
self.tree.append(Item {
start: ix,
end: ix + count,
body: ItemBody::MaybeCode(count, false),
});
begin_text = ix + count;
LoopInstruction::ContinueAndSkip(count - 1)
}
b'<' => {
self.tree.append_text(begin_text, ix);
self.tree.append(Item {
start: ix,
end: ix + 1,
body: ItemBody::MaybeHtml,
});
begin_text = ix + 1;
LoopInstruction::ContinueAndSkip(0)
}
b'!' => {
if ix + 1 < self.text.len() && bytes[ix + 1] == b'[' {
self.tree.append_text(begin_text, ix);
self.tree.append(Item {
start: ix,
end: ix + 2,
body: ItemBody::MaybeImage,
});
begin_text = ix + 2;
LoopInstruction::ContinueAndSkip(1)
} else {
LoopInstruction::ContinueAndSkip(0)
}
}
b'[' => {
self.tree.append_text(begin_text, ix);
self.tree.append(Item {
start: ix,
end: ix + 1,
body: ItemBody::MaybeLinkOpen,
});
begin_text = ix + 1;
LoopInstruction::ContinueAndSkip(0)
}
b']' => {
self.tree.append_text(begin_text, ix);
self.tree.append(Item {
start: ix,
end: ix + 1,
body: ItemBody::MaybeLinkClose(true),
});
begin_text = ix + 1;
LoopInstruction::ContinueAndSkip(0)
}
b'&' => match scan_entity(&bytes[ix..]) {
(n, Some(value)) => {
self.tree.append_text(begin_text, ix);
self.tree.append(Item {
start: ix,
end: ix + n,
body: ItemBody::SynthesizeText(self.allocs.allocate_cow(value)),
});
begin_text = ix + n;
LoopInstruction::ContinueAndSkip(n - 1)
}
_ => LoopInstruction::ContinueAndSkip(0),
},
b'|' => {
if let TableParseMode::Active = mode {
LoopInstruction::BreakAtWith(ix, None)
} else {
last_pipe_ix = ix;
pipes += 1;
LoopInstruction::ContinueAndSkip(0)
}
}
b'.' => {
if ix + 2 < bytes.len() && bytes[ix + 1] == b'.' && bytes[ix + 2] == b'.' {
self.tree.append_text(begin_text, ix);
self.tree.append(Item {
start: ix,
end: ix + 3,
body: ItemBody::SynthesizeChar('…'),
});
begin_text = ix + 3;
LoopInstruction::ContinueAndSkip(2)
} else {
LoopInstruction::ContinueAndSkip(0)
}
}
b'-' => {
let count = 1 + scan_ch_repeat(&bytes[(ix + 1)..], b'-');
if count == 1 {
LoopInstruction::ContinueAndSkip(0)
} else {
let itembody = if count == 2 {
ItemBody::SynthesizeChar('–')
} else if count == 3 {
ItemBody::SynthesizeChar('—')
} else {
let (ems, ens) = match count % 6 {
0 | 3 => (count / 3, 0),
2 | 4 => (0, count / 2),
1 => (count / 3 - 1, 2),
_ => (count / 3, 1),
};
let mut buf = String::with_capacity(3 * (ems + ens));
for _ in 0..ems {
buf.push('—');
}
for _ in 0..ens {
buf.push('–');
}
ItemBody::SynthesizeText(self.allocs.allocate_cow(buf.into()))
};
self.tree.append_text(begin_text, ix);
self.tree.append(Item {
start: ix,
end: ix + count,
body: itembody,
});
begin_text = ix + count;
LoopInstruction::ContinueAndSkip(count - 1)
}
}
c @ b'\'' | c @ b'"' => {
let string_suffix = &self.text[ix..];
let can_open = delim_run_can_open(self.text, string_suffix, 1, ix);
let can_close = delim_run_can_close(self.text, string_suffix, 1, ix);
self.tree.append_text(begin_text, ix);
self.tree.append(Item {
start: ix,
end: ix + 1,
body: ItemBody::MaybeSmartQuote(c, can_open, can_close),
});
begin_text = ix + 1;
LoopInstruction::ContinueAndSkip(0)
}
_ => LoopInstruction::ContinueAndSkip(0),
}
});
if brk.is_none() {
self.tree.append_text(begin_text, final_ix);
}
(final_ix, brk)
}
fn interrupt_paragraph_by_list(&self, suffix: &[u8]) -> bool {
scan_listitem(suffix).map_or(false, |(ix, delim, index, _)| {
self.list_nesting > 0 ||
!scan_empty_list(&suffix[ix..]) && (delim == b'*' || delim == b'-' || index == 1)
})
}
fn parse_html_block_type_1_to_5(
&mut self,
start_ix: usize,
html_end_tag: &str,
mut remaining_space: usize,
) -> usize {
let bytes = self.text.as_bytes();
let mut ix = start_ix;
loop {
let line_start_ix = ix;
ix += scan_nextline(&bytes[ix..]);
self.append_html_line(remaining_space, line_start_ix, ix);
let mut line_start = LineStart::new(&bytes[ix..]);
let n_containers = scan_containers(&self.tree, &mut line_start);
if n_containers < self.tree.spine_len() {
break;
}
if (&self.text[line_start_ix..ix]).contains(html_end_tag) {
break;
}
let next_line_ix = ix + line_start.bytes_scanned();
if next_line_ix == self.text.len() {
break;
}
ix = next_line_ix;
remaining_space = line_start.remaining_space();
}
ix
}
fn parse_html_block_type_6_or_7(
&mut self,
start_ix: usize,
mut remaining_space: usize,
) -> usize {
let bytes = self.text.as_bytes();
let mut ix = start_ix;
loop {
let line_start_ix = ix;
ix += scan_nextline(&bytes[ix..]);
self.append_html_line(remaining_space, line_start_ix, ix);
let mut line_start = LineStart::new(&bytes[ix..]);
let n_containers = scan_containers(&self.tree, &mut line_start);
if n_containers < self.tree.spine_len() || line_start.is_at_eol() {
break;
}
let next_line_ix = ix + line_start.bytes_scanned();
if next_line_ix == self.text.len() || scan_blank_line(&bytes[next_line_ix..]).is_some()
{
break;
}
ix = next_line_ix;
remaining_space = line_start.remaining_space();
}
ix
}
fn parse_indented_code_block(&mut self, start_ix: usize, mut remaining_space: usize) -> usize {
self.tree.append(Item {
start: start_ix,
end: 0,
body: ItemBody::IndentCodeBlock,
});
self.tree.push();
let bytes = self.text.as_bytes();
let mut last_nonblank_child = None;
let mut last_nonblank_ix = 0;
let mut end_ix = 0;
let mut last_line_blank = false;
let mut ix = start_ix;
loop {
let line_start_ix = ix;
ix += scan_nextline(&bytes[ix..]);
self.append_code_text(remaining_space, line_start_ix, ix);
if !last_line_blank {
last_nonblank_child = self.tree.cur();
last_nonblank_ix = ix;
end_ix = ix;
}
let mut line_start = LineStart::new(&bytes[ix..]);
let n_containers = scan_containers(&self.tree, &mut line_start);
if n_containers < self.tree.spine_len()
|| !(line_start.scan_space(4) || line_start.is_at_eol())
{
break;
}
let next_line_ix = ix + line_start.bytes_scanned();
if next_line_ix == self.text.len() {
break;
}
ix = next_line_ix;
remaining_space = line_start.remaining_space();
last_line_blank = scan_blank_line(&bytes[ix..]).is_some();
}
if let Some(child) = last_nonblank_child {
self.tree[child].next = None;
self.tree[child].item.end = last_nonblank_ix;
}
self.pop(end_ix);
ix
}
fn parse_fenced_code_block(
&mut self,
start_ix: usize,
indent: usize,
fence_ch: u8,
n_fence_char: usize,
) -> usize {
let bytes = self.text.as_bytes();
let mut info_start = start_ix + n_fence_char;
info_start += scan_whitespace_no_nl(&bytes[info_start..]);
let mut ix = info_start + scan_nextline(&bytes[info_start..]);
let info_end = ix - scan_rev_while(&bytes[info_start..ix], is_ascii_whitespace);
let info_string = unescape(&self.text[info_start..info_end]);
self.tree.append(Item {
start: start_ix,
end: 0,
body: ItemBody::FencedCodeBlock(self.allocs.allocate_cow(info_string)),
});
self.tree.push();
loop {
let mut line_start = LineStart::new(&bytes[ix..]);
let n_containers = scan_containers(&self.tree, &mut line_start);
if n_containers < self.tree.spine_len() {
break;
}
line_start.scan_space(indent);
let mut close_line_start = line_start.clone();
if !close_line_start.scan_space(4) {
let close_ix = ix + close_line_start.bytes_scanned();
if let Some(n) = scan_closing_code_fence(&bytes[close_ix..], fence_ch, n_fence_char)
{
ix = close_ix + n;
break;
}
}
let remaining_space = line_start.remaining_space();
ix += line_start.bytes_scanned();
let next_ix = ix + scan_nextline(&bytes[ix..]);
self.append_code_text(remaining_space, ix, next_ix);
ix = next_ix;
}
self.pop(ix);
ix + scan_blank_line(&bytes[ix..]).unwrap_or(0)
}
fn append_code_text(&mut self, remaining_space: usize, start: usize, end: usize) {
if remaining_space > 0 {
let cow_ix = self.allocs.allocate_cow(" "[..remaining_space].into());
self.tree.append(Item {
start,
end: start,
body: ItemBody::SynthesizeText(cow_ix),
});
}
if self.text.as_bytes()[end - 2] == b'\r' {
self.tree.append_text(start, end - 2);
self.tree.append_text(end - 1, end);
} else {
self.tree.append_text(start, end);
}
}
fn append_html_line(&mut self, remaining_space: usize, start: usize, end: usize) {
if remaining_space > 0 {
let cow_ix = self.allocs.allocate_cow(" "[..remaining_space].into());
self.tree.append(Item {
start,
end: start,
body: ItemBody::SynthesizeText(cow_ix),
});
}
if self.text.as_bytes()[end - 2] == b'\r' {
self.tree.append(Item {
start,
end: end - 2,
body: ItemBody::Html,
});
self.tree.append(Item {
start: end - 1,
end,
body: ItemBody::Html,
});
} else {
self.tree.append(Item {
start,
end,
body: ItemBody::Html,
});
}
}
fn pop(&mut self, ix: usize) {
let cur_ix = self.tree.pop().unwrap();
self.tree[cur_ix].item.end = ix;
if let ItemBody::List(true, _, _) = self.tree[cur_ix].item.body {
surgerize_tight_list(&mut self.tree, cur_ix);
}
}
fn finish_list(&mut self, ix: usize) {
if let Some(node_ix) = self.tree.peek_up() {
if let ItemBody::List(_, _, _) = self.tree[node_ix].item.body {
self.pop(ix);
self.list_nesting -= 1;
}
}
if self.last_line_blank {
if let Some(node_ix) = self.tree.peek_grandparent() {
if let ItemBody::List(ref mut is_tight, _, _) = self.tree[node_ix].item.body {
*is_tight = false;
}
}
self.last_line_blank = false;
}
}
fn continue_list(&mut self, start: usize, ch: u8, index: u64) {
if let Some(node_ix) = self.tree.peek_up() {
if let ItemBody::List(ref mut is_tight, existing_ch, _) = self.tree[node_ix].item.body {
if existing_ch == ch {
if self.last_line_blank {
*is_tight = false;
self.last_line_blank = false;
}
return;
}
}
self.finish_list(start);
}
self.tree.append(Item {
start,
end: 0,
body: ItemBody::List(true, ch, index),
});
self.list_nesting += 1;
self.tree.push();
self.last_line_blank = false;
}
fn parse_hrule(&mut self, hrule_size: usize, ix: usize) -> usize {
self.tree.append(Item {
start: ix,
end: ix + hrule_size,
body: ItemBody::Rule,
});
ix + hrule_size
}
fn parse_atx_heading(&mut self, mut ix: usize, atx_size: usize) -> usize {
let heading_ix = self.tree.append(Item {
start: ix,
end: 0,
body: ItemBody::Heading(atx_size as u32),
});
ix += atx_size;
let bytes = self.text.as_bytes();
if let Some(eol_bytes) = scan_eol(&bytes[ix..]) {
self.tree[heading_ix].item.end = ix + eol_bytes;
return ix + eol_bytes;
}
let skip_spaces = scan_whitespace_no_nl(&bytes[ix..]);
ix += skip_spaces;
let header_start = ix;
let header_node_idx = self.tree.push();
ix = self.parse_line(ix, TableParseMode::Disabled).0;
self.tree[header_node_idx].item.end = ix;
if let Some(cur_ix) = self.tree.cur() {
let header_text = &bytes[header_start..ix];
let mut limit = header_text
.iter()
.rposition(|&b| !(b == b'\n' || b == b'\r' || b == b' '))
.map_or(0, |i| i + 1);
let closer = header_text[..limit]
.iter()
.rposition(|&b| b != b'#')
.map_or(0, |i| i + 1);
if closer == 0 {
limit = closer;
} else {
let spaces = scan_rev_while(&header_text[..closer], |b| b == b' ');
if spaces > 0 {
limit = closer - spaces;
}
}
self.tree[cur_ix].item.end = limit + header_start;
}
self.tree.pop();
ix
}
fn parse_footnote(&mut self, start: usize) -> Option<usize> {
let bytes = &self.text.as_bytes()[start..];
if !bytes.starts_with(b"[^") {
return None;
}
let (mut i, label) = self.parse_refdef_label(start + 2)?;
i += 2;
if scan_ch(&bytes[i..], b':') == 0 {
return None;
}
i += 1;
self.finish_list(start);
self.tree.append(Item {
start,
end: 0,
body: ItemBody::FootnoteDefinition(self.allocs.allocate_cow(label)),
});
self.tree.push();
Some(i)
}
fn parse_refdef_label(&self, start: usize) -> Option<(usize, CowStr<'a>)> {
scan_link_label_rest(&self.text[start..], &|bytes| {
let mut line_start = LineStart::new(bytes);
let _ = scan_containers(&self.tree, &mut line_start);
let bytes_scanned = line_start.bytes_scanned();
let suffix = &bytes[bytes_scanned..];
if self.interrupt_paragraph_by_list(suffix) || scan_paragraph_interrupt(suffix) {
None
} else {
Some(bytes_scanned)
}
})
}
fn parse_refdef_total(&mut self, start: usize) -> Option<(usize, LinkLabel<'a>, LinkDef<'a>)> {
let bytes = &self.text.as_bytes()[start..];
if scan_ch(bytes, b'[') == 0 {
return None;
}
let (mut i, label) = self.parse_refdef_label(start + 1)?;
i += 1;
if scan_ch(&bytes[i..], b':') == 0 {
return None;
}
i += 1;
let (bytecount, link_def) = self.scan_refdef(start + i)?;
Some((bytecount + i, UniCase::new(label), link_def))
}
fn scan_refdef_space(&self, bytes: &[u8], mut i: usize) -> Option<(usize, usize)> {
let mut newlines = 0;
loop {
let whitespaces = scan_whitespace_no_nl(&bytes[i..]);
i += whitespaces;
if let Some(eol_bytes) = scan_eol(&bytes[i..]) {
i += eol_bytes;
newlines += 1;
if newlines > 1 {
return None;
}
} else {
break;
}
let mut line_start = LineStart::new(&bytes[i..]);
if self.tree.spine_len() != scan_containers(&self.tree, &mut line_start) {
return None;
}
i += line_start.bytes_scanned();
}
Some((i, newlines))
}
fn scan_refdef(&self, start: usize) -> Option<(usize, LinkDef<'a>)> {
let bytes = self.text.as_bytes();
let (mut i, _newlines) = self.scan_refdef_space(bytes, start)?;
let (dest_length, dest) = scan_link_dest(self.text, i, 1)?;
if dest_length == 0 {
return None;
}
let dest = unescape(dest);
i += dest_length;
let mut backup = (i - start, LinkDef { dest, title: None });
let (mut i, newlines) =
if let Some((new_i, mut newlines)) = self.scan_refdef_space(bytes, i) {
if i == self.text.len() {
newlines += 1;
}
if new_i == i && newlines == 0 {
return None;
}
if newlines > 1 {
return Some(backup);
};
(new_i, newlines)
} else {
return Some(backup);
};
if let Some((title_length, title)) = scan_refdef_title(&self.text[i..]) {
i += title_length;
backup.1.title = Some(unescape(title));
} else if newlines > 0 {
return Some(backup);
} else {
return None;
};
if let Some(bytes) = scan_blank_line(&bytes[i..]) {
backup.0 = i + bytes - start;
Some(backup)
} else if newlines > 0 {
Some(backup)
} else {
None
}
}
}
fn scan_containers(tree: &Tree<Item>, line_start: &mut LineStart) -> usize {
let mut i = 0;
for &node_ix in tree.walk_spine() {
match tree[node_ix].item.body {
ItemBody::BlockQuote => {
let save = line_start.clone();
if !line_start.scan_blockquote_marker() {
*line_start = save;
break;
}
}
ItemBody::ListItem(indent) => {
let save = line_start.clone();
if !line_start.scan_space(indent) {
if !line_start.is_at_eol() {
*line_start = save;
break;
}
}
}
_ => (),
}
i += 1;
}
i
}
fn count_header_cols(
bytes: &[u8],
mut pipes: usize,
mut start: usize,
last_pipe_ix: usize,
) -> usize {
start += scan_whitespace_no_nl(&bytes[start..]);
if bytes[start] == b'|' {
pipes -= 1;
}
if scan_blank_line(&bytes[(last_pipe_ix + 1)..]).is_some() {
pipes
} else {
pipes + 1
}
}
impl<'a> Tree<Item> {
fn append_text(&mut self, start: usize, end: usize) {
if end > start {
if let Some(ix) = self.cur() {
if ItemBody::Text == self[ix].item.body && self[ix].item.end == start {
self[ix].item.end = end;
return;
}
}
self.append(Item {
start,
end,
body: ItemBody::Text,
});
}
}
}
fn delim_run_can_open(s: &str, suffix: &str, run_len: usize, ix: usize) -> bool {
let next_char = if let Some(c) = suffix.chars().nth(run_len) {
c
} else {
return false;
};
if next_char.is_whitespace() {
return false;
}
if ix == 0 {
return true;
}
let delim = suffix.chars().next().unwrap();
if delim == '*' && !is_punctuation(next_char) {
return true;
}
let prev_char = s[..ix].chars().last().unwrap();
prev_char.is_whitespace()
|| is_punctuation(prev_char) && (delim != '\'' || ![']', ')'].contains(&prev_char))
}
fn delim_run_can_close(s: &str, suffix: &str, run_len: usize, ix: usize) -> bool {
if ix == 0 {
return false;
}
let prev_char = s[..ix].chars().last().unwrap();
if prev_char.is_whitespace() {
return false;
}
let next_char = if let Some(c) = suffix.chars().nth(run_len) {
c
} else {
return true;
};
let delim = suffix.chars().next().unwrap();
if delim == '*' && !is_punctuation(prev_char) {
return true;
}
next_char.is_whitespace() || is_punctuation(next_char)
}
fn scan_paragraph_interrupt(bytes: &[u8]) -> bool {
if scan_eol(bytes).is_some()
|| scan_hrule(bytes).is_ok()
|| scan_atx_heading(bytes).is_some()
|| scan_code_fence(bytes).is_some()
|| scan_blockquote_start(bytes).is_some()
{
return true;
}
bytes.starts_with(b"<")
&& (get_html_end_tag(&bytes[1..]).is_some()
|| is_html_tag(scan_html_block_tag(&bytes[1..]).1))
}
fn get_html_end_tag(text_bytes: &[u8]) -> Option<&'static str> {
static BEGIN_TAGS: &[&[u8]; 3] = &[b"pre", b"style", b"script"];
static ST_BEGIN_TAGS: &[&[u8]; 3] = &[b"!--", b"?", b"![CDATA["];
for (beg_tag, end_tag) in BEGIN_TAGS
.iter()
.zip(["</pre>", "</style>", "</script>"].iter())
{
let tag_len = beg_tag.len();
if text_bytes.len() < tag_len {
break;
}
if !text_bytes[..tag_len].eq_ignore_ascii_case(beg_tag) {
continue;
}
if text_bytes.len() == tag_len {
return Some(end_tag);
}
let s = text_bytes[tag_len];
if is_ascii_whitespace(s) || s == b'>' {
return Some(end_tag);
}
}
for (beg_tag, end_tag) in ST_BEGIN_TAGS.iter().zip(["-->", "?>", "]]>"].iter()) {
if text_bytes.starts_with(beg_tag) {
return Some(end_tag);
}
}
if text_bytes.len() > 1
&& text_bytes[0] == b'!'
&& text_bytes[1] >= b'A'
&& text_bytes[1] <= b'Z'
{
Some(">")
} else {
None
}
}
#[derive(Copy, Clone, Debug)]
struct InlineEl {
start: TreeIndex,
count: usize,
c: u8,
both: bool,
}
#[derive(Debug, Clone, Default)]
struct InlineStack {
stack: Vec<InlineEl>,
lower_bounds: [usize; 7],
}
impl InlineStack {
const UNDERSCORE_NOT_BOTH: usize = 0;
const ASTERISK_NOT_BOTH: usize = 1;
const ASTERISK_BASE: usize = 2;
const TILDES: usize = 5;
const UNDERSCORE_BOTH: usize = 6;
fn pop_all(&mut self, tree: &mut Tree<Item>) {
for el in self.stack.drain(..) {
for i in 0..el.count {
tree[el.start + i].item.body = ItemBody::Text;
}
}
self.lower_bounds = [0; 7];
}
fn get_lowerbound(&self, c: u8, count: usize, both: bool) -> usize {
if c == b'_' {
if both {
self.lower_bounds[InlineStack::UNDERSCORE_BOTH]
} else {
self.lower_bounds[InlineStack::UNDERSCORE_NOT_BOTH]
}
} else if c == b'*' {
let mod3_lower = self.lower_bounds[InlineStack::ASTERISK_BASE + count % 3];
if both {
mod3_lower
} else {
min(
mod3_lower,
self.lower_bounds[InlineStack::ASTERISK_NOT_BOTH],
)
}
} else {
self.lower_bounds[InlineStack::TILDES]
}
}
fn set_lowerbound(&mut self, c: u8, count: usize, both: bool, new_bound: usize) {
if c == b'_' {
if both {
self.lower_bounds[InlineStack::UNDERSCORE_BOTH] = new_bound;
} else {
self.lower_bounds[InlineStack::UNDERSCORE_NOT_BOTH] = new_bound;
}
} else if c == b'*' {
self.lower_bounds[InlineStack::ASTERISK_BASE + count % 3] = new_bound;
if !both {
self.lower_bounds[InlineStack::ASTERISK_NOT_BOTH] = new_bound;
}
} else {
self.lower_bounds[InlineStack::TILDES] = new_bound;
}
}
fn find_match(
&mut self,
tree: &mut Tree<Item>,
c: u8,
count: usize,
both: bool,
) -> Option<InlineEl> {
let lowerbound = min(self.stack.len(), self.get_lowerbound(c, count, both));
let res = self.stack[lowerbound..]
.iter()
.cloned()
.enumerate()
.rfind(|(_, el)| {
el.c == c && (!both && !el.both || (count + el.count) % 3 != 0 || count % 3 == 0)
});
if let Some((matching_ix, matching_el)) = res {
let matching_ix = matching_ix + lowerbound;
for el in &self.stack[(matching_ix + 1)..] {
for i in 0..el.count {
tree[el.start + i].item.body = ItemBody::Text;
}
}
self.stack.truncate(matching_ix);
Some(matching_el)
} else {
self.set_lowerbound(c, count, both, self.stack.len());
None
}
}
fn push(&mut self, el: InlineEl) {
self.stack.push(el)
}
}
#[derive(Debug, Clone)]
enum RefScan<'a> {
LinkLabel(CowStr<'a>, usize),
Collapsed(Option<TreeIndex>),
Failed,
}
fn scan_nodes_to_ix(
tree: &Tree<Item>,
mut node: Option<TreeIndex>,
ix: usize,
) -> Option<TreeIndex> {
while let Some(node_ix) = node {
if tree[node_ix].item.end <= ix {
node = tree[node_ix].next;
} else {
break;
}
}
node
}
fn scan_link_label<'text, 'tree>(
tree: &'tree Tree<Item>,
text: &'text str,
allow_footnote_refs: bool,
) -> Option<(usize, ReferenceLabel<'text>)> {
let bytes = &text.as_bytes();
if bytes.len() < 2 || bytes[0] != b'[' {
return None;
}
let linebreak_handler = |bytes: &[u8]| {
let mut line_start = LineStart::new(bytes);
let _ = scan_containers(tree, &mut line_start);
Some(line_start.bytes_scanned())
};
let pair = if allow_footnote_refs && b'^' == bytes[1] {
let (byte_index, cow) = scan_link_label_rest(&text[2..], &linebreak_handler)?;
(byte_index + 2, ReferenceLabel::Footnote(cow))
} else {
let (byte_index, cow) = scan_link_label_rest(&text[1..], &linebreak_handler)?;
(byte_index + 1, ReferenceLabel::Link(cow))
};
Some(pair)
}
fn scan_reference<'a, 'b>(
tree: &'a Tree<Item>,
text: &'b str,
cur: Option<TreeIndex>,
allow_footnote_refs: bool,
) -> RefScan<'b> {
let cur_ix = match cur {
None => return RefScan::Failed,
Some(cur_ix) => cur_ix,
};
let start = tree[cur_ix].item.start;
let tail = &text.as_bytes()[start..];
if tail.starts_with(b"[]") {
let closing_node = tree[cur_ix].next.unwrap();
RefScan::Collapsed(tree[closing_node].next)
} else if let Some((ix, ReferenceLabel::Link(label))) =
scan_link_label(tree, &text[start..], allow_footnote_refs)
{
RefScan::LinkLabel(label, start + ix)
} else {
RefScan::Failed
}
}
#[derive(Clone, Default)]
struct LinkStack {
inner: Vec<LinkStackEl>,
disabled_ix: usize,
}
impl LinkStack {
fn push(&mut self, el: LinkStackEl) {
self.inner.push(el);
}
fn pop(&mut self) -> Option<LinkStackEl> {
let el = self.inner.pop();
self.disabled_ix = std::cmp::min(self.disabled_ix, self.inner.len());
el
}
fn clear(&mut self) {
self.inner.clear();
self.disabled_ix = 0;
}
fn disable_all_links(&mut self) {
for el in &mut self.inner[self.disabled_ix..] {
if el.ty == LinkStackTy::Link {
el.ty = LinkStackTy::Disabled;
}
}
self.disabled_ix = self.inner.len();
}
}
#[derive(Clone, Debug)]
struct LinkStackEl {
node: TreeIndex,
ty: LinkStackTy,
}
#[derive(PartialEq, Clone, Debug)]
enum LinkStackTy {
Link,
Image,
Disabled,
}
#[derive(Clone)]
struct LinkDef<'a> {
dest: CowStr<'a>,
title: Option<CowStr<'a>>,
}
struct CodeDelims {
inner: HashMap<usize, VecDeque<TreeIndex>>,
seen_first: bool,
}
impl CodeDelims {
fn new() -> Self {
Self {
inner: Default::default(),
seen_first: false,
}
}
fn insert(&mut self, count: usize, ix: TreeIndex) {
if self.seen_first {
self.inner
.entry(count)
.or_insert_with(Default::default)
.push_back(ix);
} else {
self.seen_first = true;
}
}
fn is_populated(&self) -> bool {
!self.inner.is_empty()
}
fn find(&mut self, open_ix: TreeIndex, count: usize) -> Option<TreeIndex> {
while let Some(ix) = self.inner.get_mut(&count)?.pop_front() {
if ix > open_ix {
return Some(ix);
}
}
None
}
fn clear(&mut self) {
self.inner.clear();
self.seen_first = false;
}
}
#[derive(Copy, Clone, PartialEq, Eq, Debug)]
struct LinkIndex(usize);
#[derive(Copy, Clone, PartialEq, Eq, Debug)]
struct CowIndex(usize);
#[derive(Copy, Clone, PartialEq, Eq, Debug)]
struct AlignmentIndex(usize);
#[derive(Clone)]
struct Allocations<'a> {
refdefs: HashMap<LinkLabel<'a>, LinkDef<'a>>,
links: Vec<(LinkType, CowStr<'a>, CowStr<'a>)>,
cows: Vec<CowStr<'a>>,
alignments: Vec<Vec<Alignment>>,
}
impl<'a> Allocations<'a> {
fn new() -> Self {
Self {
refdefs: HashMap::new(),
links: Vec::with_capacity(128),
cows: Vec::new(),
alignments: Vec::new(),
}
}
fn allocate_cow(&mut self, cow: CowStr<'a>) -> CowIndex {
let ix = self.cows.len();
self.cows.push(cow);
CowIndex(ix)
}
fn allocate_link(&mut self, ty: LinkType, url: CowStr<'a>, title: CowStr<'a>) -> LinkIndex {
let ix = self.links.len();
self.links.push((ty, url, title));
LinkIndex(ix)
}
fn allocate_alignment(&mut self, alignment: Vec<Alignment>) -> AlignmentIndex {
let ix = self.alignments.len();
self.alignments.push(alignment);
AlignmentIndex(ix)
}
}
impl<'a> Index<CowIndex> for Allocations<'a> {
type Output = CowStr<'a>;
fn index(&self, ix: CowIndex) -> &Self::Output {
self.cows.index(ix.0)
}
}
impl<'a> Index<LinkIndex> for Allocations<'a> {
type Output = (LinkType, CowStr<'a>, CowStr<'a>);
fn index(&self, ix: LinkIndex) -> &Self::Output {
self.links.index(ix.0)
}
}
impl<'a> Index<AlignmentIndex> for Allocations<'a> {
type Output = Vec<Alignment>;
fn index(&self, ix: AlignmentIndex) -> &Self::Output {
self.alignments.index(ix.0)
}
}
#[derive(Clone, Default)]
pub(crate) struct HtmlScanGuard {
pub cdata: usize,
pub processing: usize,
pub declaration: usize,
}
fn special_bytes(options: &Options) -> [bool; 256] {
let mut bytes = [false; 256];
let standard_bytes = [
b'\n', b'\r', b'*', b'_', b'&', b'\\', b'[', b']', b'<', b'!', b'`',
];
for &byte in &standard_bytes {
bytes[byte as usize] = true;
}
if options.contains(Options::ENABLE_TABLES) {
bytes[b'|' as usize] = true;
}
if options.contains(Options::ENABLE_STRIKETHROUGH) {
bytes[b'~' as usize] = true;
}
if options.contains(Options::ENABLE_SMART_PUNCTUATION) {
for &byte in &[b'.', b'-', b'"', b'\''] {
bytes[byte as usize] = true;
}
}
bytes
}
pub(crate) fn create_lut(options: &Options) -> LookupTable {
#[cfg(all(target_arch = "x86_64", feature = "simd"))]
{
LookupTable {
simd: crate::simd::compute_lookup(options),
scalar: special_bytes(options),
}
}
#[cfg(not(all(target_arch = "x86_64", feature = "simd")))]
{
special_bytes(options)
}
}
pub type BrokenLinkCallback<'a> =
Option<&'a mut dyn FnMut(BrokenLink) -> Option<(CowStr<'a>, CowStr<'a>)>>;
pub struct Parser<'a> {
text: &'a str,
options: Options,
tree: Tree<Item>,
allocs: Allocations<'a>,
broken_link_callback: BrokenLinkCallback<'a>,
html_scan_guard: HtmlScanGuard,
inline_stack: InlineStack,
link_stack: LinkStack,
}
impl<'a> Parser<'a> {
pub fn new(text: &'a str) -> Parser<'a> {
Parser::new_ext(text, Options::empty())
}
pub fn new_ext(text: &'a str, options: Options) -> Parser<'a> {
Parser::new_with_broken_link_callback(text, options, None)
}
pub fn new_with_broken_link_callback(
text: &'a str,
options: Options,
broken_link_callback: BrokenLinkCallback<'a>,
) -> Parser<'a> {
let lut = create_lut(&options);
let first_pass = FirstPass::new(text, options, &lut);
let (mut tree, allocs) = first_pass.run();
tree.reset();
let inline_stack = Default::default();
let link_stack = Default::default();
let html_scan_guard = Default::default();
Parser {
text,
options,
tree,
allocs,
broken_link_callback,
inline_stack,
link_stack,
html_scan_guard,
}
}
fn handle_inline(&mut self) {
self.handle_inline_pass1();
self.handle_emphasis();
}
fn handle_inline_pass1(&mut self) {
let mut code_delims = CodeDelims::new();
let mut cur = self.tree.cur();
let mut prev = None;
let block_end = self.tree[self.tree.peek_up().unwrap()].item.end;
let block_text = &self.text[..block_end];
while let Some(mut cur_ix) = cur {
match self.tree[cur_ix].item.body {
ItemBody::MaybeHtml => {
let next = self.tree[cur_ix].next;
let autolink = if let Some(next_ix) = next {
scan_autolink(block_text, self.tree[next_ix].item.start)
} else {
None
};
if let Some((ix, uri, link_type)) = autolink {
let node = scan_nodes_to_ix(&self.tree, next, ix);
let text_node = self.tree.create_node(Item {
start: self.tree[cur_ix].item.start + 1,
end: ix - 1,
body: ItemBody::Text,
});
let link_ix = self.allocs.allocate_link(link_type, uri, "".into());
self.tree[cur_ix].item.body = ItemBody::Link(link_ix);
self.tree[cur_ix].item.end = ix;
self.tree[cur_ix].next = node;
self.tree[cur_ix].child = Some(text_node);
prev = cur;
cur = node;
if let Some(node_ix) = cur {
self.tree[node_ix].item.start = max(self.tree[node_ix].item.start, ix);
}
continue;
} else {
let inline_html = next.and_then(|next_ix| {
self.scan_inline_html(
block_text.as_bytes(),
self.tree[next_ix].item.start,
)
});
if let Some((span, ix)) = inline_html {
let node = scan_nodes_to_ix(&self.tree, next, ix);
self.tree[cur_ix].item.body = if !span.is_empty() {
let converted_string =
String::from_utf8(span).expect("invalid utf8");
ItemBody::OwnedHtml(
self.allocs.allocate_cow(converted_string.into()),
)
} else {
ItemBody::Html
};
self.tree[cur_ix].item.end = ix;
self.tree[cur_ix].next = node;
prev = cur;
cur = node;
if let Some(node_ix) = cur {
self.tree[node_ix].item.start =
max(self.tree[node_ix].item.start, ix);
}
continue;
}
}
self.tree[cur_ix].item.body = ItemBody::Text;
}
ItemBody::MaybeCode(mut search_count, preceded_by_backslash) => {
if preceded_by_backslash {
search_count -= 1;
if search_count == 0 {
self.tree[cur_ix].item.body = ItemBody::Text;
prev = cur;
cur = self.tree[cur_ix].next;
continue;
}
}
if code_delims.is_populated() {
if let Some(scan_ix) = code_delims.find(cur_ix, search_count) {
self.make_code_span(cur_ix, scan_ix, preceded_by_backslash);
} else {
self.tree[cur_ix].item.body = ItemBody::Text;
}
} else {
let mut scan = if search_count > 0 {
self.tree[cur_ix].next
} else {
None
};
while let Some(scan_ix) = scan {
if let ItemBody::MaybeCode(delim_count, _) =
self.tree[scan_ix].item.body
{
if search_count == delim_count {
self.make_code_span(cur_ix, scan_ix, preceded_by_backslash);
code_delims.clear();
break;
} else {
code_delims.insert(delim_count, scan_ix);
}
}
scan = self.tree[scan_ix].next;
}
if scan == None {
self.tree[cur_ix].item.body = ItemBody::Text;
}
}
}
ItemBody::MaybeLinkOpen => {
self.tree[cur_ix].item.body = ItemBody::Text;
self.link_stack.push(LinkStackEl {
node: cur_ix,
ty: LinkStackTy::Link,
});
}
ItemBody::MaybeImage => {
self.tree[cur_ix].item.body = ItemBody::Text;
self.link_stack.push(LinkStackEl {
node: cur_ix,
ty: LinkStackTy::Image,
});
}
ItemBody::MaybeLinkClose(could_be_ref) => {
self.tree[cur_ix].item.body = ItemBody::Text;
if let Some(tos) = self.link_stack.pop() {
if tos.ty == LinkStackTy::Disabled {
continue;
}
let next = self.tree[cur_ix].next;
if let Some((next_ix, url, title)) =
self.scan_inline_link(block_text, self.tree[cur_ix].item.end, next)
{
let next_node = scan_nodes_to_ix(&self.tree, next, next_ix);
if let Some(prev_ix) = prev {
self.tree[prev_ix].next = None;
}
cur = Some(tos.node);
cur_ix = tos.node;
let link_ix = self.allocs.allocate_link(LinkType::Inline, url, title);
self.tree[cur_ix].item.body = if tos.ty == LinkStackTy::Image {
ItemBody::Image(link_ix)
} else {
ItemBody::Link(link_ix)
};
self.tree[cur_ix].child = self.tree[cur_ix].next;
self.tree[cur_ix].next = next_node;
self.tree[cur_ix].item.end = next_ix;
if let Some(next_node_ix) = next_node {
self.tree[next_node_ix].item.start =
max(self.tree[next_node_ix].item.start, next_ix);
}
if tos.ty == LinkStackTy::Link {
self.link_stack.disable_all_links();
}
} else {
let scan_result = scan_reference(
&self.tree,
block_text,
next,
self.options.contains(Options::ENABLE_FOOTNOTES),
);
let (node_after_link, link_type) = match scan_result {
RefScan::LinkLabel(_, end_ix) => {
let reference_close_node =
scan_nodes_to_ix(&self.tree, next, end_ix - 1).unwrap();
self.tree[reference_close_node].item.body =
ItemBody::MaybeLinkClose(false);
let next_node = self.tree[reference_close_node].next;
(next_node, LinkType::Reference)
}
RefScan::Collapsed(next_node) => {
if !could_be_ref {
continue;
}
(next_node, LinkType::Collapsed)
}
RefScan::Failed => {
if !could_be_ref {
continue;
}
(next, LinkType::Shortcut)
}
};
let label: Option<(ReferenceLabel<'a>, usize)> = match scan_result {
RefScan::LinkLabel(l, end_ix) => {
Some((ReferenceLabel::Link(l), end_ix))
}
RefScan::Collapsed(..) | RefScan::Failed => {
let label_start = self.tree[tos.node].item.end - 1;
scan_link_label(
&self.tree,
&self.text[label_start..self.tree[cur_ix].item.end],
self.options.contains(Options::ENABLE_FOOTNOTES),
)
.map(|(ix, label)| (label, label_start + ix))
}
};
if let Some((ReferenceLabel::Footnote(l), end)) = label {
self.tree[tos.node].next = node_after_link;
self.tree[tos.node].child = None;
self.tree[tos.node].item.body =
ItemBody::FootnoteReference(self.allocs.allocate_cow(l));
self.tree[tos.node].item.end = end;
prev = Some(tos.node);
cur = node_after_link;
self.link_stack.clear();
continue;
} else if let Some((ReferenceLabel::Link(link_label), end)) = label {
let type_url_title = self
.allocs
.refdefs
.get(&UniCase::new(link_label.as_ref().into()))
.map(|matching_def| {
let title = matching_def
.title
.as_ref()
.cloned()
.unwrap_or_else(|| "".into());
let url = matching_def.dest.clone();
(link_type, url, title)
})
.or_else(|| {
match self.broken_link_callback.as_mut() {
Some(callback) => {
let broken_link = BrokenLink {
span: (self.tree[tos.node].item.start)..end,
link_type: link_type,
reference: link_label.as_ref(),
};
callback(broken_link).map(|(url, title)| {
(link_type.to_unknown(), url, title)
})
}
None => None,
}
});
if let Some((def_link_type, url, title)) = type_url_title {
let link_ix =
self.allocs.allocate_link(def_link_type, url, title);
self.tree[tos.node].item.body = if tos.ty == LinkStackTy::Image
{
ItemBody::Image(link_ix)
} else {
ItemBody::Link(link_ix)
};
let label_node = self.tree[tos.node].next;
self.tree[tos.node].next = node_after_link;
if label_node != cur {
self.tree[tos.node].child = label_node;
if let Some(prev_ix) = prev {
self.tree[prev_ix].next = None;
}
}
self.tree[tos.node].item.end = end;
cur = Some(tos.node);
cur_ix = tos.node;
if tos.ty == LinkStackTy::Link {
self.link_stack.disable_all_links();
}
}
}
}
}
}
_ => (),
}
prev = cur;
cur = self.tree[cur_ix].next;
}
self.link_stack.clear();
}
fn handle_emphasis(&mut self) {
let mut prev = None;
let mut prev_ix: TreeIndex;
let mut cur = self.tree.cur();
let mut single_quote_open: Option<TreeIndex> = None;
let mut double_quote_open: bool = false;
while let Some(mut cur_ix) = cur {
match self.tree[cur_ix].item.body {
ItemBody::MaybeEmphasis(mut count, can_open, can_close) => {
let c = self.text.as_bytes()[self.tree[cur_ix].item.start];
let both = can_open && can_close;
if can_close {
while let Some(el) =
self.inline_stack.find_match(&mut self.tree, c, count, both)
{
if let Some(prev_ix) = prev {
self.tree[prev_ix].next = None;
}
let match_count = min(count, el.count);
let mut end = cur_ix - 1;
let mut start = el.start + el.count;
while start > el.start + el.count - match_count {
let (inc, ty) = if c == b'~' {
(2, ItemBody::Strikethrough)
} else if start > el.start + el.count - match_count + 1 {
(2, ItemBody::Strong)
} else {
(1, ItemBody::Emphasis)
};
let root = start - inc;
end = end + inc;
self.tree[root].item.body = ty;
self.tree[root].item.end = self.tree[end].item.end;
self.tree[root].child = Some(start);
self.tree[root].next = None;
start = root;
}
prev_ix = el.start + el.count - match_count;
prev = Some(prev_ix);
cur = self.tree[cur_ix + match_count - 1].next;
self.tree[prev_ix].next = cur;
if el.count > match_count {
self.inline_stack.push(InlineEl {
start: el.start,
count: el.count - match_count,
c: el.c,
both,
})
}
count -= match_count;
if count > 0 {
cur_ix = cur.unwrap();
} else {
break;
}
}
}
if count > 0 {
if can_open {
self.inline_stack.push(InlineEl {
start: cur_ix,
count,
c,
both,
});
} else {
for i in 0..count {
self.tree[cur_ix + i].item.body = ItemBody::Text;
}
}
prev_ix = cur_ix + count - 1;
prev = Some(prev_ix);
cur = self.tree[prev_ix].next;
}
}
ItemBody::MaybeSmartQuote(c, can_open, can_close) => {
self.tree[cur_ix].item.body = match c {
b'\'' => {
if let (Some(open_ix), true) = (single_quote_open, can_close) {
self.tree[open_ix].item.body = ItemBody::SynthesizeChar('‘');
single_quote_open = None;
} else if can_open {
single_quote_open = Some(cur_ix);
}
ItemBody::SynthesizeChar('’')
}
_ => {
if can_close && double_quote_open {
double_quote_open = false;
ItemBody::SynthesizeChar('”')
} else {
if can_open && !double_quote_open {
double_quote_open = true;
}
ItemBody::SynthesizeChar('“')
}
}
};
prev = cur;
cur = self.tree[cur_ix].next;
}
_ => {
prev = cur;
cur = self.tree[cur_ix].next;
}
}
}
self.inline_stack.pop_all(&mut self.tree);
}
fn scan_inline_link(
&self,
underlying: &'a str,
mut ix: usize,
node: Option<TreeIndex>,
) -> Option<(usize, CowStr<'a>, CowStr<'a>)> {
if scan_ch(&underlying.as_bytes()[ix..], b'(') == 0 {
return None;
}
ix += 1;
ix += scan_while(&underlying.as_bytes()[ix..], is_ascii_whitespace);
let (dest_length, dest) = scan_link_dest(underlying, ix, LINK_MAX_NESTED_PARENS)?;
let dest = unescape(dest);
ix += dest_length;
ix += scan_while(&underlying.as_bytes()[ix..], is_ascii_whitespace);
let title = if let Some((bytes_scanned, t)) = self.scan_link_title(underlying, ix, node) {
ix += bytes_scanned;
ix += scan_while(&underlying.as_bytes()[ix..], is_ascii_whitespace);
t
} else {
"".into()
};
if scan_ch(&underlying.as_bytes()[ix..], b')') == 0 {
return None;
}
ix += 1;
Some((ix, dest, title))
}
fn scan_link_title(
&self,
text: &'a str,
start_ix: usize,
node: Option<TreeIndex>,
) -> Option<(usize, CowStr<'a>)> {
let bytes = text.as_bytes();
let open = match bytes.get(start_ix) {
Some(b @ b'\'') | Some(b @ b'\"') | Some(b @ b'(') => *b,
_ => return None,
};
let close = if open == b'(' { b')' } else { open };
let mut title = String::new();
let mut mark = start_ix + 1;
let mut i = start_ix + 1;
while i < bytes.len() {
let c = bytes[i];
if c == close {
let cow = if mark == 1 {
(i - start_ix + 1, text[mark..i].into())
} else {
title.push_str(&text[mark..i]);
(i - start_ix + 1, title.into())
};
return Some(cow);
}
if c == open {
return None;
}
if c == b'\n' || c == b'\r' {
if let Some(node_ix) = scan_nodes_to_ix(&self.tree, node, i + 1) {
if self.tree[node_ix].item.start > i {
title.push_str(&text[mark..i]);
title.push('\n');
i = self.tree[node_ix].item.start;
mark = i;
continue;
}
}
}
if c == b'&' {
if let (n, Some(value)) = scan_entity(&bytes[i..]) {
title.push_str(&text[mark..i]);
title.push_str(&value);
i += n;
mark = i;
continue;
}
}
if c == b'\\' && i + 1 < bytes.len() && is_ascii_punctuation(bytes[i + 1]) {
title.push_str(&text[mark..i]);
i += 1;
mark = i;
}
i += 1;
}
None
}
fn make_code_span(&mut self, open: TreeIndex, close: TreeIndex, preceding_backslash: bool) {
let first_ix = open + 1;
let last_ix = close - 1;
let bytes = self.text.as_bytes();
let mut span_start = self.tree[open].item.end;
let mut span_end = self.tree[close].item.start;
let mut buf: Option<String> = None;
if !bytes[span_start..span_end].iter().all(|&b| b == b' ') {
let opening = match bytes[span_start] {
b' ' | b'\r' | b'\n' => true,
_ => false,
};
let closing = match bytes[span_end - 1] {
b' ' | b'\r' | b'\n' => true,
_ => false,
};
let drop_enclosing_whitespace = opening && closing;
if drop_enclosing_whitespace {
span_start += 1;
if span_start < span_end {
span_end -= 1;
}
}
let mut ix = first_ix;
while ix < close {
if let ItemBody::HardBreak | ItemBody::SoftBreak = self.tree[ix].item.body {
if drop_enclosing_whitespace {
if ix == first_ix {
ix = ix + 1;
span_start = min(span_end, self.tree[ix].item.start);
continue;
} else if ix == last_ix && last_ix > first_ix {
ix = ix + 1;
continue;
}
}
let end = bytes[self.tree[ix].item.start..]
.iter()
.position(|&b| b == b'\r' || b == b'\n')
.unwrap()
+ self.tree[ix].item.start;
if let Some(ref mut buf) = buf {
buf.push_str(&self.text[self.tree[ix].item.start..end]);
buf.push(' ');
} else {
let mut new_buf = String::with_capacity(span_end - span_start);
new_buf.push_str(&self.text[span_start..end]);
new_buf.push(' ');
buf = Some(new_buf);
}
} else if let Some(ref mut buf) = buf {
let end = if ix == last_ix {
span_end
} else {
self.tree[ix].item.end
};
buf.push_str(&self.text[self.tree[ix].item.start..end]);
}
ix = ix + 1;
}
}
let cow = if let Some(buf) = buf {
buf.into()
} else {
self.text[span_start..span_end].into()
};
if preceding_backslash {
self.tree[open].item.body = ItemBody::Text;
self.tree[open].item.end = self.tree[open].item.start + 1;
self.tree[open].next = Some(close);
self.tree[close].item.body = ItemBody::Code(self.allocs.allocate_cow(cow));
self.tree[close].item.start = self.tree[open].item.start + 1;
} else {
self.tree[open].item.body = ItemBody::Code(self.allocs.allocate_cow(cow));
self.tree[open].item.end = self.tree[close].item.end;
self.tree[open].next = self.tree[close].next;
}
}
fn scan_inline_html(&mut self, bytes: &[u8], ix: usize) -> Option<(Vec<u8>, usize)> {
let c = *bytes.get(ix)?;
if c == b'!' {
Some((
vec![],
scan_inline_html_comment(bytes, ix + 1, &mut self.html_scan_guard)?,
))
} else if c == b'?' {
Some((
vec![],
scan_inline_html_processing(bytes, ix + 1, &mut self.html_scan_guard)?,
))
} else {
let (span, i) = scan_html_block_inner(
&bytes[(ix - 1)..],
Some(&|_bytes| {
let mut line_start = LineStart::new(bytes);
let _ = scan_containers(&self.tree, &mut line_start);
line_start.bytes_scanned()
}),
)?;
Some((span, i + ix - 1))
}
}
pub fn into_offset_iter(self) -> OffsetIter<'a> {
OffsetIter { inner: self }
}
}
pub(crate) enum LoopInstruction<T> {
ContinueAndSkip(usize),
BreakAtWith(usize, T),
}
#[cfg(all(target_arch = "x86_64", feature = "simd"))]
pub(crate) struct LookupTable {
pub simd: [u8; 16],
pub scalar: [bool; 256],
}
#[cfg(not(all(target_arch = "x86_64", feature = "simd")))]
type LookupTable = [bool; 256];
fn iterate_special_bytes<F, T>(
lut: &LookupTable,
bytes: &[u8],
ix: usize,
callback: F,
) -> (usize, Option<T>)
where
F: FnMut(usize, u8) -> LoopInstruction<Option<T>>,
{
#[cfg(all(target_arch = "x86_64", feature = "simd"))]
{
crate::simd::iterate_special_bytes(lut, bytes, ix, callback)
}
#[cfg(not(all(target_arch = "x86_64", feature = "simd")))]
{
scalar_iterate_special_bytes(lut, bytes, ix, callback)
}
}
pub(crate) fn scalar_iterate_special_bytes<F, T>(
lut: &[bool; 256],
bytes: &[u8],
mut ix: usize,
mut callback: F,
) -> (usize, Option<T>)
where
F: FnMut(usize, u8) -> LoopInstruction<Option<T>>,
{
while ix < bytes.len() {
let b = bytes[ix];
if lut[b as usize] {
match callback(ix, b) {
LoopInstruction::ContinueAndSkip(skip) => {
ix += skip;
}
LoopInstruction::BreakAtWith(ix, val) => {
return (ix, val);
}
}
}
ix += 1;
}
(ix, None)
}
pub struct OffsetIter<'a> {
inner: Parser<'a>,
}
impl<'a> Iterator for OffsetIter<'a> {
type Item = (Event<'a>, Range<usize>);
fn next(&mut self) -> Option<Self::Item> {
match self.inner.tree.cur() {
None => {
let ix = self.inner.tree.pop()?;
let tag = item_to_tag(&self.inner.tree[ix].item, &self.inner.allocs);
self.inner.tree.next_sibling(ix);
Some((
Event::End(tag),
self.inner.tree[ix].item.start..self.inner.tree[ix].item.end,
))
}
Some(cur_ix) => {
if self.inner.tree[cur_ix].item.body.is_inline() {
self.inner.handle_inline();
}
let node = self.inner.tree[cur_ix];
let item = node.item;
let event = item_to_event(item, self.inner.text, &self.inner.allocs);
if let Event::Start(..) = event {
self.inner.tree.push();
} else {
self.inner.tree.next_sibling(cur_ix);
}
Some((event, item.start..item.end))
}
}
}
}
fn item_to_tag<'a>(item: &Item, allocs: &Allocations<'a>) -> Tag<'a> {
match item.body {
ItemBody::Paragraph => Tag::Paragraph,
ItemBody::Emphasis => Tag::Emphasis,
ItemBody::Strong => Tag::Strong,
ItemBody::Strikethrough => Tag::Strikethrough,
ItemBody::Link(link_ix) => {
let &(ref link_type, ref url, ref title) = allocs.index(link_ix);
Tag::Link(*link_type, url.clone(), title.clone())
}
ItemBody::Image(link_ix) => {
let &(ref link_type, ref url, ref title) = allocs.index(link_ix);
Tag::Image(*link_type, url.clone(), title.clone())
}
ItemBody::Heading(level) => Tag::Heading(level),
ItemBody::FencedCodeBlock(cow_ix) => {
Tag::CodeBlock(CodeBlockKind::Fenced(allocs[cow_ix].clone()))
}
ItemBody::IndentCodeBlock => Tag::CodeBlock(CodeBlockKind::Indented),
ItemBody::BlockQuote => Tag::BlockQuote,
ItemBody::List(_, c, listitem_start) => {
if c == b'.' || c == b')' {
Tag::List(Some(listitem_start))
} else {
Tag::List(None)
}
}
ItemBody::ListItem(_) => Tag::Item,
ItemBody::TableHead => Tag::TableHead,
ItemBody::TableCell => Tag::TableCell,
ItemBody::TableRow => Tag::TableRow,
ItemBody::Table(alignment_ix) => Tag::Table(allocs[alignment_ix].clone()),
ItemBody::FootnoteDefinition(cow_ix) => Tag::FootnoteDefinition(allocs[cow_ix].clone()),
_ => panic!("unexpected item body {:?}", item.body),
}
}
fn item_to_event<'a>(item: Item, text: &'a str, allocs: &Allocations<'a>) -> Event<'a> {
let tag = match item.body {
ItemBody::Text => return Event::Text(text[item.start..item.end].into()),
ItemBody::Code(cow_ix) => return Event::Code(allocs[cow_ix].clone()),
ItemBody::SynthesizeText(cow_ix) => return Event::Text(allocs[cow_ix].clone()),
ItemBody::SynthesizeChar(c) => return Event::Text(c.into()),
ItemBody::Html => return Event::Html(text[item.start..item.end].into()),
ItemBody::OwnedHtml(cow_ix) => return Event::Html(allocs[cow_ix].clone()),
ItemBody::SoftBreak => return Event::SoftBreak,
ItemBody::HardBreak => return Event::HardBreak,
ItemBody::FootnoteReference(cow_ix) => {
return Event::FootnoteReference(allocs[cow_ix].clone())
}
ItemBody::TaskListMarker(checked) => return Event::TaskListMarker(checked),
ItemBody::Rule => return Event::Rule,
ItemBody::Paragraph => Tag::Paragraph,
ItemBody::Emphasis => Tag::Emphasis,
ItemBody::Strong => Tag::Strong,
ItemBody::Strikethrough => Tag::Strikethrough,
ItemBody::Link(link_ix) => {
let &(ref link_type, ref url, ref title) = allocs.index(link_ix);
Tag::Link(*link_type, url.clone(), title.clone())
}
ItemBody::Image(link_ix) => {
let &(ref link_type, ref url, ref title) = allocs.index(link_ix);
Tag::Image(*link_type, url.clone(), title.clone())
}
ItemBody::Heading(level) => Tag::Heading(level),
ItemBody::FencedCodeBlock(cow_ix) => {
Tag::CodeBlock(CodeBlockKind::Fenced(allocs[cow_ix].clone()))
}
ItemBody::IndentCodeBlock => Tag::CodeBlock(CodeBlockKind::Indented),
ItemBody::BlockQuote => Tag::BlockQuote,
ItemBody::List(_, c, listitem_start) => {
if c == b'.' || c == b')' {
Tag::List(Some(listitem_start))
} else {
Tag::List(None)
}
}
ItemBody::ListItem(_) => Tag::Item,
ItemBody::TableHead => Tag::TableHead,
ItemBody::TableCell => Tag::TableCell,
ItemBody::TableRow => Tag::TableRow,
ItemBody::Table(alignment_ix) => Tag::Table(allocs[alignment_ix].clone()),
ItemBody::FootnoteDefinition(cow_ix) => Tag::FootnoteDefinition(allocs[cow_ix].clone()),
_ => panic!("unexpected item body {:?}", item.body),
};
Event::Start(tag)
}
fn surgerize_tight_list(tree: &mut Tree<Item>, list_ix: TreeIndex) {
let mut list_item = tree[list_ix].child;
while let Some(listitem_ix) = list_item {
let list_item_firstborn = tree[listitem_ix].child;
if let Some(firstborn_ix) = list_item_firstborn {
if let ItemBody::Paragraph = tree[firstborn_ix].item.body {
tree[listitem_ix].child = tree[firstborn_ix].child;
}
let mut list_item_child = Some(firstborn_ix);
let mut node_to_repoint = None;
while let Some(child_ix) = list_item_child {
let repoint_ix = if let ItemBody::Paragraph = tree[child_ix].item.body {
if let Some(child_firstborn) = tree[child_ix].child {
if let Some(repoint_ix) = node_to_repoint {
tree[repoint_ix].next = Some(child_firstborn);
}
let mut child_lastborn = child_firstborn;
while let Some(lastborn_next_ix) = tree[child_lastborn].next {
child_lastborn = lastborn_next_ix;
}
child_lastborn
} else {
child_ix
}
} else {
child_ix
};
node_to_repoint = Some(repoint_ix);
tree[repoint_ix].next = tree[child_ix].next;
list_item_child = tree[child_ix].next;
}
}
list_item = tree[listitem_ix].next;
}
}
impl<'a> Iterator for Parser<'a> {
type Item = Event<'a>;
fn next(&mut self) -> Option<Event<'a>> {
match self.tree.cur() {
None => {
let ix = self.tree.pop()?;
let tag = item_to_tag(&self.tree[ix].item, &self.allocs);
self.tree.next_sibling(ix);
Some(Event::End(tag))
}
Some(cur_ix) => {
if self.tree[cur_ix].item.body.is_inline() {
self.handle_inline();
}
let node = self.tree[cur_ix];
let item = node.item;
let event = item_to_event(item, self.text, &self.allocs);
if let Event::Start(..) = event {
self.tree.push();
} else {
self.tree.next_sibling(cur_ix);
}
Some(event)
}
}
}
}
#[cfg(test)]
mod test {
use super::*;
use crate::tree::Node;
fn parser_with_extensions(text: &str) -> Parser<'_> {
let mut opts = Options::empty();
opts.insert(Options::ENABLE_TABLES);
opts.insert(Options::ENABLE_FOOTNOTES);
opts.insert(Options::ENABLE_STRIKETHROUGH);
opts.insert(Options::ENABLE_TASKLISTS);
Parser::new_ext(text, opts)
}
#[test]
#[cfg(target_pointer_width = "64")]
fn node_size() {
let node_size = std::mem::size_of::<Node<Item>>();
assert_eq!(48, node_size);
}
#[test]
#[cfg(target_pointer_width = "64")]
fn body_size() {
let body_size = std::mem::size_of::<ItemBody>();
assert_eq!(16, body_size);
}
#[test]
fn single_open_fish_bracket() {
assert_eq!(3, Parser::new("<").count());
}
#[test]
fn lone_hashtag() {
assert_eq!(2, Parser::new("#").count());
}
#[test]
fn lots_of_backslashes() {
Parser::new("\\\\\r\r").count();
Parser::new("\\\r\r\\.\\\\\r\r\\.\\").count();
}
#[test]
fn issue_320() {
parser_with_extensions(":\r\t> |\r:\r\t> |\r").count();
}
#[test]
fn issue_319() {
parser_with_extensions("|\r-]([^|\r-]([^").count();
parser_with_extensions("|\r\r=][^|\r\r=][^car").count();
}
#[test]
fn issue_303() {
parser_with_extensions("[^\r\ra]").count();
parser_with_extensions("\r\r]Z[^\x00\r\r]Z[^\x00").count();
}
#[test]
fn issue_313() {
parser_with_extensions("*]0[^\r\r*]0[^").count();
parser_with_extensions("[^\r> `][^\r> `][^\r> `][").count();
}
#[test]
fn issue_311() {
parser_with_extensions("\\\u{0d}-\u{09}\\\u{0d}-\u{09}").count();
}
#[test]
fn issue_283() {
let input = std::str::from_utf8(b"\xf0\x9b\xb2\x9f<td:^\xf0\x9b\xb2\x9f").unwrap();
parser_with_extensions(input).count();
}
#[test]
fn issue_289() {
parser_with_extensions("> - \\\n> - ").count();
parser_with_extensions("- \n\n").count();
}
#[test]
fn issue_306() {
parser_with_extensions("*\r_<__*\r_<__*\r_<__*\r_<__").count();
}
#[test]
fn issue_305() {
parser_with_extensions("_6**6*_*").count();
}
#[test]
fn another_emphasis_panic() {
parser_with_extensions("*__#_#__*").count();
}
#[test]
fn offset_iter() {
let event_offsets: Vec<_> = Parser::new("*hello* world")
.into_offset_iter()
.map(|(_ev, range)| range)
.collect();
let expected_offsets = vec![(0..13), (0..7), (1..6), (0..7), (7..13), (0..13)];
assert_eq!(expected_offsets, event_offsets);
}
#[test]
fn reference_link_offsets() {
let range =
Parser::new("# H1\n[testing][Some reference]\n\n[Some reference]: https://github.com")
.into_offset_iter()
.filter_map(|(ev, range)| match ev {
Event::Start(Tag::Link(LinkType::Reference, ..), ..) => Some(range),
_ => None,
})
.next()
.unwrap();
assert_eq!(5..30, range);
}
#[test]
fn footnote_offsets() {
let range = parser_with_extensions("Testing this[^1] out.\n\n[^1]: Footnote.")
.into_offset_iter()
.filter_map(|(ev, range)| match ev {
Event::FootnoteReference(..) => Some(range),
_ => None,
})
.next()
.unwrap();
assert_eq!(12..16, range);
}
#[test]
fn table_offset() {
let markdown = "a\n\nTesting|This|Outtt\n--|:--:|--:\nSome Data|Other data|asdf";
let event_offset = parser_with_extensions(markdown)
.into_offset_iter()
.map(|(_ev, range)| range)
.nth(3)
.unwrap();
let expected_offset = 3..59;
assert_eq!(expected_offset, event_offset);
}
#[test]
fn offset_iter_issue_378() {
let event_offsets: Vec<_> = Parser::new("a [b](c) d")
.into_offset_iter()
.map(|(_ev, range)| range)
.collect();
let expected_offsets = vec![(0..10), (0..2), (2..8), (3..4), (2..8), (8..10), (0..10)];
assert_eq!(expected_offsets, event_offsets);
}
#[test]
fn offset_iter_issue_404() {
let event_offsets: Vec<_> = Parser::new("###\n")
.into_offset_iter()
.map(|(_ev, range)| range)
.collect();
let expected_offsets = vec![(0..4), (0..4)];
assert_eq!(expected_offsets, event_offsets);
}
#[test]
fn link_def_at_eof() {
let test_str = "[My site][world]\n\n[world]: https://vincentprouillet.com";
let expected = "<p><a href=\"https://vincentprouillet.com\">My site</a></p>\n";
let mut buf = String::new();
crate::html::push_html(&mut buf, Parser::new(test_str));
assert_eq!(expected, buf);
}
#[test]
fn no_footnote_refs_without_option() {
let test_str = "a [^a]\n\n[^a]: yolo";
let expected = "<p>a <a href=\"yolo\">^a</a></p>\n";
let mut buf = String::new();
crate::html::push_html(&mut buf, Parser::new(test_str));
assert_eq!(expected, buf);
}
#[test]
fn ref_def_at_eof() {
let test_str = "[test]:\\";
let expected = "";
let mut buf = String::new();
crate::html::push_html(&mut buf, Parser::new(test_str));
assert_eq!(expected, buf);
}
#[test]
fn ref_def_cr_lf() {
let test_str = "[a]: /u\r\n\n[a]";
let expected = "<p><a href=\"/u\">a</a></p>\n";
let mut buf = String::new();
crate::html::push_html(&mut buf, Parser::new(test_str));
assert_eq!(expected, buf);
}
#[test]
fn no_dest_refdef() {
let test_str = "[a]:";
let expected = "<p>[a]:</p>\n";
let mut buf = String::new();
crate::html::push_html(&mut buf, Parser::new(test_str));
assert_eq!(expected, buf);
}
#[test]
fn broken_links_called_only_once() {
for &(markdown, expected) in &[
("See also [`g()`][crate::g].", 1),
("See also [`g()`][crate::g][].", 1),
("[brokenlink1] some other node [brokenlink2]", 2),
] {
let mut times_called = 0;
let callback = &mut |_broken_link: BrokenLink| {
times_called += 1;
None
};
let parser =
Parser::new_with_broken_link_callback(markdown, Options::empty(), Some(callback));
for _ in parser {}
assert_eq!(times_called, expected);
}
}
#[test]
fn simple_broken_link_callback() {
let test_str = "This is a link w/o def: [hello][world]";
let mut callback = |broken_link: BrokenLink| {
assert_eq!("world", broken_link.reference);
assert_eq!(&test_str[broken_link.span], "[hello][world]");
let url = "YOLO".into();
let title = "SWAG".to_owned().into();
Some((url, title))
};
let parser =
Parser::new_with_broken_link_callback(test_str, Options::empty(), Some(&mut callback));
let mut link_tag_count = 0;
for (typ, url, title) in parser.filter_map(|event| match event {
Event::Start(tag) | Event::End(tag) => match tag {
Tag::Link(typ, url, title) => Some((typ, url, title)),
_ => None,
},
_ => None,
}) {
link_tag_count += 1;
assert_eq!(typ, LinkType::ReferenceUnknown);
assert_eq!(url.as_ref(), "YOLO");
assert_eq!(title.as_ref(), "SWAG");
}
assert!(link_tag_count > 0);
}
#[test]
fn code_block_kind_check_fenced() {
let parser = Parser::new("hello\n```test\ntadam\n```");
let mut found = 0;
for (ev, _range) in parser.into_offset_iter() {
match ev {
Event::Start(Tag::CodeBlock(CodeBlockKind::Fenced(syntax))) => {
assert_eq!(syntax.as_ref(), "test");
found += 1;
}
_ => {}
}
}
assert_eq!(found, 1);
}
#[test]
fn code_block_kind_check_indented() {
let parser = Parser::new("hello\n\n ```test\n tadam\nhello");
let mut found = 0;
for (ev, _range) in parser.into_offset_iter() {
match ev {
Event::Start(Tag::CodeBlock(CodeBlockKind::Indented)) => {
found += 1;
}
_ => {}
}
}
assert_eq!(found, 1);
}
}