jer/mdbook-toc
1
Fork 0
Code Actions Activity

Use pulldown types and force deref to compare with string slice

Browse source
This commit is contained in:
Jan-Erik Rediger 2019-07-09 11:15:31 +02:00
parent 80faf141da
commit 080d0c7f0a
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/lib.rs
View file

@ -2,13 +2,13 @@ extern crate mdbook;
extern crate pulldown_cmark; extern crate pulldown_cmark;
extern crate pulldown_cmark_to_cmark; extern crate pulldown_cmark_to_cmark;
use std::borrow::Cow;
use mdbook::errors::{Error, Result}; use mdbook::errors::{Error, Result};
use mdbook::book::{Book, BookItem, Chapter}; use mdbook::book::{Book, BookItem, Chapter};
use mdbook::preprocess::{Preprocessor, PreprocessorContext}; use mdbook::preprocess::{Preprocessor, PreprocessorContext};
use pulldown_cmark::{Event, Parser}; use pulldown_cmark::{Event, Parser};
use pulldown_cmark::Tag::*; use pulldown_cmark::Tag::*;
use pulldown_cmark_to_cmark::fmt::cmark; use pulldown_cmark_to_cmark::fmt::cmark;
use pulldown_cmark::CowStr;
pub struct Toc; pub struct Toc;
@ -35,7 +35,7 @@ impl Preprocessor for Toc {
} }
} }
fn build_toc<'a>(toc: &[(i32, Cow<'a, str>)]) -> String { fn build_toc<'a>(toc: &[(i32, CowStr<'a>)]) -> String {
let mut result = String::new(); let mut result = String::new();
for (level, name) in toc { for (level, name) in toc {
@ -57,7 +57,7 @@ fn add_toc(content: &str) -> Result<String> {
for e in Parser::new(&content) { for e in Parser::new(&content) {
if let Event::Html(html) = e { if let Event::Html(html) = e {
if html == "<!-- toc -->\n" { if &*html == "<!-- toc -->\n" {
toc_found = true; toc_found = true;
} }
continue; continue;
@ -90,7 +90,7 @@ fn add_toc(content: &str) -> Result<String> {
let events = Parser::new(&content).map(|e| { let events = Parser::new(&content).map(|e| {
if let Event::Html(html) = e.clone() { if let Event::Html(html) = e.clone() {
if html == "<!-- toc -->\n" { if &*html == "<!-- toc -->\n" {
return toc_events.clone(); return toc_events.clone();
} }
} }