smark 0.1.9

The tool provides API to get markdown documents from files
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

use crate::posts::Post;
use anyhow::Result;
use glob::glob;
use std::path::PathBuf;

pub fn get_all_posts(glob_pattern: &str) -> Result<Vec<(PathBuf, Post)>> {
    let posts = glob(glob_pattern)?
        .into_iter()
        .filter_map(|path| path.ok())
        .filter_map(|path| {
            // should be /path/to/filename.md
            let post = Post::from_path(&path);
            if let Ok(post) = post {
                Some((path, post))
            } else {
                error!(
                    "Error in {:?}. Maybe this file has invalid frontmatter. Skipping this file.",
                    path
                );
                None
            }
        })
        .collect();

    Ok(posts)
}

#[cfg(test)]
mod test {
    use super::*;
    use crate::posts::template;
    use std::fs;
    use std::io::Write;
    #[test]
    fn test_get_all_posts() {
        let temp_dir = tempdir::TempDir::new("test_get_all_posts").unwrap();

        let dirs = &["a", "b", "b/d", "c", "c/f", "c/f/h"];
        let md_files = &[
            vec!["1.md", "2.md"],
            vec!["3.md", "not_md.txt"],
            vec!["4.md", "5.md", "6.md"],
            vec!["7.md", "testing.png"],
            vec!["9.md", "10.md"],
            vec!["11.md"],
        ];
        for (dir, files) in dirs.into_iter().zip(md_files.into_iter()) {
            let dir_path = temp_dir.path().join(dir);
            fs::create_dir(&dir_path).unwrap();
            for file in files.iter() {
                let mut f = fs::File::create(&dir_path.join(file)).unwrap();
                write!(f, "{}", template(&false, &None).unwrap()).unwrap();
            }
        }

        let expect_files_count = md_files
            .iter()
            .flatten()
            .filter(|x| x.ends_with(".md"))
            .count();
        let actual_files_count = get_all_posts(&format!("{}/**/*.md", temp_dir.path().display()))
            .unwrap()
            .len();

        assert_eq!(expect_files_count, actual_files_count);
    }
}