use FnMut, drop a bunch of refcells #1

Merged
cadey merged 1 commits from :use-fnmut into master 2020-06-01 04:12:56 +00:00
1 changed files with 13 additions and 20 deletions

View File

@ -1,7 +1,6 @@
use anyhow::Result;
use comrak::nodes::{AstNode, NodeValue};
use comrak::{format_commonmark, parse_document, Arena, ComrakOptions};
use std::cell::RefCell;
use std::fs::read_to_string;
use std::path::PathBuf;
@ -10,41 +9,35 @@ pub(crate) fn read(fname: PathBuf, tag: String) -> Result<String> {
let arena = Arena::new();
let mut root = parse_document(&arena, &data, &ComrakOptions::default());
let collect = RefCell::new(false);
let buf = RefCell::new(Vec::<u8>::new());
let mut collect = false;
let mut buf = Vec::<u8>::new();
iter_nodes(&mut root, &|node| {
iter_nodes(&mut root, &mut |node| {
let nd = node.data.borrow();
match nd.value {
NodeValue::Heading(ref hdr) => {
if hdr.level == 2 {
if *collect.borrow() {
collect.swap(&RefCell::new(false));
if collect {
collect = false;
}
let found_tag = String::from_utf8(nd.content.clone())?;
if found_tag == tag {
collect.swap(&RefCell::new(true));
collect = true;
}
} else {
if *collect.borrow() {
let mut apd = buf.borrow_mut();
let mut new_buf = Vec::<u8>::new();
format_commonmark(&node, &ComrakOptions::default(), &mut new_buf)?;
apd.append(&mut new_buf);
if collect {
format_commonmark(&node, &ComrakOptions::default(), &mut buf)?;
}
}
Ok(())
}
NodeValue::Item(_) => Ok(()),
_ => {
if *collect.borrow() {
let mut apd = buf.borrow_mut();
let mut new_buf = Vec::<u8>::new();
format_commonmark(&node, &ComrakOptions::default(), &mut new_buf)?;
apd.append(&mut new_buf);
if collect {
format_commonmark(&node, &ComrakOptions::default(), &mut buf)?;
}
Ok(())
@ -52,12 +45,12 @@ pub(crate) fn read(fname: PathBuf, tag: String) -> Result<String> {
}
})?;
Ok(String::from_utf8(buf.into_inner())?)
Ok(String::from_utf8(buf)?)
}
fn iter_nodes<'a, F>(node: &'a AstNode<'a>, f: &F) -> Result<()>
fn iter_nodes<'a, F>(node: &'a AstNode<'a>, f: &mut F) -> Result<()>
where
F: Fn(&'a AstNode<'a>) -> Result<()>,
F: FnMut(&'a AstNode<'a>) -> Result<()>,
{
f(node)?;
for c in node.children() {