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

use std::{
    fs, io,
    path::{Path, PathBuf},
};

pub trait Expand {
    fn expand(&self) -> io::Result<Vec<PathBuf>>;
}

impl Expand for PathBuf {
    fn expand(&self) -> io::Result<Vec<PathBuf>> {
        let mut files = vec![];

        if self.is_file() {
            files.push(self.into());
        } else if self.is_dir() {
            for entry in fs::read_dir(self)? {
                let mut inner = Self::expand(&entry?.path())?;
                files.append(&mut inner);
            }
        }

        Ok(files)
    }
}

impl Expand for Path {
    fn expand(&self) -> io::Result<Vec<PathBuf>> {
        let mut files = vec![];

        if self.is_file() {
            files.push(self.into());
        } else if self.is_dir() {
            for entry in fs::read_dir(self)? {
                let mut inner = Self::expand(&entry?.path())?;
                files.append(&mut inner);
            }
        }

        Ok(files)
    }
}