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

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

use anyhow::Result;
use xshell::Shell;

pub fn find_files<P: AsRef<Path>>(dir: P, extension: &str) -> Result<Vec<PathBuf>> {
    let mut result = Vec::new();
    let dir_path = dir.as_ref();
    find_files_recursive(dir_path, extension, &mut result)?;
    Ok(result)
}

fn find_files_recursive(dir: &Path, extension: &str, result: &mut Vec<PathBuf>) -> Result<()> {
    for entry_result in fs::read_dir(dir)? {
        let entry = entry_result?;
        let path = entry.path();

        if path.is_dir() {
            find_files_recursive(&path, extension, result)?;
        } else if path.is_file() && path.extension().map_or(false, |ext| ext == extension) {
            result.push(path);
        }
    }
    Ok(())
}

pub fn project_root() -> PathBuf {
    Path::new(env!("CARGO_MANIFEST_DIR"))
        .parent()
        .expect("Failed to find project root")
        .to_path_buf()
}

pub fn to_relative_paths<P: AsRef<Path>>(paths: Vec<PathBuf>, base_dir: P) -> Vec<PathBuf> {
    let base_path = base_dir.as_ref();
    paths
        .into_iter()
        .filter_map(|path| path.strip_prefix(base_path).ok().map(PathBuf::from))
        .collect()
}

pub fn verbose_cd<P: AsRef<Path>>(sh: &Shell, dir: P) {
    sh.change_dir(dir);
    eprintln!(
        "\n$ cd {}{}",
        sh.current_dir().display(),
        std::path::MAIN_SEPARATOR
    );
}