arbol 0.1.0

Tree-sitter rust AST query tool
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

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

use walkdir::WalkDir;

use crate::error::Result;

pub(crate) fn discover_crate_roots(root: &Path) -> Result<Vec<PathBuf>> {
    let mut crates = Vec::new();
    for entry in WalkDir::new(root).into_iter().filter_map(|e| e.ok()) {
        if entry.file_type().is_file()
            && entry.file_name() == "Cargo.toml"
            && let Some(parent) = entry.path().parent()
        {
            crates.push(parent.to_path_buf());
        }
    }
    crates.sort_by_key(|b| std::cmp::Reverse(b.as_os_str().len()));
    Ok(crates)
}

pub(crate) fn crate_for_file(file: &Path, crate_roots: &[PathBuf]) -> Option<String> {
    for root in crate_roots {
        if file.starts_with(root) {
            return Some(root.display().to_string());
        }
    }
    None
}