use globset::{Glob, GlobMatcher};
use std::ffi::OsStr;
use std::path::Path;
const GLOB_METACHARS: &[char] = &['*', '?', '[', ']', '{', '}'];
pub fn explicit_file_name_glob(pattern: &str) -> Option<&str> {
if pattern.starts_with('!') {
return None;
}
let file_name = pattern.rsplit('/').next().unwrap_or(pattern);
if file_name.is_empty() {
return None;
}
let literal_tail = match file_name.rfind(GLOB_METACHARS) {
Some(idx) => &file_name[idx + 1..],
None => file_name,
};
match literal_tail.rsplit_once('.') {
Some((_, ext)) if !ext.is_empty() => Some(file_name),
_ => None,
}
}
pub struct ExplicitIncludeMatchers {
matchers: Vec<ExplicitInclude>,
}
struct ExplicitInclude {
file_name_glob: String,
path_matcher: GlobMatcher,
}
impl ExplicitIncludeMatchers {
pub fn new(patterns: &[String]) -> Self {
let matchers = patterns
.iter()
.filter_map(|pattern| {
let file_name_glob = explicit_file_name_glob(pattern)?;
let path_glob = if let Some(anchored) = pattern.strip_prefix('/') {
anchored.to_string()
} else if pattern.contains('/') {
pattern.clone()
} else {
format!("**/{pattern}")
};
let path_matcher = globset::GlobBuilder::new(&path_glob)
.literal_separator(true)
.build()
.ok()?
.compile_matcher();
Some(ExplicitInclude {
file_name_glob: file_name_glob.to_string(),
path_matcher,
})
})
.collect();
Self { matchers }
}
pub fn is_empty(&self) -> bool {
self.matchers.is_empty()
}
pub fn file_name_globs(&self) -> impl Iterator<Item = &str> {
self.matchers.iter().map(|m| m.file_name_glob.as_str())
}
pub fn matches_relative_path(&self, path: &str) -> bool {
self.matchers.iter().any(|m| m.path_matcher.is_match(path))
}
}
pub const MARKDOWN_EXTENSIONS: &[&str] = &["md", "markdown", "mdx", "mkd", "mkdn", "mdown", "mdwn", "qmd", "rmd"];
#[inline]
pub fn is_markdown_extension(ext: &OsStr) -> bool {
ext.to_str()
.is_some_and(|s| MARKDOWN_EXTENSIONS.iter().any(|known| s.eq_ignore_ascii_case(known)))
}
#[inline]
pub fn has_markdown_extension(path: &Path) -> bool {
path.extension().is_some_and(is_markdown_extension)
}
#[derive(Debug, Clone)]
pub struct MarkdownWalkOptions {
pub respect_gitignore: bool,
pub skip_vendor_dirs: bool,
}
impl Default for MarkdownWalkOptions {
fn default() -> Self {
Self {
respect_gitignore: true,
skip_vendor_dirs: false,
}
}
}
pub fn apply_markdown_walk_options(builder: &mut ignore::WalkBuilder, options: &MarkdownWalkOptions) {
let gitignore = options.respect_gitignore;
builder
.ignore(gitignore)
.git_ignore(gitignore)
.git_global(gitignore)
.git_exclude(gitignore)
.parents(gitignore)
.hidden(false)
.require_git(false)
.add_custom_ignore_filename(".markdownlintignore");
if options.skip_vendor_dirs {
builder.filter_entry(|entry| {
let name = entry.file_name().to_str().unwrap_or("");
name != ".git" && name != "node_modules" && name != "target"
});
}
}
pub fn markdown_walk_builder(root: &Path, options: &MarkdownWalkOptions) -> ignore::WalkBuilder {
let mut builder = ignore::WalkBuilder::new(root);
apply_markdown_walk_options(&mut builder, options);
builder
}
pub fn expand_directory_pattern(pattern: &str) -> Vec<String> {
if pattern.contains('*') || pattern.contains('?') || pattern.contains('[') {
return vec![pattern.to_string()];
}
let base = pattern.trim_end_matches('/');
vec![
base.to_string(), format!("{base}/**"), ]
}
pub struct ExcludeMatchers {
matchers: Vec<(String, GlobMatcher)>,
pub invalid: Vec<(String, String)>,
}
impl ExcludeMatchers {
pub fn new(patterns: &[String]) -> Self {
let mut matchers = Vec::new();
let mut invalid = Vec::new();
for pattern in patterns.iter().flat_map(|p| expand_directory_pattern(p)) {
match Glob::new(&pattern) {
Ok(glob) => matchers.push((pattern, glob.compile_matcher())),
Err(e) => invalid.push((pattern, e.to_string())),
}
}
Self { matchers, invalid }
}
pub fn is_empty(&self) -> bool {
self.matchers.is_empty()
}
pub fn matched_pattern(&self, relative_path: &str) -> Option<&str> {
self.matchers
.iter()
.find(|(_, matcher)| matcher.is_match(relative_path))
.map(|(pattern, _)| pattern.as_str())
}
pub fn is_match(&self, relative_path: &str) -> bool {
self.matched_pattern(relative_path).is_some()
}
}
pub fn path_relative_to(path: &Path, base: &Path) -> Option<String> {
let canonical_base = base.canonicalize().ok()?;
let canonical_path = path.canonicalize().ok()?;
canonical_path.strip_prefix(&canonical_base).ok().map(|rel| {
let rel = rel.to_string_lossy();
if cfg!(windows) {
rel.replace('\\', "/")
} else {
rel.to_string()
}
})
}
#[cfg(test)]
mod tests {
use super::*;
use std::fs;
use tempfile::tempdir;
#[test]
fn markdown_extensions_match_case_insensitively() {
for ext in ["md", "MD", "Rmd", "rmd", "MarkDown", "qmd", "mdx"] {
assert!(is_markdown_extension(OsStr::new(ext)), "{ext} should match");
}
for ext in ["rs", "txt", "mdq", ""] {
assert!(!is_markdown_extension(OsStr::new(ext)), "{ext} should not match");
}
assert!(has_markdown_extension(Path::new("a/b/README.md")));
assert!(has_markdown_extension(Path::new("notebook.Rmd")));
assert!(!has_markdown_extension(Path::new("no_extension")));
assert!(!has_markdown_extension(Path::new("lib.rs")));
}
#[test]
fn walk_includes_hidden_files() {
let temp = tempdir().unwrap();
fs::create_dir_all(temp.path().join(".github")).unwrap();
fs::write(temp.path().join(".github/PULL_REQUEST_TEMPLATE.md"), "# hi").unwrap();
fs::write(temp.path().join("README.md"), "# hi").unwrap();
let files: Vec<_> = markdown_walk_builder(temp.path(), &MarkdownWalkOptions::default())
.build()
.flatten()
.filter(|e| e.file_type().is_some_and(|t| t.is_file()))
.map(|e| e.path().to_path_buf())
.collect();
assert!(files.iter().any(|p| p.ends_with(".github/PULL_REQUEST_TEMPLATE.md")));
assert!(files.iter().any(|p| p.ends_with("README.md")));
}
#[test]
fn walk_honors_gitignore_when_enabled_only() {
let temp = tempdir().unwrap();
fs::write(temp.path().join(".gitignore"), "ignored.md\n").unwrap();
fs::write(temp.path().join("ignored.md"), "# hi").unwrap();
fs::write(temp.path().join("kept.md"), "# hi").unwrap();
let walk = |respect: bool| -> Vec<std::path::PathBuf> {
markdown_walk_builder(
temp.path(),
&MarkdownWalkOptions {
respect_gitignore: respect,
..Default::default()
},
)
.build()
.flatten()
.filter(|e| e.file_type().is_some_and(|t| t.is_file()))
.map(|e| e.path().to_path_buf())
.collect()
};
let respected = walk(true);
assert!(!respected.iter().any(|p| p.ends_with("ignored.md")));
assert!(respected.iter().any(|p| p.ends_with("kept.md")));
let unrespected = walk(false);
assert!(unrespected.iter().any(|p| p.ends_with("ignored.md")));
}
#[test]
fn walk_honors_markdownlintignore() {
let temp = tempdir().unwrap();
fs::write(temp.path().join(".markdownlintignore"), "legacy.md\n").unwrap();
fs::write(temp.path().join("legacy.md"), "# hi").unwrap();
fs::write(temp.path().join("kept.md"), "# hi").unwrap();
let files: Vec<_> = markdown_walk_builder(temp.path(), &MarkdownWalkOptions::default())
.build()
.flatten()
.filter(|e| e.file_type().is_some_and(|t| t.is_file()))
.map(|e| e.path().to_path_buf())
.collect();
assert!(!files.iter().any(|p| p.ends_with("legacy.md")));
assert!(files.iter().any(|p| p.ends_with("kept.md")));
}
#[test]
fn vendor_dirs_skipped_only_when_requested() {
let temp = tempdir().unwrap();
for dir in ["node_modules", "target", "src"] {
fs::create_dir_all(temp.path().join(dir)).unwrap();
fs::write(temp.path().join(dir).join("doc.md"), "# hi").unwrap();
}
let walk = |skip: bool| -> Vec<std::path::PathBuf> {
markdown_walk_builder(
temp.path(),
&MarkdownWalkOptions {
skip_vendor_dirs: skip,
respect_gitignore: false,
},
)
.build()
.flatten()
.filter(|e| e.file_type().is_some_and(|t| t.is_file()))
.map(|e| e.path().to_path_buf())
.collect()
};
let skipped = walk(true);
assert!(!skipped.iter().any(|p| p.to_string_lossy().contains("node_modules")));
assert!(!skipped.iter().any(|p| p.to_string_lossy().contains("target")));
assert!(skipped.iter().any(|p| p.ends_with("src/doc.md")));
let unskipped = walk(false);
assert!(unskipped.iter().any(|p| p.to_string_lossy().contains("node_modules")));
}
#[test]
fn explicit_file_name_glob_extracts_literal_extensions() {
assert_eq!(explicit_file_name_glob("**/*.md.jinja"), Some("*.md.jinja"));
assert_eq!(explicit_file_name_glob("*.md.jinja"), Some("*.md.jinja"));
assert_eq!(explicit_file_name_glob("docs/*.txt"), Some("*.txt"));
assert_eq!(explicit_file_name_glob("templates/NOTES.tmpl"), Some("NOTES.tmpl"));
assert_eq!(explicit_file_name_glob("*.md"), Some("*.md"));
assert_eq!(explicit_file_name_glob("a/b/c/*.md.tmpl"), Some("*.md.tmpl"));
}
#[test]
fn explicit_file_name_glob_rejects_unpinned_patterns() {
for pattern in [
"docs/",
"docs/**",
"docs",
"*",
"**",
"**/*",
"*.*",
"*.md*",
"*.{md,jinja}",
"*.md?",
"data.[ch]",
"!drafts/*.md.jinja",
"",
"**/Makefile",
"*.",
] {
assert_eq!(explicit_file_name_glob(pattern), None, "{pattern:?} should not qualify");
}
}
#[test]
fn explicit_include_matchers_match_full_relative_paths() {
let matchers = ExplicitIncludeMatchers::new(&[
"**/*.md.jinja".to_string(),
"docs/**".to_string(),
"templates/NOTES.tmpl".to_string(),
]);
assert!(!matchers.is_empty());
assert!(matchers.matches_relative_path("test.md.jinja"));
assert!(matchers.matches_relative_path("a/b/test.md.jinja"));
assert!(matchers.matches_relative_path("templates/NOTES.tmpl"));
assert!(!matchers.matches_relative_path("docs/anything.txt"));
assert!(!matchers.matches_relative_path("test.jinja"));
assert!(!matchers.matches_relative_path("docs/NOTES.tmpl"));
assert!(!matchers.matches_relative_path("x/templates/NOTES.tmpl"));
let globs: Vec<_> = matchers.file_name_globs().collect();
assert_eq!(globs, vec!["*.md.jinja", "NOTES.tmpl"]);
}
#[test]
fn explicit_include_matchers_follow_gitignore_anchoring() {
let unanchored = ExplicitIncludeMatchers::new(&["*.md.jinja".to_string()]);
assert!(unanchored.matches_relative_path("test.md.jinja"));
assert!(unanchored.matches_relative_path("a/b/test.md.jinja"));
let anchored = ExplicitIncludeMatchers::new(&["docs/*.txt".to_string()]);
assert!(anchored.matches_relative_path("docs/a.txt"));
assert!(!anchored.matches_relative_path("docs/sub/a.txt"));
assert!(!anchored.matches_relative_path("other/docs/a.txt"));
let rooted = ExplicitIncludeMatchers::new(&["/NOTES.tmpl".to_string()]);
assert!(rooted.matches_relative_path("NOTES.tmpl"));
assert!(!rooted.matches_relative_path("docs/NOTES.tmpl"));
}
#[test]
fn explicit_include_matchers_empty_for_directory_and_wildcard_patterns() {
let matchers = ExplicitIncludeMatchers::new(&["docs/".to_string(), "**/*".to_string()]);
assert!(matchers.is_empty());
assert!(!matchers.matches_relative_path("x.md.jinja"));
}
#[test]
fn explicit_include_matchers_skip_invalid_globs() {
let matchers = ExplicitIncludeMatchers::new(&["bad[.tmpl".to_string(), "**/*.md.jinja".to_string()]);
assert!(matchers.matches_relative_path("ok.md.jinja"));
assert_eq!(matchers.file_name_globs().collect::<Vec<_>>(), vec!["*.md.jinja"]);
}
#[test]
fn exclude_matchers_expand_directory_patterns() {
let matchers = ExcludeMatchers::new(&["drafts".to_string(), "*.tmp.md".to_string()]);
assert!(matchers.is_match("drafts"));
assert!(
matchers.is_match("drafts/inner.md"),
"directory pattern must match contents"
);
assert!(matchers.is_match("note.tmp.md"));
assert!(!matchers.is_match("docs/guide.md"));
assert_eq!(matchers.matched_pattern("drafts/inner.md"), Some("drafts/**"));
assert!(matchers.invalid.is_empty());
}
#[test]
fn exclude_matchers_report_invalid_patterns() {
let matchers = ExcludeMatchers::new(&["[".to_string(), "ok.md".to_string()]);
assert_eq!(matchers.invalid.len(), 1);
assert_eq!(matchers.invalid[0].0, "[");
assert!(matchers.is_match("ok.md"));
}
#[test]
fn path_relative_to_strips_through_symlinked_base() {
let temp = tempdir().unwrap();
let base = temp.path().join("base");
fs::create_dir_all(base.join("docs")).unwrap();
fs::write(base.join("docs/a.md"), "# hi").unwrap();
assert_eq!(
path_relative_to(&base.join("docs/a.md"), &base).as_deref(),
Some("docs/a.md")
);
assert_eq!(
path_relative_to(&base.join("docs/a.md"), &base.join("docs")).as_deref(),
Some("a.md")
);
assert_eq!(path_relative_to(temp.path(), &base), None, "path outside base");
}
}