gitea-release/src/changelog.rs

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;

pub(crate) fn read(fname: PathBuf, tag: String) -> Result<String> {
    let data = read_to_string(fname)?;
    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());

    iter_nodes(&mut root, &|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));
                    }

                    let found_tag = String::from_utf8(nd.content.clone())?;

                    if found_tag == tag {
                        collect.swap(&RefCell::new(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);
                    }
                }
                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);
                }

                Ok(())
            }
        }
    })?;

    Ok(String::from_utf8(buf.into_inner())?)
}

fn iter_nodes<'a, F>(node: &'a AstNode<'a>, f: &F) -> Result<()>
where
    F: Fn(&'a AstNode<'a>) -> Result<()>,
{
    f(node)?;
    for c in node.children() {
        match c.data.borrow().value {
            NodeValue::Text(_) => {}
            NodeValue::Item(_) => {}
            NodeValue::Code(_) => {}
            _ => {
                iter_nodes(c, f)?;
            }
        }
    }

    Ok(())
}

#[cfg(test)]
mod tests {
    #[test]
    fn read_changelog() {
        let res = super::read("testdata/basic.md".into(), "0.1.0".into());
        assert!(res.is_ok());
        let delta = res.unwrap();
        assert_eq!(delta, "Hi there this is a test\\!\n### ADDED\n  - something\n")
    }
}