use std::path::{Path as StdPath, PathBuf};
use crate::extract::format::{FALLBACK_FORMAT, resolve_format};
#[derive(Debug, Clone, PartialEq, Eq)]
pub(crate) struct Target {
pub(crate) path: PathBuf,
pub(crate) language_id: &'static str,
}
#[derive(Debug, Clone)]
pub(crate) struct WalkOptions {
pub(crate) hidden: bool,
pub(crate) respect_ignore: bool,
pub(crate) follow_symlinks: bool,
pub(crate) format: Option<&'static str>,
}
impl Default for WalkOptions {
fn default() -> Self {
Self {
hidden: false,
respect_ignore: true,
follow_symlinks: false,
format: None,
}
}
}
pub(crate) fn collect(inputs: &[PathBuf], options: &WalkOptions) -> Result<Vec<Target>, String> {
let mut targets = Vec::new();
for input in inputs {
let metadata =
std::fs::metadata(input).map_err(|error| format!("{}: {error}", input.display()))?;
if metadata.is_file() {
targets.push(Target {
path: input.clone(),
language_id: options.format.unwrap_or_else(|| language_for(input)),
});
continue;
}
targets.extend(walk_directory(input, options)?);
}
targets.sort_by(|a, b| a.path.cmp(&b.path));
targets.dedup();
Ok(targets)
}
fn walk_directory(root: &StdPath, options: &WalkOptions) -> Result<Vec<Target>, String> {
let mut builder = ignore::WalkBuilder::new(root);
builder
.hidden(!options.hidden)
.git_ignore(options.respect_ignore)
.git_global(options.respect_ignore)
.git_exclude(options.respect_ignore)
.ignore(options.respect_ignore)
.parents(options.respect_ignore)
.follow_links(options.follow_symlinks);
let mut targets = Vec::new();
for entry in builder.build() {
let entry = entry.map_err(|error| format!("{}: {error}", root.display()))?;
if !entry.file_type().is_some_and(|kind| kind.is_file()) {
continue;
}
targets.push(Target {
path: entry.path().to_path_buf(),
language_id: options.format.unwrap_or_else(|| language_for(entry.path())),
});
}
Ok(targets)
}
fn language_for(path: &StdPath) -> &'static str {
path.file_name()
.and_then(|name| name.to_str())
.map_or(FALLBACK_FORMAT, |name| resolve_format(None, Some(name)))
}
#[cfg(test)]
mod tests {
use super::*;
use crate::testing::TempTree;
fn names(targets: &[Target]) -> Vec<String> {
targets
.iter()
.map(|target| {
target
.path
.file_name()
.expect("a name")
.to_string_lossy()
.into_owned()
})
.collect()
}
#[test]
fn a_directory_yields_every_file_and_names_what_will_read_it() {
let tree = TempTree::new("walk-formats");
tree.write("a.json", "{}");
tree.write("b.toml", "");
tree.write("c.md", "# no");
tree.write("d.py", "pass");
tree.write("e.yml", "a: 1");
let targets = collect(&[tree.path().to_path_buf()], &WalkOptions::default())
.expect("the walk succeeds");
assert_eq!(
names(&targets),
["a.json", "b.toml", "c.md", "d.py", "e.yml"]
);
let languages: Vec<&str> = targets.iter().map(|target| target.language_id).collect();
assert_eq!(
languages,
["json", "toml", "markdown", FALLBACK_FORMAT, "yaml"]
);
}
#[test]
fn the_order_is_stable() {
let tree = TempTree::new("walk-order");
for name in ["z.json", "a.json", "m.json"] {
tree.write(name, "{}");
}
let first = collect(&[tree.path().to_path_buf()], &WalkOptions::default())
.expect("the walk succeeds");
let second = collect(&[tree.path().to_path_buf()], &WalkOptions::default())
.expect("the walk succeeds");
assert_eq!(names(&first), ["a.json", "m.json", "z.json"]);
assert_eq!(first, second);
}
#[test]
fn gitignored_files_are_skipped_by_default_and_walked_on_request() {
let tree = TempTree::new("walk-ignore");
tree.mkdir(".git");
tree.write(".gitignore", "skipped.json\n");
tree.write("skipped.json", "{}");
tree.write("kept.json", "{}");
let default = collect(&[tree.path().to_path_buf()], &WalkOptions::default())
.expect("the walk succeeds");
assert_eq!(names(&default), ["kept.json"]);
let everything = collect(
&[tree.path().to_path_buf()],
&WalkOptions {
respect_ignore: false,
..WalkOptions::default()
},
)
.expect("the walk succeeds");
assert_eq!(names(&everything), ["kept.json", "skipped.json"]);
}
#[test]
fn a_gitignore_outside_a_repository_is_inert() {
let tree = TempTree::new("walk-ignore-nogit");
tree.write(".gitignore", "skipped.json\n");
tree.write("skipped.json", "{}");
let targets = collect(&[tree.path().to_path_buf()], &WalkOptions::default())
.expect("the walk succeeds");
assert_eq!(names(&targets), ["skipped.json"]);
}
#[test]
fn hidden_files_are_skipped_by_default_and_walked_on_request() {
let tree = TempTree::new("walk-hidden");
tree.write(".hidden.json", "{}");
tree.write("shown.json", "{}");
let default = collect(&[tree.path().to_path_buf()], &WalkOptions::default())
.expect("the walk succeeds");
assert_eq!(names(&default), ["shown.json"]);
let everything = collect(
&[tree.path().to_path_buf()],
&WalkOptions {
hidden: true,
..WalkOptions::default()
},
)
.expect("the walk succeeds");
assert_eq!(names(&everything), [".hidden.json", "shown.json"]);
}
#[test]
fn an_explicitly_named_file_beats_the_ignore_rules() {
let tree = TempTree::new("walk-explicit");
tree.write(".gitignore", "skipped.json\n");
let file = tree.write("skipped.json", "{}");
let targets = collect(&[file], &WalkOptions::default()).expect("the walk succeeds");
assert_eq!(names(&targets), ["skipped.json"]);
}
#[test]
fn an_explicitly_named_file_of_unknown_format_is_read_by_the_scan() {
let tree = TempTree::new("walk-unknown");
let file = tree.write("Dockerfile", "COPY ./src /app\n");
let targets = collect(&[file], &WalkOptions::default()).expect("the walk succeeds");
assert_eq!(names(&targets), ["Dockerfile"]);
assert_eq!(targets[0].language_id, FALLBACK_FORMAT);
}
#[test]
fn a_forced_format_overrides_the_filename() {
let tree = TempTree::new("walk-forced");
let file = tree.write("notes.md", "{}");
let targets = collect(
&[file],
&WalkOptions {
format: Some("json"),
..WalkOptions::default()
},
)
.expect("the walk succeeds");
assert_eq!(targets[0].language_id, "json");
}
#[test]
fn a_missing_input_is_refused_by_name() {
let tree = TempTree::new("walk-missing");
let error =
collect(&[tree.path().join("nope")], &WalkOptions::default()).expect_err("a refusal");
assert!(error.contains("nope"), "{error}");
}
#[test]
fn a_dotfile_resolves_by_its_whole_name() {
let tree = TempTree::new("walk-dotenv");
tree.write(".env", "A=/x.txt\n");
let targets = collect(
&[tree.path().to_path_buf()],
&WalkOptions {
hidden: true,
..WalkOptions::default()
},
)
.expect("the walk succeeds");
assert_eq!(targets.len(), 1);
assert_eq!(targets[0].language_id, "dotenv");
}
#[test]
fn naming_the_same_file_twice_examines_it_once() {
let tree = TempTree::new("walk-dedupe");
let file = tree.write("a.json", "{}");
let targets =
collect(&[file.clone(), file], &WalkOptions::default()).expect("the walk succeeds");
assert_eq!(targets.len(), 1);
}
}