// Copyright 2017 Google Inc. All rights reserved. // // Permission is hereby granted, free of charge, to any person obtaining a copy // of this software and associated documentation files (the "Software"), to deal // in the Software without restriction, including without limitation the rights // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell // copies of the Software, and to permit persons to whom the Software is // furnished to do so, subject to the following conditions: // // The above copyright notice and this permission notice shall be included in // all copies or substantial portions of the Software. // // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN // THE SOFTWARE. //! Tree-based two pass parser. use std::cmp::{max, min}; use std::collections::{HashMap, VecDeque}; use std::iter::FusedIterator; use std::num::NonZeroUsize; use std::ops::{Index, Range}; use unicase::UniCase; use crate::firstpass::run_first_pass; use crate::linklabel::{scan_link_label_rest, FootnoteLabel, LinkLabel, ReferenceLabel}; use crate::scanners::*; use crate::strings::CowStr; use crate::tree::{Tree, TreeIndex}; use crate::{ Alignment, BlockQuoteKind, CodeBlockKind, Event, HeadingLevel, LinkType, MetadataBlockKind, Options, Tag, TagEnd, }; // Allowing arbitrary depth nested parentheses inside link destinations // can create denial of service vulnerabilities if we're not careful. // The simplest countermeasure is to limit their depth, which is // explicitly allowed by the spec as long as the limit is at least 3: // https://spec.commonmark.org/0.29/#link-destination pub(crate) const LINK_MAX_NESTED_PARENS: usize = 32; #[derive(Debug, Default, Clone, Copy)] pub(crate) struct Item { pub start: usize, pub end: usize, pub body: ItemBody, } #[derive(Debug, PartialEq, Clone, Copy, Default)] pub(crate) enum ItemBody { Paragraph, Text { backslash_escaped: bool, }, SoftBreak, // true = is backlash HardBreak(bool), // These are possible inline items, need to be resolved in second pass. // repeats, can_open, can_close MaybeEmphasis(usize, bool, bool), // can_open, can_close, brace context MaybeMath(bool, bool, u8), // quote byte, can_open, can_close MaybeSmartQuote(u8, bool, bool), MaybeCode(usize, bool), // number of backticks, preceded by backslash MaybeHtml, MaybeLinkOpen, // bool indicates whether or not the preceding section could be a reference MaybeLinkClose(bool), MaybeImage, // These are inline items after resolution. Emphasis, Strong, Strikethrough, Math(CowIndex, bool), // true for display math Code(CowIndex), Link(LinkIndex), Image(LinkIndex), FootnoteReference(CowIndex), TaskListMarker(bool), // true for checked Rule, Heading(HeadingLevel, Option), // heading level FencedCodeBlock(CowIndex), IndentCodeBlock, HtmlBlock, InlineHtml, Html, OwnedHtml(CowIndex), BlockQuote(Option), List(bool, u8, u64), // is_tight, list character, list start index ListItem(usize), // indent level SynthesizeText(CowIndex), SynthesizeChar(char), FootnoteDefinition(CowIndex), MetadataBlock(MetadataBlockKind), // Tables Table(AlignmentIndex), TableHead, TableRow, TableCell, // Dummy node at the top of the tree - should not be used otherwise! #[default] Root, } impl ItemBody { fn is_inline(&self) -> bool { matches!( *self, ItemBody::MaybeEmphasis(..) | ItemBody::MaybeMath(..) | ItemBody::MaybeSmartQuote(..) | ItemBody::MaybeHtml | ItemBody::MaybeCode(..) | ItemBody::MaybeLinkOpen | ItemBody::MaybeLinkClose(..) | ItemBody::MaybeImage ) } fn is_block(&self) -> bool { matches!( *self, ItemBody::Paragraph | ItemBody::BlockQuote(..) | ItemBody::List(..) | ItemBody::ListItem(..) | ItemBody::HtmlBlock | ItemBody::Table(..) | ItemBody::TableHead | ItemBody::TableRow | ItemBody::TableCell | ItemBody::Heading(..) | ItemBody::Rule ) } } #[derive(Debug)] pub struct BrokenLink<'a> { pub span: std::ops::Range, pub link_type: LinkType, pub reference: CowStr<'a>, } /// Markdown event iterator. pub struct Parser<'input, F = DefaultBrokenLinkCallback> { text: &'input str, options: Options, tree: Tree, allocs: Allocations<'input>, broken_link_callback: Option, html_scan_guard: HtmlScanGuard, // https://github.com/pulldown-cmark/pulldown-cmark/issues/844 // Consider this example: // // [x]: xxx... // [x] // [x] // [x] // // Which expands to this HTML: // // x // x // x // // This is quadratic growth, because it's filling in the area of a square. // To prevent this, track how much it's expanded and limit it. link_ref_expansion_limit: usize, // used by inline passes. store them here for reuse inline_stack: InlineStack, link_stack: LinkStack, } impl<'input, F> std::fmt::Debug for Parser<'input, F> { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { // Only print the fields that have public types. f.debug_struct("Parser") .field("text", &self.text) .field("options", &self.options) .field( "broken_link_callback", &self.broken_link_callback.as_ref().map(|_| ..), ) .finish() } } impl<'a> BrokenLink<'a> { /// Moves the link into version with a static lifetime. /// /// The `reference` member is cloned to a Boxed or Inline version. pub fn into_static(self) -> BrokenLink<'static> { BrokenLink { span: self.span.clone(), link_type: self.link_type, reference: self.reference.into_string().into(), } } } impl<'input> Parser<'input, DefaultBrokenLinkCallback> { /// Creates a new event iterator for a markdown string without any options enabled. pub fn new(text: &'input str) -> Self { Self::new_ext(text, Options::empty()) } /// Creates a new event iterator for a markdown string with given options. pub fn new_ext(text: &'input str, options: Options) -> Self { Self::new_with_broken_link_callback(text, options, None) } } impl<'input, F: BrokenLinkCallback<'input>> Parser<'input, F> { /// In case the parser encounters any potential links that have a broken /// reference (e.g `[foo]` when there is no `[foo]: ` entry at the bottom) /// the provided callback will be called with the reference name, /// and the returned pair will be used as the link URL and title if it is not /// `None`. pub fn new_with_broken_link_callback( text: &'input str, options: Options, broken_link_callback: Option, ) -> Self { let (mut tree, allocs) = run_first_pass(text, options); 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, // always allow 100KiB link_ref_expansion_limit: text.len().max(100_000), } } /// Returns a reference to the internal `RefDefs` object, which provides access /// to the internal map of reference definitions. pub fn reference_definitions(&self) -> &RefDefs<'_> { &self.allocs.refdefs } /// Use a link label to fetch a type, url, and title. /// /// This function enforces the [`link_ref_expansion_limit`]. /// If it returns Some, it also consumes some of the fuel. /// If we're out of fuel, it immediately returns None. /// /// The URL and title are found in the [`RefDefs`] map. /// If they're not there, and a callback was provided by the user, /// the [`broken_link_callback`] will be invoked and given the opportunity /// to provide a fallback. /// /// The link type (that's "link" or "image") depends on the usage site, and /// is provided by the caller of this function. /// This function returns a new one because, if it has to invoke a callback /// to find the information, the link type is [mapped to an unknown type]. /// /// [mapped to an unknown type]: crate::LinkType::to_unknown /// [`link_ref_expansion_limit`]: Self::link_ref_expansion_limit /// [`broken_link_callback`]: Self::broken_link_callback fn fetch_link_type_url_title( &mut self, link_label: CowStr<'input>, span: Range, link_type: LinkType, ) -> Option<(LinkType, CowStr<'input>, CowStr<'input>)> { if self.link_ref_expansion_limit == 0 { return None; } let (link_type, url, title) = self .allocs .refdefs .get(link_label.as_ref()) .map(|matching_def| { // found a matching definition! 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) => { // Construct a BrokenLink struct, which will be passed to the callback let broken_link = BrokenLink { span, link_type, reference: link_label, }; callback .handle_broken_link(broken_link) .map(|(url, title)| (link_type.to_unknown(), url, title)) } None => None, } })?; // Limit expansion from link references. // This isn't a problem for footnotes, because multiple references to the same one // reuse the same node, but links/images get their HREF/SRC copied. self.link_ref_expansion_limit = self .link_ref_expansion_limit .saturating_sub(url.len() + title.len()); Some((link_type, url, title)) } /// Handle inline markup. /// /// When the parser encounters any item indicating potential inline markup, all /// inline markup passes are run on the remainder of the chain. /// /// Note: there's some potential for optimization here, but that's future work. fn handle_inline(&mut self) { self.handle_inline_pass1(); self.handle_emphasis_and_hard_break(); } /// Handle inline HTML, code spans, and links. /// /// This function handles both inline HTML and code spans, because they have /// the same precedence. It also handles links, even though they have lower /// precedence, because the URL of links must not be processed. fn handle_inline_pass1(&mut self) { let mut code_delims = CodeDelims::new(); let mut math_delims = MathDelims::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 { backslash_escaped: false, }, }); let link_ix = self.allocs .allocate_link(link_type, uri, "".into(), "".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::InlineHtml }; 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 { backslash_escaped: false, }; } ItemBody::MaybeMath(can_open, _can_close, brace_context) => { if !can_open { self.tree[cur_ix].item.body = ItemBody::Text { backslash_escaped: false, }; prev = cur; cur = self.tree[cur_ix].next; continue; } let is_display = self.tree[cur_ix].next.map_or(false, |next_ix| { matches!( self.tree[next_ix].item.body, ItemBody::MaybeMath(_can_open, _can_close, _brace_context) ) }); let result = if math_delims.is_populated() { // we have previously scanned all math environment delimiters, // so we can reuse that work math_delims.find(&self.tree, cur_ix, is_display, brace_context) } else { // we haven't previously scanned all math delimiters, // so walk the AST let mut scan = self.tree[cur_ix].next; if is_display { // a display delimiter, `$$`, is actually two delimiters // skip the second one scan = self.tree[scan.unwrap()].next; } let mut invalid = false; while let Some(scan_ix) = scan { if let ItemBody::MaybeMath(_can_open, can_close, delim_brace_context) = self.tree[scan_ix].item.body { let delim_is_display = self.tree[scan_ix].next.map_or(false, |next_ix| { matches!( self.tree[next_ix].item.body, ItemBody::MaybeMath( _can_open, _can_close, _brace_context ) ) }); if !invalid && delim_brace_context == brace_context { if (!is_display && can_close) || (is_display && delim_is_display) { // This will skip ahead past everything we // just inserted. Needed for correctness to // ensure that a new scan is done after this item. math_delims.clear(); break; } else { // Math cannot contain $, so the current item // is invalid. Keep scanning to fill math_delims. invalid = true; } } math_delims.insert( delim_is_display, delim_brace_context, scan_ix, can_close, ); } if self.tree[scan_ix].item.body.is_block() { // If this is a tight list, blocks and inlines might be // siblings. Inlines can't cross the boundary like that. scan = None; break; } scan = self.tree[scan_ix].next; } scan }; if let Some(scan_ix) = result { self.make_math_span(cur_ix, scan_ix); } else { self.tree[cur_ix].item.body = ItemBody::Text { backslash_escaped: false, }; } } 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 { backslash_escaped: false, }; prev = cur; cur = self.tree[cur_ix].next; continue; } } if code_delims.is_populated() { // we have previously scanned all codeblock delimiters, // so we can reuse that work 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 { backslash_escaped: false, }; } } else { // we haven't previously scanned all codeblock delimiters, // so walk the AST 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); } } if self.tree[scan_ix].item.body.is_block() { // If this is a tight list, blocks and inlines might be // siblings. Inlines can't cross the boundary like that. scan = None; break; } scan = self.tree[scan_ix].next; } if scan.is_none() { self.tree[cur_ix].item.body = ItemBody::Text { backslash_escaped: false, }; } } } ItemBody::MaybeLinkOpen => { self.tree[cur_ix].item.body = ItemBody::Text { backslash_escaped: false, }; self.link_stack.push(LinkStackEl { node: cur_ix, ty: LinkStackTy::Link, }); } ItemBody::MaybeImage => { self.tree[cur_ix].item.body = ItemBody::Text { backslash_escaped: false, }; self.link_stack.push(LinkStackEl { node: cur_ix, ty: LinkStackTy::Image, }); } ItemBody::MaybeLinkClose(could_be_ref) => { self.tree[cur_ix].item.body = ItemBody::Text { backslash_escaped: false, }; 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, "".into()); 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 { // ok, so its not an inline link. maybe it is a reference // to a defined link? let scan_result = scan_reference( &self.tree, block_text, next, self.options.contains(Options::ENABLE_FOOTNOTES), self.options.has_gfm_footnotes(), ); let (node_after_link, link_type) = match scan_result { // [label][reference] RefScan::LinkLabel(_, end_ix) => { // Toggle reference viability of the last closing bracket, // so that we can skip it on future iterations in case // it fails in this one. In particular, we won't call // the broken link callback twice on one reference. let reference_close_node = if let Some(node) = scan_nodes_to_ix(&self.tree, next, end_ix - 1) { node } else { continue; }; self.tree[reference_close_node].item.body = ItemBody::MaybeLinkClose(false); let next_node = self.tree[reference_close_node].next; (next_node, LinkType::Reference) } // [reference][] RefScan::Collapsed(next_node) => { // This reference has already been tried, and it's not // valid. Skip it. if !could_be_ref { continue; } (next_node, LinkType::Collapsed) } // [shortcut] // // [shortcut]: /blah RefScan::Failed | RefScan::UnexpectedFootnote => { if !could_be_ref { continue; } (next, LinkType::Shortcut) } }; // FIXME: references and labels are mixed in the naming of variables // below. Disambiguate! // (label, source_ix end) let label: Option<(ReferenceLabel<'input>, usize)> = match scan_result { RefScan::LinkLabel(l, end_ix) => { Some((ReferenceLabel::Link(l), end_ix)) } RefScan::Collapsed(..) | RefScan::Failed | RefScan::UnexpectedFootnote => { // No label? maybe it is a shortcut reference let label_start = self.tree[tos.node].item.end - 1; let label_end = self.tree[cur_ix].item.end; scan_link_label( &self.tree, &self.text[label_start..label_end], self.options.contains(Options::ENABLE_FOOTNOTES), self.options.has_gfm_footnotes(), ) .map(|(ix, label)| (label, label_start + ix)) .filter(|(_, end)| *end == label_end) } }; let id = match &label { Some( (ReferenceLabel::Link(l), _) | (ReferenceLabel::Footnote(l), _), ) => l.clone(), None => "".into(), }; // see if it's a footnote reference if let Some((ReferenceLabel::Footnote(l), end)) = label { let footref = self.allocs.allocate_cow(l); if let Some(def) = self .allocs .footdefs .get_mut(self.allocs.cows[footref.0].to_owned()) { def.use_count += 1; } if !self.options.has_gfm_footnotes() || self.allocs.footdefs.contains(&self.allocs.cows[footref.0]) { // If this came from a MaybeImage, then the `!` prefix // isn't part of the footnote reference. let footnote_ix = if tos.ty == LinkStackTy::Image { self.tree[tos.node].next = Some(cur_ix); self.tree[tos.node].child = None; self.tree[tos.node].item.body = ItemBody::SynthesizeChar('!'); self.tree[cur_ix].item.start = self.tree[tos.node].item.start + 1; self.tree[tos.node].item.end = self.tree[tos.node].item.start + 1; cur_ix } else { tos.node }; // use `next` instead of `node_after_link` because // node_after_link is calculated for a [collapsed][] link, // which footnotes don't support. self.tree[footnote_ix].next = next; self.tree[footnote_ix].child = None; self.tree[footnote_ix].item.body = ItemBody::FootnoteReference(footref); self.tree[footnote_ix].item.end = end; prev = Some(footnote_ix); cur = next; self.link_stack.clear(); continue; } } else if let Some((ReferenceLabel::Link(link_label), end)) = label { if let Some((def_link_type, url, title)) = self .fetch_link_type_url_title( link_label, (self.tree[tos.node].item.start)..end, link_type, ) { let link_ix = self.allocs.allocate_link(def_link_type, url, title, id); 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; // lets do some tree surgery to add the link to the tree // 1st: skip the label node and close node self.tree[tos.node].next = node_after_link; // then, if it exists, add the label node as a child to the link node if label_node != cur { self.tree[tos.node].child = label_node; // finally: disconnect list of children if let Some(prev_ix) = prev { self.tree[prev_ix].next = None; } } self.tree[tos.node].item.end = end; // set up cur so next node will be node_after_link cur = Some(tos.node); cur_ix = tos.node; if tos.ty == LinkStackTy::Link { self.link_stack.disable_all_links(); } } } } } } _ => { // If this is a tight list, blocks and inlines might be // siblings. Inlines can't cross the boundary like that. if let Some(cur_ix) = cur { if self.tree[cur_ix].item.body.is_block() { self.link_stack.clear(); } } } } prev = cur; cur = self.tree[cur_ix].next; } self.link_stack.clear(); } fn handle_emphasis_and_hard_break(&mut self) { let mut prev = None; let mut prev_ix: TreeIndex; let mut cur = self.tree.cur(); let mut single_quote_open: Option = 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 run_length = count; 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, run_length, both) { // have a match! if let Some(prev_ix) = prev { self.tree[prev_ix].next = None; } let match_count = min(count, el.count); // start, end are tree node indices let mut end = cur_ix - 1; let mut start = el.start + el.count; // work from the inside out while start > el.start + el.count - match_count { let inc = if start > el.start + el.count - match_count + 1 { 2 } else { 1 }; let ty = if c == b'~' { ItemBody::Strikethrough } else if inc == 2 { ItemBody::Strong } else { 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; } // set next for top most emph level 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, run_length: el.run_length, c: el.c, both: el.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, run_length, count, c, both, }); } else { for i in 0..count { self.tree[cur_ix + i].item.body = ItemBody::Text { backslash_escaped: false, }; } } 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('’') } _ /* double quote */ => { 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; } ItemBody::HardBreak(true) => { if self.tree[cur_ix].next.is_none() { self.tree[cur_ix].item.body = ItemBody::SynthesizeChar('\\'); } prev = cur; cur = self.tree[cur_ix].next; } _ => { prev = cur; // If this is a tight list, blocks and inlines might be // siblings. Inlines can't cross the boundary like that. if let Some(cur_ix) = cur { if self.tree[cur_ix].item.body.is_block() { self.inline_stack.pop_all(&mut self.tree); } } cur = self.tree[cur_ix].next; } } } self.inline_stack.pop_all(&mut self.tree); } /// Returns next byte index, url and title. fn scan_inline_link( &self, underlying: &'input str, mut ix: usize, node: Option, ) -> Option<(usize, CowStr<'input>, CowStr<'input>)> { if scan_ch(&underlying.as_bytes()[ix..], b'(') == 0 { return None; } ix += 1; let scan_separator = |ix: &mut usize| { *ix += scan_while(&underlying.as_bytes()[*ix..], is_ascii_whitespace_no_nl); if let Some(bl) = scan_eol(&underlying.as_bytes()[*ix..]) { *ix += bl; let mut line_start = LineStart::new(&underlying.as_bytes()[*ix..]); let _ = scan_containers( &self.tree, &mut line_start, self.options.has_gfm_footnotes(), ); *ix += line_start.bytes_scanned(); } *ix += scan_while(&underlying.as_bytes()[*ix..], is_ascii_whitespace_no_nl); }; scan_separator(&mut ix); let (dest_length, dest) = scan_link_dest(underlying, ix, LINK_MAX_NESTED_PARENS)?; let dest = unescape(dest, self.tree.is_in_table()); ix += dest_length; scan_separator(&mut ix); let title = if let Some((bytes_scanned, t)) = self.scan_link_title(underlying, ix, node) { ix += bytes_scanned; scan_separator(&mut ix); t } else { "".into() }; if scan_ch(&underlying.as_bytes()[ix..], b')') == 0 { return None; } ix += 1; Some((ix, dest, title)) } // returns (bytes scanned, title cow) fn scan_link_title( &self, text: &'input str, start_ix: usize, node: Option, ) -> Option<(usize, CowStr<'input>)> { 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 self.tree.is_in_table() && c == b'\\' && i + 2 < bytes.len() && bytes[i + 1] == b'\\' && bytes[i + 2] == b'|' { // this runs if there are an even number of pipes in a table // if it's odd, then it gets parsed as normal title.push_str(&text[mark..i]); i += 2; mark = i; } 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_math_span(&mut self, open: TreeIndex, mut close: TreeIndex) { let start_is_display = self.tree[open].next.filter(|&next_ix| { next_ix != close && matches!( self.tree[next_ix].item.body, ItemBody::MaybeMath(_can_open, _can_close, _brace_context) ) }); let end_is_display = self.tree[close].next.filter(|&next_ix| { matches!( self.tree[next_ix].item.body, ItemBody::MaybeMath(_can_open, _can_close, _brace_context) ) }); let is_display = start_is_display.is_some() && end_is_display.is_some(); if is_display { // This unwrap() can't panic, because if the next variable were None, end_is_display would be None close = self.tree[close].next.unwrap(); self.tree[open].next = Some(close); self.tree[open].item.end += 1; self.tree[close].item.start -= 1; } else { if self.tree[open].item.end == self.tree[close].item.start { // inline math spans cannot be empty self.tree[open].item.body = ItemBody::Text { backslash_escaped: false, }; return; } self.tree[open].next = Some(close); } let span_start = self.tree[open].item.end; let span_end = self.tree[close].item.start; let bytes = self.text.as_bytes(); let mut buf: Option = None; let mut start_ix = span_start; let mut ix = span_start; while ix < span_end { let c = bytes[ix]; if c == b'\r' || c == b'\n' { ix += 1; let buf = buf.get_or_insert_with(|| String::with_capacity(ix - span_start)); buf.push_str(&self.text[start_ix..ix]); let mut line_start = LineStart::new(&bytes[ix..]); let _ = scan_containers( &self.tree, &mut line_start, self.options.has_gfm_footnotes(), ); ix += line_start.bytes_scanned(); start_ix = ix; } else if c == b'\\' && bytes.get(ix + 1) == Some(&b'|') && self.tree.is_in_table() { let buf = buf.get_or_insert_with(|| String::with_capacity(ix + 1 - span_start)); buf.push_str(&self.text[start_ix..ix]); buf.push('|'); ix += 2; start_ix = ix; } else { ix += 1; } } let cow = if let Some(mut buf) = buf { buf.push_str(&self.text[start_ix..span_end]); buf.into() } else { self.text[span_start..span_end].into() }; self.tree[open].item.body = ItemBody::Math(self.allocs.allocate_cow(cow), is_display); self.tree[open].item.end = self.tree[close].item.end; self.tree[open].next = self.tree[close].next; } /// Make a code span. /// /// Both `open` and `close` are matching MaybeCode items. fn make_code_span(&mut self, open: TreeIndex, close: TreeIndex, preceding_backslash: bool) { let bytes = self.text.as_bytes(); let span_start = self.tree[open].item.end; let span_end = self.tree[close].item.start; let mut buf: Option = None; let mut start_ix = span_start; let mut ix = span_start; while ix < span_end { let c = bytes[ix]; if c == b'\r' || c == b'\n' { let buf = buf.get_or_insert_with(|| String::with_capacity(ix + 1 - span_start)); buf.push_str(&self.text[start_ix..ix]); buf.push(' '); ix += 1; let mut line_start = LineStart::new(&bytes[ix..]); let _ = scan_containers( &self.tree, &mut line_start, self.options.has_gfm_footnotes(), ); ix += line_start.bytes_scanned(); start_ix = ix; } else if c == b'\\' && bytes.get(ix + 1) == Some(&b'|') && self.tree.is_in_table() { let buf = buf.get_or_insert_with(|| String::with_capacity(ix + 1 - span_start)); buf.push_str(&self.text[start_ix..ix]); buf.push('|'); ix += 2; start_ix = ix; } else { ix += 1; } } let (opening, closing, all_spaces) = { let s = if let Some(buf) = &mut buf { buf.push_str(&self.text[start_ix..span_end]); &buf[..] } else { &self.text[span_start..span_end] }; ( s.as_bytes().first() == Some(&b' '), s.as_bytes().last() == Some(&b' '), s.bytes().all(|b| b == b' '), ) }; let cow: CowStr<'input> = if !all_spaces && opening && closing { if let Some(mut buf) = buf { buf.remove(0); buf.pop(); buf.into() } else { let lo = span_start + 1; let hi = (span_end - 1).max(lo); self.text[lo..hi].into() } } else 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 { backslash_escaped: true, }; 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; } } /// On success, returns a buffer containing the inline html and byte offset. /// When no bytes were skipped, the buffer will be empty and the html can be /// represented as a subslice of the input string. fn scan_inline_html(&mut self, bytes: &[u8], ix: usize) -> Option<(Vec, 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( // Subtract 1 to include the < character &bytes[(ix - 1)..], Some(&|bytes| { let mut line_start = LineStart::new(bytes); let _ = scan_containers( &self.tree, &mut line_start, self.options.has_gfm_footnotes(), ); line_start.bytes_scanned() }), )?; Some((span, i + ix - 1)) } } /// Consumes the event iterator and produces an iterator that produces /// `(Event, Range)` pairs, where the `Range` value maps to the corresponding /// range in the markdown source. pub fn into_offset_iter(self) -> OffsetIter<'input, F> { OffsetIter { inner: self } } } /// Returns number of containers scanned. pub(crate) fn scan_containers( tree: &Tree, line_start: &mut LineStart<'_>, gfm_footnotes: bool, ) -> usize { let mut i = 0; for &node_ix in tree.walk_spine() { match tree[node_ix].item.body { ItemBody::BlockQuote(..) => { // `scan_blockquote_marker` saves & restores internally if !line_start.scan_blockquote_marker() { break; } } ItemBody::ListItem(indent) => { let save = line_start.clone(); if !line_start.scan_space(indent) && !line_start.is_at_eol() { *line_start = save; break; } } ItemBody::FootnoteDefinition(..) if gfm_footnotes => { let save = line_start.clone(); if !line_start.scan_space(4) && !line_start.is_at_eol() { *line_start = save; break; } } _ => (), } i += 1; } i } impl Tree { pub(crate) fn append_text(&mut self, start: usize, end: usize, backslash_escaped: bool) { if end > start { if let Some(ix) = self.cur() { if matches!(self[ix].item.body, ItemBody::Text { .. }) && self[ix].item.end == start { self[ix].item.end = end; return; } } self.append(Item { start, end, body: ItemBody::Text { backslash_escaped }, }); } } /// Returns true if the current node is inside a table. /// /// If `cur` is an ItemBody::Table, it would return false, /// but since the `TableRow` and `TableHead` and `TableCell` /// are children of the table, anything doing inline parsing /// doesn't need to care about that. pub(crate) fn is_in_table(&self) -> bool { fn might_be_in_table(item: &Item) -> bool { item.body.is_inline() || matches!(item.body, |ItemBody::TableHead| ItemBody::TableRow | ItemBody::TableCell) } for &ix in self.walk_spine().rev() { if matches!(self[ix].item.body, ItemBody::Table(_)) { return true; } if !might_be_in_table(&self[ix].item) { return false; } } false } } #[derive(Copy, Clone, Debug)] struct InlineEl { /// offset of tree node start: TreeIndex, /// number of delimiters available for matching count: usize, /// length of the run that these delimiters came from run_length: usize, /// b'*', b'_', or b'~' c: u8, /// can both open and close both: bool, } #[derive(Debug, Clone, Default)] struct InlineStack { stack: Vec, // Lower bounds for matching indices in the stack. For example // a strikethrough delimiter will never match with any element // in the stack with index smaller than // `lower_bounds[InlineStack::TILDES]`. lower_bounds: [usize; 9], } impl InlineStack { /// These are indices into the lower bounds array. /// Not both refers to the property that the delimiter can not both /// be opener as a closer. const UNDERSCORE_NOT_BOTH: usize = 0; const ASTERISK_NOT_BOTH: usize = 1; const ASTERISK_BASE: usize = 2; const TILDES: usize = 5; const UNDERSCORE_BASE: usize = 6; fn pop_all(&mut self, tree: &mut Tree) { for el in self.stack.drain(..) { for i in 0..el.count { tree[el.start + i].item.body = ItemBody::Text { backslash_escaped: false, }; } } self.lower_bounds = [0; 9]; } fn get_lowerbound(&self, c: u8, count: usize, both: bool) -> usize { if c == b'_' { let mod3_lower = self.lower_bounds[InlineStack::UNDERSCORE_BASE + count % 3]; if both { mod3_lower } else { min( mod3_lower, 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_BASE + count % 3] = 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 truncate(&mut self, new_bound: usize) { self.stack.truncate(new_bound); for lower_bound in &mut self.lower_bounds { if *lower_bound > new_bound { *lower_bound = new_bound; } } } fn find_match( &mut self, tree: &mut Tree, c: u8, run_length: usize, both: bool, ) -> Option { let lowerbound = min(self.stack.len(), self.get_lowerbound(c, run_length, both)); let res = self.stack[lowerbound..] .iter() .cloned() .enumerate() .rfind(|(_, el)| { if c == b'~' && run_length != el.run_length { return false; } el.c == c && (!both && !el.both || (run_length + el.run_length) % 3 != 0 || run_length % 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 { backslash_escaped: false, }; } } self.truncate(matching_ix); Some(matching_el) } else { self.set_lowerbound(c, run_length, both, self.stack.len()); None } } fn trim_lower_bound(&mut self, ix: usize) { self.lower_bounds[ix] = self.lower_bounds[ix].min(self.stack.len()); } fn push(&mut self, el: InlineEl) { if el.c == b'~' { self.trim_lower_bound(InlineStack::TILDES); } self.stack.push(el) } } #[derive(Debug, Clone)] enum RefScan<'a> { // label, source ix of label end LinkLabel(CowStr<'a>, usize), // contains next node index Collapsed(Option), UnexpectedFootnote, Failed, } /// Skips forward within a block to a node which spans (ends inclusive) the given /// index into the source. fn scan_nodes_to_ix( tree: &Tree, mut node: Option, ix: usize, ) -> Option { while let Some(node_ix) = node { if tree[node_ix].item.end <= ix { node = tree[node_ix].next; } else { break; } } node } /// Scans an inline link label, which cannot be interrupted. /// Returns number of bytes (including brackets) and label on success. fn scan_link_label<'text>( tree: &Tree, text: &'text str, allow_footnote_refs: bool, gfm_footnotes: 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, gfm_footnotes); Some(line_start.bytes_scanned()) }; if allow_footnote_refs && b'^' == bytes[1] && bytes.get(2) != Some(&b']') { let linebreak_handler: &dyn Fn(&[u8]) -> Option = if gfm_footnotes { &|_| None } else { &linebreak_handler }; if let Some((byte_index, cow)) = scan_link_label_rest(&text[2..], linebreak_handler, tree.is_in_table()) { return Some((byte_index + 2, ReferenceLabel::Footnote(cow))); } } let (byte_index, cow) = scan_link_label_rest(&text[1..], &linebreak_handler, tree.is_in_table())?; Some((byte_index + 1, ReferenceLabel::Link(cow))) } fn scan_reference<'b>( tree: &Tree, text: &'b str, cur: Option, allow_footnote_refs: bool, gfm_footnotes: 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"[]") { // TODO: this unwrap is sus and should be looked at closer let closing_node = tree[cur_ix].next.unwrap(); RefScan::Collapsed(tree[closing_node].next) } else { let label = scan_link_label(tree, &text[start..], allow_footnote_refs, gfm_footnotes); match label { Some((ix, ReferenceLabel::Link(label))) => RefScan::LinkLabel(label, start + ix), Some((_ix, ReferenceLabel::Footnote(_label))) => RefScan::UnexpectedFootnote, None => RefScan::Failed, } } } #[derive(Clone, Default)] struct LinkStack { inner: Vec, disabled_ix: usize, } impl LinkStack { fn push(&mut self, el: LinkStackEl) { self.inner.push(el); } fn pop(&mut self) -> Option { 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, } /// Contains the destination URL, title and source span of a reference definition. #[derive(Clone, Debug)] pub struct LinkDef<'a> { pub dest: CowStr<'a>, pub title: Option>, pub span: Range, } /// Contains the destination URL, title and source span of a reference definition. #[derive(Clone, Debug)] pub struct FootnoteDef { pub use_count: usize, } /// Tracks tree indices of code span delimiters of each length. It should prevent /// quadratic scanning behaviours by providing (amortized) constant time lookups. struct CodeDelims { inner: HashMap>, 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_default().push_back(ix); } else { // Skip the first insert, since that delimiter will always // be an opener and not a closer. self.seen_first = true; } } fn is_populated(&self) -> bool { !self.inner.is_empty() } fn find(&mut self, open_ix: TreeIndex, count: usize) -> Option { 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; } } /// Tracks brace contexts and delimiter length for math delimiters. /// Provides amortized constant-time lookups. struct MathDelims { inner: HashMap>, } impl MathDelims { fn new() -> Self { Self { inner: Default::default(), } } fn insert( &mut self, delim_is_display: bool, brace_context: u8, ix: TreeIndex, can_close: bool, ) { self.inner .entry(brace_context) .or_default() .push_back((ix, can_close, delim_is_display)); } fn is_populated(&self) -> bool { !self.inner.is_empty() } fn find( &mut self, tree: &Tree, open_ix: TreeIndex, is_display: bool, brace_context: u8, ) -> Option { while let Some((ix, can_close, delim_is_display)) = self.inner.get_mut(&brace_context)?.pop_front() { if ix <= open_ix || (is_display && tree[open_ix].next == Some(ix)) { continue; } let can_close = can_close && tree[open_ix].item.end != tree[ix].item.start; if (!is_display && can_close) || (is_display && delim_is_display) { return Some(ix); } // if we can't use it, leave it in the queue as a tombstone for the next // thing that tries to match it self.inner .get_mut(&brace_context)? .push_front((ix, can_close, delim_is_display)); break; } None } fn clear(&mut self) { self.inner.clear(); } } #[derive(Copy, Clone, PartialEq, Eq, Debug)] pub(crate) struct LinkIndex(usize); #[derive(Copy, Clone, PartialEq, Eq, Debug)] pub(crate) struct CowIndex(usize); #[derive(Copy, Clone, PartialEq, Eq, Debug)] pub(crate) struct AlignmentIndex(usize); #[derive(Copy, Clone, PartialEq, Eq, Debug)] pub(crate) struct HeadingIndex(NonZeroUsize); #[derive(Clone)] pub(crate) struct Allocations<'a> { pub refdefs: RefDefs<'a>, pub footdefs: FootnoteDefs<'a>, links: Vec<(LinkType, CowStr<'a>, CowStr<'a>, CowStr<'a>)>, cows: Vec>, alignments: Vec>, headings: Vec>, } /// Used by the heading attributes extension. #[derive(Clone)] pub(crate) struct HeadingAttributes<'a> { pub id: Option>, pub classes: Vec>, pub attrs: Vec<(CowStr<'a>, Option>)>, } /// Keeps track of the reference definitions defined in the document. #[derive(Clone, Default, Debug)] pub struct RefDefs<'input>(pub(crate) HashMap, LinkDef<'input>>); /// Keeps track of the footnote definitions defined in the document. #[derive(Clone, Default, Debug)] pub struct FootnoteDefs<'input>(pub(crate) HashMap, FootnoteDef>); impl<'input, 'b, 's> RefDefs<'input> where 's: 'b, { /// Performs a lookup on reference label using unicode case folding. pub fn get(&'s self, key: &'b str) -> Option<&'b LinkDef<'input>> { self.0.get(&UniCase::new(key.into())) } /// Provides an iterator over all the document's reference definitions. pub fn iter(&'s self) -> impl Iterator)> { self.0.iter().map(|(k, v)| (k.as_ref(), v)) } } impl<'input, 'b, 's> FootnoteDefs<'input> where 's: 'b, { /// Performs a lookup on reference label using unicode case folding. pub fn contains(&'s self, key: &'b str) -> bool { self.0.contains_key(&UniCase::new(key.into())) } /// Performs a lookup on reference label using unicode case folding. pub fn get_mut(&'s mut self, key: CowStr<'input>) -> Option<&'s mut FootnoteDef> { self.0.get_mut(&UniCase::new(key)) } } impl<'a> Allocations<'a> { pub fn new() -> Self { Self { refdefs: RefDefs::default(), footdefs: FootnoteDefs::default(), links: Vec::with_capacity(128), cows: Vec::new(), alignments: Vec::new(), headings: Vec::new(), } } pub fn allocate_cow(&mut self, cow: CowStr<'a>) -> CowIndex { let ix = self.cows.len(); self.cows.push(cow); CowIndex(ix) } pub fn allocate_link( &mut self, ty: LinkType, url: CowStr<'a>, title: CowStr<'a>, id: CowStr<'a>, ) -> LinkIndex { let ix = self.links.len(); self.links.push((ty, url, title, id)); LinkIndex(ix) } pub fn allocate_alignment(&mut self, alignment: Vec) -> AlignmentIndex { let ix = self.alignments.len(); self.alignments.push(alignment); AlignmentIndex(ix) } pub fn allocate_heading(&mut self, attrs: HeadingAttributes<'a>) -> HeadingIndex { let ix = self.headings.len(); self.headings.push(attrs); // This won't panic. `self.headings.len()` can't be `usize::MAX` since // such a long Vec cannot fit in memory. let ix_nonzero = NonZeroUsize::new(ix.wrapping_add(1)).expect("too many headings"); HeadingIndex(ix_nonzero) } pub fn take_cow(&mut self, ix: CowIndex) -> CowStr<'a> { std::mem::replace(&mut self.cows[ix.0], "".into()) } pub fn take_link(&mut self, ix: LinkIndex) -> (LinkType, CowStr<'a>, CowStr<'a>, CowStr<'a>) { let default_link = (LinkType::ShortcutUnknown, "".into(), "".into(), "".into()); std::mem::replace(&mut self.links[ix.0], default_link) } pub fn take_alignment(&mut self, ix: AlignmentIndex) -> Vec { std::mem::take(&mut self.alignments[ix.0]) } } impl<'a> Index for Allocations<'a> { type Output = CowStr<'a>; fn index(&self, ix: CowIndex) -> &Self::Output { self.cows.index(ix.0) } } impl<'a> Index for Allocations<'a> { type Output = (LinkType, CowStr<'a>, CowStr<'a>, CowStr<'a>); fn index(&self, ix: LinkIndex) -> &Self::Output { self.links.index(ix.0) } } impl<'a> Index for Allocations<'a> { type Output = Vec; fn index(&self, ix: AlignmentIndex) -> &Self::Output { self.alignments.index(ix.0) } } impl<'a> Index for Allocations<'a> { type Output = HeadingAttributes<'a>; fn index(&self, ix: HeadingIndex) -> &Self::Output { self.headings.index(ix.0.get() - 1) } } /// A struct containing information on the reachability of certain inline HTML /// elements. In particular, for cdata elements (` { fn handle_broken_link( &mut self, link: BrokenLink<'input>, ) -> Option<(CowStr<'input>, CowStr<'input>)>; } impl<'input, T> BrokenLinkCallback<'input> for T where T: FnMut(BrokenLink<'input>) -> Option<(CowStr<'input>, CowStr<'input>)>, { fn handle_broken_link( &mut self, link: BrokenLink<'input>, ) -> Option<(CowStr<'input>, CowStr<'input>)> { self(link) } } impl<'input> BrokenLinkCallback<'input> for Box> { fn handle_broken_link( &mut self, link: BrokenLink<'input>, ) -> Option<(CowStr<'input>, CowStr<'input>)> { (**self).handle_broken_link(link) } } /// Broken link callback that does nothing. #[derive(Debug)] pub struct DefaultBrokenLinkCallback; impl<'input> BrokenLinkCallback<'input> for DefaultBrokenLinkCallback { fn handle_broken_link( &mut self, _link: BrokenLink<'input>, ) -> Option<(CowStr<'input>, CowStr<'input>)> { None } } /// Markdown event and source range iterator. /// /// Generates tuples where the first element is the markdown event and the second /// is a the corresponding range in the source string. /// /// Constructed from a `Parser` using its /// [`into_offset_iter`](struct.Parser.html#method.into_offset_iter) method. #[derive(Debug)] pub struct OffsetIter<'a, F = DefaultBrokenLinkCallback> { inner: Parser<'a, F>, } impl<'a, F: BrokenLinkCallback<'a>> OffsetIter<'a, F> { /// Returns a reference to the internal reference definition tracker. pub fn reference_definitions(&self) -> &RefDefs<'_> { self.inner.reference_definitions() } } impl<'a, F: BrokenLinkCallback<'a>> Iterator for OffsetIter<'a, F> { type Item = (Event<'a>, Range); fn next(&mut self) -> Option { match self.inner.tree.cur() { None => { let ix = self.inner.tree.pop()?; let tag_end = body_to_tag_end(&self.inner.tree[ix].item.body); self.inner.tree.next_sibling(ix); let span = self.inner.tree[ix].item.start..self.inner.tree[ix].item.end; debug_assert!(span.start <= span.end); Some((Event::End(tag_end), span)) } 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, &mut self.inner.allocs); if let Event::Start(..) = event { self.inner.tree.push(); } else { self.inner.tree.next_sibling(cur_ix); } debug_assert!(item.start <= item.end); Some((event, item.start..item.end)) } } } } fn body_to_tag_end(body: &ItemBody) -> TagEnd { match *body { ItemBody::Paragraph => TagEnd::Paragraph, ItemBody::Emphasis => TagEnd::Emphasis, ItemBody::Strong => TagEnd::Strong, ItemBody::Strikethrough => TagEnd::Strikethrough, ItemBody::Link(..) => TagEnd::Link, ItemBody::Image(..) => TagEnd::Image, ItemBody::Heading(level, _) => TagEnd::Heading(level), ItemBody::IndentCodeBlock | ItemBody::FencedCodeBlock(..) => TagEnd::CodeBlock, ItemBody::BlockQuote(..) => TagEnd::BlockQuote, ItemBody::HtmlBlock => TagEnd::HtmlBlock, ItemBody::List(_, c, _) => { let is_ordered = c == b'.' || c == b')'; TagEnd::List(is_ordered) } ItemBody::ListItem(_) => TagEnd::Item, ItemBody::TableHead => TagEnd::TableHead, ItemBody::TableCell => TagEnd::TableCell, ItemBody::TableRow => TagEnd::TableRow, ItemBody::Table(..) => TagEnd::Table, ItemBody::FootnoteDefinition(..) => TagEnd::FootnoteDefinition, ItemBody::MetadataBlock(kind) => TagEnd::MetadataBlock(kind), _ => panic!("unexpected item body {:?}", body), } } fn item_to_event<'a>(item: Item, text: &'a str, allocs: &mut 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.take_cow(cow_ix)), ItemBody::SynthesizeText(cow_ix) => return Event::Text(allocs.take_cow(cow_ix)), ItemBody::SynthesizeChar(c) => return Event::Text(c.into()), ItemBody::HtmlBlock => Tag::HtmlBlock, ItemBody::Html => return Event::Html(text[item.start..item.end].into()), ItemBody::InlineHtml => return Event::InlineHtml(text[item.start..item.end].into()), ItemBody::OwnedHtml(cow_ix) => return Event::Html(allocs.take_cow(cow_ix)), ItemBody::SoftBreak => return Event::SoftBreak, ItemBody::HardBreak(_) => return Event::HardBreak, ItemBody::FootnoteReference(cow_ix) => { return Event::FootnoteReference(allocs.take_cow(cow_ix)) } 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 (link_type, dest_url, title, id) = allocs.take_link(link_ix); Tag::Link { link_type, dest_url, title, id, } } ItemBody::Image(link_ix) => { let (link_type, dest_url, title, id) = allocs.take_link(link_ix); Tag::Image { link_type, dest_url, title, id, } } ItemBody::Heading(level, Some(heading_ix)) => { let HeadingAttributes { id, classes, attrs } = allocs.index(heading_ix); Tag::Heading { level, id: id.clone(), classes: classes.clone(), attrs: attrs.clone(), } } ItemBody::Heading(level, None) => Tag::Heading { level, id: None, classes: Vec::new(), attrs: Vec::new(), }, ItemBody::FencedCodeBlock(cow_ix) => { Tag::CodeBlock(CodeBlockKind::Fenced(allocs.take_cow(cow_ix))) } ItemBody::IndentCodeBlock => Tag::CodeBlock(CodeBlockKind::Indented), ItemBody::BlockQuote(kind) => Tag::BlockQuote(kind), 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.take_alignment(alignment_ix)), ItemBody::FootnoteDefinition(cow_ix) => Tag::FootnoteDefinition(allocs.take_cow(cow_ix)), ItemBody::MetadataBlock(kind) => Tag::MetadataBlock(kind), ItemBody::Math(cow_ix, is_display) => { return if is_display { Event::DisplayMath(allocs.take_cow(cow_ix)) } else { Event::InlineMath(allocs.take_cow(cow_ix)) } } _ => panic!("unexpected item body {:?}", item.body), }; Event::Start(tag) } impl<'a, F: BrokenLinkCallback<'a>> Iterator for Parser<'a, F> { type Item = Event<'a>; fn next(&mut self) -> Option> { match self.tree.cur() { None => { let ix = self.tree.pop()?; let tag_end = body_to_tag_end(&self.tree[ix].item.body); self.tree.next_sibling(ix); Some(Event::End(tag_end)) } 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, &mut self.allocs); if let Event::Start(ref _tag) = event { self.tree.push(); } else { self.tree.next_sibling(cur_ix); } Some(event) } } } } impl<'a, F: BrokenLinkCallback<'a>> FusedIterator for Parser<'a, F> {} #[cfg(test)] mod test { use super::*; use crate::tree::Node; // TODO: move these tests to tests/html.rs? 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::>(); assert_eq!(48, node_size); } #[test] #[cfg(target_pointer_width = "64")] fn body_size() { let body_size = std::mem::size_of::(); assert_eq!(16, body_size); } #[test] fn single_open_fish_bracket() { // dont crash assert_eq!(3, Parser::new("<").count()); } #[test] fn lone_hashtag() { // dont crash assert_eq!(2, Parser::new("#").count()); } #[test] fn lots_of_backslashes() { // dont crash Parser::new("\\\\\r\r").count(); Parser::new("\\\r\r\\.\\\\\r\r\\.\\").count(); } #[test] fn issue_320() { // dont crash parser_with_extensions(":\r\t> |\r:\r\t> |\r").count(); } #[test] fn issue_319() { // dont crash parser_with_extensions("|\r-]([^|\r-]([^").count(); parser_with_extensions("|\r\r=][^|\r\r=][^car").count(); } #[test] fn issue_303() { // dont crash parser_with_extensions("[^\r\ra]").count(); parser_with_extensions("\r\r]Z[^\x00\r\r]Z[^\x00").count(); } #[test] fn issue_313() { // dont crash parser_with_extensions("*]0[^\r\r*]0[^").count(); parser_with_extensions("[^\r> `][^\r> `][^\r> `][").count(); } #[test] fn issue_311() { // dont crash 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 - \\\n> - ").count(); parser_with_extensions("- \n\n").count(); } #[test] fn issue_306() { // dont crash parser_with_extensions("*\r_<__*\r_<__*\r_<__*\r_<__").count(); } #[test] fn issue_305() { // dont crash 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 { link_type: 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 footnote_offsets_exclamation() { let mut immediately_before_footnote = None; 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), _ => { immediately_before_footnote = Some((ev, range)); None } }) .next() .unwrap(); assert_eq!(13..17, range); if let (Event::Text(exclamation), range_exclamation) = immediately_before_footnote.as_ref().unwrap() { assert_eq!("!", &exclamation[..]); assert_eq!(&(12..13), range_exclamation); } else { panic!("what came first, then? {immediately_before_footnote:?}"); } } #[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 table_cell_span() { let markdown = "a|b|c\n--|--|--\na| |c"; let event_offset = parser_with_extensions(markdown) .into_offset_iter() .filter_map(|(ev, span)| match ev { Event::Start(Tag::TableCell) => Some(span), _ => None, }) .nth(4) .unwrap(); let expected_offset_start = "a|b|c\n--|--|--\na|".len(); assert_eq!( expected_offset_start..(expected_offset_start + 2), 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); } // FIXME: add this one regression suite #[cfg(feature = "html")] #[test] fn link_def_at_eof() { let test_str = "[My site][world]\n\n[world]: https://vincentprouillet.com"; let expected = "

My site

\n"; let mut buf = String::new(); crate::html::push_html(&mut buf, Parser::new(test_str)); assert_eq!(expected, buf); } #[cfg(feature = "html")] #[test] fn no_footnote_refs_without_option() { let test_str = "a [^a]\n\n[^a]: yolo"; let expected = "

a ^a

\n"; let mut buf = String::new(); crate::html::push_html(&mut buf, Parser::new(test_str)); assert_eq!(expected, buf); } #[cfg(feature = "html")] #[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); } #[cfg(feature = "html")] #[test] fn ref_def_cr_lf() { let test_str = "[a]: /u\r\n\n[a]"; let expected = "

a

\n"; let mut buf = String::new(); crate::html::push_html(&mut buf, Parser::new(test_str)); assert_eq!(expected, buf); } #[cfg(feature = "html")] #[test] fn no_dest_refdef() { let test_str = "[a]:"; let expected = "

[a]:

\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.as_ref()); 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, id) in parser.filter_map(|event| match event { Event::Start(tag) => match tag { Tag::Link { link_type, dest_url, title, id, } => Some((link_type, dest_url, title, id)), _ => None, }, _ => None, }) { link_tag_count += 1; assert_eq!(typ, LinkType::ReferenceUnknown); assert_eq!(url.as_ref(), "YOLO"); assert_eq!(title.as_ref(), "SWAG"); assert_eq!(id.as_ref(), "world"); } 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); } #[test] fn ref_defs() { let input = r###"[a B c]: http://example.com [another]: https://google.com text [final ONE]: http://wikipedia.org "###; let mut parser = Parser::new(input); assert!(parser.reference_definitions().get("a b c").is_some()); assert!(parser.reference_definitions().get("nope").is_none()); if let Some(_event) = parser.next() { // testing keys with shorter lifetimes than parser and its input let s = "final one".to_owned(); let link_def = parser.reference_definitions().get(&s).unwrap(); let span = &input[link_def.span.clone()]; assert_eq!(span, "[final ONE]: http://wikipedia.org"); } } #[test] fn common_lifetime_patterns_allowed<'b>() { let temporary_str = String::from("xyz"); // NOTE: this is a limitation of Rust, it doesn't allow putting lifetime parameters on the closure itself. // Hack it by attaching the lifetime to the test function instead. // TODO: why is the `'b` lifetime required at all? Changing it to `'_` breaks things :( let mut closure = |link: BrokenLink<'b>| Some(("#".into(), link.reference)); fn function(link: BrokenLink<'_>) -> Option<(CowStr<'_>, CowStr<'_>)> { Some(("#".into(), link.reference)) } for _ in Parser::new_with_broken_link_callback( "static lifetime", Options::empty(), Some(&mut closure), ) {} /* This fails to compile. Because the closure can't say `for <'a> fn(BrokenLink<'a>) -> * CowStr<'a>` and has to use the enclosing `'b` lifetime parameter, `temporary_str` lives * shorter than `'b`. I think this is unlikely to occur in real life, and if it does, the * fix is simple: move it out to a function that allows annotating the lifetimes. */ //for _ in Parser::new_with_broken_link_callback(&temporary_str, Options::empty(), Some(&mut callback)) { //} for _ in Parser::new_with_broken_link_callback( "static lifetime", Options::empty(), Some(&mut function), ) {} for _ in Parser::new_with_broken_link_callback( &temporary_str, Options::empty(), Some(&mut function), ) {} } }