trillium-include-dir-impl 0.1.1

temporary fork of the include_dir_impl crate for use in trillium.rs
Documentation 
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

use crate::{file::File, timestamp_to_tokenstream, Error};
use proc_macro2::TokenStream;
use quote::{quote, ToTokens};
use std::{
    fs::Metadata,
    path::{Path, PathBuf},
};

#[derive(Debug, Clone)]
pub(crate) struct Dir {
    root_rel_path: PathBuf,
    abs_path: PathBuf,
    files: Vec<File>,
    dirs: Vec<Dir>,
    metadata: Metadata,
}

impl PartialEq for Dir {
    fn eq(&self, other: &Self) -> bool {
        self.root_rel_path == other.root_rel_path
            && self.abs_path == other.abs_path
            && self.files == other.files
            && self.dirs == other.dirs
    }
}

impl Dir {
    pub fn from_disk<Q: AsRef<Path>, P: Into<PathBuf>>(root: Q, path: P) -> Result<Dir, Error> {
        let abs_path = path.into();
        let root = root.as_ref();

        let root_rel_path = abs_path.strip_prefix(&root)?.to_path_buf();
        let metadata = std::fs::metadata(&abs_path)?;

        if !abs_path.exists() {
            return Err("The directory doesn't exist".into());
        }

        let mut files = Vec::new();
        let mut dirs = Vec::new();

        for entry in abs_path.read_dir()? {
            let entry = entry?.path();

            if entry.is_file() {
                files.push(File::from_disk(&root, entry)?);
            } else if entry.is_dir() {
                dirs.push(Dir::from_disk(&root, entry)?);
            }
        }

        Ok(Dir {
            root_rel_path,
            abs_path,
            files,
            dirs,
            metadata,
        })
    }
}

impl ToTokens for Dir {
    fn to_tokens(&self, tokens: &mut TokenStream) {
        let root_rel_path = self.root_rel_path.display().to_string();
        let files = &self.files;
        let dirs = &self.dirs;
        let created = timestamp_to_tokenstream(self.metadata.created());
        let modified = timestamp_to_tokenstream(self.metadata.modified());
        let accessed = timestamp_to_tokenstream(self.metadata.accessed());

        let tok = quote! {
            Dir {
                path: #root_rel_path,
                files: &[#(
                    #files
                 ),*],
                dirs: &[#(
                    #dirs
                 ),*],
                created: #created,
                modified: #modified,
                accessed: #accessed,
            }
        };

        tok.to_tokens(tokens);
    }
}